mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-29 22:06:38 +08:00
ryan
9fa38dcfdc
- Implement cached queries in GetActivePushEventByKey and GetActivePushChannelByName. - Set TTL for cached items to 24 hours via activePushEventCacheTTL and activePushChannelCacheTTL constants. - Implement GORM hooks (AfterSave and AfterDelete) on PushEvent and PushChannel to auto-evict Redis caches, guaranteeing cache consistency. - Evict Redis caches manually inside API handlers for Create/Update/Delete/Toggle event/channel endpoints. - Update events.go to query models through the new caching methods.
99 lines
3.1 KiB
Go
99 lines
3.1 KiB
Go
// Copyright 2026 Arctel.net
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package model
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/Rain-kl/Wavelet/internal/db"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// PushEvent 系统通知事件模型
|
|
type PushEvent struct {
|
|
ID uint64 `json:"id" gorm:"primaryKey;autoIncrement"`
|
|
EventKey string `json:"event_key" gorm:"uniqueIndex;size:80;not null"` // 如 admin_login
|
|
Name string `json:"name" gorm:"size:100;not null"` // 如 管理员登录
|
|
TaskType string `json:"task_type" gorm:"size:100;index;not null;default:''"` // 关联的异步任务类型
|
|
Channels []string `json:"channels" gorm:"type:text;serializer:json;not null"` // 推送渠道列表，如 ["lark"]
|
|
Targets []string `json:"targets" gorm:"type:text;serializer:json;not null"` // 推送目标用户/邮箱列表
|
|
Template string `json:"template" gorm:"type:text;not null"` // 消息模板 JSON
|
|
Enabled bool `json:"enabled" gorm:"index;not null;default:false"` // 是否启用
|
|
CreatedAt time.Time `json:"created_at" gorm:"autoCreateTime;index"`
|
|
UpdatedAt time.Time `json:"updated_at" gorm:"autoUpdateTime;index"`
|
|
}
|
|
|
|
// TableName 指定 GORM 表名
|
|
func (PushEvent) TableName() string {
|
|
return "w_push_events"
|
|
}
|
|
|
|
// Validate 基础校验
|
|
func (pe *PushEvent) Validate() error {
|
|
pe.EventKey = strings.TrimSpace(pe.EventKey)
|
|
pe.Name = strings.TrimSpace(pe.Name)
|
|
pe.Template = strings.TrimSpace(pe.Template)
|
|
|
|
if pe.EventKey == "" {
|
|
return errors.New("event key is required")
|
|
}
|
|
if pe.Name == "" {
|
|
return errors.New("event name is required")
|
|
}
|
|
if pe.Template == "" {
|
|
return errors.New("event template is required")
|
|
}
|
|
if pe.Enabled && len(pe.Channels) == 0 {
|
|
return errors.New("cannot enable event without any push channels configured")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
const activePushEventCacheTTL = 24 * time.Hour
|
|
|
|
// GetActivePushEventByKey 获取启用的通知事件 (优先从 Redis 缓存获取)
|
|
func GetActivePushEventByKey(ctx context.Context, key string) (*PushEvent, error) {
|
|
cacheKey := "push:event:active:" + key
|
|
var event PushEvent
|
|
if db.Redis != nil {
|
|
if err := db.GetJSON(ctx, cacheKey, &event); err == nil {
|
|
return &event, nil
|
|
}
|
|
}
|
|
|
|
err := db.DB(ctx).Where("event_key = ? AND enabled = ?", key, true).First(&event).Error
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if db.Redis != nil {
|
|
// 缓存有效时间设置为 24 小时
|
|
_ = db.SetJSON(ctx, cacheKey, event, activePushEventCacheTTL)
|
|
}
|
|
|
|
return &event, nil
|
|
}
|
|
|
|
// DeleteActivePushEventCache 清理启用通知事件的缓存
|
|
func DeleteActivePushEventCache(ctx context.Context, key string) {
|
|
if db.Redis != nil {
|
|
_ = db.Redis.Del(ctx, db.PrefixedKey("push:event:active:"+key)).Err()
|
|
}
|
|
}
|
|
|
|
// AfterSave GORM 保存后钩子，用于自动清理缓存
|
|
func (pe *PushEvent) AfterSave(tx *gorm.DB) error {
|
|
DeleteActivePushEventCache(tx.Statement.Context, pe.EventKey)
|
|
return nil
|
|
}
|
|
|
|
// AfterDelete GORM 删除后钩子，用于自动清理缓存
|
|
func (pe *PushEvent) AfterDelete(tx *gorm.DB) error {
|
|
DeleteActivePushEventCache(tx.Statement.Context, pe.EventKey)
|
|
return nil
|
|
}
|