mayfly-go/server/internal/msg/msgx/msgx.go

package msgx

import (
	"fmt"
	"mayfly-go/pkg/model"
)

type MsgType int8
type ChannelType string

const (
	MsgTypeText     MsgType = 1
	MsgTypeMarkdown MsgType = 2
	MsgTypeHtml     MsgType = 3
)

const (
	ChannelTypeEmail     ChannelType = "email"
	ChannelTypeDingBot   ChannelType = "dingBot"
	ChannelTypeQywxBot   ChannelType = "qywxBot"
	ChannelTypeFeishuBot ChannelType = "feishuBot"
)

const (
	ReceiverKey = "receiver"
)

// Send 发送消息
func Send(channel *Channel, msg *Msg) error {
	sender, err := GetMsgSender(channel.Type)
	if err != nil {
		return err
	}
	return sender.Send(channel, msg)
}

type Receiver struct {
	model.ExtraData

	Mobile string
	Email  string
}

type Msg struct {
	model.ExtraData

	Title   string         // 消息title
	Type    MsgType        // 消息类型
	Content string         // 消息内容
	Params  map[string]any // 消息参数(替换消息中的占位符)

	Receivers []Receiver // 消息接收人
}

// Channel 消息发送渠道信息
type Channel struct {
	model.ExtraData

	Type ChannelType // 渠道类型
	Name string
	URL  string
}

// MsgSender 定义消息发送接口
type MsgSender interface {
	// Send 发送消息
	Send(channel *Channel, msg *Msg) error
}

var messageSenders = make(map[ChannelType]MsgSender)

// RegisterMsgSender 注册消息发送器
func RegisterMsgSender(channel ChannelType, sender MsgSender) {
	messageSenders[channel] = sender
}

// GetMsgSender 获取消息发送器
func GetMsgSender(channel ChannelType) (MsgSender, error) {
	sender, ok := messageSenders[channel]
	if !ok {
		return nil, fmt.Errorf("unsupported message channel %s", channel)
	}
	return sender, nil
}