feat: wrapper asynq

This commit is contained in:
tiglog 2023-08-16 07:27:28 +08:00
parent 47f14f94ab
commit 3ecd10b367
2 changed files with 34 additions and 4 deletions

View File

@ -29,7 +29,7 @@ func Init(addr, username, password string, database int) {
var cli *asynq.Client
func Client() *asynq.Client {
func GetClient() *asynq.Client {
onceCli.Do(func() {
cli = asynq.NewClient(redisOpt)
})
@ -38,7 +38,7 @@ func Client() *asynq.Client {
var svc *asynq.Server
func ServerWithArgs(concurrency int, fn asynq.ErrorHandlerFunc) *asynq.Server {
func GetServerWithArgs(concurrency int, fn asynq.ErrorHandlerFunc) *asynq.Server {
onceSvc.Do(func() {
svc = asynq.NewServer(
redisOpt,
@ -51,7 +51,7 @@ func ServerWithArgs(concurrency int, fn asynq.ErrorHandlerFunc) *asynq.Server {
return svc
}
func ServerWithConfig(cfg asynq.Config) *asynq.Server {
func GetServerWithConfig(cfg asynq.Config) *asynq.Server {
onceSvc.Do(func() {
svc = asynq.NewServer(
redisOpt,
@ -61,7 +61,7 @@ func ServerWithConfig(cfg asynq.Config) *asynq.Server {
return svc
}
func Server() *asynq.Server {
func GetServer() *asynq.Server {
onceSvc.Do(func() {
svc = asynq.NewServer(
redisOpt,

30
gqueue/wrapper.go Normal file
View File

@ -0,0 +1,30 @@
//
// wrapper.go
// Copyright (C) 2023 tiglog <me@tiglog.com>
//
// Distributed under terms of the MIT license.
//
package gqueue
import "github.com/hibiken/asynq"
type Task = asynq.Task
type Client = asynq.Client
type Server = asynq.Server
var GetRetryCount = asynq.GetRetryCount
var GetTaskID = asynq.GetTaskID
var GetQueueName = asynq.GetQueueName
var GetMaxRetry = asynq.GetMaxRetry
var ProcessIn = asynq.ProcessIn
var ProcessAt = asynq.ProcessAt
var MaxRetry = asynq.MaxRetry
var Timeout = asynq.Timeout
var SkipRetry = asynq.SkipRetry
var NewTask = asynq.NewTask
var NewServeMux = asynq.NewServeMux
var NewClient = asynq.NewClient