mirror of https://github.com/dtm-labs/dtm.git
yedf2
5 years ago
10 changed files with 327 additions and 65 deletions
@ -0,0 +1,90 @@ |
|||||
|
package examples |
||||
|
|
||||
|
import ( |
||||
|
"context" |
||||
|
"database/sql" |
||||
|
|
||||
|
"github.com/gin-gonic/gin" |
||||
|
"github.com/yedf/dtm/dtmcli" |
||||
|
"github.com/yedf/dtm/dtmgrpc" |
||||
|
"google.golang.org/protobuf/types/known/emptypb" |
||||
|
) |
||||
|
|
||||
|
func init() { |
||||
|
addSample("grpc_saga_barrier", func() string { |
||||
|
req := dtmcli.MustMarshal(&TransReq{Amount: 30}) |
||||
|
gid := dtmgrpc.MustGenGid(DtmGrpcServer) |
||||
|
saga := dtmgrpc.NewSaga(DtmGrpcServer, gid). |
||||
|
Add(BusiGrpc+"/examples.Busi/TransOutBSaga", BusiGrpc+"/examples.Busi/TransOutRevertBSaga", req). |
||||
|
Add(BusiGrpc+"/examples.Busi/TransInBSaga", BusiGrpc+"/examples.Busi/TransOutRevertBSaga", req) |
||||
|
err := saga.Submit() |
||||
|
dtmcli.FatalIfError(err) |
||||
|
return saga.Gid |
||||
|
}) |
||||
|
} |
||||
|
|
||||
|
func sagaGrpcBarrierAdjustBalance(sdb *sql.Tx, uid int, amount int) (interface{}, error) { |
||||
|
_, err := dtmcli.StxExec(sdb, "update dtm_busi.user_account set balance = balance + ? where user_id = ?", amount, uid) |
||||
|
return dtmcli.ResultSuccess, err |
||||
|
|
||||
|
} |
||||
|
|
||||
|
func (s *busiServer) TransInBSaga(ctx context.Context, in *dtmgrpc.BusiRequest) (*emptypb.Empty, error) { |
||||
|
req := TransReq{} |
||||
|
dtmcli.MustUnmarshal(in.BusiData, &req) |
||||
|
return &emptypb.Empty{}, handleGrpcBusiness(in, MainSwitch.TransInResult.Fetch(), req.TransInResult, dtmcli.GetFuncName()) |
||||
|
} |
||||
|
|
||||
|
func (s *busiServer) TransOutBSaga(ctx context.Context, in *dtmgrpc.BusiRequest) (*emptypb.Empty, error) { |
||||
|
req := TransReq{} |
||||
|
dtmcli.MustUnmarshal(in.BusiData, &req) |
||||
|
return &emptypb.Empty{}, handleGrpcBusiness(in, MainSwitch.TransOutResult.Fetch(), req.TransOutResult, dtmcli.GetFuncName()) |
||||
|
} |
||||
|
|
||||
|
func (s *busiServer) TransInRevertBSaga(ctx context.Context, in *dtmgrpc.BusiRequest) (*emptypb.Empty, error) { |
||||
|
req := TransReq{} |
||||
|
dtmcli.MustUnmarshal(in.BusiData, &req) |
||||
|
return &emptypb.Empty{}, handleGrpcBusiness(in, MainSwitch.TransInRevertResult.Fetch(), "", dtmcli.GetFuncName()) |
||||
|
} |
||||
|
|
||||
|
func (s *busiServer) TransOutRevertBSaga(ctx context.Context, in *dtmgrpc.BusiRequest) (*emptypb.Empty, error) { |
||||
|
req := TransReq{} |
||||
|
dtmcli.MustUnmarshal(in.BusiData, &req) |
||||
|
return &emptypb.Empty{}, handleGrpcBusiness(in, MainSwitch.TransOutRevertResult.Fetch(), "", dtmcli.GetFuncName()) |
||||
|
} |
||||
|
|
||||
|
func sagaBarrierTransIn(c *gin.Context) (interface{}, error) { |
||||
|
req := reqFrom(c) |
||||
|
if req.TransInResult != "" { |
||||
|
return req.TransInResult, nil |
||||
|
} |
||||
|
barrier := MustBarrierFromGin(c) |
||||
|
return barrier.Call(sdbGet(), func(sdb *sql.Tx) (interface{}, error) { |
||||
|
return sagaBarrierAdjustBalance(sdb, 1, req.Amount) |
||||
|
}) |
||||
|
} |
||||
|
|
||||
|
func sagaBarrierTransInCompensate(c *gin.Context) (interface{}, error) { |
||||
|
barrier := MustBarrierFromGin(c) |
||||
|
return barrier.Call(sdbGet(), func(sdb *sql.Tx) (interface{}, error) { |
||||
|
return sagaBarrierAdjustBalance(sdb, 1, -reqFrom(c).Amount) |
||||
|
}) |
||||
|
} |
||||
|
|
||||
|
func sagaBarrierTransOut(c *gin.Context) (interface{}, error) { |
||||
|
req := reqFrom(c) |
||||
|
if req.TransInResult != "" { |
||||
|
return req.TransInResult, nil |
||||
|
} |
||||
|
barrier := MustBarrierFromGin(c) |
||||
|
return barrier.Call(sdbGet(), func(sdb *sql.Tx) (interface{}, error) { |
||||
|
return sagaBarrierAdjustBalance(sdb, 2, -req.Amount) |
||||
|
}) |
||||
|
} |
||||
|
|
||||
|
func sagaBarrierTransOutCompensate(c *gin.Context) (interface{}, error) { |
||||
|
barrier := MustBarrierFromGin(c) |
||||
|
return barrier.Call(sdbGet(), func(sdb *sql.Tx) (interface{}, error) { |
||||
|
return sagaBarrierAdjustBalance(sdb, 2, reqFrom(c).Amount) |
||||
|
}) |
||||
|
} |
||||
@ -1,24 +0,0 @@ |
|||||
package examples |
|
||||
|
|
||||
import ( |
|
||||
"github.com/yedf/dtm/dtmcli" |
|
||||
) |
|
||||
|
|
||||
func init() { |
|
||||
addSample("saga_wait", func() string { |
|
||||
dtmcli.Logf("a saga busi transaction begin") |
|
||||
req := &TransReq{ |
|
||||
Amount: 30, |
|
||||
TransInResult: "SUCCESS", |
|
||||
TransOutResult: "SUCCESS", |
|
||||
} |
|
||||
saga := dtmcli.NewSaga(DtmServer, dtmcli.MustGenGid(DtmServer)). |
|
||||
Add(Busi+"/TransOut", Busi+"/TransOutRevert", req). |
|
||||
Add(Busi+"/TransIn", Busi+"/TransInRevert", req) |
|
||||
saga.WaitResult = true // 设置为等待结果模式，后面的submit调用，会等待服务器处理这个事务。如果Submit正常返回，那么整个全局事务已成功完成
|
|
||||
err := saga.Submit() |
|
||||
dtmcli.Logf("result gid is: %s", saga.Gid) |
|
||||
dtmcli.FatalIfError(err) |
|
||||
return saga.Gid |
|
||||
}) |
|
||||
} |
|
||||
Loading…
Reference in new issue