Method CreateBatch

api/service/batch_upload_manager.go:195–261

CreateBatch 创建批量上传会话

(eid, userID int64, req *BatchInitRequest)

// CreateBatch 创建批量上传会话
func (m *BatchUploadManager) CreateBatch(eid, userID int64, req *BatchInitRequest) (*BatchUpload, []DuplicateFileInfo, error) {
	dirManager := NewDirectoryManager()
	if err := dirManager.CreateDirectoryTree(eid, req.LibraryID, req.FileStructure, req.BasePath, userID, req.OriginType, req.OriginSource, req.OriginRefID); err != nil {
		return nil, nil, fmt.Errorf("创建目录结构失败: %v", err)
	}

	batchID := m.generateBatchID()
	uploadToken := m.generateUploadToken()

	files := make(map[string]*FileUpload)
	var duplicateFiles []DuplicateFileInfo

	for _, item := range req.FileStructure {
		if !item.IsDirectory {
			fileID := m.generateFileID()
			files[fileID] = &FileUpload{
				ID:           fileID,
				RelativePath: item.RelativePath,
				Status:       "queued",
				Progress:     0,
				TotalSize:    item.Size,
				StartTime:    time.Now(),
				UpdateTime:   time.Now(),
				SpeedHistory: make([]int64, 0),
			}

			filePath := dirManager.GetDirectoryPath(req.BasePath, item.RelativePath)
			if shouldAppendMarkdownSuffix() && filepath.Ext(filePath) != ".md" {
				filePath += ".md"
			}
			existingFile, err := model.GetFileByPathAndLibraryNotDeleted(eid, req.LibraryID, filePath)
			if err == nil && existingFile != nil {
				duplicateFiles = append(duplicateFiles, DuplicateFileInfo{
					RelativePath: item.RelativePath,
					ExistingID:   existingFile.ID,
				})
			}
		}
	}

	batch := &BatchUpload{
		ID:            batchID,
		LibraryID:     req.LibraryID,
		UserID:        userID,
		EID:           eid,
		Status:        "init",
		TotalFiles:    req.TotalFiles,
		UploadedFiles: 0,
		FailedFiles:   0,
		Files:         files,
		UploadToken:   uploadToken,
		BasePath:      req.BasePath,
		OriginType:    strings.TrimSpace(req.OriginType),
		OriginSource:  strings.TrimSpace(req.OriginSource),
		OriginRefID:   req.OriginRefID,
		CreatedAt:     time.Now(),
		UpdatedAt:     time.Now(),
	}