@@ -41,11 +41,11 @@ func (g *Generator) generateServiceClient(gfile *protogen.GeneratedFile, file *p
4141 gfile .P ("name string" )
4242 gfile .P ("}" )
4343
44- if g .standalone {
45- gfile .P ("func New" , serviceName , "Client(name string, c " , microClientPackage .Ident ("Client" ), ") " , file .GoImportPath .Ident (serviceName ), "Client {" )
46- } else {
47- gfile .P ("func New" , serviceName , "Client(name string, c " , microClientPackage .Ident ("Client" ), ") " , serviceName , "Client {" )
48- }
44+ // if g.standalone {
45+ // gfile.P("func New", serviceName, "Client(name string, c ", microClientPackage.Ident("Client"), ") ", file.GoImportPath.Ident(serviceName), "Client {")
46+ // } else {
47+ gfile .P ("func New" , serviceName , "Client(name string, c " , microClientPackage .Ident ("Client" ), ") " , serviceName , "Client {" )
48+ // }
4949 gfile .P ("return &" , unexport (serviceName ), "Client{c: c, name: name}" )
5050 gfile .P ("}" )
5151 gfile .P ()
@@ -326,11 +326,11 @@ func (g *Generator) generateServiceClientMethods(gfile *protogen.GeneratedFile,
326326func (g * Generator ) generateServiceServer (gfile * protogen.GeneratedFile , file * protogen.File , service * protogen.Service ) {
327327 serviceName := service .GoName
328328 gfile .P ("type " , unexport (serviceName ), "Server struct {" )
329- if g .standalone {
330- gfile .P (file .GoImportPath .Ident (serviceName ), "Server" )
331- } else {
332- gfile .P (serviceName , "Server" )
333- }
329+ // if g.standalone {
330+ // gfile.P(file.GoImportPath.Ident(serviceName), "Server")
331+ // } else {
332+ gfile .P (serviceName , "Server" )
333+ // }
334334 gfile .P ("}" )
335335 gfile .P ()
336336}
@@ -487,11 +487,11 @@ func (g *Generator) generateServiceServerMethods(gfile *protogen.GeneratedFile,
487487
488488func (g * Generator ) generateServiceRegister (gfile * protogen.GeneratedFile , file * protogen.File , service * protogen.Service , component string ) {
489489 serviceName := service .GoName
490- if g .standalone {
491- gfile .P ("func Register" , serviceName , "Server(s " , microServerPackage .Ident ("Server" ), ", sh " , file .GoImportPath .Ident (serviceName ), "Server, opts ..." , microServerPackage .Ident ("HandlerOption" ), ") error {" )
492- } else {
493- gfile .P ("func Register" , serviceName , "Server(s " , microServerPackage .Ident ("Server" ), ", sh " , serviceName , "Server, opts ..." , microServerPackage .Ident ("HandlerOption" ), ") error {" )
494- }
490+ // if g.standalone {
491+ // gfile.P("func Register", serviceName, "Server(s ", microServerPackage.Ident("Server"), ", sh ", file.GoImportPath.Ident(serviceName), "Server, opts ...", microServerPackage.Ident("HandlerOption"), ") error {")
492+ // } else {
493+ gfile .P ("func Register" , serviceName , "Server(s " , microServerPackage .Ident ("Server" ), ", sh " , serviceName , "Server, opts ..." , microServerPackage .Ident ("HandlerOption" ), ") error {" )
494+ // }
495495 gfile .P ("type " , unexport (serviceName ), " interface {" )
496496 var generate bool
497497 for _ , method := range service .Methods {
@@ -574,7 +574,8 @@ func (g *Generator) generateClientFuncSignature(gfile *protogen.GeneratedFile, s
574574 args = append (args , "*" , gfile .QualifiedGoIdent (method .Output .GoIdent ))
575575 } else {
576576 // TODO
577- args = append (args , gfile .QualifiedGoIdent (protogen.GoIdent {GoName : serviceName + "_" + method .GoName + "Client" , GoImportPath : file .GoImportPath }))
577+ // args = append(args, gfile.QualifiedGoIdent(protogen.GoIdent{GoName: serviceName + "_" + method.GoName + "Client", GoImportPath: method.Output.GoIdent.GoImportPath}))
578+ args = append (args , serviceName + "_" + method .GoName + "Client" )
578579 }
579580 args = append (args , ", error) {" )
580581 gfile .P (args ... )
0 commit comments