Method GetBatchProgress

api/service/progress_storage.go:297–387

GetBatchProgress 获取批次进度

(batchID string, params *ProgressQueryParams)

// GetBatchProgress 获取批次进度
func (s *ProgressStorage) GetBatchProgress(batchID string, params *ProgressQueryParams) (*BatchProgressResponse, error) {
	batch, err := s.LoadBatch(batchID)
	if err != nil {
		return nil, err
	}

	// 计算批次进度信息
	batchProgress := s.calculateBatchProgress(batch)

	response := &BatchProgressResponse{
		BatchID:       batchID,
		Status:        batch.Status,
		BatchProgress: batchProgress,
		LastUpdate:    batch.UpdatedAt.UnixMilli(),
	}

	// 如果需要详细信息
	if params.Detail {
		files := make(map[string]*FileProgress)
		errors := make([]FileError, 0)

		for FileUploadID, fileUpload := range batch.Files {
			// 如果指定了特定文件ID，只返回该文件
			if params.FileUploadID != "" && FileUploadID != params.FileUploadID {
				continue
			}

			// 如果指定了时间戳，只返回更新时间晚于该时间戳的文件
			if params.Since > 0 && fileUpload.UpdateTime.UnixMilli() <= params.Since {
				continue
			}

			fileProgress := &FileProgress{
				FileUploadID: FileUploadID,
				RelativePath: fileUpload.RelativePath,
				Status:       fileUpload.Status,
				Progress:     fileUpload.Progress,
				UploadedSize: fileUpload.UploadedSize,
				TotalSize:    fileUpload.TotalSize,
				Speed:        fileUpload.Speed,
				ETA:          fileUpload.ETA,
				Error:        fileUpload.Error,
			}

			// 如果有数据库ID，从数据库获取文件信息
			if fileUpload.DatabaseID != 0 {
				// 从数据库获取文件详细信息
				dbFile, err := model.GetUploadFileByID(fileUpload.DatabaseID)
				if err == nil && dbFile != nil {
					// 添加数据库ID到响应中
					fileProgress.DatabaseID = dbFile.ID
					// 如果有错误信息，确保包含在响应中
					if fileProgress.Status == "failed" && (dbFile.Error != "" && fileProgress.Error == "") {
						fileProgress.Error = dbFile.Error
					}
				}
			}