feat(gqueue): 可以配置 server

This commit is contained in:
tiglog 2023-07-05 18:57:31 +08:00
parent 017a2dfc49
commit b3ed0b7429

View File

@ -38,11 +38,36 @@ func Client() *asynq.Client {
var svc *asynq.Server var svc *asynq.Server
func Server() *asynq.Server { func ServerWithArgs(concurrency int, fn asynq.ErrorHandlerFunc) *asynq.Server {
onceSvc.Do(func() { onceSvc.Do(func() {
svc = asynq.NewServer( svc = asynq.NewServer(
redisOpt, redisOpt,
asynq.Config{Concurrency: 10}, asynq.Config{
Concurrency: concurrency,
ErrorHandler: fn,
},
)
})
return svc
}
func ServerWithConfig(cfg asynq.Config) *asynq.Server {
onceSvc.Do(func() {
svc = asynq.NewServer(
redisOpt,
cfg,
)
})
return svc
}
func Server() *asynq.Server {
onceSvc.Do(func() {
svc = asynq.NewServer(
redisOpt,
asynq.Config{
Concurrency: 0,
},
) )
}) })
return svc return svc