common/rag/eino/base_task.go

package eino

import (
	"time"

	"gitea.com/red-future/common/beans"
)

// BaseTask 任务基类 - MongoDB版本
type BaseTask struct {
	beans.MongoBaseDO `bson:",inline"`
	// 任务信息
	TaskType TaskType     `bson:"taskType" json:"taskType"`
	Status   TaskStatus   `bson:"status" json:"status"`
	Priority TaskPriority `bson:"priority,omitempty" json:"priority,omitempty"`
	// 进度
	TotalItems     int64   `bson:"totalItems" json:"totalItems"`
	ProcessedItems int64   `bson:"processedItems" json:"processedItems"`
	Progress       float64 `bson:"progress" json:"progress"`
	// 结果
	StartTime    *time.Time `bson:"startTime" json:"startTime"`
	EndTime      *time.Time `bson:"endTime,omitempty" json:"endTime,omitempty"`
	Duration     int64      `bson:"duration,omitempty" json:"duration,omitempty"`
	SuccessCount int64      `bson:"successCount" json:"successCount"`
	FailCount    int64      `bson:"failCount" json:"failCount"`
	// 其他
	Executor string `bson:"executor,omitempty" json:"executor,omitempty"`
}

// SQLBaseTask 任务基类 - SQL版本
type SQLBaseTask struct {
	beans.SQLBaseDO
	// 任务信息
	TaskType TaskType     `json:"taskType"`
	Status   TaskStatus   `json:"status"`
	Priority TaskPriority `json:"priority,omitempty"`
	// 进度
	TotalItems     int64   `json:"totalItems"`
	ProcessedItems int64   `json:"processedItems"`
	Progress       float64 `json:"progress"`
	// 结果
	StartTime    *time.Time `json:"startTime"`
	EndTime      *time.Time `json:"endTime,omitempty"`
	Duration     int64      `json:"duration,omitempty"`
	SuccessCount int64      `json:"successCount"`
	FailCount    int64      `json:"failCount"`
	// 其他
	Executor string `json:"executor,omitempty"`
}