@@ -144,6 +144,7 @@ func newServiceInfo(hasStreaming bool, keepStreamingMethods bool, keepNonStreami
144144 return svcInfo
145145}
146146
147+ {{- $referToKitexServerInterface := and .StreamX .HasStreaming}}
147148{{range .AllMethods}}
148149{{- $isStreaming := or .ClientStreaming .ServerStreaming}}
149150{{- $streamingUnary := (eq .StreamingMode "unary")}}
@@ -153,6 +154,10 @@ func newServiceInfo(hasStreaming bool, keepStreamingMethods bool, keepNonStreami
153154{{- $bidiSide := and .ClientStreaming .ServerStreaming}}
154155{{- $arg := ""}}
155156{{- $handlerFunc := ""}}
157+ {{- $serverInterfaceName := printf "%s.%s" .PkgRefName .ServiceName }}
158+ {{- if $referToKitexServerInterface}}
159+ {{- $serverInterfaceName = .ServiceName}}{{/* when streamx is enabled and there are streaming methods, refer to Server Interface defined in service/server.go */}}
160+ {{- end}}
156161{{- $mode := ""}}
157162 {{- if $streamingUnary -}} {{- $mode = "serviceinfo.StreamingUnary" }} {{- $handlerFunc = "InvokeUnaryHandler" }}
158163 {{- else if $serverSide -}} {{- $mode = "serviceinfo.StreamingServer" }} {{- $handlerFunc = "InvokeServerStreamHandler" }}
@@ -228,8 +233,8 @@ func {{LowerFirst .Name}}Handler(ctx context.Context, handler interface{}, arg,
228233 {{- end}}
229234 {{if gt .ArgsLength 0}}realArg := {{else}}_ = {{end}}arg.(*{{if not .GenArgResultStruct}}{{.PkgRefName}}.{{end}}{{.ArgStructName}})
230235 {{if or (not .Void) .Exceptions}}realResult := result.(*{{if not .GenArgResultStruct}}{{.PkgRefName}}.{{end}}{{.ResStructName}}){{end}}
231- {{if .Void}}err := handler.({{.PkgRefName}}.{{.ServiceName }}).{{.Name}}(ctx{{range .Args}}, realArg.{{.Name}}{{end}})
232- {{else}}success, err := handler.({{.PkgRefName}}.{{.ServiceName }}).{{.Name}}(ctx{{range .Args}}, realArg.{{.Name}}{{end}})
236+ {{if .Void}}err := handler.({{$serverInterfaceName }}).{{.Name}}(ctx{{range .Args}}, realArg.{{.Name}}{{end}})
237+ {{else}}success, err := handler.({{$serverInterfaceName }}).{{.Name}}(ctx{{range .Args}}, realArg.{{.Name}}{{end}})
233238 {{end -}}
234239 if err != nil {
235240 {{- if $HandlerReturnKeepResp }}
0 commit comments