feat: 增加推送通知配置
This commit is contained in:
@@ -0,0 +1,69 @@
|
||||
package push
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"io"
|
||||
"net/http"
|
||||
"net/url"
|
||||
"strings"
|
||||
"time"
|
||||
)
|
||||
|
||||
const defaultBarkServer = "https://api.day.app"
|
||||
|
||||
// BarkConfig 是 iOS Bark 推送的配置。
|
||||
type BarkConfig struct {
|
||||
DeviceKey string
|
||||
Server string // 可选,默认 https://api.day.app
|
||||
}
|
||||
|
||||
// BarkProvider 实现了通过 Bark 发送 iOS 推送。
|
||||
type BarkProvider struct {
|
||||
deviceKey string
|
||||
server string
|
||||
client *http.Client
|
||||
}
|
||||
|
||||
// NewBarkProvider 创建 Bark 推送 Provider。
|
||||
// DeviceKey 为空时返回 ErrProviderConfigInvalid。
|
||||
func NewBarkProvider(cfg BarkConfig) (*BarkProvider, error) {
|
||||
if strings.TrimSpace(cfg.DeviceKey) == "" {
|
||||
return nil, ErrProviderConfigInvalid
|
||||
}
|
||||
server := strings.TrimSpace(cfg.Server)
|
||||
if server == "" {
|
||||
server = defaultBarkServer
|
||||
}
|
||||
server = strings.TrimRight(server, "/")
|
||||
return &BarkProvider{
|
||||
deviceKey: strings.TrimSpace(cfg.DeviceKey),
|
||||
server: server,
|
||||
client: &http.Client{Timeout: 10 * time.Second},
|
||||
}, nil
|
||||
}
|
||||
|
||||
func (p *BarkProvider) Name() string { return "bark" }
|
||||
|
||||
func (p *BarkProvider) Send(ctx context.Context, msg Message) error {
|
||||
title := url.PathEscape(msg.Title)
|
||||
body := url.PathEscape(msg.Content)
|
||||
reqURL := fmt.Sprintf("%s/%s/%s/%s", p.server, p.deviceKey, title, body)
|
||||
|
||||
req, err := http.NewRequestWithContext(ctx, http.MethodPost, reqURL, nil)
|
||||
if err != nil {
|
||||
return fmt.Errorf("bark: create request: %w", err)
|
||||
}
|
||||
|
||||
resp, err := p.client.Do(req)
|
||||
if err != nil {
|
||||
return fmt.Errorf("bark: send failed: %w", err)
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
if resp.StatusCode >= 400 {
|
||||
bodyBytes, _ := io.ReadAll(resp.Body)
|
||||
return fmt.Errorf("bark: http %d: %s", resp.StatusCode, string(bodyBytes))
|
||||
}
|
||||
return nil
|
||||
}
|
||||
@@ -0,0 +1,12 @@
|
||||
package push
|
||||
|
||||
import "context"
|
||||
|
||||
// NoopProvider 是一个空实现,未配置任何推送渠道时使用。
|
||||
type NoopProvider struct{}
|
||||
|
||||
func NewNoopProvider() *NoopProvider { return &NoopProvider{} }
|
||||
|
||||
func (p *NoopProvider) Name() string { return "noop" }
|
||||
|
||||
func (p *NoopProvider) Send(_ context.Context, _ Message) error { return nil }
|
||||
@@ -0,0 +1,23 @@
|
||||
package push
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
)
|
||||
|
||||
var ErrProviderConfigInvalid = errors.New("push provider config invalid")
|
||||
|
||||
// Message 描述一条待发送的推送消息。
|
||||
type Message struct {
|
||||
Title string
|
||||
Content string
|
||||
}
|
||||
|
||||
// Provider 是站外推送渠道的统一接口。
|
||||
// 实现方负责具体的 HTTP 调用(Bark、WPush 等)。
|
||||
type Provider interface {
|
||||
// Name 返回渠道标识,用于日志。
|
||||
Name() string
|
||||
// Send 发送一条推送消息。失败时返回 error。
|
||||
Send(ctx context.Context, msg Message) error
|
||||
}
|
||||
@@ -0,0 +1,73 @@
|
||||
package push
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io"
|
||||
"net/http"
|
||||
"net/url"
|
||||
"strings"
|
||||
"time"
|
||||
)
|
||||
|
||||
const wpushAPIURL = "https://api.wpush.cn/api/v1/send"
|
||||
|
||||
// WPushConfig 是 WPush 推送的配置。
|
||||
type WPushConfig struct {
|
||||
APIKey string
|
||||
}
|
||||
|
||||
// WPushProvider 实现了通过 WPush 发送推送。
|
||||
// 支持微信公众号、飞书、钉钉、企业微信等多种渠道,取决于用户在 WPush 侧的配置。
|
||||
type WPushProvider struct {
|
||||
apiKey string
|
||||
client *http.Client
|
||||
}
|
||||
|
||||
// NewWPushProvider 创建 WPush 推送 Provider。
|
||||
// APIKey 为空时返回 ErrProviderConfigInvalid。
|
||||
func NewWPushProvider(cfg WPushConfig) (*WPushProvider, error) {
|
||||
if strings.TrimSpace(cfg.APIKey) == "" {
|
||||
return nil, ErrProviderConfigInvalid
|
||||
}
|
||||
return &WPushProvider{
|
||||
apiKey: strings.TrimSpace(cfg.APIKey),
|
||||
client: &http.Client{Timeout: 10 * time.Second},
|
||||
}, nil
|
||||
}
|
||||
|
||||
func (p *WPushProvider) Name() string { return "wpush" }
|
||||
|
||||
func (p *WPushProvider) Send(ctx context.Context, msg Message) error {
|
||||
form := url.Values{}
|
||||
form.Set("apikey", p.apiKey)
|
||||
form.Set("title", msg.Title)
|
||||
form.Set("content", msg.Content)
|
||||
|
||||
req, err := http.NewRequestWithContext(ctx, http.MethodPost, wpushAPIURL, strings.NewReader(form.Encode()))
|
||||
if err != nil {
|
||||
return fmt.Errorf("wpush: create request: %w", err)
|
||||
}
|
||||
req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
|
||||
|
||||
resp, err := p.client.Do(req)
|
||||
if err != nil {
|
||||
return fmt.Errorf("wpush: send failed: %w", err)
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
bodyBytes, _ := io.ReadAll(resp.Body)
|
||||
|
||||
var result struct {
|
||||
Code int `json:"code"`
|
||||
Message string `json:"message"`
|
||||
}
|
||||
if err := json.Unmarshal(bodyBytes, &result); err != nil {
|
||||
return fmt.Errorf("wpush: parse response: %w (body: %s)", err, string(bodyBytes))
|
||||
}
|
||||
if result.Code != 0 {
|
||||
return fmt.Errorf("wpush: api error code=%d: %s", result.Code, result.Message)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
@@ -0,0 +1,100 @@
|
||||
package adminpush
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"time"
|
||||
)
|
||||
|
||||
// ── 数据库模型 ──
|
||||
|
||||
type pushChannel struct {
|
||||
ID uint64 `gorm:"primaryKey" json:"id"`
|
||||
Name string `gorm:"size:64;not null" json:"name"`
|
||||
Type string `gorm:"size:32;not null" json:"type"`
|
||||
Config json.RawMessage `gorm:"type:json;not null" json:"config"`
|
||||
Enabled bool `gorm:"not null;default:true" json:"enabled"`
|
||||
CreatedAt time.Time `json:"created_at"`
|
||||
UpdatedAt time.Time `json:"updated_at"`
|
||||
}
|
||||
|
||||
func (pushChannel) TableName() string { return "push_channels" }
|
||||
|
||||
type pushRule struct {
|
||||
ID uint64 `gorm:"primaryKey" json:"id"`
|
||||
Event string `gorm:"size:64;not null;uniqueIndex" json:"event"`
|
||||
Enabled bool `gorm:"not null;default:true" json:"enabled"`
|
||||
Threshold int `gorm:"not null;default:5" json:"threshold"`
|
||||
MessageTemplate string `gorm:"size:255;not null;default:''" json:"message_template"`
|
||||
CreatedAt time.Time `json:"created_at"`
|
||||
UpdatedAt time.Time `json:"updated_at"`
|
||||
}
|
||||
|
||||
func (pushRule) TableName() string { return "push_rules" }
|
||||
|
||||
// ── DTO ──
|
||||
|
||||
type ChannelDTO struct {
|
||||
ID uint64 `json:"id"`
|
||||
Name string `json:"name"`
|
||||
Type string `json:"type"`
|
||||
Config json.RawMessage `json:"config"`
|
||||
Enabled bool `json:"enabled"`
|
||||
CreatedAt time.Time `json:"created_at"`
|
||||
UpdatedAt time.Time `json:"updated_at"`
|
||||
}
|
||||
|
||||
type RuleDTO struct {
|
||||
ID uint64 `json:"id"`
|
||||
Event string `json:"event"`
|
||||
Enabled bool `json:"enabled"`
|
||||
Threshold int `json:"threshold"`
|
||||
MessageTemplate string `json:"message_template"`
|
||||
CreatedAt time.Time `json:"created_at"`
|
||||
UpdatedAt time.Time `json:"updated_at"`
|
||||
}
|
||||
|
||||
// ── 请求 ──
|
||||
|
||||
type CreateChannelRequest struct {
|
||||
Name string `json:"name" binding:"required"`
|
||||
Type string `json:"type" binding:"required"`
|
||||
Config json.RawMessage `json:"config" binding:"required"`
|
||||
}
|
||||
|
||||
type UpdateChannelRequest struct {
|
||||
Name *string `json:"name"`
|
||||
Config *json.RawMessage `json:"config"`
|
||||
Enabled *bool `json:"enabled"`
|
||||
}
|
||||
|
||||
type UpdateRuleRequest struct {
|
||||
Enabled *bool `json:"enabled"`
|
||||
Threshold *int `json:"threshold"`
|
||||
MessageTemplate *string `json:"message_template"`
|
||||
}
|
||||
|
||||
// ── 转换 ──
|
||||
|
||||
func toChannelDTO(row pushChannel) ChannelDTO {
|
||||
return ChannelDTO{
|
||||
ID: row.ID,
|
||||
Name: row.Name,
|
||||
Type: row.Type,
|
||||
Config: row.Config,
|
||||
Enabled: row.Enabled,
|
||||
CreatedAt: row.CreatedAt,
|
||||
UpdatedAt: row.UpdatedAt,
|
||||
}
|
||||
}
|
||||
|
||||
func toRuleDTO(row pushRule) RuleDTO {
|
||||
return RuleDTO{
|
||||
ID: row.ID,
|
||||
Event: row.Event,
|
||||
Enabled: row.Enabled,
|
||||
Threshold: row.Threshold,
|
||||
MessageTemplate: row.MessageTemplate,
|
||||
CreatedAt: row.CreatedAt,
|
||||
UpdatedAt: row.UpdatedAt,
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,169 @@
|
||||
package adminpush
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"strconv"
|
||||
|
||||
"hfb_sys/backend/internal/middleware"
|
||||
"hfb_sys/backend/pkg/response"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
)
|
||||
|
||||
type Handler struct {
|
||||
service *Service
|
||||
}
|
||||
|
||||
func NewHandler(service *Service) *Handler {
|
||||
return &Handler{service: service}
|
||||
}
|
||||
|
||||
// ── 渠道 ──
|
||||
|
||||
func (h *Handler) ListChannels(c *gin.Context) {
|
||||
items, err := h.service.ListChannels(c.Request.Context())
|
||||
if err != nil {
|
||||
writePushError(c, err)
|
||||
return
|
||||
}
|
||||
response.OK(c, gin.H{"items": items})
|
||||
}
|
||||
|
||||
func (h *Handler) CreateChannel(c *gin.Context) {
|
||||
if _, ok := currentAdminID(c); !ok {
|
||||
response.Unauthorized(c, "缺少管理员上下文")
|
||||
return
|
||||
}
|
||||
var req CreateChannelRequest
|
||||
if err := c.ShouldBindJSON(&req); err != nil {
|
||||
response.BadRequest(c, "名称、类型和配置不能为空")
|
||||
return
|
||||
}
|
||||
item, err := h.service.CreateChannel(c.Request.Context(), req)
|
||||
if err != nil {
|
||||
writePushError(c, err)
|
||||
return
|
||||
}
|
||||
response.Created(c, item)
|
||||
}
|
||||
|
||||
func (h *Handler) TestChannel(c *gin.Context) {
|
||||
if _, ok := currentAdminID(c); !ok {
|
||||
response.Unauthorized(c, "缺少管理员上下文")
|
||||
return
|
||||
}
|
||||
id, err := parseID(c)
|
||||
if err != nil {
|
||||
response.BadRequest(c, "ID 不正确")
|
||||
return
|
||||
}
|
||||
if err := h.service.TestChannel(c.Request.Context(), id); err != nil {
|
||||
writePushError(c, err)
|
||||
return
|
||||
}
|
||||
response.OK(c, gin.H{"sent": true})
|
||||
}
|
||||
|
||||
func (h *Handler) UpdateChannel(c *gin.Context) {
|
||||
if _, ok := currentAdminID(c); !ok {
|
||||
response.Unauthorized(c, "缺少管理员上下文")
|
||||
return
|
||||
}
|
||||
id, err := parseID(c)
|
||||
if err != nil {
|
||||
response.BadRequest(c, "ID 不正确")
|
||||
return
|
||||
}
|
||||
var req UpdateChannelRequest
|
||||
if err := c.ShouldBindJSON(&req); err != nil {
|
||||
response.BadRequest(c, "请求格式不正确")
|
||||
return
|
||||
}
|
||||
item, err := h.service.UpdateChannel(c.Request.Context(), id, req)
|
||||
if err != nil {
|
||||
writePushError(c, err)
|
||||
return
|
||||
}
|
||||
response.OK(c, item)
|
||||
}
|
||||
|
||||
func (h *Handler) DeleteChannel(c *gin.Context) {
|
||||
if _, ok := currentAdminID(c); !ok {
|
||||
response.Unauthorized(c, "缺少管理员上下文")
|
||||
return
|
||||
}
|
||||
id, err := parseID(c)
|
||||
if err != nil {
|
||||
response.BadRequest(c, "ID 不正确")
|
||||
return
|
||||
}
|
||||
if err := h.service.DeleteChannel(c.Request.Context(), id); err != nil {
|
||||
writePushError(c, err)
|
||||
return
|
||||
}
|
||||
response.OK(c, gin.H{"deleted": true})
|
||||
}
|
||||
|
||||
// ── 规则 ──
|
||||
|
||||
func (h *Handler) ListRules(c *gin.Context) {
|
||||
items, err := h.service.ListRules(c.Request.Context())
|
||||
if err != nil {
|
||||
writePushError(c, err)
|
||||
return
|
||||
}
|
||||
response.OK(c, gin.H{"items": items})
|
||||
}
|
||||
|
||||
func (h *Handler) UpdateRule(c *gin.Context) {
|
||||
if _, ok := currentAdminID(c); !ok {
|
||||
response.Unauthorized(c, "缺少管理员上下文")
|
||||
return
|
||||
}
|
||||
id, err := parseID(c)
|
||||
if err != nil {
|
||||
response.BadRequest(c, "ID 不正确")
|
||||
return
|
||||
}
|
||||
var req UpdateRuleRequest
|
||||
if err := c.ShouldBindJSON(&req); err != nil {
|
||||
response.BadRequest(c, "请求格式不正确")
|
||||
return
|
||||
}
|
||||
item, err := h.service.UpdateRule(c.Request.Context(), id, req)
|
||||
if err != nil {
|
||||
writePushError(c, err)
|
||||
return
|
||||
}
|
||||
response.OK(c, item)
|
||||
}
|
||||
|
||||
// ── 辅助 ──
|
||||
|
||||
func parseID(c *gin.Context) (uint64, error) {
|
||||
return strconv.ParseUint(c.Param("id"), 10, 64)
|
||||
}
|
||||
|
||||
func currentAdminID(c *gin.Context) (uint64, bool) {
|
||||
value, ok := c.Get(middleware.ContextAdminID)
|
||||
if !ok {
|
||||
return 0, false
|
||||
}
|
||||
adminID, ok := value.(uint64)
|
||||
return adminID, ok
|
||||
}
|
||||
|
||||
func writePushError(c *gin.Context, err error) {
|
||||
switch {
|
||||
case errors.Is(err, ErrDependencyUnavailable):
|
||||
response.ServiceUnavailable(c, "数据库未连接")
|
||||
case errors.Is(err, ErrChannelNotFound):
|
||||
response.NotFound(c, "推送渠道不存在")
|
||||
case errors.Is(err, ErrRuleNotFound):
|
||||
response.NotFound(c, "推送规则不存在")
|
||||
case errors.Is(err, ErrInvalidChannel):
|
||||
response.BadRequest(c, "渠道类型或配置不正确")
|
||||
default:
|
||||
response.InternalServerError(c, "推送服务暂时不可用")
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,164 @@
|
||||
package adminpush
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"errors"
|
||||
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
type Repository struct {
|
||||
db *gorm.DB
|
||||
}
|
||||
|
||||
func NewRepository(db *gorm.DB) *Repository {
|
||||
return &Repository{db: db}
|
||||
}
|
||||
|
||||
// ── 渠道 CRUD ──
|
||||
|
||||
func (r *Repository) ListChannels(ctx context.Context) ([]ChannelDTO, error) {
|
||||
var rows []pushChannel
|
||||
if err := r.db.WithContext(ctx).Order("id ASC").Find(&rows).Error; err != nil {
|
||||
return nil, err
|
||||
}
|
||||
items := make([]ChannelDTO, 0, len(rows))
|
||||
for _, row := range rows {
|
||||
items = append(items, toChannelDTO(row))
|
||||
}
|
||||
return items, nil
|
||||
}
|
||||
|
||||
func (r *Repository) CreateChannel(ctx context.Context, req CreateChannelRequest) (*ChannelDTO, error) {
|
||||
cfgJSON, err := json.Marshal(req.Config)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
row := pushChannel{
|
||||
Name: req.Name,
|
||||
Type: req.Type,
|
||||
Config: cfgJSON,
|
||||
Enabled: true,
|
||||
}
|
||||
if err := r.db.WithContext(ctx).Create(&row).Error; err != nil {
|
||||
return nil, err
|
||||
}
|
||||
dto := toChannelDTO(row)
|
||||
return &dto, nil
|
||||
}
|
||||
|
||||
func (r *Repository) UpdateChannel(ctx context.Context, id uint64, req UpdateChannelRequest) (*ChannelDTO, error) {
|
||||
var row pushChannel
|
||||
if err := r.db.WithContext(ctx).First(&row, id).Error; err != nil {
|
||||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||||
return nil, ErrChannelNotFound
|
||||
}
|
||||
return nil, err
|
||||
}
|
||||
if req.Name != nil {
|
||||
row.Name = *req.Name
|
||||
}
|
||||
if req.Config != nil {
|
||||
cfgJSON, err := json.Marshal(req.Config)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
row.Config = cfgJSON
|
||||
}
|
||||
if req.Enabled != nil {
|
||||
row.Enabled = *req.Enabled
|
||||
}
|
||||
if err := r.db.WithContext(ctx).Save(&row).Error; err != nil {
|
||||
return nil, err
|
||||
}
|
||||
dto := toChannelDTO(row)
|
||||
return &dto, nil
|
||||
}
|
||||
|
||||
func (r *Repository) DeleteChannel(ctx context.Context, id uint64) error {
|
||||
result := r.db.WithContext(ctx).Delete(&pushChannel{}, id)
|
||||
if result.Error != nil {
|
||||
return result.Error
|
||||
}
|
||||
if result.RowsAffected == 0 {
|
||||
return ErrChannelNotFound
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// ── 规则 CRUD ──
|
||||
|
||||
func (r *Repository) ListRules(ctx context.Context) ([]RuleDTO, error) {
|
||||
var rows []pushRule
|
||||
if err := r.db.WithContext(ctx).Order("id ASC").Find(&rows).Error; err != nil {
|
||||
return nil, err
|
||||
}
|
||||
items := make([]RuleDTO, 0, len(rows))
|
||||
for _, row := range rows {
|
||||
items = append(items, toRuleDTO(row))
|
||||
}
|
||||
return items, nil
|
||||
}
|
||||
|
||||
func (r *Repository) UpdateRule(ctx context.Context, id uint64, req UpdateRuleRequest) (*RuleDTO, error) {
|
||||
var row pushRule
|
||||
if err := r.db.WithContext(ctx).First(&row, id).Error; err != nil {
|
||||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||||
return nil, ErrRuleNotFound
|
||||
}
|
||||
return nil, err
|
||||
}
|
||||
if req.Enabled != nil {
|
||||
row.Enabled = *req.Enabled
|
||||
}
|
||||
if req.Threshold != nil {
|
||||
row.Threshold = *req.Threshold
|
||||
}
|
||||
if req.MessageTemplate != nil {
|
||||
row.MessageTemplate = *req.MessageTemplate
|
||||
}
|
||||
if err := r.db.WithContext(ctx).Save(&row).Error; err != nil {
|
||||
return nil, err
|
||||
}
|
||||
dto := toRuleDTO(row)
|
||||
return &dto, nil
|
||||
}
|
||||
|
||||
// ── 推送调用方使用 ──
|
||||
|
||||
func (r *Repository) GetChannel(ctx context.Context, id uint64) (*ChannelDTO, error) {
|
||||
var row pushChannel
|
||||
if err := r.db.WithContext(ctx).First(&row, id).Error; err != nil {
|
||||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||||
return nil, ErrChannelNotFound
|
||||
}
|
||||
return nil, err
|
||||
}
|
||||
dto := toChannelDTO(row)
|
||||
return &dto, nil
|
||||
}
|
||||
|
||||
func (r *Repository) GetActiveChannels(ctx context.Context) ([]ChannelDTO, error) {
|
||||
var rows []pushChannel
|
||||
if err := r.db.WithContext(ctx).Where("enabled = ?", true).Find(&rows).Error; err != nil {
|
||||
return nil, err
|
||||
}
|
||||
items := make([]ChannelDTO, 0, len(rows))
|
||||
for _, row := range rows {
|
||||
items = append(items, toChannelDTO(row))
|
||||
}
|
||||
return items, nil
|
||||
}
|
||||
|
||||
func (r *Repository) GetRule(ctx context.Context, event string) (*RuleDTO, error) {
|
||||
var row pushRule
|
||||
if err := r.db.WithContext(ctx).Where("event = ?", event).First(&row).Error; err != nil {
|
||||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||||
return nil, nil
|
||||
}
|
||||
return nil, err
|
||||
}
|
||||
dto := toRuleDTO(row)
|
||||
return &dto, nil
|
||||
}
|
||||
@@ -0,0 +1,149 @@
|
||||
package adminpush
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"strings"
|
||||
|
||||
"hfb_sys/backend/internal/integrations/push"
|
||||
)
|
||||
|
||||
var (
|
||||
ErrDependencyUnavailable = errors.New("dependency unavailable")
|
||||
ErrChannelNotFound = errors.New("push channel not found")
|
||||
ErrRuleNotFound = errors.New("push rule not found")
|
||||
ErrInvalidChannel = errors.New("invalid channel type or config")
|
||||
)
|
||||
|
||||
type Service struct {
|
||||
repo *Repository
|
||||
}
|
||||
|
||||
func NewService(repo *Repository) *Service {
|
||||
return &Service{repo: repo}
|
||||
}
|
||||
|
||||
func (s *Service) ListChannels(ctx context.Context) ([]ChannelDTO, error) {
|
||||
if s.repo == nil {
|
||||
return nil, ErrDependencyUnavailable
|
||||
}
|
||||
return s.repo.ListChannels(ctx)
|
||||
}
|
||||
|
||||
func (s *Service) CreateChannel(ctx context.Context, req CreateChannelRequest) (*ChannelDTO, error) {
|
||||
if s.repo == nil {
|
||||
return nil, ErrDependencyUnavailable
|
||||
}
|
||||
if req.Name == "" || req.Type == "" {
|
||||
return nil, ErrInvalidChannel
|
||||
}
|
||||
if err := validateChannelConfig(req.Type, req.Config); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return s.repo.CreateChannel(ctx, req)
|
||||
}
|
||||
|
||||
func (s *Service) UpdateChannel(ctx context.Context, id uint64, req UpdateChannelRequest) (*ChannelDTO, error) {
|
||||
if s.repo == nil {
|
||||
return nil, ErrDependencyUnavailable
|
||||
}
|
||||
if req.Config != nil {
|
||||
current, err := s.repo.GetChannel(ctx, id)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if err := validateChannelConfig(current.Type, *req.Config); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
}
|
||||
return s.repo.UpdateChannel(ctx, id, req)
|
||||
}
|
||||
|
||||
func (s *Service) DeleteChannel(ctx context.Context, id uint64) error {
|
||||
if s.repo == nil {
|
||||
return ErrDependencyUnavailable
|
||||
}
|
||||
return s.repo.DeleteChannel(ctx, id)
|
||||
}
|
||||
|
||||
func (s *Service) ListRules(ctx context.Context) ([]RuleDTO, error) {
|
||||
if s.repo == nil {
|
||||
return nil, ErrDependencyUnavailable
|
||||
}
|
||||
return s.repo.ListRules(ctx)
|
||||
}
|
||||
|
||||
func (s *Service) UpdateRule(ctx context.Context, id uint64, req UpdateRuleRequest) (*RuleDTO, error) {
|
||||
if s.repo == nil {
|
||||
return nil, ErrDependencyUnavailable
|
||||
}
|
||||
if req.Threshold != nil && *req.Threshold < 1 {
|
||||
return nil, ErrInvalidChannel
|
||||
}
|
||||
return s.repo.UpdateRule(ctx, id, req)
|
||||
}
|
||||
|
||||
func (s *Service) TestChannel(ctx context.Context, id uint64) error {
|
||||
if s.repo == nil {
|
||||
return ErrDependencyUnavailable
|
||||
}
|
||||
ch, err := s.repo.GetChannel(ctx, id)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
provider, err := buildProvider(ch.Type, ch.Config)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return provider.Send(ctx, push.Message{
|
||||
Title: "推送测试",
|
||||
Content: "这是一条测试消息,如果你收到了说明推送配置正确。",
|
||||
})
|
||||
}
|
||||
|
||||
func validateChannelConfig(channelType string, cfg json.RawMessage) error {
|
||||
switch channelType {
|
||||
case "bark":
|
||||
var c struct {
|
||||
DeviceKey string `json:"device_key"`
|
||||
}
|
||||
if err := json.Unmarshal(cfg, &c); err != nil || strings.TrimSpace(c.DeviceKey) == "" {
|
||||
return ErrInvalidChannel
|
||||
}
|
||||
case "wpush":
|
||||
var c struct {
|
||||
APIKey string `json:"api_key"`
|
||||
}
|
||||
if err := json.Unmarshal(cfg, &c); err != nil || strings.TrimSpace(c.APIKey) == "" {
|
||||
return ErrInvalidChannel
|
||||
}
|
||||
default:
|
||||
return ErrInvalidChannel
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func buildProvider(channelType string, cfg json.RawMessage) (push.Provider, error) {
|
||||
switch channelType {
|
||||
case "bark":
|
||||
var c struct {
|
||||
DeviceKey string `json:"device_key"`
|
||||
Server string `json:"server"`
|
||||
}
|
||||
if err := json.Unmarshal(cfg, &c); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return push.NewBarkProvider(push.BarkConfig{DeviceKey: c.DeviceKey, Server: c.Server})
|
||||
case "wpush":
|
||||
var c struct {
|
||||
APIKey string `json:"api_key"`
|
||||
}
|
||||
if err := json.Unmarshal(cfg, &c); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return push.NewWPushProvider(push.WPushConfig{APIKey: c.APIKey})
|
||||
default:
|
||||
return nil, errors.New("unknown channel type")
|
||||
}
|
||||
}
|
||||
@@ -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
|
||||
}
|
||||
|
||||
@@ -14,6 +14,7 @@ import (
|
||||
"hfb_sys/backend/internal/modules/adminfinance"
|
||||
"hfb_sys/backend/internal/modules/adminmgr"
|
||||
"hfb_sys/backend/internal/modules/adminnotification"
|
||||
"hfb_sys/backend/internal/modules/adminpush"
|
||||
"hfb_sys/backend/internal/modules/adminrole"
|
||||
"hfb_sys/backend/internal/modules/adminuser"
|
||||
"hfb_sys/backend/internal/modules/announcement"
|
||||
@@ -152,6 +153,12 @@ func New(cfg config.Config, deps Dependencies, logger *zap.Logger) *gin.Engine {
|
||||
}
|
||||
adminNotificationService := adminnotification.NewService(adminNotificationRepo)
|
||||
adminNotificationHandler := adminnotification.NewHandler(adminNotificationService)
|
||||
var adminPushRepo *adminpush.Repository
|
||||
if deps.DB != nil {
|
||||
adminPushRepo = adminpush.NewRepository(deps.DB)
|
||||
}
|
||||
adminPushService := adminpush.NewService(adminPushRepo)
|
||||
adminPushHandler := adminpush.NewHandler(adminPushService)
|
||||
userHandler := user.NewHandler(userRepo)
|
||||
var realnameRepo *realname.Repository
|
||||
if deps.DB != nil {
|
||||
@@ -555,6 +562,13 @@ func New(cfg config.Config, deps Dependencies, logger *zap.Logger) *gin.Engine {
|
||||
adminRoutes.GET("/notifications/unread-count", requirePerm("notification:view"), adminNotificationHandler.UnreadCount)
|
||||
adminRoutes.PUT("/notifications/read-all", requirePerm("notification:view"), adminNotificationHandler.MarkAllRead)
|
||||
adminRoutes.POST("/notifications/:id/read", requirePerm("notification:view"), adminNotificationHandler.MarkRead)
|
||||
adminRoutes.GET("/push-channels", requirePerm("system_config:view"), adminPushHandler.ListChannels)
|
||||
adminRoutes.POST("/push-channels", requirePerm("system_config:update"), adminPushHandler.CreateChannel)
|
||||
adminRoutes.PUT("/push-channels/:id", requirePerm("system_config:update"), adminPushHandler.UpdateChannel)
|
||||
adminRoutes.DELETE("/push-channels/:id", requirePerm("system_config:update"), adminPushHandler.DeleteChannel)
|
||||
adminRoutes.POST("/push-channels/:id/test", requirePerm("system_config:update"), adminPushHandler.TestChannel)
|
||||
adminRoutes.GET("/push-rules", requirePerm("system_config:view"), adminPushHandler.ListRules)
|
||||
adminRoutes.PUT("/push-rules/:id", requirePerm("system_config:update"), adminPushHandler.UpdateRule)
|
||||
adminRoutes.GET("/audit-logs", requirePerm("audit_log:view"), adminAuditHandler.List)
|
||||
if chatHubHandler != nil {
|
||||
adminRoutes.GET("/chats/events", requirePerm("chat:view"), chatHubHandler.AdminEvents)
|
||||
|
||||
Reference in New Issue
Block a user