before use touch
This commit is contained in:
yedongfu
parent
6a3c429dd1
commit
8865ed7136
304
dtmsvr/trans.go
Normal file
304
dtmsvr/trans.go
Normal file
@ -0,0 +1,304 @@
|
|||||||
|
package dtmsvr
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/yedf/dtm/common"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Trans interface {
|
||||||
|
GetDataBranches() []TransBranchModel
|
||||||
|
ProcessOnce(db *common.MyDb, branches []TransBranchModel) error
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetTrans(trans *TransGlobalModel) Trans {
|
||||||
|
if trans.TransType == "saga" {
|
||||||
|
return &TransSaga{TransGlobalModel: trans}
|
||||||
|
} else if trans.TransType == "tcc" {
|
||||||
|
return &TransTcc{TransGlobalModel: trans}
|
||||||
|
} else if trans.TransType == "xa" {
|
||||||
|
return &TransXa{TransGlobalModel: trans}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
type TransSaga struct {
|
||||||
|
*TransGlobalModel
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *TransSaga) GetDataBranches() []TransBranchModel {
|
||||||
|
nsteps := []TransBranchModel{}
|
||||||
|
steps := []M{}
|
||||||
|
common.MustUnmarshalString(t.Data, &steps)
|
||||||
|
for _, step := range steps {
|
||||||
|
for _, branchType := range []string{"compensate", "action"} {
|
||||||
|
nsteps = append(nsteps, TransBranchModel{
|
||||||
|
Gid: t.Gid,
|
||||||
|
Branch: fmt.Sprintf("%d", len(nsteps)+1),
|
||||||
|
Data: step["data"].(string),
|
||||||
|
Url: step[branchType].(string),
|
||||||
|
BranchType: branchType,
|
||||||
|
Status: "prepared",
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nsteps
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *TransSaga) ProcessOnce(db *common.MyDb, branches []TransBranchModel) error {
|
||||||
|
gid := t.Gid
|
||||||
|
current := 0 // 当前正在处理的步骤
|
||||||
|
for ; current < len(branches); current++ {
|
||||||
|
step := branches[current]
|
||||||
|
if step.BranchType == "compensate" && step.Status == "prepared" || step.BranchType == "action" && step.Status == "finished" {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if step.BranchType == "action" && step.Status == "prepared" {
|
||||||
|
resp, err := common.RestyClient.R().SetBody(step.Data).SetQueryParam("gid", step.Gid).Post(step.Url)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
body := resp.String()
|
||||||
|
|
||||||
|
db.Must().Model(&TransGlobalModel{}).Where("gid=?", gid).Update("gid", gid) // 更新update_time，避免被定时任务再次
|
||||||
|
if strings.Contains(body, "SUCCESS") {
|
||||||
|
writeTransLog(gid, "step finished", "finished", step.Branch, "")
|
||||||
|
dbr := db.Must().Model(&step).Where("status=?", "prepared").Updates(M{
|
||||||
|
"status": "finished",
|
||||||
|
"finish_time": time.Now(),
|
||||||
|
})
|
||||||
|
checkAffected(dbr)
|
||||||
|
} else if strings.Contains(body, "FAIL") {
|
||||||
|
writeTransLog(gid, "step rollbacked", "rollbacked", step.Branch, "")
|
||||||
|
dbr := db.Must().Model(&step).Where("status=?", "prepared").Updates(M{
|
||||||
|
"status": "rollbacked",
|
||||||
|
"rollback_time": time.Now(),
|
||||||
|
})
|
||||||
|
checkAffected(dbr)
|
||||||
|
break
|
||||||
|
} else {
|
||||||
|
return fmt.Errorf("unknown response: %s, will be retried", body)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if current == len(branches) { // saga 事务完成
|
||||||
|
writeTransLog(gid, "saga finished", "finished", "", "")
|
||||||
|
dbr := db.Must().Model(&TransGlobalModel{}).Where("gid=? and status=?", gid, "committed").Updates(M{
|
||||||
|
"status": "finished",
|
||||||
|
"finish_time": time.Now(),
|
||||||
|
})
|
||||||
|
checkAffected(dbr)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
for current = current - 1; current >= 0; current-- {
|
||||||
|
step := branches[current]
|
||||||
|
if step.BranchType != "compensate" || step.Status != "prepared" {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
resp, err := common.RestyClient.R().SetBody(step.Data).SetQueryParam("gid", step.Gid).Post(step.Url)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
body := resp.String()
|
||||||
|
if strings.Contains(body, "SUCCESS") {
|
||||||
|
writeTransLog(gid, "step rollbacked", "rollbacked", step.Branch, "")
|
||||||
|
dbr := db.Must().Model(&step).Where("status=?", step.Status).Updates(M{
|
||||||
|
"status": "rollbacked",
|
||||||
|
"rollback_time": time.Now(),
|
||||||
|
})
|
||||||
|
checkAffected(dbr)
|
||||||
|
} else {
|
||||||
|
return fmt.Errorf("expect compensate return SUCCESS")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if current != -1 {
|
||||||
|
return fmt.Errorf("saga current not -1")
|
||||||
|
}
|
||||||
|
writeTransLog(gid, "saga rollbacked", "rollbacked", "", "")
|
||||||
|
dbr := db.Must().Model(&TransGlobalModel{}).Where("status=? and gid=?", "committed", gid).Updates(M{
|
||||||
|
"status": "rollbacked",
|
||||||
|
"rollback_time": time.Now(),
|
||||||
|
})
|
||||||
|
checkAffected(dbr)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
type TransTcc struct {
|
||||||
|
*TransGlobalModel
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *TransTcc) GetDataBranches() []TransBranchModel {
|
||||||
|
nsteps := []TransBranchModel{}
|
||||||
|
steps := []M{}
|
||||||
|
common.MustUnmarshalString(t.Data, &steps)
|
||||||
|
for _, step := range steps {
|
||||||
|
for _, branchType := range []string{"rollback", "commit", "prepare"} {
|
||||||
|
nsteps = append(nsteps, TransBranchModel{
|
||||||
|
Gid: t.Gid,
|
||||||
|
Branch: fmt.Sprintf("%d", len(nsteps)+1),
|
||||||
|
Data: step["data"].(string),
|
||||||
|
Url: step[branchType].(string),
|
||||||
|
BranchType: branchType,
|
||||||
|
Status: "prepared",
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nsteps
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *TransTcc) ProcessOnce(db *common.MyDb, branches []TransBranchModel) error {
|
||||||
|
gid := t.Gid
|
||||||
|
current := 0 // 当前正在处理的步骤
|
||||||
|
for ; current < len(branches); current++ {
|
||||||
|
step := branches[current]
|
||||||
|
if step.BranchType == "prepare" && step.Status == "finished" || step.BranchType != "commit" && step.Status == "prepared" {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if step.BranchType == "prepare" && step.Status == "prepared" {
|
||||||
|
resp, err := common.RestyClient.R().SetBody(step.Data).SetQueryParam("gid", step.Gid).Post(step.Url)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
body := resp.String()
|
||||||
|
db.Must().Model(&TransGlobalModel{}).Where("gid=?", gid).Update("gid", gid) // 更新update_time，避免被定时任务再次
|
||||||
|
if strings.Contains(body, "SUCCESS") {
|
||||||
|
writeTransLog(gid, "step finished", "finished", step.Branch, "")
|
||||||
|
dbr := db.Must().Model(&step).Where("status=?", "prepared").Updates(M{
|
||||||
|
"status": "finished",
|
||||||
|
"finish_time": time.Now(),
|
||||||
|
})
|
||||||
|
checkAffected(dbr)
|
||||||
|
} else if strings.Contains(body, "FAIL") {
|
||||||
|
writeTransLog(gid, "step rollbacked", "rollbacked", step.Branch, "")
|
||||||
|
dbr := db.Must().Model(&step).Where("status=?", "prepared").Updates(M{
|
||||||
|
"status": "rollbacked",
|
||||||
|
"rollback_time": time.Now(),
|
||||||
|
})
|
||||||
|
checkAffected(dbr)
|
||||||
|
break
|
||||||
|
} else {
|
||||||
|
return fmt.Errorf("unknown response: %s, will be retried", body)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
//////////////////////////////////////////////////
|
||||||
|
if current == len(branches) { // tcc 事务完成
|
||||||
|
writeTransLog(gid, "saga finished", "finished", "", "")
|
||||||
|
dbr := db.Must().Model(&TransGlobalModel{}).Where("gid=? and status=?", gid, "committed").Updates(M{
|
||||||
|
"status": "finished",
|
||||||
|
"finish_time": time.Now(),
|
||||||
|
})
|
||||||
|
checkAffected(dbr)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
for current = current - 1; current >= 0; current-- {
|
||||||
|
step := branches[current]
|
||||||
|
if step.BranchType != "compensate" || step.Status != "prepared" {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
resp, err := common.RestyClient.R().SetBody(step.Data).SetQueryParam("gid", step.Gid).Post(step.Url)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
body := resp.String()
|
||||||
|
if strings.Contains(body, "SUCCESS") {
|
||||||
|
writeTransLog(gid, "step rollbacked", "rollbacked", step.Branch, "")
|
||||||
|
dbr := db.Must().Model(&step).Where("status=?", step.Status).Updates(M{
|
||||||
|
"status": "rollbacked",
|
||||||
|
"rollback_time": time.Now(),
|
||||||
|
})
|
||||||
|
checkAffected(dbr)
|
||||||
|
} else {
|
||||||
|
return fmt.Errorf("expect compensate return SUCCESS")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if current != -1 {
|
||||||
|
return fmt.Errorf("saga current not -1")
|
||||||
|
}
|
||||||
|
writeTransLog(gid, "saga rollbacked", "rollbacked", "", "")
|
||||||
|
dbr := db.Must().Model(&TransGlobalModel{}).Where("status=? and gid=?", "committed", gid).Updates(M{
|
||||||
|
"status": "rollbacked",
|
||||||
|
"rollback_time": time.Now(),
|
||||||
|
})
|
||||||
|
checkAffected(dbr)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
type TransXa struct {
|
||||||
|
*TransGlobalModel
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *TransXa) GetDataBranches() []TransBranchModel {
|
||||||
|
return []TransBranchModel{}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *TransXa) ProcessOnce(db *common.MyDb, branches []TransBranchModel) error {
|
||||||
|
gid := t.Gid
|
||||||
|
if t.Status == "finished" {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
if t.Status == "committed" {
|
||||||
|
for _, branch := range branches {
|
||||||
|
if branch.Status == "finished" {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
db.Must().Model(&TransGlobalModel{}).Where("gid=?", gid).Update("gid", gid) // 更新update_time，避免被定时任务再次
|
||||||
|
resp, err := common.RestyClient.R().SetBody(M{
|
||||||
|
"branch": branch.Branch,
|
||||||
|
"action": "commit",
|
||||||
|
"gid": branch.Gid,
|
||||||
|
}).Post(branch.Url)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
body := resp.String()
|
||||||
|
if !strings.Contains(body, "SUCCESS") {
|
||||||
|
return fmt.Errorf("bad response: %s", body)
|
||||||
|
}
|
||||||
|
writeTransLog(gid, "step finished", "finished", branch.Branch, "")
|
||||||
|
db.Must().Model(&branch).Where("status=?", "prepared").Updates(M{
|
||||||
|
"status": "finished",
|
||||||
|
"finish_time": time.Now(),
|
||||||
|
})
|
||||||
|
}
|
||||||
|
writeTransLog(gid, "xa finished", "finished", "", "")
|
||||||
|
db.Must().Model(&TransGlobalModel{}).Where("gid=? and status=?", gid, "committed").Updates(M{
|
||||||
|
"status": "finished",
|
||||||
|
"finish_time": time.Now(),
|
||||||
|
})
|
||||||
|
} else if t.Status == "prepared" { // 未commit直接处理的情况为回滚场景
|
||||||
|
for _, branch := range branches {
|
||||||
|
if branch.Status == "rollbacked" {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
db.Must().Model(&TransGlobalModel{}).Where("gid=?", gid).Update("gid", gid) // 更新update_time，避免被定时任务再次
|
||||||
|
resp, err := common.RestyClient.R().SetBody(M{
|
||||||
|
"branch": branch.Branch,
|
||||||
|
"action": "rollback",
|
||||||
|
"gid": branch.Gid,
|
||||||
|
}).Post(branch.Url)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
body := resp.String()
|
||||||
|
if !strings.Contains(body, "SUCCESS") {
|
||||||
|
return fmt.Errorf("bad response: %s", body)
|
||||||
|
}
|
||||||
|
writeTransLog(gid, "step rollbacked", "rollbacked", branch.Branch, "")
|
||||||
|
db.Must().Model(&branch).Where("status=?", "prepared").Updates(M{
|
||||||
|
"status": "rollbacked",
|
||||||
|
"finish_time": time.Now(),
|
||||||
|
})
|
||||||
|
}
|
||||||
|
writeTransLog(gid, "xa rollbacked", "rollbacked", "", "")
|
||||||
|
db.Must().Model(&TransGlobalModel{}).Where("gid=? and status=?", gid, "prepared").Updates(M{
|
||||||
|
"status": "rollbacked",
|
||||||
|
"finish_time": time.Now(),
|
||||||
|
})
|
||||||
|
} else {
|
||||||
|
return fmt.Errorf("bad trans status: %s", t.Status)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
77
tcc.go
Normal file
77
tcc.go
Normal file
@ -0,0 +1,77 @@
|
|||||||
|
package dtm
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
|
||||||
|
"github.com/sirupsen/logrus"
|
||||||
|
"github.com/yedf/dtm/common"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Tcc struct {
|
||||||
|
TccData
|
||||||
|
Server string
|
||||||
|
}
|
||||||
|
|
||||||
|
type TccData struct {
|
||||||
|
Gid string `json:"gid"`
|
||||||
|
TransType string `json:"trans_type"`
|
||||||
|
Steps []TccStep `json:"steps"`
|
||||||
|
QueryPrepared string `json:"query_prepared"`
|
||||||
|
}
|
||||||
|
type TccStep struct {
|
||||||
|
Prepare string `json:"prepare"`
|
||||||
|
Commit string `json:"commit"`
|
||||||
|
Rollback string `json:"rollback"`
|
||||||
|
Data string `json:"data"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func TccNew(server string, gid string, queryPrepared string) *Saga {
|
||||||
|
return &Saga{
|
||||||
|
SagaData: SagaData{
|
||||||
|
Gid: gid,
|
||||||
|
TransType: "tcc",
|
||||||
|
QueryPrepared: queryPrepared,
|
||||||
|
},
|
||||||
|
Server: server,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
func (s *Tcc) Add(prepare string, commit string, rollback string, data interface{}) error {
|
||||||
|
logrus.Printf("tcc %s Add %s %s %s %v", s.Gid, prepare, commit, rollback, data)
|
||||||
|
d, err := json.Marshal(data)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
step := TccStep{
|
||||||
|
Prepare: prepare,
|
||||||
|
Commit: commit,
|
||||||
|
Rollback: rollback,
|
||||||
|
Data: string(d),
|
||||||
|
}
|
||||||
|
s.Steps = append(s.Steps, step)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Tcc) Prepare() error {
|
||||||
|
logrus.Printf("preparing %s body: %v", s.Gid, &s.TccData)
|
||||||
|
resp, err := common.RestyClient.R().SetBody(&s.TccData).Post(fmt.Sprintf("%s/prepare", s.Server))
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if resp.StatusCode() != 200 {
|
||||||
|
return fmt.Errorf("prepare failed: %v", resp.Body())
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Tcc) Commit() error {
|
||||||
|
logrus.Printf("committing %s body: %v", s.Gid, &s.TccData)
|
||||||
|
resp, err := common.RestyClient.R().SetBody(&s.TccData).Post(fmt.Sprintf("%s/commit", s.Server))
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if resp.StatusCode() != 200 {
|
||||||
|
return fmt.Errorf("commit failed: %v", resp.Body())
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
Loading…
x
Reference in New Issue
Block a user