JCS-pub/common/pkgs/storage/local/local.go

package local

import (
	"fmt"

	clitypes "gitlink.org.cn/cloudream/storage2/client/types"
	"gitlink.org.cn/cloudream/storage2/common/pkgs/storage/factory/reg"
	"gitlink.org.cn/cloudream/storage2/common/pkgs/storage/types"
	"gitlink.org.cn/cloudream/storage2/common/pkgs/storage/utils"
	cortypes "gitlink.org.cn/cloudream/storage2/coordinator/types"
)

func init() {
	reg.RegisterBuilder[*cortypes.LocalStorageType](func(detail *clitypes.UserSpaceDetail) types.StorageBuilder {
		return &builder{
			detail: detail,
		}
	})
}

type builder struct {
	types.EmptyBuilder
	detail *clitypes.UserSpaceDetail
}

func (b *builder) FeatureDesc() types.FeatureDesc {
	return types.FeatureDesc{
		HasBypassWrite: true,
		HasBypassRead:  true,
	}
}

func (b *builder) CreateShardStore() (types.ShardStore, error) {
	return NewShardStore(b.detail)
}

func (b *builder) CreatePublicStore() (types.PublicStore, error) {
	return NewPublicStore(b.detail)
}

func (b *builder) CreateMultiparter() (types.Multiparter, error) {
	feat := utils.FindFeature[*cortypes.MultipartUploadFeature](b.detail.Storage)
	if feat == nil {
		return nil, fmt.Errorf("feature %T not found", cortypes.MultipartUploadFeature{})
	}

	return &Multiparter{
		feat: feat,
	}, nil
}

func (b *builder) CreateS2STransfer() (types.S2STransfer, error) {
	feat := utils.FindFeature[*cortypes.S2STransferFeature](b.detail.Storage)
	if feat == nil {
		return nil, fmt.Errorf("feature %T not found", cortypes.S2STransferFeature{})
	}

	return &S2STransfer{
		detail: b.detail,
	}, nil
}