fix templates

Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org>
This commit is contained in:
Василий Толстов 2021-02-19 16:56:19 +03:00
parent 2583a8e29b
commit a4bed53e0f
3 changed files with 13 additions and 12 deletions

File diff suppressed because one or more lines are too long

View File

@ -5,6 +5,7 @@ package {{goPkgLastElement .File | splitArray ";" | last | replace "." "_"}}
import ( import (
"context" "context"
micro_api "github.com/unistack-org/micro/v3/api"
micro_client "github.com/unistack-org/micro/v3/client" micro_client "github.com/unistack-org/micro/v3/client"
micro_server "github.com/unistack-org/micro/v3/server" micro_server "github.com/unistack-org/micro/v3/server"
) )
@ -100,7 +101,7 @@ func (x *{{$ServiceName | lowerFirst}}Service{{.Name}}) RecvMsg(m interface{}) e
func (x *{{$ServiceName | lowerFirst}}Service{{.Name}}) Send(m *{{$reqMethod}}) error { func (x *{{$ServiceName | lowerFirst}}Service{{.Name}}) Send(m *{{$reqMethod}}) error {
return x.stream.Send(m) return x.stream.Send(m)
} }
{{- end}} {{end}}
{{ if .ServerStreaming -}} {{ if .ServerStreaming -}}
func (x *{{$ServiceName | lowerFirst}}Service{{.Name}}) Recv() (*{{$rspMethod}}, error) { func (x *{{$ServiceName | lowerFirst}}Service{{.Name}}) Recv() (*{{$rspMethod}}, error) {
@ -110,10 +111,10 @@ func (x *{{$ServiceName | lowerFirst}}Service{{.Name}}) Recv() (*{{$rspMethod}},
} }
return m, nil return m, nil
} }
{{- end }} {{ end }}
{{- end}} {{- end}}
{{- end -}} {{- end}}
// Micro server stuff // Micro server stuff
@ -186,8 +187,8 @@ func (x *{{$ServiceName | lowerFirst}}{{.Name}}Stream) Recv() (*{{$reqMethod}},
} }
{{end}} {{end}}
{{- end}} {{end}}
{{- end}} {{end}}
// Register{{$ServiceName}}Handler registers server handler // Register{{$ServiceName}}Handler registers server handler
func Register{{$ServiceName}}Handler(s micro_server.Server, sh {{$ServiceName}}Handler, opts ...micro_server.HandlerOption) error { func Register{{$ServiceName}}Handler(s micro_server.Server, sh {{$ServiceName}}Handler, opts ...micro_server.HandlerOption) error {

View File

@ -151,7 +151,7 @@ func (x *{{$ServiceName | lowerFirst}}Service{{.Name}}) Recv() (*{{$rspMethod}},
{{- end }} {{- end }}
{{- end}} {{- end}}
{{- end -}} {{end}}
// Micro server stuff // Micro server stuff