Merge pull request #341 from TarsCloud/fix/lbbniu/trace

修复接口返回void生成代码错误
TarsCloud · Mar 10, 2022 · 3299ba4 · 3299ba4
2 parents abcf0c6 + 649f866
commit 3299ba4
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tars/tools/tars2go/gen_go.go b/tars/tools/tars2go/gen_go.go
@@ -1291,10 +1291,14 @@ _resp := new(requestf.ResponsePacket)
 			}
 		}
 		if withContext && !withoutTrace {
+			traceParamFlag := "traceParamFlag := traceData.NeedTraceParam(trace.EstCR, uint(0))"
+			if isOut || fun.HasRet {
+				traceParamFlag = "traceParamFlag := traceData.NeedTraceParam(trace.EstCR, uint(_is.Len()))"
+			}
 			c.WriteString(`
 if ok && traceData.TraceCall {
 	var traceParam string
-	traceParamFlag := traceData.NeedTraceParam(trace.EstCR, uint(_is.Len()))
+	` + traceParamFlag + `
 	if traceParamFlag == trace.EnpNormal {
 		value := map[string]interface{}{}
 `)