fix drpc method passing #31
4
drpc.go
4
drpc.go
@ -30,6 +30,9 @@ type drpcClient struct {
|
||||
|
||||
func (d *drpcClient) call(ctx context.Context, addr string, req client.Request, rsp interface{}, opts client.CallOptions) error {
|
||||
var header map[string]string
|
||||
if strings.HasPrefix(addr, "http") {
|
||||
addr = addr[strings.Index(addr, ":")+3:]
|
||||
}
|
||||
|
||||
if md, ok := metadata.FromOutgoingContext(ctx); ok {
|
||||
header = make(map[string]string, len(md))
|
||||
@ -90,6 +93,7 @@ func (d *drpcClient) call(ctx context.Context, addr string, req client.Request,
|
||||
*/
|
||||
ch := make(chan error, 1)
|
||||
_ = dialCtx
|
||||
|
||||
// rc, err := net.DialContext(ctx, "tcp", addr)
|
||||
rc, err := net.Dial("tcp", addr)
|
||||
if err != nil {
|
||||
|
21
request.go
21
request.go
@ -18,24 +18,19 @@ type drpcRequest struct {
|
||||
}
|
||||
|
||||
// service Struct.Method /service.Struct/Method
|
||||
func methodToDRPC(service, method string) string {
|
||||
// no method or already grpc method
|
||||
func methodToDRPC(service string, method string) string {
|
||||
// no method or already drpc method
|
||||
if len(method) == 0 || method[0] == '/' {
|
||||
return method
|
||||
}
|
||||
|
||||
// assume method is Foo.Bar
|
||||
mParts := strings.Split(method, ".")
|
||||
if len(mParts) != 2 {
|
||||
return method
|
||||
idx := strings.LastIndex(method, ".")
|
||||
if len(method) < 3 || idx < 2 {
|
||||
return fmt.Sprintf("/%s.%s", strings.Title(service), method)
|
||||
}
|
||||
|
||||
if len(service) == 0 {
|
||||
return fmt.Sprintf("/%s/%s", mParts[0], mParts[1])
|
||||
}
|
||||
|
||||
// return /pkg.Foo/Bar
|
||||
return fmt.Sprintf("/%s.%s/%s", strings.Title(service), mParts[0], mParts[1])
|
||||
drpcService := method[:idx]
|
||||
drpcMethod := method[idx+1:]
|
||||
return fmt.Sprintf("/%s/%s", drpcService, drpcMethod)
|
||||
}
|
||||
|
||||
func newDRPCRequest(service, method string, request interface{}, contentType string, reqOpts ...client.RequestOption) client.Request {
|
||||
|
@ -12,7 +12,7 @@ func TestMethodToDRPC(t *testing.T) {
|
||||
}{
|
||||
{
|
||||
"helloworld",
|
||||
"Greeter.SayHello",
|
||||
"Greeter/SayHello",
|
||||
"/Helloworld.Greeter/SayHello",
|
||||
},
|
||||
{
|
||||
|
Loading…
Reference in New Issue
Block a user