215 lines
6.5 KiB
Go
215 lines
6.5 KiB
Go
package task
|
||
|
||
import (
|
||
"errors"
|
||
|
||
"context"
|
||
"go-common/app/admin/main/aegis/model/common"
|
||
libtime "go-common/library/time"
|
||
)
|
||
|
||
//...
|
||
const (
|
||
ActionConsumerOff = int8(0)
|
||
ActionConsumerOn = int8(1)
|
||
|
||
TaskStateInit = int8(0)
|
||
TaskStateDispatch = int8(1)
|
||
TaskStateDelay = int8(2)
|
||
TaskStateSubmit = int8(3)
|
||
TaskStateRscSb = int8(4)
|
||
TaskStateClosed = int8(5)
|
||
|
||
TaskConfigAssign = int8(1)
|
||
TaskConfigRangeWeight = int8(2)
|
||
TaskConfigEqualWeight = int8(3)
|
||
|
||
TaskRoleMember = int8(1) //组员
|
||
TaskRoleLeader = int8(2) //组长
|
||
TaskNoRole = int8(0) //无身份
|
||
)
|
||
|
||
// ErrEmpty empty pool
|
||
var (
|
||
ErrEmpty = errors.New("empty pool")
|
||
ErrRole = errors.New("不在用户组内")
|
||
)
|
||
|
||
// Task ..
|
||
type Task struct {
|
||
ID int64 `form:"id" json:"id" gorm:"AUTO_INCREMENT;primary_key;"`
|
||
BusinessID int64 `form:"business_id" json:"business_id" gorm:"column:business_id"`
|
||
FlowID int64 `form:"flow_id" json:"flow_id" gorm:"column:flow_id"`
|
||
RID int64 `form:"rid" json:"rid" gorm:"column:rid"`
|
||
AdminID int64 `form:"admin_id" json:"admin_id" gorm:"column:admin_id"`
|
||
UID int64 `form:"uid" json:"uid" gorm:"column:uid"`
|
||
MID int64 `form:"mid" json:"mid" gorm:"column:mid"`
|
||
State int8 `form:"state" json:"state" gorm:"column:state"`
|
||
Weight int64 `form:"weight" json:"weight" gorm:"column:weight"`
|
||
Utime int64 `form:"utime" json:"utime" gorm:"column:utime"`
|
||
Gtime common.IntTime `form:"gtime" json:"gtime" gorm:"column:gtime"`
|
||
Fans int64 `form:"fans" json:"fans" gorm:"column:fans"`
|
||
Group string `form:"group" json:"group" gorm:"column:group"`
|
||
Reason string `form:"reason" json:"reason" grom:"column:reason"`
|
||
Ctime common.IntTime `form:"ctime" json:"ctime" gorm:"column:ctime"`
|
||
Mtime common.IntTime `form:"mtime" json:"mtime" gorm:"column:mtime"`
|
||
}
|
||
|
||
// TableName ...
|
||
func (t Task) TableName() string {
|
||
return "task"
|
||
}
|
||
|
||
// TempOptions 中间参数
|
||
type TempOptions struct {
|
||
BisLeader bool // 是否组长
|
||
NoCache bool // 不使用缓存
|
||
Action string
|
||
}
|
||
|
||
// NextOptions options for Next
|
||
type NextOptions struct {
|
||
common.BaseOptions
|
||
TempOptions
|
||
SeizeCount int64 `form:"seize_count" default:"10"` // 抢占多少个
|
||
DispatchCount int64 `form:"dispatch_count" default:"1"` // 领取多少个
|
||
}
|
||
|
||
// ListOptions options for List
|
||
type ListOptions struct {
|
||
common.BaseOptions
|
||
common.Pager
|
||
TempOptions
|
||
BisShow bool // 用于列表展示还是直接派发
|
||
State int8 `form:"state"`
|
||
}
|
||
|
||
// SubmitOptions options for Submit
|
||
type SubmitOptions struct {
|
||
common.BaseOptions
|
||
TempOptions
|
||
TaskID int64 `form:"task_id"`
|
||
Utime uint64
|
||
OldUID int64
|
||
OldState int8
|
||
}
|
||
|
||
// DelayOptions options for Delay
|
||
type DelayOptions struct {
|
||
common.BaseOptions
|
||
TaskID int64 `form:"task_id"`
|
||
Reason string `form:"reason"`
|
||
}
|
||
|
||
// ConfigOption .
|
||
type ConfigOption struct {
|
||
common.BaseOptions
|
||
ID int64 `form:"id"`
|
||
Btime string `form:"btime"`
|
||
Etime string `form:"etime"`
|
||
Description string `form:"description"`
|
||
ConfType int8 `form:"conf_type" validate:"required"`
|
||
ConfJSON string `form:"conf_json" validate:"required"`
|
||
}
|
||
|
||
// Config .
|
||
type Config struct {
|
||
ID int64 `form:"id" json:"id" gorm:"AUTO_INCREMENT;primary_key;"`
|
||
ConfJSON string `json:"conf_json" gorm:"column:conf_json"`
|
||
ConfType int8 `form:"conf_type" json:"conf_type" gorm:"column:conf_type"`
|
||
BusinessID int64 `form:"business_id" json:"business_id" gorm:"column:business_id"`
|
||
FlowID int64 `form:"flow_id" json:"flow_id" gorm:"column:flow_id"`
|
||
Btime libtime.Time `form:"btime" json:"btime" gorm:"column:btime"`
|
||
Etime libtime.Time `form:"etime" json:"etime" gorm:"column:etime"`
|
||
State int8 `form:"state" json:"state" gorm:"column:state"`
|
||
UID int64 `form:"uid" json:"uid" gorm:"column:uid"`
|
||
Uname string `form:"uname" json:"uname" gorm:"column:uname"`
|
||
Description string `form:"description" json:"description" gorm:"column:description"`
|
||
Ctime libtime.Time `form:"ctime" json:"ctime" gorm:"column:ctime"`
|
||
Mtime libtime.Time `form:"mtime" json:"mtime" gorm:"column:mtime"`
|
||
}
|
||
|
||
// TableName for orm
|
||
func (c Config) TableName() string {
|
||
return "task_config"
|
||
}
|
||
|
||
// EqualWeightConfig 等值权重
|
||
type EqualWeightConfig struct {
|
||
Name string `json:"name"` // taskid 或者 mid
|
||
IDs string `json:"ids"`
|
||
Weight int64 `json:"weight"`
|
||
Type int8 `json:"type"` // 周期或者定值
|
||
}
|
||
|
||
// RangeWeightConfig 权重
|
||
type RangeWeightConfig struct {
|
||
Name string `json:"name"`
|
||
Range []*RangeConfig `json:"range"`
|
||
}
|
||
|
||
// RangeConfig 范围配置
|
||
type RangeConfig struct {
|
||
Threshold int64 `json:"threshold"`
|
||
Weight int64 `json:"weight"`
|
||
}
|
||
|
||
// AssignConfig 指派
|
||
type AssignConfig struct {
|
||
MIDs []int64 `json:"mids"`
|
||
UIDs []int64 `json:"uids"`
|
||
}
|
||
|
||
// QueryParams 配置筛选参数
|
||
type QueryParams struct {
|
||
common.Pager
|
||
ConfType int8 `form:"conf_type"`
|
||
State int8 `form:"state"`
|
||
BusinessID int64 `form:"business_id"`
|
||
FlowID int64 `form:"flow_id"`
|
||
Btime string `form:"mtime_from"`
|
||
Etime string `form:"mtime_to"`
|
||
|
||
ConfName string `form:"conf_name"` // 筛选配置具体类型，fans,group,waittime,mid,taskid
|
||
IDFilter string `form:"id_filter"` // 筛选具体的ID
|
||
TypeFilter string `form:"type_filter"` // 筛选动态或静态权重
|
||
}
|
||
|
||
// History 任务日志
|
||
type History struct {
|
||
TaskID int64 `json:"task_id"`
|
||
AdminID int64
|
||
UID int64
|
||
Reason string
|
||
Uname string
|
||
Action int8
|
||
}
|
||
|
||
// UnDOStat undo stat
|
||
type UnDOStat struct {
|
||
Assign int64 `json:"assign_count" gorm:"column:assign"`
|
||
Delay int64 `json:"delay_count" gorm:"column:delay"`
|
||
Normal int64 `json:"normal_count" gorm:"column:normal"`
|
||
}
|
||
|
||
// Stat 列表页最上方
|
||
type Stat struct {
|
||
Normal int64 `json:"normal_count" gorm:"column:normal"`
|
||
Assign int64 `json:"assign_count" gorm:"column:assign"`
|
||
|
||
DelayTotal int64 `json:"delay_total" gorm:"column:delayTotal"`
|
||
DelayPersonal int64 `json:"delay_personal" gorm:"column:delayPersonal"`
|
||
|
||
ReviewTotal int64 `json:"review_total"`
|
||
ReviewPersonal int64 `json:"review_personal"`
|
||
}
|
||
|
||
// RangeFunc .
|
||
type RangeFunc func(context.Context, *ListOptions) (map[int64]*Task, int64, []int64, []int64, error)
|
||
|
||
// RemoveFunc .
|
||
type RemoveFunc func(context.Context, *common.BaseOptions, ...interface{}) error
|
||
|
||
// ListFuncDB .
|
||
type ListFuncDB func(context.Context, map[int64]*Task, []int64, ...interface{}) (map[int64]struct{}, error)
|