102 lines
3.2 KiB
Go
102 lines
3.2 KiB
Go
package examples
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/sirupsen/logrus"
|
|
"github.com/yedf/dtm/common"
|
|
"github.com/yedf/dtm/dtmcli"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// 事务参与者的服务地址
|
|
const SagaBarrierBusiApi = "/api/busi_saga_barrier"
|
|
|
|
var SagaBarrierBusi = fmt.Sprintf("http://localhost:%d%s", SagaBarrierBusiPort, SagaBarrierBusiApi)
|
|
|
|
func SagaBarrierMain() {
|
|
go SagaBarrierStartSvr()
|
|
SagaBarrierFireRequest()
|
|
time.Sleep(1000 * time.Second)
|
|
}
|
|
|
|
func SagaBarrierStartSvr() {
|
|
logrus.Printf("saga barrier examples starting")
|
|
app := common.GetGinApp()
|
|
SagaBarrierAddRoute(app)
|
|
app.Run(fmt.Sprintf(":%d", SagaBarrierBusiPort))
|
|
}
|
|
|
|
func SagaBarrierFireRequest() {
|
|
logrus.Printf("a busi transaction begin")
|
|
req := &TransReq{
|
|
Amount: 30,
|
|
TransInResult: "SUCCESS",
|
|
TransOutResult: "SUCCESS",
|
|
}
|
|
saga := dtmcli.NewSaga(DtmServer).
|
|
Add(SagaBarrierBusi+"/TransOut", SagaBarrierBusi+"/TransOutCompensate", req).
|
|
Add(SagaBarrierBusi+"/TransIn", SagaBarrierBusi+"/TransInCompensate", req)
|
|
logrus.Printf("busi trans submit")
|
|
err := saga.Submit()
|
|
e2p(err)
|
|
}
|
|
|
|
// api
|
|
|
|
func SagaBarrierAddRoute(app *gin.Engine) {
|
|
app.POST(SagaBarrierBusiApi+"/TransIn", common.WrapHandler(sagaBarrierTransIn))
|
|
app.POST(SagaBarrierBusiApi+"/TransInCompensate", common.WrapHandler(sagaBarrierTransInCompensate))
|
|
app.POST(SagaBarrierBusiApi+"/TransOut", common.WrapHandler(sagaBarrierTransOut))
|
|
app.POST(SagaBarrierBusiApi+"/TransOutCompensate", common.WrapHandler(sagaBarrierTransOutCompensate))
|
|
logrus.Printf("examples listening at %d", SagaBarrierBusiPort)
|
|
}
|
|
|
|
var SagaBarrierTransInResult = ""
|
|
var SagaBarrierTransOutResult = ""
|
|
var SagaBarrierTransInCompensateResult = ""
|
|
var SagaBarrierTransOutCompensateResult = ""
|
|
|
|
func sagaBarrierTransIn(c *gin.Context) (interface{}, error) {
|
|
gid := c.Query("gid")
|
|
req := reqFrom(c)
|
|
res := common.OrString(SagaBarrierTransInResult, req.TransInResult, "SUCCESS")
|
|
logrus.Printf("%s TransIn: %v result: %s", gid, req, res)
|
|
return M{"result": res}, nil
|
|
}
|
|
|
|
func sagaBarrierTransInCompensate(c *gin.Context) (interface{}, error) {
|
|
gid := c.Query("gid")
|
|
req := reqFrom(c)
|
|
res := common.OrString(SagaBarrierTransInCompensateResult, "SUCCESS")
|
|
logrus.Printf("%s TransInCompensate: %v result: %s", gid, req, res)
|
|
return M{"result": res}, nil
|
|
}
|
|
|
|
func sagaBarrierTransOut(c *gin.Context) (interface{}, error) {
|
|
gid := c.Query("gid")
|
|
lid := c.Query("lid")
|
|
req := reqFrom(c)
|
|
return dtmcli.ThroughBarrierCall(dbGet().ToSqlDB(), "saga", gid, lid, "action", func(sdb *sql.DB) (interface{}, error) {
|
|
db := common.SqlDB2DB(sdb)
|
|
dbr := db.Model(&UserAccount{}).Where("user_id = ?", c.Query("user_id")).
|
|
Update("balance", gorm.Expr("balance - ?", req.Amount))
|
|
return nil, dbr.Error
|
|
})
|
|
|
|
// res := common.OrString(SagaBarrierTransOutResult, req.TransOutResult, "SUCCESS")
|
|
// logrus.Printf("%s TransOut: %v result: %s", gid, req, res)
|
|
// return M{"result": res}, nil
|
|
}
|
|
|
|
func sagaBarrierTransOutCompensate(c *gin.Context) (interface{}, error) {
|
|
gid := c.Query("gid")
|
|
req := reqFrom(c)
|
|
res := common.OrString(SagaBarrierTransOutCompensateResult, "SUCCESS")
|
|
logrus.Printf("%s TransOutCompensate: %v result: %s", gid, req, res)
|
|
return M{"result": res}, nil
|
|
}
|