wechat-job/internal/mqs/amq/task/mqtask/register.go

package mqtask

import (
	"github.com/hibiken/asynq"
	"github.com/suyuan32/simple-admin-job/internal/mqs/amq/handler/amq/wxhook"

	"github.com/suyuan32/simple-admin-job/internal/mqs/amq/handler/amq/base"
	"github.com/suyuan32/simple-admin-job/internal/mqs/amq/types/pattern"
)

// Register adds task to cron. | 在此处定义任务处理逻辑，注册worker.
func (m *MQTask) Register() {
	mux := asynq.NewServeMux()

	// define the handler | 定义处理逻辑
	mux.Handle(pattern.RecordHelloWorld, base.NewHelloWorldHandler(m.svcCtx))

	mux.Handle(pattern.RecordSayMorning, wxhook.NewSayMorningHandler(m.svcCtx))

	mux.Handle(pattern.RecordSendWx, wxhook.NewSendWxHandler(m.svcCtx))

	m.mux = mux
}