58 lines
1.3 KiB
Go
58 lines
1.3 KiB
Go
package worker
|
|
|
|
import (
|
|
"ferry/pkg/logger"
|
|
"fmt"
|
|
|
|
"github.com/spf13/viper"
|
|
|
|
"github.com/RichardKnop/machinery/v1"
|
|
taskConfig "github.com/RichardKnop/machinery/v1/config"
|
|
"github.com/RichardKnop/machinery/v1/tasks"
|
|
)
|
|
|
|
var AsyncTaskCenter *machinery.Server
|
|
|
|
func StartServer() {
|
|
tc, err := NewTaskCenter()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
AsyncTaskCenter = tc
|
|
}
|
|
|
|
func NewTaskCenter() (*machinery.Server, error) {
|
|
cnf := &taskConfig.Config{
|
|
Broker: fmt.Sprintf("redis://%v:%v",
|
|
viper.GetString("settings.redis.host"),
|
|
viper.GetString("settings.redis.port"),
|
|
),
|
|
DefaultQueue: "ServerTasksQueue",
|
|
ResultBackend: "eager",
|
|
}
|
|
server, err := machinery.NewServer(cnf)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
initAsyncTaskMap()
|
|
return server, server.RegisterTasks(asyncTaskMap)
|
|
}
|
|
|
|
func NewAsyncTaskWorker(concurrency int) *machinery.Worker {
|
|
consumerTag := "TaskWorker"
|
|
worker := AsyncTaskCenter.NewWorker(consumerTag, concurrency)
|
|
errorHandler := func(err error) {
|
|
logger.Error("执行失败: ", err)
|
|
}
|
|
preTaskHandler := func(signature *tasks.Signature) {
|
|
logger.Info("开始执行: ", signature.Name)
|
|
}
|
|
postTaskHandler := func(signature *tasks.Signature) {
|
|
logger.Info("执行结束: ", signature.Name)
|
|
}
|
|
worker.SetPostTaskHandler(postTaskHandler)
|
|
worker.SetErrorHandler(errorHandler)
|
|
worker.SetPreTaskHandler(preTaskHandler)
|
|
return worker
|
|
}
|