GetBatch 获取批量上传会话
(batchID string)
| 262 | |
| 263 | // GetBatch 获取批量上传会话 |
| 264 | func (m *BatchUploadManager) GetBatch(batchID string) (*BatchUpload, error) { |
| 265 | m.mu.RLock() |
| 266 | batch, exists := m.batches[batchID] |
| 267 | m.mu.RUnlock() |
| 268 | |
| 269 | if exists { |
| 270 | return batch, nil |
| 271 | } |
| 272 | |
| 273 | // 从Redis加载 |
| 274 | batch, err := m.progressStorage.LoadBatch(batchID) |
| 275 | if err != nil { |
| 276 | return nil, fmt.Errorf("批次不存在: %s", batchID) |
| 277 | } |
| 278 | |
| 279 | m.mu.Lock() |
| 280 | m.batches[batchID] = batch |
| 281 | m.mu.Unlock() |
| 282 | |
| 283 | return batch, nil |
| 284 | } |
| 285 | |
| 286 | // ValidateUploadToken 验证上传令牌 |
| 287 | func (m *BatchUploadManager) ValidateUploadToken(batchID, token string) bool { |
no test coverage detected