ut

type BasePage struct {
	Page     int64 `json:"page,omitempty"`
	PageSize int64 `json:"page_size,omitempty"`
}

type BaseQuery struct {
	*BasePage
	*BaseSort
}

type BaseSort struct {
	SortDesc []string `json:"sort_desc,omitempty"`
	SortAsc  []string `json:"sort_asc,omitempty"`
}

type ConcurrentMap[K comparable, V any] struct {
	// contains filtered or unexported fields
}

type FuncCalcTime struct {
	Name string

	Stage     []timeItem
	LinePrint bool
	// contains filtered or unexported fields
}

type GenericSlice[T any] []T

type Geo struct {
	Type        string    `json:"type" bson:"type,omitempty" mapstructure:"type"` // Point
	Coordinates []float64 `json:"coordinates" bson:"coordinates,omitempty"`       // lng,lat
}

type GeoItem struct {
	Field   string  `json:"field,omitempty"`    // 获取的字段
	ToField string  `json:"to_field,omitempty"` // 距离字段名
	Lng     float64 `json:"lng,omitempty"`      // 范围 -180 到 180
	Lat     float64 `json:"lat,omitempty"`      // 范围 -90 到 90
	GeoMax  int64   `json:"geo_max,omitempty"`  // 单位为米
	GeoMin  int64   `json:"geo_min,omitempty"`  // 单位为米
}

type GormQueryResult[T any] struct {
	Count int64 `json:"count"`
	Datas []*T  `json:"datas"`
}

type Kov struct {
	Key   string `json:"key,omitempty" bson:"key,omitempty"` // 格式是 "abc.die.ede"
	Op    string `json:"op,omitempty" bson:"op,omitempty"`   // 格式为英文 看 AllowOps
	Value any    `json:"value,omitempty" bson:"value,omitempty" jsonschema:"type=string"`
}

type LocalWork[T any, R any] struct {
	Name string `json:"name"` // 任务名称
	Tid  string `json:"tid"`  // 任务id 需唯一

	NowCount     atomic.Int64 // 当前已执行个数
	FailCount    atomic.Int64 // 失败次数
	Running      bool         // 当前执行状态
	Concurrency  uint         // 线程数量
	ForceStop    bool         // 是否被强制停止
	PrintProcess bool         // 是否打印进度 默认开启

	SuccessStopLimit uint // 成功多少条后停止

	ChanData []T // 待处理的数据

	Results   ThreadSafeArray[R]
	OnSuccess func(work *LocalWork[T, R], results []R) error

	// 权重调度相关
	WeightedClients []WeightedClient // 带权重的客户端列表
	// contains filtered or unexported fields
}

type ModelCtxMapperPack struct {
	InjectData map[string]any `json:"inject_data,omitempty"` // 注入数据 默认覆盖
	DropKeys   []string       `json:"drop_keys,omitempty"`   // 需要丢弃的key
}

type MongoFacetResult struct {
	Count int64 `json:"count,omitempty"`
	Data  any   `json:"data"`
}

type Pk struct {
	LocalKey      string `json:"local_key,omitempty"`
	RemoteModelId string `json:"remote_model_id,omitempty"`
	RemoteKey     string `json:"remote_key,omitempty"`
	Alias         string `json:"alias,omitempty"`
	EmptyReturn   bool   `json:"empty_return,omitempty"`
	Unwind        bool   `json:"unwind,omitempty"`
}

type PruneCtxQuery struct {
	// contains filtered or unexported fields
}

type QueryFull struct {
	*QueryParse
	*BaseQuery
	Pks      []*Pk `json:"pks,omitempty"`
	GetCount bool  `json:"get_count,omitempty"` // 获取过滤总数
}

type QueryParse struct {
	And  []*Kov   `json:"and,omitempty"`
	Or   []*Kov   `json:"or,omitempty"`
	Geos *GeoItem `json:"geos,omitempty"`
}

type RedisWork[T any] struct {
	PlatformName      string                                  // 平台名
	PlatformEng       string                                  // 平台英文名
	Name              string                                  // 任务名称
	Description       string                                  // 更多描述
	BulkIds           []string                                // id列表
	BulkStart         int64                                   // id起始值
	BulkEnd           int64                                   // id截止值
	DefaultStartCount int64                                   // 默认启动数
	DefaultEndCount   int64                                   // 默认截止数值
	NowCount          atomic.Int64                            // 当前已执行个数
	FileName          string                                  // 保存文件名
	Concurrency       uint                                    // 线程数量
	Results           []T                                     // 结果保存
	Running           bool                                    // 当前执行状态
	FetchCall         func(id string) ([]T, error)            // 获取单个
	FetchError        func(id string, err error) ([]T, error) // 获取错误
	ForceStop         bool                                    // 是否被强制停止
	RangeReverse      bool                                    // 范围旋转 默认从小到大 如果设置为true 则是从大到小
	InjectData        map[string]any                          //
	OnSuccess         func(self *RedisWork[T]) error          // 当成功时执行
	OnStartup         func(self *RedisWork[T]) error          // 启动时
	OnRangeStart      func(scopes []string, self *RedisWork[T]) []string
	OnRangSuccess     func(self *RedisWork[T], results []T, resultMap map[string][]T) error
	OnSingleSuccess   func(self *RedisWork[T], results []T, target string) error
	ItemDelayStart    time.Duration // 每获取一个的间隔时间起始点
	ItemDelayEnd      time.Duration // 每获取一个的间隔时间终止点
	RangeDelayStart   time.Duration
	RangeDelayEnd     time.Duration
	Stat              RedisWorkStat
	// contains filtered or unexported fields
}

type RedisWorkStat struct {
	AllCount     int64
	RangerCount  int64
	BulkCount    int64
	SuccessCount int64
	FailCount    int64
}

type StructConverter struct {
	CustomIsZero map[reflect.Type]func(reflect.Value, reflect.StructField) bool
	SpecialTypes []reflect.Type // 特殊结构体类型列表
}

type ThreadSafeArray[T any] struct {
	// contains filtered or unexported fields
}

type TrieMatch struct {
	// contains filtered or unexported fields
}

type TrieNode struct {
	// contains filtered or unexported fields
}

type WeightedClient struct {
	Client any  `json:"client"` // The client instance (can be any type)
	Weight uint `json:"weight"` // The weight of this client (higher means more work assigned)
}