feat: 添加队列工作者配置,增强作业提供者功能,支持并发控制和健康检查
This commit is contained in:
@@ -17,7 +17,19 @@ func DefaultProvider() container.ProviderContainer {
|
||||
}
|
||||
}
|
||||
|
||||
type Config struct{}
|
||||
type Config struct {
|
||||
// Optional per-queue worker concurrency. If empty, defaults apply.
|
||||
QueueWorkers QueueWorkersConfig
|
||||
}
|
||||
|
||||
// QueueWorkers allows configuring worker concurrency per queue.
|
||||
// Key is the queue name, value is MaxWorkers. If empty, defaults are used.
|
||||
// Example TOML:
|
||||
//
|
||||
// [Job]
|
||||
// # high=20, default=10, low=5
|
||||
// # QueueWorkers = { high = 20, default = 10, low = 5 }
|
||||
type QueueWorkersConfig map[string]int
|
||||
|
||||
const (
|
||||
PriorityDefault = river.PriorityDefault
|
||||
@@ -31,3 +43,25 @@ const (
|
||||
QueueDefault = river.QueueDefault
|
||||
QueueLow = "low"
|
||||
)
|
||||
|
||||
// queueConfig returns a river.QueueConfig map built from QueueWorkers or defaults.
|
||||
func (c *Config) queueConfig() map[string]river.QueueConfig {
|
||||
cfg := map[string]river.QueueConfig{}
|
||||
if c == nil || len(c.QueueWorkers) == 0 {
|
||||
cfg[QueueHigh] = river.QueueConfig{MaxWorkers: 10}
|
||||
cfg[QueueDefault] = river.QueueConfig{MaxWorkers: 10}
|
||||
cfg[QueueLow] = river.QueueConfig{MaxWorkers: 10}
|
||||
return cfg
|
||||
}
|
||||
for name, n := range c.QueueWorkers {
|
||||
if n <= 0 {
|
||||
n = 1
|
||||
}
|
||||
cfg[name] = river.QueueConfig{MaxWorkers: n}
|
||||
}
|
||||
|
||||
if _, ok := cfg[QueueDefault]; !ok {
|
||||
cfg[QueueDefault] = river.QueueConfig{MaxWorkers: 10}
|
||||
}
|
||||
return cfg
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user