rag/internal/providers/asynq.go
2024-06-13 09:16:48 +08:00

23 lines
463 B
Go

package providers
import "github.com/hibiken/asynq"
var AsynQClient *asynq.Client
var AsynQServer *asynq.Server
func getAsynQRedisOpt() asynq.RedisClientOpt {
return asynq.RedisClientOpt{
Addr: Config.Redis.Addr,
Password: Config.Redis.Pass,
DB: 0,
}
}
func InitAsynQClient() {
AsynQClient = asynq.NewClient(getAsynQRedisOpt())
}
func InitAsynQServer() {
AsynQServer = asynq.NewServer(getAsynQRedisOpt(), asynq.Config{Concurrency: 10})
}