dtm/dtmcli/tcc.go

package dtmcli

import (
	"fmt"

	"github.com/gin-gonic/gin"
	"github.com/go-resty/resty/v2"
)

// Tcc struct of tcc
type Tcc struct {
	Gid string
	TransBase
}

// TccGlobalFunc type of global tcc call
type TccGlobalFunc func(tcc *Tcc) (*resty.Response, error)

// TccGlobalTransaction begin a tcc global transaction
// dtm dtm服务器地址
// gid 全局事务id
// tccFunc tcc事务函数，里面会定义全局事务的分支
func TccGlobalTransaction(dtm string, gid string, tccFunc TccGlobalFunc) (rerr error) {
	data := &M{
		"gid":        gid,
		"trans_type": "tcc",
	}
	tcc := &Tcc{TransBase: TransBase{Dtm: dtm}, Gid: gid}
	rerr = tcc.CallDtm(data, "prepare")
	if rerr != nil {
		return rerr
	}
	// 小概率情况下，prepare成功了，但是由于网络状况导致上面Failure，那么不执行下面defer的内容，等待超时后再回滚标记事务失败，也没有问题
	defer func() {
		x := recover()
		operation := If(x == nil && rerr == nil, "submit", "abort").(string)
		err := tcc.CallDtm(data, operation)
		if rerr == nil {
			rerr = err
		}
		if x != nil {
			panic(x)
		}
	}()
	resp, rerr := tccFunc(tcc)
	rerr = CheckResponse(resp, rerr)
	return
}

// TccFromReq tcc from request info
func TccFromReq(c *gin.Context) (*Tcc, error) {
	tcc := &Tcc{
		TransBase: *TransBaseFromReq(c),
		Gid:       c.Query("gid"),
	}
	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
// 函数首先注册子事务的所有分支，成功后调用try分支，返回try分支的调用结果
func (t *Tcc) CallBranch(body interface{}, tryURL string, confirmURL string, cancelURL string) (*resty.Response, error) {
	branchID := t.NewBranchID()
	err := t.CallDtm(&M{
		"gid":        t.Gid,
		"branch_id":  branchID,
		"trans_type": "tcc",
		"status":     "prepared",
		"data":       string(MustMarshal(body)),
		"try":        tryURL,
		"confirm":    confirmURL,
		"cancel":     cancelURL,
	}, "registerTccBranch")
	if err != nil {
		return nil, err
	}
	resp, err := RestyClient.R().
		SetBody(body).
		SetQueryParams(MS{
			"dtm":         t.Dtm,
			"gid":         t.Gid,
			"branch_id":   branchID,
			"trans_type":  "tcc",
			"branch_type": "try",
		}).
		Post(tryURL)
	return resp, CheckResponse(resp, err)
}