break
c7f9e005af
Some checks failed
构建并部署到测试环境（无 SSH） / build-and-deploy (push) Failing after 1h43m42s
- 迁移 000221：新增 tb_business_user_group、tb_business_user_group_member、tb_shop_business_owner_import_task，成员一账号一行由部分唯一索引保证，店铺所属组按当前负责人实时推导，不回填历史分组。 - 用户组 CRUD、成员改组/清空归属、店铺批量交接（原子失败不部分写入）。 - 店铺负责人 CSV 导入任务：逐行独立事务、逐行明细、任务级与行级失败分离。 - 读侧推导与筛选：未分组、业务线、停用组可筛出并带停用标记。 - 补齐操作审计动作与资源、openapi 清单、发布门禁巡检表清单。 - 归档 add-shop-salesperson-groups 变更并同步 openspec/specs/business-user-group，补齐 AUG26-003 验证证据链。
118 lines
2.6 KiB
Go
118 lines
2.6 KiB
Go
package storage
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"path/filepath"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/google/uuid"
|
|
|
|
"github.com/break/junhong_cmp_fiber/pkg/config"
|
|
"github.com/break/junhong_cmp_fiber/pkg/constants"
|
|
"github.com/break/junhong_cmp_fiber/pkg/errors"
|
|
)
|
|
|
|
type Service struct {
|
|
provider Provider
|
|
config *config.StorageConfig
|
|
}
|
|
|
|
func NewService(provider Provider, cfg *config.StorageConfig) *Service {
|
|
return &Service{
|
|
provider: provider,
|
|
config: cfg,
|
|
}
|
|
}
|
|
|
|
func (s *Service) GenerateFileKey(purpose, fileName string) (string, error) {
|
|
mapping, ok := PurposeMappings[purpose]
|
|
if !ok {
|
|
return "", errors.New(errors.CodeInvalidParam, "不支持的文件用途: %s", purpose)
|
|
}
|
|
|
|
ext := filepath.Ext(fileName)
|
|
if ext == "" {
|
|
ext = ".bin"
|
|
}
|
|
if (purpose == constants.StoragePurposeAssetPackageBatchOrder || purpose == constants.StoragePurposeDeviceBatchAllocation || purpose == constants.StoragePurposeShopBusinessOwnerImport) && !strings.EqualFold(ext, ".csv") {
|
|
return "", errors.New(errors.CodeInvalidParam, "批量业务文件必须为CSV格式")
|
|
}
|
|
|
|
now := time.Now()
|
|
id := uuid.New().String()
|
|
|
|
key := fmt.Sprintf("%s/%04d/%02d/%02d/%s%s",
|
|
mapping.Prefix,
|
|
now.Year(),
|
|
now.Month(),
|
|
now.Day(),
|
|
id,
|
|
strings.ToLower(ext),
|
|
)
|
|
|
|
return key, nil
|
|
}
|
|
|
|
func (s *Service) GetUploadURL(ctx context.Context, purpose, fileName, contentType string) (*PresignResult, error) {
|
|
fileKey, err := s.GenerateFileKey(purpose, fileName)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if contentType == "" {
|
|
if mapping, ok := PurposeMappings[purpose]; ok && mapping.ContentType != "" {
|
|
contentType = mapping.ContentType
|
|
} else {
|
|
contentType = "application/octet-stream"
|
|
}
|
|
}
|
|
|
|
expires := s.config.Presign.UploadExpires
|
|
if expires == 0 {
|
|
expires = 15 * time.Minute
|
|
}
|
|
|
|
url, err := s.provider.GetUploadURL(ctx, fileKey, contentType, expires)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &PresignResult{
|
|
URL: url,
|
|
FileKey: fileKey,
|
|
ExpiresIn: int(expires.Seconds()),
|
|
}, nil
|
|
}
|
|
|
|
func (s *Service) GetDownloadURL(ctx context.Context, fileKey string) (*PresignResult, error) {
|
|
expires := s.config.Presign.DownloadExpires
|
|
if expires == 0 {
|
|
expires = 24 * time.Hour
|
|
}
|
|
|
|
url, err := s.provider.GetDownloadURL(ctx, fileKey, expires)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &PresignResult{
|
|
URL: url,
|
|
FileKey: fileKey,
|
|
ExpiresIn: int(expires.Seconds()),
|
|
}, nil
|
|
}
|
|
|
|
func (s *Service) DownloadToTemp(ctx context.Context, fileKey string) (string, func(), error) {
|
|
return s.provider.DownloadToTemp(ctx, fileKey)
|
|
}
|
|
|
|
func (s *Service) Provider() Provider {
|
|
return s.provider
|
|
}
|
|
|
|
func (s *Service) Bucket() string {
|
|
return s.config.S3.Bucket
|
|
}
|