63 lines
1.7 KiB
Go
63 lines
1.7 KiB
Go
|
package grpc
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
_ "go.unistack.org/micro-server-grpc/v3/proto"
|
||
|
|
"go.unistack.org/micro/v3/server"
|
||
|
|
"google.golang.org/grpc"
|
||
|
|
"google.golang.org/protobuf/reflect/protoreflect"
|
||
|
|
"google.golang.org/protobuf/reflect/protoregistry"
|
||
|
|
)
|
||
|
|
|
||
|
|
type reflector struct{}
|
||
|
|
|
||
|
|
func (r *reflector) FindFileByPath(path string) (protoreflect.FileDescriptor, error) {
|
||
|
|
fd, err := protoregistry.GlobalFiles.FindFileByPath(path)
|
||
|
|
if err != nil {
|
||
|
|
fmt.Printf("err: %v\n", err)
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
return fd, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *reflector) FindDescriptorByName(name protoreflect.FullName) (protoreflect.Descriptor, error) {
|
||
|
|
fd, err := protoregistry.GlobalFiles.FindDescriptorByName(name)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
return fd, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *reflector) GetServiceInfo() map[string]grpc.ServiceInfo {
|
||
|
|
fmt.Printf("GetServiceInfo\n")
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *reflector) FindExtensionByName(field protoreflect.FullName) (protoreflect.ExtensionType, error) {
|
||
|
|
fmt.Printf("FindExtensionByName field %#+v\n", field)
|
||
|
|
return nil, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *reflector) FindExtensionByNumber(message protoreflect.FullName, field protoreflect.FieldNumber) (protoreflect.ExtensionType, error) {
|
||
|
|
fmt.Printf("FindExtensionByNumber message %#+v field %#+v\n", message, field)
|
||
|
|
return nil, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *reflector) RangeExtensionsByMessage(message protoreflect.FullName, f func(protoreflect.ExtensionType) bool) {
|
||
|
|
fmt.Printf("RangeExtensionsByMessage\n")
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestReflector(t *testing.T) {
|
||
|
|
srv := NewServer(Reflection(&reflector{}), server.Address(":12345"))
|
||
|
|
if err := srv.Init(); err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
if err := srv.Start(); err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
t.Logf("addr %s", srv.Options().Address)
|
||
|
|
select {}
|
||
|
|
}
|