support push choice

This commit is contained in:
akrike
2025-02-05 13:46:16 +08:00
parent 4fa227ee82
commit 06cda6821a
3 changed files with 9 additions and 3 deletions

View File

@@ -31,3 +31,8 @@ func (p *pushRepository) DeletePushConfig(id int) error {
Id: int64(id),
}).Error
}
func (p *pushRepository) GetPushConfigByIds(ids []int) (result []model.Push) {
db.Model(&model.Push{}).Where("id in ?", ids).Find(&result)
return
}

View File

@@ -56,6 +56,7 @@ type ProcessBase struct {
statuPush bool
logReport bool
cgroupEnable bool
PushIds []int
memoryLimit *float32
cpuLimit *float32
}
@@ -240,7 +241,7 @@ func (p *ProcessBase) push(message string) {
"{$message}": message,
"{$status}": strconv.Itoa(int(p.State.State)),
}
PushService.Push(messagePlaceholders)
PushService.Push(p.Config.PushIds, messagePlaceholders)
}
}

View File

@@ -13,8 +13,8 @@ type pushService struct{}
var PushService = new(pushService)
func (p *pushService) Push(placeholders map[string]string) {
pl := repository.PushRepository.GetPushList()
func (p *pushService) Push(ids []int, placeholders map[string]string) {
pl := repository.PushRepository.GetPushConfigByIds(ids)
for _, v := range pl {
if v.Enable {
if v.Method == http.MethodGet {