feat(jzero): ivm init support stream

.template/ivm/init/logic-body.tpl

@@ -0,0 +1,24 @@
+    logic := {{ .OldService }}logic.New{{ .LogicTypeName | FirstUpper }}(l.ctx, l.svcCtx)
+    marshal, err := proto.Marshal(in)
+    if err != nil {
+    	return nil, err
+    }
+    var oldIn {{ .OldService }}pb.{{ .RequestTypeName }}
+    err = proto.Unmarshal(marshal, &oldIn)
+    if err != nil {
+    	return nil, err
+    }
+    result, err := logic.{{ .MethodName }}(&oldIn)
+    if err != nil {
+    	return nil, err
+    }
+    marshal, err = proto.Marshal(result)
+    if err != nil {
+    	return nil, err
+    }
+    var newResp {{ .Service }}pb.{{ .ResponseTypeName }}
+    err = proto.Unmarshal(marshal, &newResp)
+    if err != nil {
+    	return nil, err
+    }
+    return &newResp, nil

.template/ivm/init/logic-client-server-stream-adaptor.tpl

@@ -0,0 +1,49 @@
+type {{ .MethodName | FirstUpper }}ServerAdapter struct {
+	{{ .Service }}pb.{{ .Service | FirstUpper }}_{{ .MethodName | FirstUpper }}Server
+}
+
+func (s {{ .MethodName | FirstUpper }}ServerAdapter) Send(response *{{ .OldService }}pb.{{ .ResponseTypeName }}) error {
+	marshal, err := proto.Marshal(response)
+	if err != nil {
+		return err
+	}
+	var newResp {{ .Service }}pb.{{ .ResponseTypeName }}
+	err = proto.Unmarshal(marshal, &newResp)
+	if err != nil {
+		return err
+	}
+
+	err = s.SendMsg(&newResp)
+	if err != nil {
+		if err == io.EOF {
+			return nil
+		}
+		return err
+	}
+
+	return nil
+}
+
+func (s {{ .MethodName | FirstUpper }}ServerAdapter) Recv() (*{{ .OldService }}pb.{{ .RequestTypeName }}, error) {
+	for {
+		newIn, err := s.{{ .Service | FirstUpper }}_{{ .MethodName | FirstUpper }}Server.Recv()
+		if err == io.EOF {
+			return nil, io.EOF
+		}
+		if err != nil {
+			return nil, err
+		}
+
+		marshal, err := proto.Marshal(newIn)
+		if err != nil {
+			return nil, err
+		}
+		var oldIn {{ .OldService }}pb.{{ .RequestTypeName }}
+		err = proto.Unmarshal(marshal, &oldIn)
+		if err != nil {
+			return nil, err
+		}
+
+		return &oldIn, nil
+	}
+}

.template/ivm/init/logic-client-server-stream-body.tpl

@@ -0,0 +1,8 @@
+logic := {{ .OldService }}logic.New{{ .LogicTypeName }}(l.ctx, l.svcCtx)
+
+// Create the adapter
+adapter := &{{ .MethodName | FirstUpper }}ServerAdapter{
+    stream,
+}
+
+return logic.{{ .MethodName | FirstUpper }}(adapter)

.template/ivm/init/logic-client-stream-adaptor.tpl

@@ -0,0 +1,41 @@
+type {{ .MethodName | FirstUpper }}ServerAdapter struct {
+	{{ .Service }}pb.{{ .Service | FirstUpper }}_{{ .MethodName | FirstUpper }}Server
+}
+
+func (s *{{ .MethodName | FirstUpper }}ServerAdapter) SendAndClose(response *{{ .OldService }}pb.{{ .ResponseTypeName }}) error {
+	marshal, err := proto.Marshal(response)
+	if err != nil {
+		return err
+	}
+
+	var newResp {{ .Service }}pb.{{ .ResponseTypeName }}
+	err = proto.Unmarshal(marshal, &newResp)
+	if err != nil {
+		return err
+	}
+	return s.SendMsg(&newResp)
+}
+
+func (s *{{ .MethodName | FirstUpper }}ServerAdapter) Recv() (*{{ .OldService }}pb.{{ .RequestTypeName }}, error) {
+	for {
+		newIn, err := s.{{ .Service | FirstUpper }}_{{ .MethodName | FirstUpper }}Server.Recv()
+		if err == io.EOF {
+			return nil, io.EOF
+		}
+		if err != nil {
+			return nil, err
+		}
+
+		marshal, err := proto.Marshal(newIn)
+		if err != nil {
+			return nil, err
+		}
+		var oldIn {{ .OldService }}pb.{{ .RequestTypeName }}
+		err = proto.Unmarshal(marshal, &oldIn)
+		if err != nil {
+			return nil, err
+		}
+
+		return &oldIn, nil
+	}
+}

.template/ivm/init/logic-client-stream-body.tpl

@@ -0,0 +1,8 @@
+logic := {{ .OldService }}logic.New{{ .LogicTypeName }}(l.ctx, l.svcCtx)
+
+// Create the adapter
+adapter := &{{ .MethodName | FirstUpper }}ServerAdapter{
+    stream,
+}
+
+return logic.{{ .MethodName | FirstUpper }}(adapter)

.template/ivm/init/logic-server-stream-adaptor.tpl

@@ -0,0 +1,25 @@
+type {{ .MethodName | FirstUpper }}ServerAdapter struct {
+	{{ .Service }}pb.{{ .Service | FirstUpper }}_{{ .MethodName | FirstUpper }}Server
+}
+
+func (s *{{ .MethodName | FirstUpper }}ServerAdapter) Send(response *{{ .OldService }}pb.{{ .ResponseTypeName }}) error {
+	marshal, err := proto.Marshal(response)
+	if err != nil {
+		return err
+	}
+	var newResp {{ .Service }}pb.{{ .ResponseTypeName }}
+	err = proto.Unmarshal(marshal, &newResp)
+	if err != nil {
+		return err
+	}
+
+	err = s.SendMsg(&newResp)
+	if err != nil {
+		if err == io.EOF {
+			return nil
+		}
+		return err
+	}
+
+	return nil
+}

.template/ivm/init/logic-server-stream-body.tpl

@@ -0,0 +1,19 @@
+logic := {{ .OldService }}logic.New{{ .LogicTypeName }}(l.ctx, l.svcCtx)
+
+// Create the adapter
+adapter := &{{ .MethodName | FirstUpper }}ServerAdapter{
+	stream,
+}
+
+marshal, err := proto.Marshal(in)
+if err != nil {
+	return err
+}
+
+var oldIn {{ .OldService }}pb.{{ .RequestTypeName }}
+err = proto.Unmarshal(marshal, &oldIn)
+if err != nil {
+	return err
+}
+
+return logic.{{ .MethodName | FirstUpper }}(&oldIn, adapter)

internal/gen/genapi.go

@@ -45,8 +45,10 @@ type LogicFile struct {
 	Handler string
 	Path    string
 
-	ClientStream bool
-	ServerStream bool
+	RequestTypeName  string
+	ResponseTypeName string
+	ClientStream     bool
+	ServerStream     bool
 }
 
 func (ja *JzeroApi) Gen() error {

internal/gen/genrpc.go

@@ -232,11 +232,13 @@ func (jr *JzeroRpc) GetAllLogicFiles(protoSpec rpcparser.Proto) ([]LogicFile, er
 			fp := filepath.Join(jr.Wd, "internal", "logic", strings.ToLower(service.Name), namingFormat+".go")
 
 			f := LogicFile{
-				Path:         fp,
-				Handler:      rpc.Name,
-				Group:        service.Name,
-				ClientStream: rpc.StreamsRequest,
-				ServerStream: rpc.StreamsReturns,
+				Path:             fp,
+				Handler:          rpc.Name,
+				Group:            service.Name,
+				ClientStream:     rpc.StreamsRequest,
+				ServerStream:     rpc.StreamsReturns,
+				ResponseTypeName: rpc.ReturnsType,
+				RequestTypeName:  rpc.RequestType,
 			}
 
 			logicFiles = append(logicFiles, f)

internal/ivm/ivminit/init.go

@@ -98,7 +98,7 @@ func Init(command *cobra.Command, args []string) error {
 
 	for i, fp := range newVersionProtoFilepath {
 		oldFp := protoFiles[i]
-		err = ivmInit.setUpdateProtoLogic(fp, oldFp)
+		err = ivmInit.updateProtoLogic(fp, oldFp)
 		if err != nil {
 			return err
 		}