fusen/proto
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

支持 stream 序列化 删除gateway的一些垃圾代码

Browse Source
This commit is contained in:
huangsimin@fusen.cn
2023-12-07 15:34:40 +08:00
parent a2873f273b
commit 2ed4499221
6 changed files with 94 additions and 66 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
goutils/proto_build/tpls/auto_grpc_gateway_nacos.tpl
View File

@@ -1,10 +0,0 @@
package service
import (
"context"
"net/http"
"github.com/grpc-ecosystem/grpc-gateway/v2/runtime"
"google.golang.org/grpc"
)

20
goutils/proto_build/tpls/auto_grpc_gateway_register.tpl
View File

@@ -1,20 +0,0 @@
package logic
import (
"context"
"{{.ProjectName}}/gen/go/service"
"github.com/grpc-ecosystem/grpc-gateway/v2/runtime"
"google.golang.org/grpc"
)
func AutoRegisterHandler(ctx context.Context, mux *runtime.ServeMux, opts ...grpc.DialOption) error {
var err error
{{range .FuncNames}}
err = service.{{.}}(ctx, mux, opts...)
if err != nil {
return err
}
{{end}}
return nil
}

15
goutils/proto_build/tpls/logic_fusen_handler.tpl
View File

@@ -1,10 +1,23 @@
package {{.PackageName}}
import (
{{if eq .MethodType "rpc"}}
"context"
{{end}}
"{{.ProjectName}}/gen/go/service"
)
{{if eq .MethodType "rpc"}}
func (l *{{.StructName}}) {{.MethodName}}Logic({{range $index, $param := .Params}}{{if $index}}, {{end}}{{$param}}{{end}}) (resp {{.MethodReturn}},err error) {
return resp, err
}
}
{{else if eq .MethodType "stream"}}
func (l *{{.StructName}}) {{.MethodName}}Logic(stream {{range $index, $param := .Params}}{{if $index}}, {{end}}{{$param}}{{end}}) (err error) {
return err
}
{{end}}

10
goutils/proto_build/tpls/logic_grpc_struct.tpl
View File

@@ -45,9 +45,19 @@ type {{.MethodName}}Handler struct {
type {{.MethodName}}HandlerMust struct{}
{{if eq .MethodType "rpc"}}
func (lgrpc *{{.StructName}}Grpc) {{.MethodName}}({{range $index, $param := .Params}}{{if $index}}, {{end}}{{$param}}{{end}}) ({{.MethodReturn}}, error) {
return New{{.StructName}}(ctx).{{.MethodName}}Logic({{range $index, $param := .ParamsName}}{{if $index}}, {{end}}{{$param}}{{end}})
}
{{else if eq .MethodType "stream"}}
func (lgrpc *{{.StructName}}Grpc) {{.MethodName}}(stream {{range $index, $param := .Params}}{{if $index}}, {{end}}{{$param}}{{end}}) error {
return New{{.StructName}}(stream.Context()).{{.MethodName}}Logic(stream)
}
{{end}}
{{end}}