ChineseSubFinder/internal/pkg/notify_center/NotifyCenter.go

package notify_center

import (
	"github.com/allanpk716/ChineseSubFinder/internal/pkg/log_helper"
	"github.com/go-resty/resty/v2"
	"net/url"
	"sync"
)

type NotifyCenter struct {
	webhookUrl string
	infos map[string]string
	mu sync.Mutex
}

func NewNotifyCenter(webhookUrl string) *NotifyCenter {
	n := NotifyCenter{webhookUrl: webhookUrl}
	n.infos = make(map[string]string)
	return &n
}

func (n *NotifyCenter) Add(groupName, infoContent string) {
	if n == nil {
		return
	}
	n.mu.Lock()
	defer n.mu.Unlock()
	n.infos[groupName] = infoContent
}

func (n *NotifyCenter) Send() {
	if n == nil || n.webhookUrl == "" {
		return
	}
	client := resty.New()
	for s, s2 := range n.infos {
		_, err := client.R().Get(n.webhookUrl + s + "/" + url.QueryEscape(s2))
		if err != nil {
			log_helper.GetLogger().Errorln("NewNotifyCenter.Send", err)
			return
		}
	}
}

func (n *NotifyCenter) Clear() {
	if n == nil {
		return
	}
	n.infos = make(map[string]string)
}

var Notify *NotifyCenter