aiforge/modules/worker/worker.go

package worker

import (
	"code.gitea.io/gitea/modules/setting"
	"github.com/RichardKnop/machinery/v1"
	mchConf "github.com/RichardKnop/machinery/v1/config"
)

var (
	AsyncTaskCenter *machinery.Server
)

func init() {
	tc, err := NewTaskCenter()
	if err != nil {
		panic(err)
	}
	AsyncTaskCenter = tc
}

func NewTaskCenter() (*machinery.Server, error) {
	cnf := &mchConf.Config{
		Broker:        setting.Broker,
		DefaultQueue:  setting.DefaultQueue,
		ResultBackend: setting.ResultBackend,
	}
	// Create server instance
	return machinery.NewServer(cnf)
}