fusen/proto
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

修复stream中的错误

Browse Source
This commit is contained in:
huangsimin@fusen.cn 2023-12-07 16:20:19 +08:00
parent 76a781d006
commit 61c10dab23
1 changed files with 9 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
goutils/proto_build/main.go
View File

@ -1082,19 +1082,17 @@ func parseGoFile(filePath string) (ClientParams []*ClientParam) {
} else if iface, ok := typeSpec.Type.(*ast.InterfaceType); ok { } else if iface, ok := typeSpec.Type.(*ast.InterfaceType); ok {
interfaceName := typeSpec.Name.Name interfaceName := typeSpec.Name.Name
if len(iface.Methods.List) == 3 { if len(iface.Methods.List) >= 3 {
if len(iface.Methods.List[0].Names) == 1 && len(iface.Methods.List[1].Names) == 1 { if len(iface.Methods.List[2].Names) == 0 {
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)
continue // log.Println(name0, name1)
} if getTypeString(iface.Methods.List[2].Type) == "grpc.ClientStream" {
// MethodType = "stream"
// log.Println(interfaceName)
continue
} }
} }
} }