120 lines
4.5 KiB
Go
120 lines
4.5 KiB
Go
package examples
|
||
|
||
import (
|
||
"database/sql"
|
||
"fmt"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
"github.com/sirupsen/logrus"
|
||
"github.com/yedf/dtm/common"
|
||
"github.com/yedf/dtm/dtmcli"
|
||
)
|
||
|
||
// TccBarrierFireRequest 1
|
||
func TccBarrierFireRequest() string {
|
||
logrus.Printf("tcc transaction begin")
|
||
gid, err := dtmcli.TccGlobalTransaction(DtmServer, func(tcc *dtmcli.Tcc) (rerr error) {
|
||
res1, rerr := tcc.CallBranch(&TransReq{Amount: 30}, Busi+"/TccBTransOutTry", Busi+"/TccBTransOutConfirm", Busi+"/TccBTransOutCancel")
|
||
if rerr != nil {
|
||
return
|
||
}
|
||
if res1.StatusCode() != 200 {
|
||
return fmt.Errorf("bad status code: %d", res1.StatusCode())
|
||
}
|
||
res2, rerr := tcc.CallBranch(&TransReq{Amount: 30}, Busi+"/TccBTransInTry", Busi+"/TccBTransInConfirm", Busi+"/TccBTransInCancel")
|
||
if rerr != nil {
|
||
return
|
||
}
|
||
if res2.StatusCode() != 200 {
|
||
return fmt.Errorf("bad status code: %d", res2.StatusCode())
|
||
}
|
||
logrus.Printf("tcc returns: %s, %s", res1.String(), res2.String())
|
||
return
|
||
})
|
||
e2p(err)
|
||
return gid
|
||
}
|
||
|
||
// TccBarrierAddRoute 1
|
||
func TccBarrierAddRoute(app *gin.Engine) {
|
||
app.POST(BusiAPI+"/TccBTransInTry", common.WrapHandler(tccBarrierTransInTry))
|
||
app.POST(BusiAPI+"/TccBTransInConfirm", common.WrapHandler(tccBarrierTransInConfirm))
|
||
app.POST(BusiAPI+"/TccBTransInCancel", common.WrapHandler(tccBarrierTransInCancel))
|
||
app.POST(BusiAPI+"/TccBTransOutTry", common.WrapHandler(tccBarrierTransOutTry))
|
||
app.POST(BusiAPI+"/TccBTransOutConfirm", common.WrapHandler(tccBarrierTransOutConfirm))
|
||
app.POST(BusiAPI+"/TccBTransOutCancel", common.WrapHandler(TccBarrierTransOutCancel))
|
||
logrus.Printf("examples listening at %d", BusiPort)
|
||
}
|
||
|
||
const transInUID = 1
|
||
const transOutUID = 2
|
||
|
||
func adjustTrading(sdb *sql.DB, uid int, amount int) (interface{}, error) {
|
||
db := common.SQLDB2DB(sdb)
|
||
dbr := db.Exec("update dtm_busi.user_account_trading t join dtm_busi.user_account a on t.user_id=a.user_id and t.user_id=? set t.trading_balance=t.trading_balance + ? where a.balance + t.trading_balance + ? >= 0", uid, amount, amount)
|
||
if dbr.Error == nil && dbr.RowsAffected == 0 {
|
||
return nil, fmt.Errorf("update error, maybe balance not enough")
|
||
}
|
||
return common.MS{"dtm_server": "SUCCESS"}, nil
|
||
}
|
||
|
||
func adjustBalance(sdb *sql.DB, uid int, amount int) (interface{}, error) {
|
||
db := common.SQLDB2DB(sdb)
|
||
dbr := db.Exec("update dtm_busi.user_account_trading t join dtm_busi.user_account a on t.user_id=a.user_id and t.user_id=? set t.trading_balance=t.trading_balance + ?", uid, -amount, -amount)
|
||
if dbr.Error == nil && dbr.RowsAffected == 1 {
|
||
dbr = db.Exec("update dtm_busi.user_account set balance=balance+? where user_id=?", amount, uid)
|
||
}
|
||
if dbr.Error != nil {
|
||
return nil, dbr.Error
|
||
}
|
||
if dbr.RowsAffected == 0 {
|
||
return nil, fmt.Errorf("update 0 rows")
|
||
}
|
||
return common.MS{"dtm_result": "SUCCESS"}, nil
|
||
}
|
||
|
||
// TCC下，转入
|
||
func tccBarrierTransInTry(c *gin.Context) (interface{}, error) {
|
||
req := reqFrom(c) // 去重构一下，改成可以重复使用的输入
|
||
if req.TransInResult != "" {
|
||
return req.TransInResult, nil
|
||
}
|
||
return dtmcli.ThroughBarrierCall(dbGet().ToSQLDB(), dtmcli.TransInfoFromReq(c), func(sdb *sql.DB) (interface{}, error) {
|
||
return adjustTrading(sdb, transInUID, req.Amount)
|
||
})
|
||
}
|
||
|
||
func tccBarrierTransInConfirm(c *gin.Context) (interface{}, error) {
|
||
return dtmcli.ThroughBarrierCall(dbGet().ToSQLDB(), dtmcli.TransInfoFromReq(c), func(sdb *sql.DB) (interface{}, error) {
|
||
return adjustBalance(sdb, transInUID, reqFrom(c).Amount)
|
||
})
|
||
}
|
||
|
||
func tccBarrierTransInCancel(c *gin.Context) (interface{}, error) {
|
||
return dtmcli.ThroughBarrierCall(dbGet().ToSQLDB(), dtmcli.TransInfoFromReq(c), func(sdb *sql.DB) (interface{}, error) {
|
||
return adjustTrading(sdb, transInUID, -reqFrom(c).Amount)
|
||
})
|
||
}
|
||
|
||
func tccBarrierTransOutTry(c *gin.Context) (interface{}, error) {
|
||
req := reqFrom(c)
|
||
if req.TransInResult != "" {
|
||
return req.TransInResult, nil
|
||
}
|
||
return dtmcli.ThroughBarrierCall(dbGet().ToSQLDB(), dtmcli.TransInfoFromReq(c), func(sdb *sql.DB) (interface{}, error) {
|
||
return adjustTrading(sdb, transOutUID, -req.Amount)
|
||
})
|
||
}
|
||
|
||
func tccBarrierTransOutConfirm(c *gin.Context) (interface{}, error) {
|
||
return dtmcli.ThroughBarrierCall(dbGet().ToSQLDB(), dtmcli.TransInfoFromReq(c), func(sdb *sql.DB) (interface{}, error) {
|
||
return adjustBalance(sdb, transOutUID, -reqFrom(c).Amount)
|
||
})
|
||
}
|
||
|
||
func TccBarrierTransOutCancel(c *gin.Context) (interface{}, error) {
|
||
return dtmcli.ThroughBarrierCall(dbGet().ToSQLDB(), dtmcli.TransInfoFromReq(c), func(sdb *sql.DB) (interface{}, error) {
|
||
return adjustTrading(sdb, transOutUID, reqFrom(c).Amount)
|
||
})
|
||
}
|