github.com/alimy/mir/v4@v4.1.0/internal/generator/templates/hertz_iface.tmpl

// Code generated by go-mir. DO NOT EDIT.
// versions:
// - mir {{ .VerInfo.MirVer }}

package {{ .PkgName }}

import (
	"context"
    "net/http"

    "github.com/alimy/mir/v4"
    "{{if notEmptyStr .EngineInfo.PkgName }}{{ .EngineInfo.PkgName }}{{else}}github.com/cloudwego/hertz{{end}}/pkg/app"
	"{{if notEmptyStr .EngineInfo.PkgName }}{{ .EngineInfo.PkgName }}{{else}}github.com/cloudwego/hertz{{end}}/pkg/route"
	{{ declareImports .Imports }}
)
{{- if .DeclareCoreInterface }}
type _binding_ interface {
	Bind(context.Context, *app.RequestContext) mir.Error
}

type _render_ interface {
	Render(context.Context, *app.RequestContext)
}

type _default_ interface {
	Bind(context.Context, *app.RequestContext, any) mir.Error
	Render(context.Context, *app.RequestContext, any, mir.Error)
}
{{- end }}

{{ declareTypes .InOuts .PkgPath .Imports }}

{{if notEmptyStr .Comment }}// {{.Comment}}{{end}}
type {{.TypeName}} interface {
	_default_

{{if notEmptyStr .Chain }}// Chain provide handlers chain for hertz
    {{.Chain}}() []app.HandlerFunc
{{end}}
{{range .Fields}}    {{if .JustUseContext }}{{ .MethodName}}(context.Context, *app.RequestContext){{else}}{{.MethodName}}({{if .IsUseContext }}context.Context, *app.RequestContext{{if notEmptyStr .InName }}, {{end}}{{end}}{{if notEmptyStr .InName }}*{{ .InName }}{{end}}) {{if notEmptyStr .OutName }}(*{{ .OutName}}, mir.Error){{else}}mir.Error{{end}}{{end}}
{{end}}

    mustEmbedUnimplemented{{.TypeName}}Servant()
}
{{if .IsUseFieldChain }}
type {{.TypeName}}Chain interface {
{{range .ChainFields }}    Chain{{.MethodName}}() []app.HandlerFunc
{{end}}

	mustEmbedUnimplemented{{.TypeName}}Chain()
}
{{end}}

// Register{{.TypeName}}Servant register {{.TypeName}} servant to hertz
func Register{{.TypeName}}Servant(e *route.Engine, s {{.TypeName}}{{if .IsUseFieldChain }},  m ...{{.TypeName}}Chain{{end}}) {
{{- if .IsUseFieldChain -}}
	var cc {{.TypeName}}Chain
	if len(m) > 0 {
		cc = m[0]
	} else {
		cc = &Unimplemented{{.TypeName}}Chain{}
	}
{{- end}}
{{if notEmptyStr .Group }}    router := e.Group("{{.Group}}"){{else}}    router := e{{end}}
{{if notEmptyStr .Chain }}    // use chain for router
    middlewares := s.{{.Chain}}()
    router.Use(middlewares...)
{{end}}
    // register routes info to router
{{range .Fields}}{{if .NotHttpAny }}    router.Handle("{{.HttpMethod}}", "{{.Path}}", {{if .IsFieldChain }}append(cc.Chain{{.MethodName}}(), {{end}}{{if .JustUseContext}}s.{{ .MethodName}}{{else}}func(c context.Context, ctx *app.RequestContext) {
	{{- if $.WatchCtxDone }}
		select {
		case <- c.Done():
			return
		default:
		}
	{{end -}}
	{{if notEmptyStr .InName -}}
	req := new({{.InName}})
	{{if .IsBindIn -}}
	var bv _binding_ = req
	if err := bv.Bind(c, ctx); err != nil {
	{{- else -}}
	if err := s.Bind(c, ctx, req); err != nil {
	{{- end }}
		s.Render(c, ctx, nil, err)
		return
	}
	{{- end }}
	{{if notEmptyStr .OutName -}}
	resp, err := s.{{ .MethodName}}({{if .IsUseContext }}c, ctx{{if notEmptyStr .InName }}, {{end}}{{end}}{{if notEmptyStr .InName}}req{{end}})
	{{if .IsRenderOut -}}
	if err != nil {
		s.Render(c, ctx, nil, err)
		return
	}
	var rv _render_ = resp
	rv.Render(c, ctx)
	{{- else -}}
	s.Render(c, ctx, resp, err)
	{{- end }}
	{{- else -}}
	s.Render(c, ctx, nil, s.{{.MethodName}}({{if .IsUseContext }}c, ctx{{if notEmptyStr .InName }}, {{end}}{{end}}{{if notEmptyStr .InName}}req{{end}}))
	{{- end }}
	}{{end}}{{if .IsFieldChain }})...{{end}})
	{{else if .JustHttpAny}}    router.Any("{{.Path}}", {{if .IsFieldChain }}append(cc.Chain{{.MethodName}}(), {{end}}{{if .JustUseContext}}s.{{ .MethodName}}{{else}}func(c context.Context, ctx *app.RequestContext) {
	{{- if $.WatchCtxDone }}
		select {
		case <- c.Done():
			return
		default:
		}
	{{end -}}
	{{if notEmptyStr .InName -}}
	req := new({{.InName}})
	{{if .IsBindIn -}}
	var bv _binding_ = req
	if err := bv.Bind(c, ctx); err != nil {
	{{- else -}}
	if err := s.Bind(c, ctx, req); err != nil {
	{{- end }}
		s.Render(c, ctx, nil, err)
		return
	}
	{{- end }}
	{{if notEmptyStr .OutName -}}
	resp, err := s.{{ .MethodName}}({{if .IsUseContext }}c, ctx{{if notEmptyStr .InName }}, {{end}}{{end}}{{if notEmptyStr .InName}}req{{end}})
	{{if .IsRenderOut -}}
	if err != nil {
		s.Render(c, ctx, nil, err)
		return
	}
	var rv _render_ = resp
	rv.Render(c, ctx)
	{{- else -}}
	s.Render(c, ctx, resp, err)
	{{- end }}
	{{- else -}}
	s.Render(c, ctx, nil, s.{{.MethodName}}({{if .IsUseContext }}c, ctx{{if notEmptyStr .InName }}, {{end}}{{end}}{{if notEmptyStr .InName}}req{{end}}))
	{{- end }}
	}{{end}}{{if .IsFieldChain }})...{{end}})
	{{else}}{{$field := .}}    {
        	h := {{if .IsFieldChain }}append(cc.Chain{{.MethodName}}(), {{end}}{{if .JustUseContext}}s.{{ .MethodName}}{{else}}func(c context.Context, ctx *app.RequestContext) {
				{{- if $.WatchCtxDone }}
				select {
				case <- c.Done():
					return
				default:
				}
				{{end -}}
	{{if notEmptyStr .InName -}}
	req := new({{.InName}})
	{{if .IsBindIn -}}
	var bv _binding_ = req
	if err := bv.Bind(c, ctx); err != nil {
	{{- else -}}
	if err := s.Bind(c, ctx, req); err != nil {
	{{- end }}
		s.Render(c, ctx, nil, err)
		return
	}
	{{- end }}
	{{if notEmptyStr .OutName -}}
	resp, err := s.{{ .MethodName}}({{if .IsUseContext }}c, ctx{{if notEmptyStr .InName }}, {{end}}{{end}}{{if notEmptyStr .InName}}req{{end}})
	{{if .IsRenderOut -}}
	if err != nil {
		s.Render(c, ctx, nil, err)
		return
	}
	var rv _render_ = resp
	rv.Render(c, ctx)
	{{- else -}}
	s.Render(c, ctx, resp, err)
	{{- end }}
	{{- else -}}
	s.Render(c, ctx, nil, s.{{.MethodName}}({{if .IsUseContext }}c, ctx{{if notEmptyStr .InName }}, {{end}}{{end}}{{if notEmptyStr .InName}}req{{end}}))
	{{- end }}
	    }{{end}}{{if .IsFieldChain }}){{end}}
		{{range .AnyHttpMethods}}        router.Handle("{{.}}", "{{$field.Path}}", h{{if $field.IsFieldChain }}...{{end}})
		{{end}}    }
	{{end}}
{{- end -}}
}

{{ $unimplementedServant := print "Unimplemented" .TypeName "Servant" }}
// {{$unimplementedServant}} can be embedded to have forward compatible implementations.
type {{$unimplementedServant}} struct {}

{{if notEmptyStr .Chain }}
func ({{$unimplementedServant}}){{.Chain}}() []app.HandlerFunc {
    return nil
}

{{end}}
{{range .Fields}}
func ({{$unimplementedServant}}){{if .JustUseContext}}{{ .MethodName}}(c context.Context, ctx *app.RequestContext){{else}}{{.MethodName}}({{if .IsUseContext }}c context.Context, ctx *app.RequestContext{{if notEmptyStr .InName }}, {{end}}{{end}}{{if notEmptyStr .InName }}req *{{ .InName }}{{end}}) {{if notEmptyStr .OutName }}(*{{ .OutName}}, mir.Error){{else}}mir.Error{{end}}{{end}} {
	{{if .JustUseContext -}}
		ctx.String(http.StatusNotImplemented, http.StatusText(http.StatusNotImplemented))
	{{else -}}
	return {{if notEmptyStr .OutName }}nil, {{end}}mir.Errorln(http.StatusNotImplemented, http.StatusText(http.StatusNotImplemented))
	{{end -}}
}

{{end}}
func ({{$unimplementedServant}})mustEmbedUnimplemented{{.TypeName}}Servant() {}
{{if .IsUseFieldChain }}
{{ $unimplementedChain := print "Unimplemented" .TypeName "Chain" }}
// {{$unimplementedChain}} can be embedded to have forward compatible implementations.
type {{$unimplementedChain}} struct {
}

{{range .Fields}}{{if .IsFieldChain }}func (b *{{$unimplementedChain}}) Chain{{.MethodName}}() []app.HandlerFunc {
	return nil
}
{{end}}
{{end}}
func (b *{{$unimplementedChain}})mustEmbedUnimplemented{{.TypeName}}Chain() {}
{{end}}