opensource/dtm
3
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

saga use save status ok

Browse Source
This commit is contained in:
yedongfu 2021-05-27 19:01:04 +08:00
parent 8865ed7136
commit 8a72a4d629
2 changed files with 17 additions and 32 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
dtmsvr/trans.go
View File

@ -48,7 +48,6 @@ func (t *TransSaga) GetDataBranches() []TransBranchModel {
} }
func (t *TransSaga) ProcessOnce(db *common.MyDb, branches []TransBranchModel) error { func (t *TransSaga) ProcessOnce(db *common.MyDb, branches []TransBranchModel) error {
gid := t.Gid
current := 0 // 当前正在处理的步骤 current := 0 // 当前正在处理的步骤
for ; current < len(branches); current++ { for ; current < len(branches); current++ {
step := branches[current] step := branches[current]
@ -62,21 +61,11 @@ func (t *TransSaga) ProcessOnce(db *common.MyDb, branches []TransBranchModel) er
} }
body := resp.String() body := resp.String()
db.Must().Model(&TransGlobalModel{}).Where("gid=?", gid).Update("gid", gid) // 更新update_time，避免被定时任务再次 t.touch(db.Must())
if strings.Contains(body, "SUCCESS") { if strings.Contains(body, "SUCCESS") {
writeTransLog(gid, "step finished", "finished", step.Branch, "") step.saveStatus(db.Must(), "finished")
dbr := db.Must().Model(&step).Where("status=?", "prepared").Updates(M{
"status": "finished",
"finish_time": time.Now(),
})
checkAffected(dbr)
} else if strings.Contains(body, "FAIL") { } else if strings.Contains(body, "FAIL") {
writeTransLog(gid, "step rollbacked", "rollbacked", step.Branch, "") step.saveStatus(db.Must(), "rollbacked")
dbr := db.Must().Model(&step).Where("status=?", "prepared").Updates(M{
"status": "rollbacked",
"rollback_time": time.Now(),
})
checkAffected(dbr)
break break
} else { } else {
return fmt.Errorf("unknown response: %s, will be retried", body) return fmt.Errorf("unknown response: %s, will be retried", body)
@ -84,12 +73,7 @@ func (t *TransSaga) ProcessOnce(db *common.MyDb, branches []TransBranchModel) er
} }
} }
if current == len(branches) { // saga 事务完成 if current == len(branches) { // saga 事务完成
writeTransLog(gid, "saga finished", "finished", "", "") t.saveStatus(db.Must(), "finished")
dbr := db.Must().Model(&TransGlobalModel{}).Where("gid=? and status=?", gid, "committed").Updates(M{
"status": "finished",
"finish_time": time.Now(),
})
checkAffected(dbr)
return nil return nil
} }
for current = current - 1; current >= 0; current-- { for current = current - 1; current >= 0; current-- {
@ -103,12 +87,7 @@ func (t *TransSaga) ProcessOnce(db *common.MyDb, branches []TransBranchModel) er
} }
body := resp.String() body := resp.String()
if strings.Contains(body, "SUCCESS") { if strings.Contains(body, "SUCCESS") {
writeTransLog(gid, "step rollbacked", "rollbacked", step.Branch, "") step.saveStatus(db.Must(), "rollbacked")
dbr := db.Must().Model(&step).Where("status=?", step.Status).Updates(M{
"status": "rollbacked",
"rollback_time": time.Now(),
})
checkAffected(dbr)
} else { } else {
return fmt.Errorf("expect compensate return SUCCESS") return fmt.Errorf("expect compensate return SUCCESS")
} }
@ -116,12 +95,7 @@ func (t *TransSaga) ProcessOnce(db *common.MyDb, branches []TransBranchModel) er
if current != -1 { if current != -1 {
return fmt.Errorf("saga current not -1") return fmt.Errorf("saga current not -1")
} }
writeTransLog(gid, "saga rollbacked", "rollbacked", "", "") t.saveStatus(db.Must(), "rollbacked")
dbr := db.Must().Model(&TransGlobalModel{}).Where("status=? and gid=?", "committed", gid).Updates(M{
"status": "rollbacked",
"rollback_time": time.Now(),
})
checkAffected(dbr)
return nil return nil
} }

11
dtmsvr/types.go
View File

@ -31,6 +31,17 @@ func (t *TransGlobalModel) touch(db *common.MyDb) *gorm.DB {
return db.Model(&TransGlobalModel{}).Where("gid=?", t.Gid).Update("gid", t.Gid) // 更新update_time，避免被定时任务再次 return db.Model(&TransGlobalModel{}).Where("gid=?", t.Gid).Update("gid", t.Gid) // 更新update_time，避免被定时任务再次
} }
func (t *TransGlobalModel) saveStatus(db *common.MyDb, status string) *gorm.DB {
writeTransLog(t.Gid, "step change", status, "", "")
dbr := db.Must().Model(t).Where("status=?", t.Status).Updates(M{
"status": status,
"finish_time": time.Now(),
})
checkAffected(dbr)
t.Status = status
return dbr
}
type TransBranchModel struct { type TransBranchModel struct {
common.ModelBase common.ModelBase
Gid string Gid string