75 lines
2.4 KiB
Go
75 lines
2.4 KiB
Go
package dtmsvr
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/yedf/dtm/dtmcli"
|
|
"gorm.io/gorm/clause"
|
|
)
|
|
|
|
func svcSubmit(t *TransGlobal, waitResult bool) (interface{}, error) {
|
|
db := dbGet()
|
|
dbt := TransFromDb(db, t.Gid)
|
|
if dbt != nil && dbt.Status != dtmcli.StatusPrepared && dbt.Status != dtmcli.StatusSubmitted {
|
|
return M{"dtm_result": dtmcli.ResultFailure, "message": fmt.Sprintf("current status %s, cannot sumbmit", dbt.Status)}, nil
|
|
}
|
|
t.Status = dtmcli.StatusSubmitted
|
|
t.saveNew(db)
|
|
return t.Process(db, waitResult), nil
|
|
|
|
}
|
|
|
|
func svcPrepare(t *TransGlobal) (interface{}, error) {
|
|
t.Status = dtmcli.StatusPrepared
|
|
t.saveNew(dbGet())
|
|
return dtmcli.MapSuccess, nil
|
|
}
|
|
|
|
func svcAbort(t *TransGlobal, waitResult bool) (interface{}, error) {
|
|
db := dbGet()
|
|
dbt := TransFromDb(db, t.Gid)
|
|
if t.TransType != "xa" && t.TransType != "tcc" || dbt.Status != dtmcli.StatusPrepared && dbt.Status != "aborting" {
|
|
return M{"dtm_result": dtmcli.ResultFailure, "message": fmt.Sprintf("trans type: %s current status %s, cannot abort", dbt.TransType, dbt.Status)}, nil
|
|
}
|
|
return dbt.Process(db, waitResult), nil
|
|
}
|
|
|
|
func svcRegisterTccBranch(branch *TransBranch, data dtmcli.MS) (interface{}, error) {
|
|
db := dbGet()
|
|
dbt := TransFromDb(db, branch.Gid)
|
|
if dbt.Status != dtmcli.StatusPrepared {
|
|
return M{"dtm_result": dtmcli.ResultFailure, "message": fmt.Sprintf("current status: %s cannot register branch", dbt.Status)}, nil
|
|
}
|
|
|
|
branches := []TransBranch{*branch, *branch, *branch}
|
|
for i, b := range []string{dtmcli.BranchCancel, dtmcli.BranchConfirm, dtmcli.BranchTry} {
|
|
branches[i].BranchType = b
|
|
branches[i].URL = data[b]
|
|
}
|
|
|
|
db.Must().Clauses(clause.OnConflict{
|
|
DoNothing: true,
|
|
}).Create(branches)
|
|
global := TransGlobal{Gid: branch.Gid}
|
|
global.touch(dbGet(), config.TransCronInterval)
|
|
return dtmcli.MapSuccess, nil
|
|
}
|
|
|
|
func svcRegisterXaBranch(branch *TransBranch) (interface{}, error) {
|
|
branch.Status = dtmcli.StatusPrepared
|
|
db := dbGet()
|
|
dbt := TransFromDb(db, branch.Gid)
|
|
if dbt.Status != dtmcli.StatusPrepared {
|
|
return M{"dtm_result": dtmcli.ResultFailure, "message": fmt.Sprintf("current status: %s cannot register branch", dbt.Status)}, nil
|
|
}
|
|
branches := []TransBranch{*branch, *branch}
|
|
branches[0].BranchType = dtmcli.BranchRollback
|
|
branches[1].BranchType = dtmcli.BranchCommit
|
|
db.Must().Clauses(clause.OnConflict{
|
|
DoNothing: true,
|
|
}).Create(branches)
|
|
global := TransGlobal{Gid: branch.Gid}
|
|
global.touch(db, config.TransCronInterval)
|
|
return dtmcli.MapSuccess, nil
|
|
}
|