七月迭代短暂完结,还有很多后端的关键东西没有弄,这是一版赶时间做的东西

pkg/storage/service.go

@@ -10,6 +10,7 @@ import (
 	"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"
 )
 
@@ -35,6 +36,9 @@ func (s *Service) GenerateFileKey(purpose, fileName string) (string, error) {
 	if ext == "" {
 		ext = ".bin"
 	}
+	if (purpose == constants.StoragePurposeAssetPackageBatchOrder || purpose == constants.StoragePurposeDeviceBatchAllocation) && !strings.EqualFold(ext, ".csv") {
+		return "", errors.New(errors.CodeInvalidParam, "批量业务文件必须为CSV格式")
+	}
 
 	now := time.Now()
 	id := uuid.New().String()

pkg/storage/types.go

@@ -1,5 +1,7 @@
 package storage
 
+import "github.com/break/junhong_cmp_fiber/pkg/constants"
+
 type PresignResult struct {
 	URL       string `json:"url"`
 	FileKey   string `json:"file_key"`
@@ -15,4 +17,6 @@ var PurposeMappings = map[string]PurposeMapping{
 	"iot_import": {Prefix: "imports", ContentType: "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"},
 	"export":     {Prefix: "exports", ContentType: "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"},
 	"attachment": {Prefix: "attachments", ContentType: ""},
+	constants.StoragePurposeAssetPackageBatchOrder: {Prefix: constants.AssetPackageBatchOrderStoragePrefix, ContentType: "text/csv"},
+	constants.StoragePurposeDeviceBatchAllocation:  {Prefix: constants.DeviceBatchAllocationStoragePrefix, ContentType: "text/csv"},
 }