JCS-pub/common/pkgs/publock/reqbuilder/package.go

package reqbuilder

import (
	"strconv"

	"gitlink.org.cn/cloudream/jcs-pub/common/pkgs/publock/lockprovider"
	"gitlink.org.cn/cloudream/jcs-pub/common/pkgs/publock/types"
	jcstypes "gitlink.org.cn/cloudream/jcs-pub/common/types"
)

type PackageLockReqBuilder struct {
	*LockRequestBuilder
}

func (b *LockRequestBuilder) Package() *PackageLockReqBuilder {
	return &PackageLockReqBuilder{LockRequestBuilder: b}
}
func (b *PackageLockReqBuilder) Buzy(pkgID jcstypes.PackageID) *PackageLockReqBuilder {
	b.locks = append(b.locks, types.Lock{
		Path:   b.makePath(pkgID),
		Name:   lockprovider.PackageBuzyLock,
		Target: lockprovider.NewEmptyTarget(),
	})
	return b
}

func (b *PackageLockReqBuilder) Pin(pkgID jcstypes.PackageID) *PackageLockReqBuilder {
	b.locks = append(b.locks, types.Lock{
		Path:   b.makePath(pkgID),
		Name:   lockprovider.PackagePinLock,
		Target: lockprovider.NewEmptyTarget(),
	})
	return b
}

func (b *PackageLockReqBuilder) makePath(pkgID jcstypes.PackageID) []string {
	return []string{lockprovider.PackageLockPathPrefix, strconv.FormatInt(int64(pkgID), 10)}
}