saga should refactor next
This commit is contained in:
yedongfu
parent
fa8abb239f
commit
faca20d756
105
examples/main_saga.go
Normal file
105
examples/main_saga.go
Normal file
@ -0,0 +1,105 @@
|
||||
package examples
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"time"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
"github.com/sirupsen/logrus"
|
||||
"github.com/yedf/dtm"
|
||||
"github.com/yedf/dtm/common"
|
||||
)
|
||||
|
||||
// 事务参与者的服务地址
|
||||
const SagaBusiPort = 8081
|
||||
const SagaBusiApi = "/api/busi_saga"
|
||||
|
||||
var SagaBusi = fmt.Sprintf("http://localhost:%d%s", SagaBusiPort, SagaBusiApi)
|
||||
|
||||
func SagaMain() {
|
||||
go SagaStartSvr()
|
||||
sagaFireRequest()
|
||||
time.Sleep(1000 * time.Second)
|
||||
}
|
||||
|
||||
func SagaStartSvr() {
|
||||
logrus.Printf("saga examples starting")
|
||||
app := common.GetGinApp()
|
||||
SagaAddRoute(app)
|
||||
app.Run(fmt.Sprintf(":%d", SagaBusiPort))
|
||||
}
|
||||
|
||||
func sagaFireRequest() {
|
||||
gid := common.GenGid()
|
||||
logrus.Printf("busi transaction begin: %s", gid)
|
||||
req := &TransReq{
|
||||
Amount: 30,
|
||||
TransInResult: "SUCCESS",
|
||||
TransOutResult: "SUCCESS",
|
||||
}
|
||||
saga := dtm.SagaNew(DtmServer, gid)
|
||||
|
||||
saga.Add(SagaBusi+"/TransOut", SagaBusi+"/TransOutCompensate", req)
|
||||
saga.Add(SagaBusi+"/TransIn", SagaBusi+"/TransInCompensate", req)
|
||||
err := saga.Prepare(SagaBusi + "/TransQuery")
|
||||
e2p(err)
|
||||
logrus.Printf("busi trans commit")
|
||||
err = saga.Commit()
|
||||
e2p(err)
|
||||
}
|
||||
|
||||
// api
|
||||
|
||||
func SagaAddRoute(app *gin.Engine) {
|
||||
app.POST(SagaBusiApi+"/TransIn", common.WrapHandler(sagaTransIn))
|
||||
app.POST(SagaBusiApi+"/TransInCompensate", common.WrapHandler(sagaTransInCompensate))
|
||||
app.POST(SagaBusiApi+"/TransOut", common.WrapHandler(SagaTransOut))
|
||||
app.POST(SagaBusiApi+"/TransOutCompensate", common.WrapHandler(sagaTransOutCompensate))
|
||||
app.GET(SagaBusiApi+"/TransQuery", common.WrapHandler(sagaTransQuery))
|
||||
logrus.Printf("examples listening at %d", SagaBusiPort)
|
||||
}
|
||||
|
||||
var SagaTransInResult = ""
|
||||
var SagaTransOutResult = ""
|
||||
var SagaTransInCompensateResult = ""
|
||||
var SagaTransOutCompensateResult = ""
|
||||
var SagaTransQueryResult = ""
|
||||
|
||||
func sagaTransIn(c *gin.Context) (interface{}, error) {
|
||||
gid := c.Query("gid")
|
||||
req := transReqFromContext(c)
|
||||
res := common.OrString(SagaTransInResult, req.TransInResult, "SUCCESS")
|
||||
logrus.Printf("%s TransIn: %v result: %s", gid, req, res)
|
||||
return M{"result": res}, nil
|
||||
}
|
||||
|
||||
func sagaTransInCompensate(c *gin.Context) (interface{}, error) {
|
||||
gid := c.Query("gid")
|
||||
req := transReqFromContext(c)
|
||||
res := common.OrString(SagaTransInCompensateResult, "SUCCESS")
|
||||
logrus.Printf("%s TransInCompensate: %v result: %s", gid, req, res)
|
||||
return M{"result": res}, nil
|
||||
}
|
||||
|
||||
func SagaTransOut(c *gin.Context) (interface{}, error) {
|
||||
gid := c.Query("gid")
|
||||
req := transReqFromContext(c)
|
||||
res := common.OrString(SagaTransOutResult, req.TransOutResult, "SUCCESS")
|
||||
logrus.Printf("%s TransOut: %v result: %s", gid, req, res)
|
||||
return M{"result": res}, nil
|
||||
}
|
||||
|
||||
func sagaTransOutCompensate(c *gin.Context) (interface{}, error) {
|
||||
gid := c.Query("gid")
|
||||
req := transReqFromContext(c)
|
||||
res := common.OrString(SagaTransOutCompensateResult, "SUCCESS")
|
||||
logrus.Printf("%s TransOutCompensate: %v result: %s", gid, req, res)
|
||||
return M{"result": res}, nil
|
||||
}
|
||||
|
||||
func sagaTransQuery(c *gin.Context) (interface{}, error) {
|
||||
gid := c.Query("gid")
|
||||
logrus.Printf("%s TransQuery", gid)
|
||||
res := common.OrString(SagaTransQueryResult, "SUCCESS")
|
||||
return M{"result": res}, nil
|
||||
}
|
||||
125
examples/main_tcc.go
Normal file
125
examples/main_tcc.go
Normal file
@ -0,0 +1,125 @@
|
||||
package examples
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"time"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
"github.com/sirupsen/logrus"
|
||||
"github.com/yedf/dtm"
|
||||
"github.com/yedf/dtm/common"
|
||||
)
|
||||
|
||||
// 事务参与者的服务地址
|
||||
const TccBusiPort = 8083
|
||||
const TccBusiApi = "/api/busi_tcc"
|
||||
|
||||
var TccBusi = fmt.Sprintf("http://localhost:%d%s", TccBusiPort, TccBusiApi)
|
||||
|
||||
func TccMain() {
|
||||
go TccStartSvr()
|
||||
tccFireRequest()
|
||||
time.Sleep(1000 * time.Second)
|
||||
}
|
||||
|
||||
func TccStartSvr() {
|
||||
logrus.Printf("tcc examples starting")
|
||||
app := common.GetGinApp()
|
||||
TccAddRoute(app)
|
||||
app.Run(fmt.Sprintf(":%d", TccBusiPort))
|
||||
}
|
||||
|
||||
func tccFireRequest() {
|
||||
gid := common.GenGid()
|
||||
logrus.Printf("busi transaction begin: %s", gid)
|
||||
req := &TransReq{
|
||||
Amount: 30,
|
||||
TransInResult: "SUCCESS",
|
||||
TransOutResult: "SUCCESS",
|
||||
}
|
||||
tcc := dtm.TccNew(DtmServer, gid)
|
||||
|
||||
tcc.Add(TccBusi+"/TransOutTry", TccBusi+"/TransOutConfirm", TccBusi+"/TransOutCancel", req)
|
||||
tcc.Add(TccBusi+"/TransInTry", TccBusi+"/TransInConfirm", TccBusi+"/TransOutCancel", req)
|
||||
err := tcc.Prepare(TccBusi + "/TransQuery")
|
||||
e2p(err)
|
||||
logrus.Printf("busi trans commit")
|
||||
err = tcc.Commit()
|
||||
e2p(err)
|
||||
}
|
||||
|
||||
// api
|
||||
|
||||
func TccAddRoute(app *gin.Engine) {
|
||||
app.POST(TccBusiApi+"/TransInTry", common.WrapHandler(tccTransInTry))
|
||||
app.POST(TccBusiApi+"/TransInConfirm", common.WrapHandler(tccTransInConfirm))
|
||||
app.POST(TccBusiApi+"/TransInCancel", common.WrapHandler(tccTransCancel))
|
||||
app.POST(TccBusiApi+"/TransOutTry", common.WrapHandler(tccTransOutTry))
|
||||
app.POST(TccBusiApi+"/TransOutConfirm", common.WrapHandler(tccTransOutConfirm))
|
||||
app.POST(TccBusiApi+"/TransOutCancel", common.WrapHandler(tccTransOutCancel))
|
||||
app.GET(TccBusiApi+"/TransQuery", common.WrapHandler(tccTransQuery))
|
||||
logrus.Printf("examples listening at %d", TccBusiPort)
|
||||
}
|
||||
|
||||
var TccTransInTryResult = ""
|
||||
var TccTransOutTryResult = ""
|
||||
var TccTransInCancelResult = ""
|
||||
var TccTransOutCancelResult = ""
|
||||
var TccTransInConfirmResult = ""
|
||||
var TccTransOutConfirmResult = ""
|
||||
var TccTransQueryResult = ""
|
||||
|
||||
func tccTransInTry(c *gin.Context) (interface{}, error) {
|
||||
gid := c.Query("gid")
|
||||
req := transReqFromContext(c)
|
||||
res := common.OrString(TccTransInTryResult, req.TransInResult, "SUCCESS")
|
||||
logrus.Printf("%s TransInTry: %v result: %s", gid, req, res)
|
||||
return M{"result": res}, nil
|
||||
}
|
||||
|
||||
func tccTransInConfirm(c *gin.Context) (interface{}, error) {
|
||||
gid := c.Query("gid")
|
||||
req := transReqFromContext(c)
|
||||
res := common.OrString(TccTransInConfirmResult, "SUCCESS")
|
||||
logrus.Printf("%s tccTransInConfirm: %v result: %s", gid, req, res)
|
||||
return M{"result": res}, nil
|
||||
}
|
||||
|
||||
func tccTransCancel(c *gin.Context) (interface{}, error) {
|
||||
gid := c.Query("gid")
|
||||
req := transReqFromContext(c)
|
||||
res := common.OrString(TccTransInCancelResult, "SUCCESS")
|
||||
logrus.Printf("%s tccTransCancel: %v result: %s", gid, req, res)
|
||||
return M{"result": res}, nil
|
||||
}
|
||||
|
||||
func tccTransOutTry(c *gin.Context) (interface{}, error) {
|
||||
gid := c.Query("gid")
|
||||
req := transReqFromContext(c)
|
||||
res := common.OrString(TccTransOutTryResult, req.TransOutResult, "SUCCESS")
|
||||
logrus.Printf("%s TransOut: %v result: %s", gid, req, res)
|
||||
return M{"result": res}, nil
|
||||
}
|
||||
|
||||
func tccTransOutConfirm(c *gin.Context) (interface{}, error) {
|
||||
gid := c.Query("gid")
|
||||
req := transReqFromContext(c)
|
||||
res := common.OrString(TccTransOutConfirmResult, "SUCCESS")
|
||||
logrus.Printf("%s TransOutConfirm: %v result: %s", gid, req, res)
|
||||
return M{"result": res}, nil
|
||||
}
|
||||
|
||||
func tccTransOutCancel(c *gin.Context) (interface{}, error) {
|
||||
gid := c.Query("gid")
|
||||
req := transReqFromContext(c)
|
||||
res := common.OrString(TccTransOutCancelResult, "SUCCESS")
|
||||
logrus.Printf("%s tccTransOutCancel: %v result: %s", gid, req, res)
|
||||
return M{"result": res}, nil
|
||||
}
|
||||
|
||||
func tccTransQuery(c *gin.Context) (interface{}, error) {
|
||||
gid := c.Query("gid")
|
||||
logrus.Printf("%s TransQuery", gid)
|
||||
res := common.OrString(TccTransQueryResult, "SUCCESS")
|
||||
return M{"result": res}, nil
|
||||
}
|
||||
116
examples/main_xa.go
Normal file
116
examples/main_xa.go
Normal file
@ -0,0 +1,116 @@
|
||||
package examples
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"time"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
"github.com/sirupsen/logrus"
|
||||
"github.com/yedf/dtm"
|
||||
"github.com/yedf/dtm/common"
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
// 事务参与者的服务地址
|
||||
const XaBusiPort = 8082
|
||||
const XaBusiApi = "/api/busi_xa"
|
||||
|
||||
var XaBusi = fmt.Sprintf("http://localhost:%d%s", XaBusiPort, XaBusiApi)
|
||||
|
||||
var XaClient *dtm.XaClient = nil
|
||||
|
||||
type UserAccount struct {
|
||||
common.ModelBase
|
||||
UserId int
|
||||
Balance string
|
||||
}
|
||||
|
||||
func (u *UserAccount) TableName() string { return "user_account" }
|
||||
|
||||
func dbGet() *common.MyDb {
|
||||
return common.DbGet(Config.Mysql)
|
||||
}
|
||||
|
||||
func XaMain() {
|
||||
go XaStartSvr()
|
||||
time.Sleep(100 * time.Millisecond)
|
||||
xaFireRequest()
|
||||
time.Sleep(1000 * time.Second)
|
||||
}
|
||||
|
||||
func XaStartSvr() {
|
||||
common.InitApp(&Config)
|
||||
logrus.Printf("xa examples starting")
|
||||
app := common.GetGinApp()
|
||||
XaClient = dtm.XaClientNew(DtmServer, Config.Mysql, app, XaBusi+"/xa")
|
||||
XaAddRoute(app)
|
||||
app.Run(fmt.Sprintf(":%d", XaBusiPort))
|
||||
}
|
||||
|
||||
func xaFireRequest() {
|
||||
gid := common.GenGid()
|
||||
err := XaClient.XaGlobalTransaction(gid, func() (rerr error) {
|
||||
defer common.P2E(&rerr)
|
||||
req := GenTransReq(30, false, false)
|
||||
resp, err := common.RestyClient.R().SetBody(req).SetQueryParams(map[string]string{
|
||||
"gid": gid,
|
||||
"user_id": "1",
|
||||
}).Post(XaBusi + "/TransOut")
|
||||
common.CheckRestySuccess(resp, err)
|
||||
resp, err = common.RestyClient.R().SetBody(req).SetQueryParams(map[string]string{
|
||||
"gid": gid,
|
||||
"user_id": "2",
|
||||
}).Post(XaBusi + "/TransOut")
|
||||
common.CheckRestySuccess(resp, err)
|
||||
return nil
|
||||
})
|
||||
e2p(err)
|
||||
}
|
||||
|
||||
// api
|
||||
func XaAddRoute(app *gin.Engine) {
|
||||
app.POST(XaBusiApi+"/TransIn", common.WrapHandler(xaTransIn))
|
||||
app.POST(XaBusiApi+"/TransOut", common.WrapHandler(xaTransOut))
|
||||
}
|
||||
|
||||
func xaTransIn(c *gin.Context) (interface{}, error) {
|
||||
err := XaClient.XaLocalTransaction(c.Query("gid"), func(db *common.MyDb) (rerr error) {
|
||||
req := transReqFromContext(c)
|
||||
if req.TransInResult != "SUCCESS" {
|
||||
return fmt.Errorf("tranIn failed")
|
||||
}
|
||||
dbr := db.Model(&UserAccount{}).Where("user_id = ?", c.Query("user_id")).
|
||||
Update("balance", gorm.Expr("balance - ?", req.Amount))
|
||||
return dbr.Error
|
||||
})
|
||||
e2p(err)
|
||||
return M{"result": "SUCCESS"}, nil
|
||||
}
|
||||
|
||||
func xaTransOut(c *gin.Context) (interface{}, error) {
|
||||
err := XaClient.XaLocalTransaction(c.Query("gid"), func(db *common.MyDb) (rerr error) {
|
||||
req := transReqFromContext(c)
|
||||
if req.TransOutResult != "SUCCESS" {
|
||||
return fmt.Errorf("tranOut failed")
|
||||
}
|
||||
dbr := db.Model(&UserAccount{}).Where("user_id = ?", c.Query("user_id")).
|
||||
Update("balance", gorm.Expr("balance + ?", req.Amount))
|
||||
return dbr.Error
|
||||
})
|
||||
e2p(err)
|
||||
return M{"result": "SUCCESS"}, nil
|
||||
}
|
||||
|
||||
func ResetXaData() {
|
||||
db := dbGet()
|
||||
db.Must().Exec("truncate user_account")
|
||||
db.Must().Exec("insert into user_account (user_id, balance) values (1, 10000), (2, 10000)")
|
||||
type XaRow struct {
|
||||
Data string
|
||||
}
|
||||
xas := []XaRow{}
|
||||
db.Must().Raw("xa recover").Scan(&xas)
|
||||
for _, xa := range xas {
|
||||
db.Must().Exec(fmt.Sprintf("xa rollback '%s'", xa.Data))
|
||||
}
|
||||
}
|
||||
Loading…
x
Reference in New Issue
Block a user