50 lines
1.4 KiB
Go
50 lines
1.4 KiB
Go
package dtmsvr
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/yedf/dtm/common"
|
|
)
|
|
|
|
type TransTccProcessor struct {
|
|
*TransGlobal
|
|
}
|
|
|
|
func init() {
|
|
registorProcessorCreator("tcc", func(trans *TransGlobal) TransProcessor { return &TransTccProcessor{TransGlobal: trans} })
|
|
}
|
|
|
|
func (t *TransTccProcessor) GenBranches() []TransBranch {
|
|
return []TransBranch{}
|
|
}
|
|
|
|
func (t *TransTccProcessor) ExecBranch(db *common.DB, branch *TransBranch) {
|
|
resp, err := common.RestyClient.R().SetBody(branch.Data).SetQueryParam("gid", branch.Gid).Post(branch.Url)
|
|
e2p(err)
|
|
body := resp.String()
|
|
if strings.Contains(body, "SUCCESS") {
|
|
t.touch(db, config.TransCronInterval)
|
|
branch.changeStatus(db, "succeed")
|
|
} else if branch.BranchType == "try" && strings.Contains(body, "FAIL") {
|
|
t.touch(db, config.TransCronInterval)
|
|
branch.changeStatus(db, "failed")
|
|
} else {
|
|
panic(fmt.Errorf("unknown response: %s, will be retried", body))
|
|
}
|
|
}
|
|
|
|
func (t *TransTccProcessor) ProcessOnce(db *common.DB, branches []TransBranch) {
|
|
if t.Status == "succeed" || t.Status == "failed" {
|
|
return
|
|
}
|
|
branchType := common.If(t.Status == "submitted", "confirm", "cancel").(string)
|
|
for current := len(branches) - 1; current >= -1; current-- {
|
|
if current == -1 { // 已全部处理完
|
|
t.changeStatus(db, common.If(t.Status == "submitted", "succeed", "failed").(string))
|
|
} else if branches[current].BranchType == branchType {
|
|
t.ExecBranch(db, &branches[current])
|
|
}
|
|
}
|
|
}
|