JCS-pub/common/pkgs/publock/reqbuilder/user_space.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 UserSpaceLockReqBuilder struct {
	*LockRequestBuilder
}

func (b *LockRequestBuilder) UserSpace() *UserSpaceLockReqBuilder {
	return &UserSpaceLockReqBuilder{LockRequestBuilder: b}
}
func (b *UserSpaceLockReqBuilder) Buzy(spaceID jcstypes.UserSpaceID) *UserSpaceLockReqBuilder {
	b.locks = append(b.locks, types.Lock{
		Path:   b.makePath(spaceID),
		Name:   lockprovider.UserSpaceBuzyLock,
		Target: lockprovider.NewEmptyTarget(),
	})
	return b
}

func (b *UserSpaceLockReqBuilder) GC(spaceID jcstypes.UserSpaceID) *UserSpaceLockReqBuilder {
	b.locks = append(b.locks, types.Lock{
		Path:   b.makePath(spaceID),
		Name:   lockprovider.UserSpaceGCLock,
		Target: lockprovider.NewEmptyTarget(),
	})
	return b
}

func (b *UserSpaceLockReqBuilder) makePath(hubID jcstypes.UserSpaceID) []string {
	return []string{lockprovider.UserSpaceLockPathPrefix, strconv.FormatInt(int64(hubID), 10)}
}