From 61c10dab237d619793ed02b6f29f2da7e9f57e76 Mon Sep 17 00:00:00 2001 From: "huangsimin@fusen.cn" Date: Thu, 7 Dec 2023 16:20:19 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8Dstream=E4=B8=AD=E7=9A=84?= =?UTF-8?q?=E9=94=99=E8=AF=AF?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- goutils/proto_build/main.go | 20 +++++++++----------- 1 file changed, 9 insertions(+), 11 deletions(-) diff --git a/goutils/proto_build/main.go b/goutils/proto_build/main.go index 87f296e..1c8bde7 100644 --- a/goutils/proto_build/main.go +++ b/goutils/proto_build/main.go @@ -1082,19 +1082,17 @@ func parseGoFile(filePath string) (ClientParams []*ClientParam) { } else if iface, ok := typeSpec.Type.(*ast.InterfaceType); ok { interfaceName := typeSpec.Name.Name - if len(iface.Methods.List) == 3 { - if len(iface.Methods.List[0].Names) == 1 && len(iface.Methods.List[1].Names) == 1 { - name0 := iface.Methods.List[0].Names[0].Name - name1 := iface.Methods.List[1].Names[0].Name - if name0 == "Send" && name1 == "CloseAndRecv" { - // log.Println(name0, name1) - if getTypeString(iface.Methods.List[2].Type) == "grpc.ClientStream" { - // MethodType = "stream" - // log.Println(interfaceName) + if len(iface.Methods.List) >= 3 { + if len(iface.Methods.List[2].Names) == 0 { - continue - } + // log.Println(name0, name1) + if getTypeString(iface.Methods.List[2].Type) == "grpc.ClientStream" { + // MethodType = "stream" + // log.Println(interfaceName) + + continue } + } }