JCS-pub/hub/internal/rpc/user_space.go

package rpc

import (
	"context"

	"gitlink.org.cn/cloudream/common/consts/errorcode"
	"gitlink.org.cn/cloudream/common/pkgs/logger"
	"gitlink.org.cn/cloudream/jcs-pub/common/pkgs/rpc"
	hubrpc "gitlink.org.cn/cloudream/jcs-pub/common/pkgs/rpc/hub"
)

func (svc *Service) BaseStoreListAll(context context.Context, msg *hubrpc.BaseStoreListAll) (*hubrpc.BaseStoreListAllResp, *rpc.CodeError) {
	pub, err := svc.stgPool.GetBaseStore(&msg.UserSpace)
	if err != nil {
		return nil, rpc.Failed(errorcode.OperationFailed, err.Error())
	}

	es, err := pub.ListAll(msg.Path)
	if err != nil {
		return nil, rpc.Failed(errorcode.OperationFailed, err.Error())
	}

	return &hubrpc.BaseStoreListAllResp{
		Entries: es,
	}, nil
}

func (svc *Service) BaseStoreMkdirs(context context.Context, msg *hubrpc.BaseStoreMkdirs) (*hubrpc.BaseStoreMkdirsResp, *rpc.CodeError) {
	pub, err := svc.stgPool.GetBaseStore(&msg.UserSpace)
	if err != nil {
		return nil, rpc.Failed(errorcode.OperationFailed, err.Error())
	}

	var suc []bool
	for _, p := range msg.Pathes {
		if err := pub.Mkdir(p); err != nil {
			suc = append(suc, false)
			logger.Warnf("userspace %v mkdir %s: %v", msg.UserSpace, p, err)
		} else {
			suc = append(suc, true)
		}
	}

	return &hubrpc.BaseStoreMkdirsResp{
		Successes: suc,
	}, nil
}