Skip to content

Commit 3d92ee0

Browse files
committed
standalone checks
Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org>
1 parent 750c949 commit 3d92ee0

File tree

1 file changed

+17
-16
lines changed

1 file changed

+17
-16
lines changed

util.go

Lines changed: 17 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -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,
326326
func (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

488488
func (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

Comments
 (0)