You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
94 lines
2.5 KiB
94 lines
2.5 KiB
2 years ago
|
{{$svrType := .ServiceType}}
|
||
|
|
{{$svrName := .ServiceName}}
|
||
|
|
|
||
|
|
{{- range .MethodSets}}
|
||
|
|
const Operation{{$svrType}}{{.OriginalName}} = "/{{$svrName}}/{{.OriginalName}}"
|
||
|
|
{{- end}}
|
||
|
|
|
||
|
|
type {{.ServiceType}}HTTPServer interface {
|
||
|
|
{{- range .MethodSets}}
|
||
|
|
{{- if ne .Comment ""}}
|
||
|
|
{{.Comment}}
|
||
|
|
{{- end}}
|
||
|
|
{{.Name}}(context.Context, *{{.Request}}) (*{{.Reply}}, error)
|
||
|
|
{{- end}}
|
||
|
|
}
|
||
|
|
|
||
|
|
func Register{{.ServiceType}}HTTPServer(s *http.Server, srv {{.ServiceType}}HTTPServer) {
|
||
|
|
r := s.Route("/")
|
||
|
|
{{- range .Methods}}
|
||
|
|
r.{{.Method}}("{{.Path}}", _{{$svrType}}_{{.Name}}{{.Num}}_HTTP_Handler(srv))
|
||
|
|
{{- end}}
|
||
|
|
}
|
||
|
|
|
||
|
|
{{range .Methods}}
|
||
|
|
func _{{$svrType}}_{{.Name}}{{.Num}}_HTTP_Handler(srv {{$svrType}}HTTPServer) func(ctx http.Context) error {
|
||
|
|
return func(ctx http.Context) error {
|
||
|
|
var in {{.Request}}
|
||
|
|
{{- if .HasBody}}
|
||
|
|
if err := ctx.Bind(&in{{.Body}}); err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
{{- if not (eq .Body "")}}
|
||
|
|
if err := ctx.BindQuery(&in); err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
{{- end}}
|
||
|
|
{{- else}}
|
||
|
|
if err := ctx.BindQuery(&in{{.Body}}); err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
{{- end}}
|
||
|
|
{{- if .HasVars}}
|
||
|
|
if err := ctx.BindVars(&in); err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
{{- end}}
|
||
|
|
http.SetOperation(ctx,Operation{{$svrType}}{{.OriginalName}})
|
||
|
|
h := ctx.Middleware(func(ctx context.Context, req interface{}) (interface{}, error) {
|
||
|
|
return srv.{{.Name}}(ctx, req.(*{{.Request}}))
|
||
|
|
})
|
||
|
|
out, err := h(ctx, &in)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
reply := out.(*{{.Reply}})
|
||
|
|
return ctx.Result(200, reply{{.ResponseBody}})
|
||
|
|
}
|
||
|
|
}
|
||
|
|
{{end}}
|
||
|
|
|
||
|
|
type {{.ServiceType}}HTTPClient interface {
|
||
|
|
{{- range .MethodSets}}
|
||
|
|
{{.Name}}(ctx context.Context, req *{{.Request}}, opts ...http.CallOption) (rsp *{{.Reply}}, err error)
|
||
|
|
{{- end}}
|
||
|
|
}
|
||
|
|
|
||
|
|
type {{.ServiceType}}HTTPClientImpl struct{
|
||
|
|
cc *http.Client
|
||
|
|
}
|
||
|
|
|
||
|
|
func New{{.ServiceType}}HTTPClient (client *http.Client) {{.ServiceType}}HTTPClient {
|
||
|
|
return &{{.ServiceType}}HTTPClientImpl{client}
|
||
|
|
}
|
||
|
|
|
||
|
|
{{range .MethodSets}}
|
||
|
|
func (c *{{$svrType}}HTTPClientImpl) {{.Name}}(ctx context.Context, in *{{.Request}}, opts ...http.CallOption) (*{{.Reply}}, error) {
|
||
|
|
var out {{.Reply}}
|
||
|
|
pattern := "{{.Path}}"
|
||
|
|
path := binding.EncodeURL(pattern, in, {{not .HasBody}})
|
||
|
|
opts = append(opts, http.Operation(Operation{{$svrType}}{{.OriginalName}}))
|
||
|
|
opts = append(opts, http.PathTemplate(pattern))
|
||
|
|
{{if .HasBody -}}
|
||
|
|
err := c.cc.Invoke(ctx, "{{.Method}}", path, in{{.Body}}, &out{{.ResponseBody}}, opts...)
|
||
|
|
{{else -}}
|
||
|
|
err := c.cc.Invoke(ctx, "{{.Method}}", path, nil, &out{{.ResponseBody}}, opts...)
|
||
|
|
{{end -}}
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
return &out, err
|
||
|
|
}
|
||
|
|
{{end}}
|