From 16cc990fdda29a3926393cf34e8d6e83171d5bf9 Mon Sep 17 00:00:00 2001
From: neosu <280097765@qq.com>
Date: Wed, 4 Aug 2021 17:46:51 +0800
Subject: [PATCH] fix context missing (#872)

Co-authored-by: suzhenpeng <suzhenpeng@ecoplants.tech>
---
 tools/goctl/rpc/generator/genserver.go | 17 +++++++++++++----
 1 file changed, 13 insertions(+), 4 deletions(-)

diff --git a/tools/goctl/rpc/generator/genserver.go b/tools/goctl/rpc/generator/genserver.go
index 180ab789..fb5aba0b 100644
--- a/tools/goctl/rpc/generator/genserver.go
+++ b/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
 }