dtm/dtmcli/types.go

package dtmcli

import (
	"fmt"

	"github.com/yedf/dtm/common"
)

// GenGid generate a new gid
func GenGid(server string) string {
	res := common.MS{}
	_, err := common.RestyClient.R().SetResult(&res).Get(server + "/newGid")
	e2p(err)
	return res["gid"]
}

// IDGenerator used to generate a branch id
type IDGenerator struct {
	parentID string
	branchID int
}

// NewBranchID generate a branch id
func (g *IDGenerator) NewBranchID() string {
	if g.branchID >= 99 {
		panic(fmt.Errorf("branch id is larger than 99"))
	}
	if len(g.parentID) >= 20 {
		panic(fmt.Errorf("total branch id is longer than 20"))
	}
	g.branchID = g.branchID + 1
	return g.parentID + fmt.Sprintf("%02d", g.branchID)
}