diff --git a/test/busi/barrier.go b/test/busi/barrier.go index 40a8a16..556e6aa 100644 --- a/test/busi/barrier.go +++ b/test/busi/barrier.go @@ -88,7 +88,7 @@ func init() { if req.TransOutResult != "" { return dtmcli.String2DtmError(req.TransOutResult) } - if req.Store == "redis" { + if req.Store == Redis { return MustBarrierFromGin(c).RedisCheckAdjustAmount(RedisGet(), GetRedisAccountKey(TransOutUID), req.Amount, 7*86400) } @@ -97,7 +97,7 @@ func init() { }) })) app.POST(BusiAPI+"/TccBTransOutConfirm", dtmutil.WrapHandler2(func(c *gin.Context) interface{} { - if reqFrom(c).Store == "redis" { + if reqFrom(c).Store == Redis { return nil } return MustBarrierFromGin(c).Call(txGet(), func(tx *sql.Tx) error { @@ -111,7 +111,7 @@ func init() { // TccBarrierTransOutCancel will be use in test func TccBarrierTransOutCancel(c *gin.Context) interface{} { req := reqFrom(c) - if req.Store == "redis" { + if req.Store == Redis { return MustBarrierFromGin(c).RedisCheckAdjustAmount(RedisGet(), GetRedisAccountKey(TransOutUID), -req.Amount, 7*86400) } return MustBarrierFromGin(c).Call(txGet(), func(tx *sql.Tx) error { diff --git a/test/busi/busi.go b/test/busi/busi.go index 623de06..c39dc27 100644 --- a/test/busi/busi.go +++ b/test/busi/busi.go @@ -19,6 +19,9 @@ const TransOutUID = 1 // TransInUID 2 const TransInUID = 2 +// Redis 1 +const Redis = "redis" + func handleGrpcBusiness(in *BusiReq, result1 string, result2 string, busi string) error { res := dtmimp.OrString(result1, result2, dtmcli.ResultSuccess) logger.Debugf("grpc busi %s %v %s %s result: %s", busi, in, result1, result2, res)