feat: 增加推送通知配置
This commit is contained in:
yml2213
@@ -1,12 +1,16 @@
|
||||
package chat
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"fmt"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"gorm.io/gorm"
|
||||
"gorm.io/gorm/clause"
|
||||
"hfb_sys/backend/internal/integrations/push"
|
||||
"hfb_sys/backend/internal/model"
|
||||
"hfb_sys/backend/internal/modules/supportgroup"
|
||||
)
|
||||
@@ -120,10 +124,17 @@ func EnsureListingConversation(tx *gorm.DB, listing model.RentalListing, preferr
|
||||
}
|
||||
|
||||
// 7. 库存预警检查
|
||||
// 站内信在事务内写入；外部推送在事务提交后发送，避免事务回滚时误发。
|
||||
// 这里通过 goroutine 延迟 1 秒发送，给事务提交留出时间。
|
||||
// 极端情况下事务回滚仍可能误发，但仅是通知层面的轻微不一致，可接受。
|
||||
if qrcode != nil {
|
||||
if err := checkQrCodeStockAndAlert(tx, &conversation); err != nil {
|
||||
// 预警失败不阻塞建群,仅记录日志
|
||||
// TODO: 添加日志
|
||||
if alert, err := checkQrCodeStockAndAlert(tx, &conversation); err == nil && alert != nil {
|
||||
go func(a pushAlert) {
|
||||
time.Sleep(time.Second)
|
||||
for _, p := range a.providers {
|
||||
_ = p.Send(context.Background(), a.message)
|
||||
}
|
||||
}(*alert)
|
||||
}
|
||||
}
|
||||
|
||||
@@ -315,72 +326,134 @@ func markQrCodeAsUsed(tx *gorm.DB, qrcodeID uint64, conversationID uint64) error
|
||||
}).Error
|
||||
}
|
||||
|
||||
func checkQrCodeStockAndAlert(tx *gorm.DB, conversation *model.ChatConversation) error {
|
||||
// 获取库存阈值
|
||||
// pushAlert 事务提交后需要发送的外部推送。
|
||||
type pushAlert struct {
|
||||
providers []push.Provider
|
||||
message push.Message
|
||||
}
|
||||
|
||||
func checkQrCodeStockAndAlert(tx *gorm.DB, conversation *model.ChatConversation) (*pushAlert, error) {
|
||||
// 从 push_rules 获取规则
|
||||
threshold := int64(5)
|
||||
var cfg model.SystemConfig
|
||||
if err := tx.Where("`key` = ?", "chat.qrcode_low_stock_threshold").First(&cfg).Error; err == nil && cfg.Value != "" {
|
||||
// 尝试解析为数字
|
||||
if val, err := parseThreshold(cfg.Value); err == nil {
|
||||
threshold = val
|
||||
ruleEnabled := true
|
||||
messageTemplate := "企业微信群二维码库存不足（剩余 {{.Count}} 张），请及时补充"
|
||||
var rule struct {
|
||||
Enabled bool `gorm:"column:enabled"`
|
||||
Threshold int `gorm:"column:threshold"`
|
||||
MessageTemplate string `gorm:"column:message_template"`
|
||||
}
|
||||
if err := tx.Table("push_rules").Where("event = ?", "qrcode_low_stock").First(&rule).Error; err == nil {
|
||||
ruleEnabled = rule.Enabled
|
||||
threshold = int64(rule.Threshold)
|
||||
if rule.MessageTemplate != "" {
|
||||
messageTemplate = rule.MessageTemplate
|
||||
}
|
||||
}
|
||||
|
||||
if !ruleEnabled {
|
||||
return nil, nil
|
||||
}
|
||||
|
||||
// 统计未使用的二维码数量
|
||||
var count int64
|
||||
if err := tx.Model(&model.ChatQrCode{}).
|
||||
Where("status = ?", QrCodeStatusUnused).
|
||||
Where("expires_at IS NULL OR expires_at > ?", time.Now()).
|
||||
Count(&count).Error; err != nil {
|
||||
return err
|
||||
return nil, err
|
||||
}
|
||||
|
||||
// 如果低于阈值,发送预警给所有客服
|
||||
if count <= threshold {
|
||||
// 查询所有 cs 角色的客服
|
||||
var csAdmins []model.AdminUser
|
||||
if err := tx.Table("admin_users").
|
||||
Joins("JOIN admin_user_roles ON admin_users.id = admin_user_roles.admin_user_id").
|
||||
Joins("JOIN roles ON admin_user_roles.role_id = roles.id").
|
||||
Where("roles.code = ? AND admin_users.status = ?", "cs", "active").
|
||||
Select("admin_users.id").
|
||||
Find(&csAdmins).Error; err != nil {
|
||||
return err
|
||||
if count > threshold {
|
||||
return nil, nil
|
||||
}
|
||||
|
||||
// 低于阈值：写入站内信（事务内）
|
||||
var csAdmins []model.AdminUser
|
||||
if err := tx.Table("admin_users").
|
||||
Joins("JOIN admin_user_roles ON admin_users.id = admin_user_roles.admin_user_id").
|
||||
Joins("JOIN roles ON admin_user_roles.role_id = roles.id").
|
||||
Where("roles.code = ? AND admin_users.status = ?", "cs", "active").
|
||||
Select("admin_users.id").
|
||||
Find(&csAdmins).Error; err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
alertContent := strings.ReplaceAll(messageTemplate, "{{.Count}}", fmt.Sprintf("%d", count))
|
||||
|
||||
entries := make([]map[string]interface{}, 0, len(csAdmins))
|
||||
now := time.Now()
|
||||
for _, admin := range csAdmins {
|
||||
entries = append(entries, map[string]interface{}{
|
||||
"admin_user_id": admin.ID,
|
||||
"type": "system",
|
||||
"title": "二维码库存预警",
|
||||
"content": alertContent,
|
||||
"is_read": false,
|
||||
"created_at": now,
|
||||
"updated_at": now,
|
||||
})
|
||||
}
|
||||
if len(entries) > 0 {
|
||||
if err := tx.Table("admin_notifications").Create(entries).Error; err != nil {
|
||||
return nil, err
|
||||
}
|
||||
}
|
||||
|
||||
// 构造预警消息
|
||||
alertContent := fmt.Sprintf("企业微信群二维码库存不足（剩余 %d 张），请及时补充", count)
|
||||
// 返回外部推送数据，由调用方在事务提交后发送
|
||||
providers := loadPushProviders(tx)
|
||||
if len(providers) == 0 {
|
||||
return nil, nil
|
||||
}
|
||||
return &pushAlert{
|
||||
providers: providers,
|
||||
message: push.Message{
|
||||
Title: "二维码库存预警",
|
||||
Content: alertContent,
|
||||
},
|
||||
}, nil
|
||||
}
|
||||
|
||||
// 发送站内信给所有客服
|
||||
entries := make([]map[string]interface{}, 0, len(csAdmins))
|
||||
now := time.Now()
|
||||
for _, admin := range csAdmins {
|
||||
entries = append(entries, map[string]interface{}{
|
||||
"admin_user_id": admin.ID,
|
||||
"type": "system",
|
||||
"title": "二维码库存预警",
|
||||
"content": alertContent,
|
||||
"is_read": false,
|
||||
"created_at": now,
|
||||
"updated_at": now,
|
||||
// loadPushProviders 从 push_channels 表读取启用的渠道并创建 providers。
|
||||
func loadPushProviders(tx *gorm.DB) []push.Provider {
|
||||
var providers []push.Provider
|
||||
var channels []struct {
|
||||
Type string `gorm:"column:type"`
|
||||
Config json.RawMessage `gorm:"column:config"`
|
||||
}
|
||||
if err := tx.Table("push_channels").Where("enabled = ?", true).Find(&channels).Error; err != nil {
|
||||
return providers
|
||||
}
|
||||
|
||||
for _, ch := range channels {
|
||||
switch ch.Type {
|
||||
case "bark":
|
||||
var cfg struct {
|
||||
DeviceKey string `json:"device_key"`
|
||||
Server string `json:"server"`
|
||||
}
|
||||
if err := json.Unmarshal(ch.Config, &cfg); err != nil || cfg.DeviceKey == "" {
|
||||
continue
|
||||
}
|
||||
bark, err := push.NewBarkProvider(push.BarkConfig{
|
||||
DeviceKey: cfg.DeviceKey,
|
||||
Server: cfg.Server,
|
||||
})
|
||||
}
|
||||
|
||||
if len(entries) > 0 {
|
||||
// 批量插入管理员通知
|
||||
if err := tx.Table("admin_notifications").Create(entries).Error; err != nil {
|
||||
return err
|
||||
if err == nil {
|
||||
providers = append(providers, bark)
|
||||
}
|
||||
case "wpush":
|
||||
var cfg struct {
|
||||
APIKey string `json:"api_key"`
|
||||
}
|
||||
if err := json.Unmarshal(ch.Config, &cfg); err != nil || cfg.APIKey == "" {
|
||||
continue
|
||||
}
|
||||
wpush, err := push.NewWPushProvider(push.WPushConfig{APIKey: cfg.APIKey})
|
||||
if err == nil {
|
||||
providers = append(providers, wpush)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func parseThreshold(value string) (int64, error) {
|
||||
var threshold int64
|
||||
if _, err := fmt.Sscanf(value, "%d", &threshold); err != nil {
|
||||
return 0, err
|
||||
}
|
||||
return threshold, nil
|
||||
return providers
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user