瀏覽代碼

fix context missing (#872)

Co-authored-by: suzhenpeng <suzhenpeng@ecoplants.tech>
neosu 3 年之前
父節點
當前提交
16cc990fdd
共有 1 個文件被更改,包括 13 次插入4 次删除
  1. 13 4
      tools/goctl/rpc/generator/genserver.go

+ 13 - 4
tools/goctl/rpc/generator/genserver.go

@@ -73,11 +73,20 @@ func (g *DefaultGenerator) GenServer(ctx DirContext, proto parser.Proto, cfg *co
 		return err
 	}
 
+	notStream := false
+	for _, rpc := range service.RPC {
+		if !rpc.StreamsRequest && !rpc.StreamsReturns {
+			notStream = true
+			break
+		}
+	}
+
 	err = util.With("server").GoFmt(true).Parse(text).SaveTo(map[string]interface{}{
-		"head":    head,
-		"server":  stringx.From(service.Name).ToCamel(),
-		"imports": strings.Join(imports.KeysStr(), util.NL),
-		"funcs":   strings.Join(funcList, util.NL),
+		"head":      head,
+		"server":    stringx.From(service.Name).ToCamel(),
+		"imports":   strings.Join(imports.KeysStr(), util.NL),
+		"funcs":     strings.Join(funcList, util.NL),
+		"notStream": notStream,
 	}, serverFile, true)
 	return err
 }