fix context missing (#872)

Co-authored-by: suzhenpeng <suzhenpeng@ecoplants.tech>
master
neosu 3 years ago committed by GitHub
parent 00061c2e5b
commit 16cc990fdd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -73,11 +73,20 @@ func (g *DefaultGenerator) GenServer(ctx DirContext, proto parser.Proto, cfg *co
return err 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{}{ err = util.With("server").GoFmt(true).Parse(text).SaveTo(map[string]interface{}{
"head": head, "head": head,
"server": stringx.From(service.Name).ToCamel(), "server": stringx.From(service.Name).ToCamel(),
"imports": strings.Join(imports.KeysStr(), util.NL), "imports": strings.Join(imports.KeysStr(), util.NL),
"funcs": strings.Join(funcList, util.NL), "funcs": strings.Join(funcList, util.NL),
"notStream": notStream,
}, serverFile, true) }, serverFile, true)
return err return err
} }

Loading…
Cancel
Save