更新

Author: guonaihong

Makefile

@@ -1,4 +1,5 @@
 build:
+	go build ./cmd/h2o/h2o.go
 
 codemsg:
 	go build ./cmd/h2o/h2o.go

cmd/h2o/h2o.go

@@ -8,6 +8,7 @@ import (
 	"github.com/antlabs/h2o/codemsg"
 	"github.com/antlabs/h2o/http"
 	"github.com/antlabs/h2o/pb"
+	"github.com/antlabs/h2o/transport"
 	"github.com/antlabs/tostruct/json"
 	"github.com/antlabs/tostruct/option"
 	"github.com/antlabs/tostruct/yaml"
@@ -77,6 +78,8 @@ type H2O struct {
 	HTTP http.HTTP `clop:"subcommand" usage:"gen http code"`
 	//子命令，生成protobuf
 	PB pb.Pb `clop:"subcommand" usage:"gen protobuf code"`
+	//transport
+	Transport transport.Transport `clop:"subcommand" usage:"transport"`
 }
 
 func main() {

pb/pb.go

@@ -82,7 +82,7 @@ func (b *Pb) SubMain() {
 			var typeOut bytes.Buffer
 			tmplType.Gen(&typeOut)
 			tmplPb.PbType = typeOut.String()
-			tmplPb.Func = append(tmplPb.Func, Func{Name: h.Handler, ReqName: h.Req.Name, RespName: h.Resp.Name})
+			tmplPb.Func = append(tmplPb.Func, Func{RpcName: h.Handler, ReqName: h.Req.Name, RespName: h.Resp.Name})
 
 		}
 

pb/pb.tmpl

@@ -12,7 +12,7 @@ message {{.URLName}} {
 {{- .PbType}}
 service {{.ServiceName}} {
   {{- range $value := .Func}}
-  rpc {{.Name}} ({{.ReqName}}) returns ({{.RespName}});
+  rpc {{.RpcName}} ({{.ReqName}}) returns ({{.RespName}});
   {{- end}}
 }
 

pb/pb_tmpl.go

@@ -25,7 +25,7 @@ type PbTmpl struct {
 }
 
 type Func struct {
-	Name     string //service名
+	RpcName  string //service名
 	ReqName  string //service 请求参数名
 	RespName string //service 响应参数名
 }

transport/gozero_h2o_http_client_logic.go

@@ -0,0 +1,38 @@
+package transport
+
+import (
+	_ "embed"
+	"html/template"
+	"io"
+	//"github.com/Masterminds/sprig"
+)
+
+var (
+	//go:embed gozero_h2o_http_client_logic.tmpl
+	goZeroH2oHTTPClientLogic string
+)
+
+type transportGoZeroHTTPClientTmpl struct {
+	PackageName       string //包名
+	GoZeroBaseURL     string //
+	HTTPClientBaseURL string //
+	Func              Func   //func
+}
+
+type Func struct {
+	RpcName  string //service名
+	ReqName  string //service 请求参数名
+	RespName string //service 响应参数名
+}
+
+func newFuncTemplate() *template.Template {
+	tmpl := goZeroH2oHTTPClientLogic
+	return template.Must(template.New("h2o-transport-gozero-http-client-tmpl").Parse(tmpl))
+}
+
+func (g *transportGoZeroHTTPClientTmpl) Gen(w io.Writer) error {
+
+	tmpl := newFuncTemplate()
+
+	return tmpl.Execute(w, *g)
+}

transport/gozero_h2o_http_client_logic.tmpl

@@ -0,0 +1,41 @@
+package {{.PackageName}}
+
+import (
+	"context"
+
+  "github.com/antlabs/deepcopy"
+  pb{{.PackageName}} "{{.GoZeroBaseURL}}/{{.PackageName}}"
+   "{{.HTTPClientBaseURL}}/{{.PackageName}}"
+	"github.com/zeromicro/go-zero/core/logx"
+)
+
+{{- $RpcName := .Func.RpcName}}
+{{- $ReqName := .Func.ReqName}}
+{{- $RespName := .Func.RespName}}
+type {{$RpcName}}Logic struct {
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+	logx.Logger
+}
+
+func New{{$RpcName}}Logic(ctx context.Context,svcCtx *svc.ServiceContext) *{{$RpcName}} {
+	return &{{$RpcName}}{
+		ctx:    ctx,
+		svcCtx: svcCtx,
+		Logger: logx.WithContext(ctx),
+	}
+}
+
+func (l *{{$RpcName}}Logic) {{$RpcName}} (in *pb{{$ReqName}}) (*pb{{.PackageName}}.{{$RespName}}, error) {
+  var req {{.PackageName}}.{{$RpcName}}Req
+  deepcopy.Copy(&req, in).Do()
+  resp, err := {{.PackageName}}.{{$RpcName}}Ctx(req)
+  if err != nil {
+    return nil, err
+  }
+
+  var out pb{{.PackageName}}{{$RespName}}Resp
+  deepcopy.Copy(&out, resp).Do()
+	return out, nil
+}
+

transport/gozero_h2o_logic.tmpl

@@ -0,0 +1,14 @@
+package transport
+
+import (
+	"os"
+	"testing"
+)
+
+func Test_GoZeroHTTPClient(t *testing.T) {
+	tmpl := transportGoZeroHTTPClientTmpl{PackageName: "users", GoZeroBaseURL: "gitlab.xx/server", HTTPClientBaseURL: "gitlab.xx/client"}
+	tmpl.Func = Func{RpcName: "CreateUser",
+		ReqName:  "CreateUserReq",
+		RespName: "CreateUserResp"}
+	tmpl.Gen(os.Stdout)
+}

transport/gozero_tmpl_test.go

@@ -0,0 +1,85 @@
+package transport
+
+import (
+	"bytes"
+	"fmt"
+	"os"
+
+	"github.com/antlabs/h2o/internal/save"
+	"github.com/antlabs/h2o/parser"
+)
+
+const (
+	goZero     = "go-zero"
+	httpClient = "h2o-http-client"
+)
+
+// 这是主模块
+// TransportTmpl模块，就是选用一种服务端接入数据，然后选用一种客户端把数据发走
+type Transport struct {
+	// 来自哪个服务端
+	FromType string `clop:"long" usage:"server template type" default:"go-zero"`
+	// 推到哪个客户端
+	ToType string `clop:"long" usage:"client template type" default:"h2o-http-client"`
+	// 设置服务端包的基本url
+	FromBaseURL string `clop:"--from-base-url" usage:"from base url"`
+	// 设置客户端包的基本url
+	ToBaseURL string `clop:"--to-base-url" usage:"to base url"`
+	// 解析
+	File []string `clop:"short;long;greedy" usage:"Parsing dst files" valid:"required"`
+	// 生成目录
+	Dir string `clop:"short;long" usage:"gen dir" default:"."`
+}
+
+// 入口
+func (t *Transport) SubMain() {
+
+	for _, f := range t.File {
+
+		//var out bytes.Buffer
+		c, err := parser.Parser(f)
+		if err != nil {
+			fmt.Printf("h2o.HTTP.parser %s\n", err)
+			return
+		}
+
+		for _, h := range c.Muilt {
+
+			h.ModifyHandler()
+
+			//protobuf.Marshal()
+			h.Req.Name, h.Resp.Name = h.GetReqName(), h.GetRespName()
+
+			if h.Handler == "" {
+				panic("h.Handler is empty")
+			}
+
+			if t.FromType == goZero && t.ToType == httpClient {
+
+				tmpl := transportGoZeroHTTPClientTmpl{
+					PackageName:       c.Package,
+					GoZeroBaseURL:     t.FromBaseURL,
+					HTTPClientBaseURL: t.ToBaseURL}
+
+				tmpl.Func = Func{
+					RpcName:  h.Handler,
+					ReqName:  h.Req.Name,
+					RespName: h.Resp.Name}
+
+				save.TmplFile(getLogicName(t.Dir, c.Package, h.Handler), true, func() []byte {
+					var typeBuf bytes.Buffer
+					tmpl.Gen(&typeBuf)
+					return typeBuf.Bytes()
+				})
+			}
+		}
+
+	}
+}
+
+func getLogicName(dir string, packageName string, handler string) string {
+	prefix := fmt.Sprintf("%s/internal/logic/%s/%s", dir, packageName, handler)
+
+	os.MkdirAll(prefix, 0755)
+	return fmt.Sprintf("%s/%s", prefix, handler)
+}