opensource/dtm
3
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

grpc msg ok

Browse Source
This commit is contained in:
yedf2 2021-08-10 14:03:12 +08:00
parent a49b09a981
commit 0bc435e01c
18 changed files with 212 additions and 66 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
dtmcli/types.go
View File

@ -89,6 +89,9 @@ func (tb *TransBase) CallDtm(body interface{}, operation string) error {
// ErrFailure 表示返回失败，要求回滚
var ErrFailure = errors.New("transaction FAILURE")
// ErrPending 表示暂时失败，要求重试
var ErrPending = errors.New("transaction PENDING")
// ResultSuccess 表示返回成功，可以进行下一步
var ResultSuccess = M{"dtm_result": "SUCCESS"}

7
dtmcli/utils.go
View File

@ -264,8 +264,13 @@ func CheckResult(res interface{}, err error) error {
if ok {
return CheckResponse(resp, err)
}
if res != nil && strings.Contains(MustMarshalString(res), "FAILURE") {
if res != nil {
str := MustMarshalString(res)
if strings.Contains(str, "FAILURE") {
return ErrFailure
} else if strings.Contains(str, "PENDING") {
return ErrPending
}
}
return err
}

20
dtmgrpc/dtmgrpc.pb.go
View File

@ -418,13 +418,17 @@ var file_dtmgrpc_dtmgrpc_proto_rawDesc = []byte{
0x01, 0x20, 0x01, 0x28, 0x09, 0x52, 0x09, 0x44, 0x74, 0x6d, 0x52, 0x65, 0x73, 0x75, 0x6c, 0x74,
0x12, 0x1e, 0x0a, 0x0a, 0x44, 0x74, 0x6d, 0x4d, 0x65, 0x73, 0x73, 0x61, 0x67, 0x65, 0x18, 0x02,
0x20, 0x01, 0x28, 0x09, 0x52, 0x0a, 0x44, 0x74, 0x6d, 0x4d, 0x65, 0x73, 0x73, 0x61, 0x67, 0x65,
0x32, 0x3e, 0x0a, 0x03, 0x44, 0x74, 0x6d, 0x12, 0x37, 0x0a, 0x06, 0x53, 0x75, 0x62, 0x6d, 0x69,
0x32, 0x78, 0x0a, 0x03, 0x44, 0x74, 0x6d, 0x12, 0x37, 0x0a, 0x06, 0x53, 0x75, 0x62, 0x6d, 0x69,
0x74, 0x12, 0x13, 0x2e, 0x64, 0x74, 0x6d, 0x67, 0x72, 0x70, 0x63, 0x2e, 0x44, 0x74, 0x6d, 0x52,
0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x1a, 0x16, 0x2e, 0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2e,
0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, 0x75, 0x66, 0x2e, 0x45, 0x6d, 0x70, 0x74, 0x79, 0x22, 0x00,
0x42, 0x1d, 0x5a, 0x1b, 0x67, 0x69, 0x74, 0x68, 0x75, 0x62, 0x2e, 0x63, 0x6f, 0x6d, 0x2f, 0x79,
0x65, 0x64, 0x66, 0x2f, 0x64, 0x74, 0x6d, 0x2f, 0x64, 0x74, 0x6d, 0x67, 0x72, 0x70, 0x63, 0x62,
0x06, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x33,
0x12, 0x38, 0x0a, 0x07, 0x50, 0x72, 0x65, 0x70, 0x61, 0x72, 0x65, 0x12, 0x13, 0x2e, 0x64, 0x74,
0x6d, 0x67, 0x72, 0x70, 0x63, 0x2e, 0x44, 0x74, 0x6d, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74,
0x1a, 0x16, 0x2e, 0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62,
0x75, 0x66, 0x2e, 0x45, 0x6d, 0x70, 0x74, 0x79, 0x22, 0x00, 0x42, 0x1d, 0x5a, 0x1b, 0x67, 0x69,
0x74, 0x68, 0x75, 0x62, 0x2e, 0x63, 0x6f, 0x6d, 0x2f, 0x79, 0x65, 0x64, 0x66, 0x2f, 0x64, 0x74,
0x6d, 0x2f, 0x64, 0x74, 0x6d, 0x67, 0x72, 0x70, 0x63, 0x62, 0x06, 0x70, 0x72, 0x6f, 0x74, 0x6f,
0x33,
}
var (
@ -455,9 +459,11 @@ var file_dtmgrpc_dtmgrpc_proto_depIdxs = []int32{
0, // 1: dtmgrpc.BusiRequest.info:type_name -> dtmgrpc.DtmTransInfo
6, // 2: dtmgrpc.BusiRequest.Extra:type_name -> dtmgrpc.BusiRequest.ExtraEntry
1, // 3: dtmgrpc.Dtm.Submit:input_type -> dtmgrpc.DtmRequest
7, // 4: dtmgrpc.Dtm.Submit:output_type -> google.protobuf.Empty
4, // [4:5] is the sub-list for method output_type
3, // [3:4] is the sub-list for method input_type
1, // 4: dtmgrpc.Dtm.Prepare:input_type -> dtmgrpc.DtmRequest
7, // 5: dtmgrpc.Dtm.Submit:output_type -> google.protobuf.Empty
7, // 6: dtmgrpc.Dtm.Prepare:output_type -> google.protobuf.Empty
5, // [5:7] is the sub-list for method output_type
3, // [3:5] is the sub-list for method input_type
3, // [3:3] is the sub-list for extension type_name
3, // [3:3] is the sub-list for extension extendee
0, // [0:3] is the sub-list for field type_name

1
dtmgrpc/dtmgrpc.proto
View File

@ -8,6 +8,7 @@ package dtmgrpc;
// The dtm service definition.
service Dtm {
rpc Submit(DtmRequest) returns (google.protobuf.Empty) {}
rpc Prepare(DtmRequest) returns (google.protobuf.Empty) {}
}

36
dtmgrpc/dtmgrpc_grpc.pb.go
View File

@ -20,6 +20,7 @@ const _ = grpc.SupportPackageIsVersion7
// For semantics around ctx use and closing/ending streaming RPCs, please refer to https://pkg.go.dev/google.golang.org/grpc/?tab=doc#ClientConn.NewStream.
type DtmClient interface {
Submit(ctx context.Context, in *DtmRequest, opts ...grpc.CallOption) (*emptypb.Empty, error)
Prepare(ctx context.Context, in *DtmRequest, opts ...grpc.CallOption) (*emptypb.Empty, error)
}
type dtmClient struct {
@ -39,11 +40,21 @@ func (c *dtmClient) Submit(ctx context.Context, in *DtmRequest, opts ...grpc.Cal
return out, nil
}
func (c *dtmClient) Prepare(ctx context.Context, in *DtmRequest, opts ...grpc.CallOption) (*emptypb.Empty, error) {
out := new(emptypb.Empty)
err := c.cc.Invoke(ctx, "/dtmgrpc.Dtm/Prepare", in, out, opts...)
if err != nil {
return nil, err
}
return out, nil
}
// DtmServer is the server API for Dtm service.
// All implementations must embed UnimplementedDtmServer
// for forward compatibility
type DtmServer interface {
Submit(context.Context, *DtmRequest) (*emptypb.Empty, error)
Prepare(context.Context, *DtmRequest) (*emptypb.Empty, error)
mustEmbedUnimplementedDtmServer()
}
@ -54,6 +65,9 @@ type UnimplementedDtmServer struct {
func (UnimplementedDtmServer) Submit(context.Context, *DtmRequest) (*emptypb.Empty, error) {
return nil, status.Errorf(codes.Unimplemented, "method Submit not implemented")
}
func (UnimplementedDtmServer) Prepare(context.Context, *DtmRequest) (*emptypb.Empty, error) {
return nil, status.Errorf(codes.Unimplemented, "method Prepare not implemented")
}
func (UnimplementedDtmServer) mustEmbedUnimplementedDtmServer() {}
// UnsafeDtmServer may be embedded to opt out of forward compatibility for this service.
@ -85,6 +99,24 @@ func _Dtm_Submit_Handler(srv interface{}, ctx context.Context, dec func(interfac
return interceptor(ctx, in, info, handler)
}
func _Dtm_Prepare_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
in := new(DtmRequest)
if err := dec(in); err != nil {
return nil, err
}
if interceptor == nil {
return srv.(DtmServer).Prepare(ctx, in)
}
info := &grpc.UnaryServerInfo{
Server: srv,
FullMethod: "/dtmgrpc.Dtm/Prepare",
}
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
return srv.(DtmServer).Prepare(ctx, req.(*DtmRequest))
}
return interceptor(ctx, in, info, handler)
}
// Dtm_ServiceDesc is the grpc.ServiceDesc for Dtm service.
// It's only intended for direct use with grpc.RegisterService,
// and not to be introspected or modified (even as a copy)
@ -96,6 +128,10 @@ var Dtm_ServiceDesc = grpc.ServiceDesc{
MethodName: "Submit",
Handler: _Dtm_Submit_Handler,
},
{
MethodName: "Prepare",
Handler: _Dtm_Prepare_Handler,
},
},
Streams: []grpc.StreamDesc{},
Metadata: "dtmgrpc/dtmgrpc.proto",

12
dtmgrpc/message.go
View File

@ -44,3 +44,15 @@ func (s *MsgGrpc) Submit() error {
})
return err
}
// Prepare prepare the msg
func (s *MsgGrpc) Prepare(queryPrepared string) error {
s.QueryPrepared = dtmcli.OrString(queryPrepared, s.QueryPrepared)
_, err := MustGetDtmClient(s.Dtm).Prepare(context.Background(), &DtmRequest{
Gid: s.Gid,
TransType: s.TransType,
QueryPrepared: s.QueryPrepared,
Data: dtmcli.MustMarshalString(&s.Steps),
})
return err
}

13
dtmgrpc/type.go
View File

@ -7,6 +7,8 @@ import (
"github.com/yedf/dtm/dtmcli"
grpc "google.golang.org/grpc"
codes "google.golang.org/grpc/codes"
status "google.golang.org/grpc/status"
)
var clients = map[string]*grpc.ClientConn{}
@ -69,3 +71,14 @@ func GrpcClientLog(ctx context.Context, method string, req, reply interface{}, c
}
return err
}
// Result2Error 将通用的result转成grpc的error
func Result2Error(res interface{}, err error) error {
e := dtmcli.CheckResult(res, err)
if e == dtmcli.ErrFailure {
return status.New(codes.Aborted, fmt.Sprintf("failure: res: %v, err: %s", res, e.Error())).Err()
} else if e == dtmcli.ErrPending {
return status.New(codes.Unavailable, fmt.Sprintf("failure: res: %v, err: %s", res, e.Error())).Err()
}
return e
}

12
dtmsvr/api.go
View File

@ -1,6 +1,10 @@
package dtmsvr
import "fmt"
import (
"fmt"
"github.com/yedf/dtm/dtmcli"
)
func svcSubmit(t *TransGlobal, waitResult bool) (interface{}, error) {
db := dbGet()
@ -13,3 +17,9 @@ func svcSubmit(t *TransGlobal, waitResult bool) (interface{}, error) {
return t.Process(db, waitResult), nil
}
func svcPrepare(t *TransGlobal) (interface{}, error) {
t.Status = "prepared"
t.saveNew(dbGet())
return dtmcli.ResultSuccess, nil
}

10
dtmsvr/api_grpc.go
View File

@ -3,6 +3,7 @@ package dtmsvr
import (
"context"
"github.com/yedf/dtm/dtmgrpc"
pb "github.com/yedf/dtm/dtmgrpc"
"google.golang.org/protobuf/types/known/emptypb"
)
@ -13,6 +14,11 @@ type dtmServer struct {
}
func (s *dtmServer) Submit(ctx context.Context, in *pb.DtmRequest) (*emptypb.Empty, error) {
svcSubmit(TransFromDtmRequest(in), in.WaitResult)
return &emptypb.Empty{}, nil
r, err := svcSubmit(TransFromDtmRequest(in), in.WaitResult)
return &emptypb.Empty{}, dtmgrpc.Result2Error(r, err)
}
func (s *dtmServer) Prepare(ctx context.Context, in *pb.DtmRequest) (*emptypb.Empty, error) {
r, err := svcPrepare(TransFromDtmRequest(in))
return &emptypb.Empty{}, dtmgrpc.Result2Error(r, err)
}

22
dtmsvr/trans.go
View File

@ -158,18 +158,18 @@ func (t *TransGlobal) setNextCron(expireIn int64) []string {
return []string{"next_cron_interval", "next_cron_time"}
}
func (t *TransGlobal) getBranchResult(branch *TransBranch) string {
func (t *TransGlobal) getURLResult(url string, branchID, branchType string, branchData []byte) string {
if t.Protocol == "grpc" {
server, method := dtmgrpc.GetServerAndMethod(branch.URL)
server, method := dtmgrpc.GetServerAndMethod(url)
conn := dtmgrpc.MustGetGrpcConn(server)
err := conn.Invoke(context.Background(), method, &dtmgrpc.BusiRequest{
Info: &dtmgrpc.DtmTransInfo{
Gid: t.Gid,
TransType: t.TransType,
BranchID: branch.BranchID,
BranchType: branch.BranchType,
BranchID: branchID,
BranchType: branchType,
},
AppData: []byte(branch.Data),
AppData: []byte(branchData),
}, &emptypb.Empty{})
if err == nil {
return "SUCCESS"
@ -178,19 +178,23 @@ func (t *TransGlobal) getBranchResult(branch *TransBranch) string {
}
return err.Error()
}
resp, err := dtmcli.RestyClient.R().SetBody(branch.Data).
resp, err := dtmcli.RestyClient.R().SetBody(branchData).
SetQueryParams(dtmcli.MS{
"gid": t.Gid,
"trans_type": t.TransType,
"branch_id": branch.BranchID,
"branch_type": branch.BranchType,
"branch_id": branchID,
"branch_type": branchType,
}).
SetHeader("Content-type", "application/json").
Post(branch.URL)
Post(url)
e2p(err)
return resp.String()
}
func (t *TransGlobal) getBranchResult(branch *TransBranch) string {
return t.getURLResult(branch.URL, branch.BranchID, branch.BranchType, []byte(branch.Data))
}
func (t *TransGlobal) execBranch(db *common.DB, branch *TransBranch) {
body := t.getBranchResult(branch)
if strings.Contains(body, "SUCCESS") {

14
dtmsvr/trans_grpc_msg_test.go
View File

@ -1,6 +1,7 @@
package dtmsvr
import (
"fmt"
"testing"
"time"
@ -25,10 +26,13 @@ func grpcMsgNormal(t *testing.T) {
func grpcMsgPending(t *testing.T) {
msg := genGrpcMsg("grpc-msg-pending")
examples.MainSwitch.TransInResult.SetOnce("PENDING")
err := msg.Submit()
err := msg.Prepare(fmt.Sprintf("%s/examples.Busi/CanSubmit", examples.BusiGrpc))
assert.Nil(t, err)
WaitTransProcessed(msg.Gid)
examples.MainSwitch.CanSubmitResult.SetOnce("PENDING")
CronTransOnce(60 * time.Second)
assert.Equal(t, "prepared", getTransStatus(msg.Gid))
examples.MainSwitch.TransInResult.SetOnce("PENDING")
CronTransOnce(60 * time.Second)
assert.Equal(t, "submitted", getTransStatus(msg.Gid))
CronTransOnce(60 * time.Second)
assert.Equal(t, "succeed", getTransStatus(msg.Gid))
@ -37,7 +41,7 @@ func grpcMsgPending(t *testing.T) {
func genGrpcMsg(gid string) *dtmgrpc.MsgGrpc {
req := dtmcli.MustMarshal(&examples.TransReq{Amount: 30})
return dtmgrpc.NewMsgGrpc(examples.DtmGrpcServer, gid).
Add(examples.BusiPb+"/examples.Busi/TransOut", req).
Add(examples.BusiPb+"/examples.Busi/TransIn", req)
Add(examples.BusiGrpc+"/examples.Busi/TransOut", req).
Add(examples.BusiGrpc+"/examples.Busi/TransIn", req)
}

4
dtmsvr/trans_msg.go
View File

@ -36,9 +36,7 @@ func (t *TransGlobal) mayQueryPrepared(db *common.DB) {
if t.Status != "prepared" {
return
}
resp, err := dtmcli.RestyClient.R().SetQueryParam("gid", t.Gid).Get(t.QueryPrepared)
e2p(err)
body := resp.String()
body := t.getURLResult(t.QueryPrepared, "", "", nil)
if strings.Contains(body, "SUCCESS") {
t.changeStatus(db, "submitted")
} else {

66
examples/busi.pb.go
View File

@ -26,28 +26,32 @@ var File_examples_busi_proto protoreflect.FileDescriptor
var file_examples_busi_proto_rawDesc = []byte{
0x0a, 0x13, 0x65, 0x78, 0x61, 0x6d, 0x70, 0x6c, 0x65, 0x73, 0x2f, 0x62, 0x75, 0x73, 0x69, 0x2e,
0x70, 0x72, 0x6f, 0x74, 0x6f, 0x12, 0x08, 0x65, 0x78, 0x61, 0x6d, 0x70, 0x6c, 0x65, 0x73, 0x1a,
0x11, 0x64, 0x74, 0x6d, 0x70, 0x62, 0x2f, 0x64, 0x74, 0x6d, 0x70, 0x62, 0x2e, 0x70, 0x72, 0x6f,
0x74, 0x6f, 0x1a, 0x1b, 0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2f, 0x70, 0x72, 0x6f, 0x74, 0x6f,
0x62, 0x75, 0x66, 0x2f, 0x65, 0x6d, 0x70, 0x74, 0x79, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x32,
0xf8, 0x01, 0x0a, 0x04, 0x42, 0x75, 0x73, 0x69, 0x12, 0x37, 0x0a, 0x07, 0x54, 0x72, 0x61, 0x6e,
0x73, 0x49, 0x6e, 0x12, 0x12, 0x2e, 0x64, 0x74, 0x6d, 0x70, 0x62, 0x2e, 0x42, 0x75, 0x73, 0x69,
0x15, 0x64, 0x74, 0x6d, 0x67, 0x72, 0x70, 0x63, 0x2f, 0x64, 0x74, 0x6d, 0x67, 0x72, 0x70, 0x63,
0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x1a, 0x1b, 0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2f, 0x70,
0x72, 0x6f, 0x74, 0x6f, 0x62, 0x75, 0x66, 0x2f, 0x65, 0x6d, 0x70, 0x74, 0x79, 0x2e, 0x70, 0x72,
0x6f, 0x74, 0x6f, 0x32, 0xbd, 0x02, 0x0a, 0x04, 0x42, 0x75, 0x73, 0x69, 0x12, 0x3b, 0x0a, 0x09,
0x43, 0x61, 0x6e, 0x53, 0x75, 0x62, 0x6d, 0x69, 0x74, 0x12, 0x14, 0x2e, 0x64, 0x74, 0x6d, 0x67,
0x72, 0x70, 0x63, 0x2e, 0x42, 0x75, 0x73, 0x69, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x1a,
0x16, 0x2e, 0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, 0x75,
0x66, 0x2e, 0x45, 0x6d, 0x70, 0x74, 0x79, 0x22, 0x00, 0x12, 0x39, 0x0a, 0x07, 0x54, 0x72, 0x61,
0x6e, 0x73, 0x49, 0x6e, 0x12, 0x14, 0x2e, 0x64, 0x74, 0x6d, 0x67, 0x72, 0x70, 0x63, 0x2e, 0x42,
0x75, 0x73, 0x69, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x1a, 0x16, 0x2e, 0x67, 0x6f, 0x6f,
0x67, 0x6c, 0x65, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, 0x75, 0x66, 0x2e, 0x45, 0x6d, 0x70,
0x74, 0x79, 0x22, 0x00, 0x12, 0x3a, 0x0a, 0x08, 0x54, 0x72, 0x61, 0x6e, 0x73, 0x4f, 0x75, 0x74,
0x12, 0x14, 0x2e, 0x64, 0x74, 0x6d, 0x67, 0x72, 0x70, 0x63, 0x2e, 0x42, 0x75, 0x73, 0x69, 0x52,
0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x1a, 0x16, 0x2e, 0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2e,
0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, 0x75, 0x66, 0x2e, 0x45, 0x6d, 0x70, 0x74, 0x79, 0x22, 0x00,
0x12, 0x3f, 0x0a, 0x0d, 0x54, 0x72, 0x61, 0x6e, 0x73, 0x49, 0x6e, 0x52, 0x65, 0x76, 0x65, 0x72,
0x74, 0x12, 0x14, 0x2e, 0x64, 0x74, 0x6d, 0x67, 0x72, 0x70, 0x63, 0x2e, 0x42, 0x75, 0x73, 0x69,
0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x1a, 0x16, 0x2e, 0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65,
0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, 0x75, 0x66, 0x2e, 0x45, 0x6d, 0x70, 0x74, 0x79, 0x22,
0x00, 0x12, 0x38, 0x0a, 0x08, 0x54, 0x72, 0x61, 0x6e, 0x73, 0x4f, 0x75, 0x74, 0x12, 0x12, 0x2e,
0x64, 0x74, 0x6d, 0x70, 0x62, 0x2e, 0x42, 0x75, 0x73, 0x69, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73,
0x74, 0x1a, 0x16, 0x2e, 0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f,
0x62, 0x75, 0x66, 0x2e, 0x45, 0x6d, 0x70, 0x74, 0x79, 0x22, 0x00, 0x12, 0x3d, 0x0a, 0x0d, 0x54,
0x72, 0x61, 0x6e, 0x73, 0x49, 0x6e, 0x52, 0x65, 0x76, 0x65, 0x72, 0x74, 0x12, 0x12, 0x2e, 0x64,
0x74, 0x6d, 0x70, 0x62, 0x2e, 0x42, 0x75, 0x73, 0x69, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74,
0x1a, 0x16, 0x2e, 0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62,
0x75, 0x66, 0x2e, 0x45, 0x6d, 0x70, 0x74, 0x79, 0x22, 0x00, 0x12, 0x3e, 0x0a, 0x0e, 0x54, 0x72,
0x61, 0x6e, 0x73, 0x4f, 0x75, 0x74, 0x52, 0x65, 0x76, 0x65, 0x72, 0x74, 0x12, 0x12, 0x2e, 0x64,
0x74, 0x6d, 0x70, 0x62, 0x2e, 0x42, 0x75, 0x73, 0x69, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74,
0x1a, 0x16, 0x2e, 0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62,
0x75, 0x66, 0x2e, 0x45, 0x6d, 0x70, 0x74, 0x79, 0x22, 0x00, 0x42, 0x1e, 0x5a, 0x1c, 0x67, 0x69,
0x74, 0x68, 0x75, 0x62, 0x2e, 0x63, 0x6f, 0x6d, 0x2f, 0x79, 0x65, 0x64, 0x66, 0x2f, 0x64, 0x74,
0x6d, 0x2f, 0x65, 0x78, 0x61, 0x6d, 0x70, 0x6c, 0x65, 0x73, 0x62, 0x06, 0x70, 0x72, 0x6f, 0x74,
0x6f, 0x33,
0x00, 0x12, 0x40, 0x0a, 0x0e, 0x54, 0x72, 0x61, 0x6e, 0x73, 0x4f, 0x75, 0x74, 0x52, 0x65, 0x76,
0x65, 0x72, 0x74, 0x12, 0x14, 0x2e, 0x64, 0x74, 0x6d, 0x67, 0x72, 0x70, 0x63, 0x2e, 0x42, 0x75,
0x73, 0x69, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x1a, 0x16, 0x2e, 0x67, 0x6f, 0x6f, 0x67,
0x6c, 0x65, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, 0x75, 0x66, 0x2e, 0x45, 0x6d, 0x70, 0x74,
0x79, 0x22, 0x00, 0x42, 0x1e, 0x5a, 0x1c, 0x67, 0x69, 0x74, 0x68, 0x75, 0x62, 0x2e, 0x63, 0x6f,
0x6d, 0x2f, 0x79, 0x65, 0x64, 0x66, 0x2f, 0x64, 0x74, 0x6d, 0x2f, 0x65, 0x78, 0x61, 0x6d, 0x70,
0x6c, 0x65, 0x73, 0x62, 0x06, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x33,
}
var file_examples_busi_proto_goTypes = []interface{}{
@ -55,16 +59,18 @@ var file_examples_busi_proto_goTypes = []interface{}{
(*emptypb.Empty)(nil), // 1: google.protobuf.Empty
}
var file_examples_busi_proto_depIdxs = []int32{
0, // 0: examples.Busi.TransIn:input_type -> dtmgrpc.BusiRequest
0, // 1: examples.Busi.TransOut:input_type -> dtmgrpc.BusiRequest
0, // 2: examples.Busi.TransInRevert:input_type -> dtmgrpc.BusiRequest
0, // 3: examples.Busi.TransOutRevert:input_type -> dtmgrpc.BusiRequest
1, // 4: examples.Busi.TransIn:output_type -> google.protobuf.Empty
1, // 5: examples.Busi.TransOut:output_type -> google.protobuf.Empty
1, // 6: examples.Busi.TransInRevert:output_type -> google.protobuf.Empty
1, // 7: examples.Busi.TransOutRevert:output_type -> google.protobuf.Empty
4, // [4:8] is the sub-list for method output_type
0, // [0:4] is the sub-list for method input_type
0, // 0: examples.Busi.CanSubmit:input_type -> dtmgrpc.BusiRequest
0, // 1: examples.Busi.TransIn:input_type -> dtmgrpc.BusiRequest
0, // 2: examples.Busi.TransOut:input_type -> dtmgrpc.BusiRequest
0, // 3: examples.Busi.TransInRevert:input_type -> dtmgrpc.BusiRequest
0, // 4: examples.Busi.TransOutRevert:input_type -> dtmgrpc.BusiRequest
1, // 5: examples.Busi.CanSubmit:output_type -> google.protobuf.Empty
1, // 6: examples.Busi.TransIn:output_type -> google.protobuf.Empty
1, // 7: examples.Busi.TransOut:output_type -> google.protobuf.Empty
1, // 8: examples.Busi.TransInRevert:output_type -> google.protobuf.Empty
1, // 9: examples.Busi.TransOutRevert:output_type -> google.protobuf.Empty
5, // [5:10] is the sub-list for method output_type
0, // [0:5] is the sub-list for method input_type
0, // [0:0] is the sub-list for extension type_name
0, // [0:0] is the sub-list for extension extendee
0, // [0:0] is the sub-list for field type_name

1
examples/busi.proto
View File

@ -8,6 +8,7 @@ import "google/protobuf/empty.proto";
// The dtm service definition.
service Busi {
rpc CanSubmit(dtmgrpc.BusiRequest) returns (google.protobuf.Empty) {}
rpc TransIn(dtmgrpc.BusiRequest) returns (google.protobuf.Empty) {}
rpc TransOut(dtmgrpc.BusiRequest) returns (google.protobuf.Empty) {}
rpc TransInRevert(dtmgrpc.BusiRequest) returns (google.protobuf.Empty) {}

36
examples/busi_grpc.pb.go
View File

@ -20,6 +20,7 @@ const _ = grpc.SupportPackageIsVersion7
//
// For semantics around ctx use and closing/ending streaming RPCs, please refer to https://pkg.go.dev/google.golang.org/grpc/?tab=doc#ClientConn.NewStream.
type BusiClient interface {
CanSubmit(ctx context.Context, in *dtmgrpc.BusiRequest, opts ...grpc.CallOption) (*emptypb.Empty, error)
TransIn(ctx context.Context, in *dtmgrpc.BusiRequest, opts ...grpc.CallOption) (*emptypb.Empty, error)
TransOut(ctx context.Context, in *dtmgrpc.BusiRequest, opts ...grpc.CallOption) (*emptypb.Empty, error)
TransInRevert(ctx context.Context, in *dtmgrpc.BusiRequest, opts ...grpc.CallOption) (*emptypb.Empty, error)
@ -34,6 +35,15 @@ func NewBusiClient(cc grpc.ClientConnInterface) BusiClient {
return &busiClient{cc}
}
func (c *busiClient) CanSubmit(ctx context.Context, in *dtmgrpc.BusiRequest, opts ...grpc.CallOption) (*emptypb.Empty, error) {
out := new(emptypb.Empty)
err := c.cc.Invoke(ctx, "/examples.Busi/CanSubmit", in, out, opts...)
if err != nil {
return nil, err
}
return out, nil
}
func (c *busiClient) TransIn(ctx context.Context, in *dtmgrpc.BusiRequest, opts ...grpc.CallOption) (*emptypb.Empty, error) {
out := new(emptypb.Empty)
err := c.cc.Invoke(ctx, "/examples.Busi/TransIn", in, out, opts...)
@ -74,6 +84,7 @@ func (c *busiClient) TransOutRevert(ctx context.Context, in *dtmgrpc.BusiRequest
// All implementations must embed UnimplementedBusiServer
// for forward compatibility
type BusiServer interface {
CanSubmit(context.Context, *dtmgrpc.BusiRequest) (*emptypb.Empty, error)
TransIn(context.Context, *dtmgrpc.BusiRequest) (*emptypb.Empty, error)
TransOut(context.Context, *dtmgrpc.BusiRequest) (*emptypb.Empty, error)
TransInRevert(context.Context, *dtmgrpc.BusiRequest) (*emptypb.Empty, error)
@ -85,6 +96,9 @@ type BusiServer interface {
type UnimplementedBusiServer struct {
}
func (UnimplementedBusiServer) CanSubmit(context.Context, *dtmgrpc.BusiRequest) (*emptypb.Empty, error) {
return nil, status.Errorf(codes.Unimplemented, "method CanSubmit not implemented")
}
func (UnimplementedBusiServer) TransIn(context.Context, *dtmgrpc.BusiRequest) (*emptypb.Empty, error) {
return nil, status.Errorf(codes.Unimplemented, "method TransIn not implemented")
}
@ -110,6 +124,24 @@ func RegisterBusiServer(s grpc.ServiceRegistrar, srv BusiServer) {
s.RegisterService(&Busi_ServiceDesc, srv)
}
func _Busi_CanSubmit_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
in := new(dtmgrpc.BusiRequest)
if err := dec(in); err != nil {
return nil, err
}
if interceptor == nil {
return srv.(BusiServer).CanSubmit(ctx, in)
}
info := &grpc.UnaryServerInfo{
Server: srv,
FullMethod: "/examples.Busi/CanSubmit",
}
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
return srv.(BusiServer).CanSubmit(ctx, req.(*dtmgrpc.BusiRequest))
}
return interceptor(ctx, in, info, handler)
}
func _Busi_TransIn_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
in := new(dtmgrpc.BusiRequest)
if err := dec(in); err != nil {
@ -189,6 +221,10 @@ var Busi_ServiceDesc = grpc.ServiceDesc{
ServiceName: "examples.Busi",
HandlerType: (*BusiServer)(nil),
Methods: []grpc.MethodDesc{
{
MethodName: "CanSubmit",
Handler: _Busi_CanSubmit_Handler,
},
{
MethodName: "TransIn",
Handler: _Busi_TransIn_Handler,

4
examples/main_base.go
View File

@ -14,8 +14,8 @@ const (
BusiAPI = "/api/busi"
// BusiPort busi server port
BusiPort = 8081
// BusiPbPort busi server port
BusiPbPort = 60081
// BusiGrpcPort busi server port
BusiGrpcPort = 60081
)
// Busi busi service url prefix

11
examples/main_grpc.go
View File

@ -13,8 +13,8 @@ import (
"google.golang.org/protobuf/types/known/emptypb"
)
// BusiPb busi service grpc address
var BusiPb string = fmt.Sprintf("localhost:%d", BusiPbPort)
// BusiGrpc busi service grpc address
var BusiGrpc string = fmt.Sprintf("localhost:%d", BusiGrpcPort)
// DtmClient grpc client for dtm
var DtmClient dtmgrpc.DtmClient = nil
@ -26,7 +26,7 @@ func GrpcStartup() {
DtmClient = dtmgrpc.NewDtmClient(conn)
dtmcli.Logf("dtm client inited")
lis, err := net.Listen("tcp", fmt.Sprintf(":%d", BusiPbPort))
lis, err := net.Listen("tcp", fmt.Sprintf(":%d", BusiGrpcPort))
dtmcli.FatalIfError(err)
s := grpc.NewServer(grpc.UnaryInterceptor(dtmgrpc.GrpcServerLog))
RegisterBusiServer(s, &busiServer{})
@ -54,6 +54,11 @@ type busiServer struct {
UnimplementedBusiServer
}
func (s *busiServer) CanSubmit(ctx context.Context, in *dtmgrpc.BusiRequest) (*emptypb.Empty, error) {
res := MainSwitch.CanSubmitResult.Fetch()
return &emptypb.Empty{}, dtmgrpc.Result2Error(res, nil)
}
func (s *busiServer) TransIn(ctx context.Context, in *dtmgrpc.BusiRequest) (*emptypb.Empty, error) {
req := TransReq{}
dtmcli.MustUnmarshal(in.AppData, &req)

4
examples/main_grpc_msg.go
View File

@ -15,8 +15,8 @@ func MsgGrpcSetup(app *gin.Engine) {
func MsgGrpcFireRequest() string {
req := dtmcli.MustMarshal(&TransReq{Amount: 30})
msg := dtmgrpc.NewMsgGrpc(DtmGrpcServer, dtmcli.MustGenGid(DtmServer)).
Add(BusiPb+"/examples.Busi/TransOut", req).
Add(BusiPb+"/examples.Busi/TransIn", req)
Add(BusiGrpc+"/examples.Busi/TransOut", req).
Add(BusiGrpc+"/examples.Busi/TransIn", req)
err := msg.Submit()
e2p(err)
return msg.Gid