98 lines
2.4 KiB
Go
98 lines
2.4 KiB
Go
package dtmcli
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/go-resty/resty/v2"
|
|
"github.com/sirupsen/logrus"
|
|
"github.com/yedf/dtm/common"
|
|
)
|
|
|
|
// Tcc struct of tcc
|
|
type Tcc struct {
|
|
IDGenerator
|
|
Dtm string
|
|
Gid string
|
|
}
|
|
|
|
// TccGlobalFunc type of global tcc call
|
|
type TccGlobalFunc func(tcc *Tcc) error
|
|
|
|
// TccGlobalTransaction begin a tcc global transaction
|
|
func TccGlobalTransaction(dtm string, tccFunc TccGlobalFunc) (gid string, rerr error) {
|
|
gid = GenGid(dtm)
|
|
data := &M{
|
|
"gid": gid,
|
|
"trans_type": "tcc",
|
|
}
|
|
defer func() {
|
|
var err error
|
|
if x := recover(); x != nil || rerr != nil {
|
|
_, err = common.RestyClient.R().SetBody(data).Post(dtm + "/abort")
|
|
} else {
|
|
_, err = common.RestyClient.R().SetBody(data).Post(dtm + "/submit")
|
|
}
|
|
if err != nil {
|
|
logrus.Errorf("submitting or abort global transaction error: %v", err)
|
|
}
|
|
}()
|
|
tcc := &Tcc{Dtm: dtm, Gid: gid}
|
|
resp, rerr := common.RestyClient.R().SetBody(data).Post(tcc.Dtm + "/prepare")
|
|
if rerr != nil {
|
|
return
|
|
}
|
|
if !strings.Contains(resp.String(), "SUCCESS") {
|
|
rerr = fmt.Errorf("bad response: %s", resp.String())
|
|
}
|
|
rerr = tccFunc(tcc)
|
|
return
|
|
}
|
|
|
|
// TccFromReq tcc from request info
|
|
func TccFromReq(c *gin.Context) (*Tcc, error) {
|
|
tcc := &Tcc{
|
|
Dtm: c.Query("dtm"),
|
|
Gid: c.Query("gid"),
|
|
IDGenerator: IDGenerator{parentID: c.Query("branch_id")},
|
|
}
|
|
if tcc.Dtm == "" || tcc.Gid == "" {
|
|
return nil, fmt.Errorf("bad tcc info. dtm: %s, gid: %s", tcc.Dtm, tcc.Gid)
|
|
}
|
|
return tcc, nil
|
|
}
|
|
|
|
// CallBranch call a tcc branch
|
|
func (t *Tcc) CallBranch(body interface{}, tryURL string, confirmURL string, cancelURL string) (*resty.Response, error) {
|
|
branchID := t.NewBranchID()
|
|
resp, err := common.RestyClient.R().
|
|
SetBody(&M{
|
|
"gid": t.Gid,
|
|
"branch_id": branchID,
|
|
"trans_type": "tcc",
|
|
"status": "prepared",
|
|
"data": string(common.MustMarshal(body)),
|
|
"try": tryURL,
|
|
"confirm": confirmURL,
|
|
"cancel": cancelURL,
|
|
}).
|
|
Post(t.Dtm + "/registerTccBranch")
|
|
if err != nil {
|
|
return resp, err
|
|
}
|
|
if !strings.Contains(resp.String(), "SUCCESS") {
|
|
return nil, fmt.Errorf("registerTccBranch failed: %s", resp.String())
|
|
}
|
|
return common.RestyClient.R().
|
|
SetBody(body).
|
|
SetQueryParams(common.MS{
|
|
"dtm": t.Dtm,
|
|
"gid": t.Gid,
|
|
"branch_id": branchID,
|
|
"trans_type": "tcc",
|
|
"branch_type": "try",
|
|
}).
|
|
Post(tryURL)
|
|
}
|