package models import ( "bytes" "encoding/json" "errors" "fmt" "github.com/codinl/go-logger" orm "go-admin/common/global" "regexp" "text/template" "time" ) const ( AnnouncementMessageType = 1 // 公告消息 BusinessMessageType = 2 // 业务消息 TargetTypeByAll = 1 // 接收对象类型：1=全员 TargetTypeByRole = 2 // 接收对象类型：2=角色 TargetTypeByUser = 3 // 接收对象类型：3=用户 MessageStatusOnDraft = 0 // 草稿 MessageStatusOnStart = 1 // 启用 MessageStatusOnDisable = 2 // 禁用 MessageStatusOnExpire = 3 // 已过期 MessageStart = 1 // 启用 MessageStop = 2 // 禁用 BizTypeOnAllot = "stock_transfer" // 库存调拨 EventOnAllot = "wait_receive" //待收货 ) // SystemMessageConfig 消息配置主表（公告 + 业务统一） type SystemMessageConfig struct { Model MessageType uint8 `json:"message_type"` // 消息类型：1=公告，2=业务 Title string `json:"title"` // 消息标题 ContentTemplate string `json:"content_template"` // 消息模板内容，支持{{变量}} Level uint8 `json:"level"` // 消息等级：1=普通，2=重要，3=紧急 DisplayMode uint8 `json:"display_mode"` // 展示方式：1=弹窗，2=滚动，3=弹窗+滚动 StartTime *time.Time `json:"start_time,omitempty"` // 公告生效时间 EndTime *time.Time `json:"end_time,omitempty"` // 公告失效时间 BizType string `json:"biz_type,omitempty"` // 业务类型 Event string `json:"event,omitempty"` // 事件类型 Status uint8 `json:"status"` // 状态：0=草稿，1=启用，2=禁用，3=已过期 } // SystemMessageTarget 接收对象配置表（支持角色/部门/用户/全员） type SystemMessageTarget struct { Model MessageConfigID uint `json:"message_config_id"` // 对应消息配置 ID TargetType uint8 `json:"target_type"` // 接收对象类型：1=全员，2=角色，3=用户 TargetID int64 `json:"target_id,omitempty"` // 对象ID（角色ID/用户ID） } // SystemEventRegistry 业务事件注册表（用于事件枚举管理） type SystemEventRegistry struct { Model BizType string `json:"biz_type"` // 业务类型 Event string `json:"event"` // 事件名 BizName string `json:"biz_name"` // 业务名称（展示用） EventName string `json:"event_name"` // 事件名称（展示用） Description string `json:"description"` // 说明 TemplateVariables string `json:"template_variables"` // 可用变量 JSON 字符串 IsEnabled bool `json:"is_enabled"` // 是否启用 } type TemplateVarItem struct { Key string `json:"key"` Desc string `json:"desc"` } // SystemUserMessage 用户收到的消息记录表（实际发送记录） type SystemUserMessage struct { Model UserID int64 `json:"user_id"` // 接收用户ID MessageConfigID uint `json:"message_config_id"` // 对应消息配置ID，可为空（被删时保留记录） MessageType uint8 `json:"message_type"` // 消息类型：1=公告，2=业务 Title string `json:"title"` // 消息标题快照 Content string `json:"content"` // 渲染后内容 IsRead bool `json:"is_read"` // 是否阅读 ReadTime *time.Time `json:"read_time,omitempty"` // 阅读时间 IsVisible bool `json:"is_visible"` // 是否展示给用户 } // SysMessageCreateReq 新增公告请求结构体 type SysMessageCreateReq struct { Title string `json:"title" binding:"required"` // 消息标题 ContentTemplate string `json:"content_template" binding:"required"` // 消息内容 Level uint8 `json:"level" binding:"required,oneof=1 2 3"` // 消息等级：1=普通，2=重要，3=紧急 DisplayMode uint8 `json:"display_mode" binding:"required,oneof=1 2 3"` // 展示方式：1=弹窗，2=滚动，3=弹窗+滚动 StartTime string `json:"start_time,omitempty"` // 公告生效时间 EndTime string `json:"end_time,omitempty"` // 公告失效时间 Status uint8 `json:"status"` // 状态：0=草稿，1=启用 TargetType uint8 `json:"target_type" binding:"required,oneof=1 2 3"` // 接收对象类型：1=全员，2=角色，3=用户 TargetID []int64 `json:"target_id,omitempty"` // 对象ID（角色ID/用户ID） } // SysMessageCreateResp 创建成功响应 type SysMessageCreateResp struct { ID uint32 `json:"id"` } // SysMessageEditReq 编辑公告消息-入参 type SysMessageEditReq struct { ID uint `json:"id" binding:"required"` // 公告ID（必须） Title string `json:"title" binding:"required"` // 标题 ContentTemplate string `json:"content_template" binding:"required"` // 内容模板 Level uint8 `json:"level" binding:"required,oneof=1 2 3"` // 重要等级 DisplayMode uint8 `json:"display_mode" binding:"required,oneof=1 2 3"` // 展示模式 StartTime string `json:"start_time"` // 生效时间（可空） EndTime string `json:"end_time"` // 失效时间（可空） Status uint8 `json:"status"` // 状态：0 草稿、1 启用 TargetType uint8 `json:"target_type" binding:"required,oneof=1 2 3"` // 接收对象类型：1=全员，2=角色，3=用户 TargetID []int64 `json:"target_id,omitempty"` // 接收对象ID } // SysMessageListReq 公告消息列表-入参 type SysMessageListReq struct { Title string `form:"title"` // 消息标题 Level uint8 `form:"level"` // 消息等级：1=普通，2=重要，3=紧急 DisplayMode uint8 `form:"display_mode"` // 展示方式：1=弹窗，2=滚动，3=弹窗+滚动 Status uint8 `form:"status"` // 状态：0=草稿，1=启用，2=禁用，3=已过期 TargetType uint8 `form:"target_type"` // 从接收对象配置表中筛选 Page int `form:"page_index,default=1"` // 页码 PageSize int `form:"page_size,default=10"` // 每页条数 } // SysMessageListResp 公告消息列表-出参 type SysMessageListResp struct { List []SysAnnouncementItem `json:"list"` Total int64 `json:"total"` // 总条数 Page int `json:"page_index"` // 页码 PageSize int `json:"page_size"` // 每页条数 } // SysAnnouncementItem 公告消息展示结构体 type SysAnnouncementItem struct { ID uint32 `json:"id"` CreatedAt time.Time `json:"created_at"` UpdatedAt time.Time `json:"updated_at"` Title string `json:"title"` // 消息标题 ContentTemplate string `json:"content_template"` // 消息内容 Level uint8 `json:"level"` // 消息等级：1=普通，2=重要，3=紧急 DisplayMode uint8 `json:"display_mode"` // 展示方式：1=弹窗，2=滚动，3=弹窗+滚动 StartTime *time.Time `json:"start_time,omitempty"` // 公告生效时间 EndTime *time.Time `json:"end_time,omitempty"` // 公告失效时间 Status uint8 `json:"status"` // 状态：0=草稿，1=启用，2=禁用，3=已过期 TargetType uint8 `json:"target_type"` // 接收对象类型：1=全员，2=角色，3=用户 TargetID []int64 `json:"target_id,omitempty"` // 对象ID（角色ID/用户ID） } // SysMessageSetStatusReq 设置公告消息状态 type SysMessageSetStatusReq struct { ID uint `json:"id" binding:"required"` Status uint8 `json:"status" binding:"required"` // 目标状态：1=启用，2=禁用 } // SysMessageDeleteReq 删除公告消息 type SysMessageDeleteReq struct { ID uint `json:"id" binding:"required"` } // SysMessageDetailReq 公告详情请求 type SysMessageDetailReq struct { ID uint `json:"id" binding:"required"` } // SysMessageDetailResp 公告详情响应 type SysMessageDetailResp struct { ID uint `json:"id"` CreatedAt time.Time `json:"created_at"` UpdatedAt time.Time `json:"updated_at"` Title string `json:"title"` // 消息标题 Content string `json:"content"` // 消息内容 Level uint8 `json:"level"` // 消息等级：1=普通，2=重要，3=紧急 DisplayMode uint8 `json:"display_mode"` // 展示方式：1=弹窗，2=滚动，3=弹窗+滚动 Status uint8 `json:"status"` // 状态：0=草稿，1=启用，2=禁用，3=已过期 StartTime *time.Time `json:"start_time,omitempty"` // 公告生效时间 EndTime *time.Time `json:"end_time,omitempty"` // 公告失效时间 TargetType uint8 `json:"target_type"` // 接收对象类型：1=全员，2=角色，3=用户 TargetID []int64 `json:"target_id,omitempty"` // 对象ID（角色ID/用户ID） } // UserMessageListReq 用户消息列表-入参 type UserMessageListReq struct { IsRead int `json:"is_read"` // 0-全部；1-已读；2-未读 Page int `json:"page_index"` // 页码 PageSize int `json:"page_size"` // 每页数量 } // UserMessageListResp 用户消息列表-出参 type UserMessageListResp struct { List []UserMessageDetailRespItem `json:"list"` Total int64 `json:"total"` Page int `json:"page_index"` PageSize int `json:"page_size"` } type UserMessageDetailRespItem struct { ID uint `json:"id"` CreatedAt time.Time `json:"created_at"` // 接收时间 UserID int64 `json:"user_id"` // 接收用户ID MessageConfigID uint `json:"message_config_id"` // 对应消息配置ID，可为空（被删时保留记录） MessageType uint8 `json:"message_type"` // 消息类型：1=公告，2=业务 Title string `json:"title"` // 消息标题快照 Content string `json:"content"` // 渲染后内容 IsRead bool `json:"is_read"` // 是否阅读 ReadTime *time.Time `json:"read_time,omitempty"` // 阅读时间 IsVisible bool `json:"is_visible"` // 是否展示给用户 Level uint8 `json:"level,omitempty"` // 公告消息字段，消息等级：1=普通，2=重要，3=紧急 DisplayMode uint8 `json:"display_mode,omitempty"` // 公告消息字段，展示方式：1=弹窗，2=滚动，3=弹窗+滚动 StartTime *time.Time `json:"start_time,omitempty"` // 公告消息字段，公告生效时间 EndTime *time.Time `json:"end_time,omitempty"` // 公告消息字段，公告失效时间 } // UserMessageSetStatusReq 用户阅读消息-入参 type UserMessageSetStatusReq struct { ID []uint `json:"id"` // 消息ID（可选，read_all 为 false 时必传） ReadAll bool `json:"read_all"` // 是否设为全部已读 } // BusMessageCreateReq 新增业务消息-入参 type BusMessageCreateReq struct { Title string `json:"title" binding:"required"` // 消息标题 ContentTemplate string `json:"content_template" binding:"required"` // 消息模板 BizType string `json:"biz_type" binding:"required"` // 业务类型 Event string `json:"event" binding:"required"` // 事件类型 RoleID []int64 `json:"role_id" binding:"required"` // 接收对象：角色ID列表 Status uint8 `json:"status" binding:"required"` // 消息状态：1启用，2禁用 } // BusMessageCreateResp 新增业务消息-出参 type BusMessageCreateResp struct { ID uint `json:"id"` } // BusMessageListReq 业务消息列表 - 入参 type BusMessageListReq struct { BizType string `json:"biz_type"` // 业务类型 Event string `json:"event"` // 事件类型 Status uint8 `json:"status"` // 状态：1启用，2禁用 RoleID []int64 `json:"role_id"` // 角色ID Page int `json:"page_index"` // 页码 PageSize int `json:"page_size"` // 每页条数 } // BusMessageListResp 业务消息列表 - 出参 type BusMessageListResp struct { List []BusMessageItem `json:"list"` Total int64 `json:"total"` Page int `json:"page_index"` PageSize int `json:"page_size"` } // BusMessageItem 单条业务消息展示项 type BusMessageItem struct { ID uint32 `json:"id"` CreatedAt time.Time `json:"created_at"` UpdatedAt time.Time `json:"updated_at"` Title string `json:"title"` // 消息标题 ContentTemplate string `json:"content_template"` // 消息模版 BizType string `json:"biz_type"` // 业务类型 BizName string `json:"biz_name"` // 业务名称 Event string `json:"event"` // 事件类型 EventName string `json:"event_name"` // 事件名称 Status uint8 `json:"status"` // 状态：1启用，2禁用 RoleID []int64 `json:"role_id"` // 角色ID } // BusMessageEditReq 编辑业务消息 - 入参 type BusMessageEditReq struct { ID uint32 `json:"id" binding:"required"` // 消息ID Title string `json:"title" binding:"required"` // 消息标题 BizType string `json:"biz_type" binding:"required"` // 业务类型 Event string `json:"event" binding:"required"` // 事件类型 ContentTemplate string `json:"content_template" binding:"required"` // 消息模版 Status uint8 `json:"status" binding:"required"` // 状态：1启用，2禁用 RoleID []int64 `json:"role_id" binding:"required"` // 接收对象：角色ID列表 } // BusMessageSetStatusReq 设置业务消息状态 type BusMessageSetStatusReq struct { ID uint `json:"id" binding:"required"` Status uint8 `json:"status" binding:"required"` // 目标状态：1=启用，2=禁用 } // BusMessageDeleteReq 删除业务消息 type BusMessageDeleteReq struct { ID uint `json:"id" binding:"required"` } // BusMessageDetailReq 业务消息详情 - 入参 type BusMessageDetailReq struct { ID uint64 `json:"id" binding:"required"` // 业务消息ID } // BizTypeListReq 查询业务类型列表 - 入参 type BizTypeListReq struct { Page int `json:"page_index"` PageSize int `json:"page_size"` } // BizTypeListResp 查询业务类型列表 - 出参 type BizTypeListResp struct { List []BizTypeItem `json:"list"` // 业务类型列表 Total int64 `json:"total"` // 总数 Page int `json:"page_index"` // 当前页 PageSize int `json:"page_size"` // 页大小 } // BizTypeItem 单条业务类型展示项 type BizTypeItem struct { BizType string `json:"biz_type"` // 业务类型编码 BizName string `json:"biz_name"` // 业务名称 } // EventListReq 查询事件类型列表 - 入参 type EventListReq struct { BizType string `json:"biz_type" binding:"required"` Page int `json:"page_index"` PageSize int `json:"page_size"` } // EventListResp 查询事件类型列表 - 出参 type EventListResp struct { List []EventItem `json:"list"` Total int64 `json:"total"` Page int `json:"page_index"` PageSize int `json:"page_size"` } // EventItem 单条事件类型展示项 type EventItem struct { Event string `json:"event"` // 事件类型编码 EventName string `json:"event_name"` // 业务类型编码 } // TemplateVarReq 查询模板变量字段列表 - 入参 type TemplateVarReq struct { BizType string `json:"biz_type" binding:"required"` // 业务类型 Event string `json:"event" binding:"required"` // 事件类型 Page int `json:"page_index"` // 当前页 PageSize int `json:"page_size"` // 每页大小 } // TemplateVarListResp 查询模板变量字段列表 - 出参 type TemplateVarListResp struct { List []TemplateVariable `json:"list"` // 模板变量列表 Total int64 `json:"total"` // 总数 Page int `json:"page_index"` // 当前页 PageSize int `json:"page_size"` // 每页条数 } // TemplateVariable 模板变量项 type TemplateVariable struct { Key string `json:"key"` // 模板变量键 Desc string `json:"desc"` // 模板变量描述 } // UserMessageDeleteReq 用户消息-删除 请求参数 type UserMessageDeleteReq struct { IDs []uint `json:"ids"` // 消息ID列表（DeleteAll=false时必填） DeleteAll bool `json:"delete_all"` // 是否删除全部 } // ValidateBizEventAndTemplate 校验业务类型、事件类型以及模板变量是否合法 func ValidateBizEventAndTemplate(bizType, event, contentTemplate string) error { var registry SystemEventRegistry err := orm.Eloquent.Where("biz_type = ? AND event = ? AND is_enabled = ?", bizType, event, true).First(®istry).Error if err != nil { return errors.New("业务类型或事件类型无效") } // 支持结构体数组格式：[{"key":"user_name","desc":"用户名"}] var allowedItems []TemplateVarItem if err := json.Unmarshal([]byte(registry.TemplateVariables), &allowedItems); err != nil { return errors.New("事件注册表中变量格式非法") } // 提取合法 key allowedMap := make(map[string]bool) for _, item := range allowedItems { allowedMap[item.Key] = true } // 提取模板中的变量名 varRegex := regexp.MustCompile(`\{\{\s*(\w+)\s*\}\}`) matches := varRegex.FindAllStringSubmatch(contentTemplate, -1) for _, match := range matches { varName := match[1] if !allowedMap[varName] { return fmt.Errorf("模板中使用了未注册变量：%s", varName) } } return nil } // TriggerBizEvent 业务消息-触发消息模版 func TriggerBizEvent(bizType, event string, inputVars map[string]interface{}, optionalReceivers ...uint32) error { // 1. 查找启用的消息配置 var cfg SystemMessageConfig err := orm.Eloquent.Where("message_type = ? AND biz_type = ? AND event = ? AND status = 1", BusinessMessageType, bizType, event).Find(&cfg).Error if err != nil { return nil // 无需发送 } // 2. 渲染模板内容 content, err := RenderTemplate(cfg.ContentTemplate, inputVars) if err != nil { return fmt.Errorf("模板渲染失败: %w", err) } // 查找门店ID var storeId uint32 if len(optionalReceivers) > 0 { if event == EventOnAllot { // 库存调拨-待收货消息 storeId = optionalReceivers[0] } } else { fmt.Println("没有传入 optionalReceivers，使用默认配置") } // 3. 查找接收用户 receiverIDs, err := GetTargetUserIds(cfg.ID, storeId) if err != nil { return fmt.Errorf("获取接收人失败: %w", err) } // 4. 写入 user_message 表 var msgs []SystemUserMessage for _, uid := range receiverIDs { msgs = append(msgs, SystemUserMessage{ UserID: uid, MessageConfigID: uint(cfg.ID), MessageType: BusinessMessageType, Title: cfg.Title, Content: content, IsRead: false, IsVisible: true, }) } return orm.Eloquent.Create(&msgs).Error } // GetTargetUserIds 获取接收用户函数模板（支持角色） func GetTargetUserIds(configID uint32, storeId uint32) ([]int64, error) { // 1.查询角色绑定 var targets []SystemMessageTarget err := orm.Eloquent.Where("message_config_id = ? AND target_type = ?", configID, TargetTypeByRole). Find(&targets).Error if err != nil { return nil, err } // 2.查询该模版配置的所有角色 var roleIDs []int64 for _, t := range targets { roleIDs = append(roleIDs, t.TargetID) } // 3.查询该角色下的用户（包含字段 uid, store_data） var users []SysUser err = orm.Eloquent.Table("sys_user"). Where("role_id IN ? AND deleted_at IS NULL", roleIDs). Find(&users).Error if err != nil { return nil, err } // 如果 storeId == 0，直接返回所有 uid if storeId == 0 { var allUserIDs []int64 for _, user := range users { allUserIDs = append(allUserIDs, int64(user.Uid)) } return allUserIDs, nil } // 4.筛选出门店有效用户 now := time.Now() var validUserIDs []int64 for _, user := range users { var storeList []StoreInfo if err := json.Unmarshal([]byte(user.StoreData), &storeList); err != nil { continue } for _, store := range storeList { // 解析时间 expireTime, err := time.Parse(StoreDateTimeFormat, store.ExpireTime) if err != nil { continue } // 包含当天有效时间 expireTime = expireTime.Add(24*time.Hour - time.Second) if store.StoreID == int(storeId) && expireTime.After(now) { validUserIDs = append(validUserIDs, int64(user.UserId)) break } } } return validUserIDs, nil } // RenderTemplate 渲染模版内容 func RenderTemplate(templateStr string, data map[string]interface{}) (string, error) { // 创建模板实例 tmpl, err := template.New("msg").Option("missingkey=zero").Parse(templateStr) if err != nil { return "", err } // 渲染 var buf bytes.Buffer err = tmpl.Execute(&buf, data) if err != nil { return "", err } return buf.String(), nil } // CheckExpiredSystemMessages 检查公告消息是否过期，如果过期则更新状态为3 func CheckExpiredSystemMessages() { now := time.Now() // 查询所有状态为启用（1）、类型为公告（1）、且 EndTime 不为空、已过期的消息 var expiredMessages []SystemMessageConfig err := orm.Eloquent. Table("system_message_config"). Where("message_type = ? AND status = ? AND end_time IS NOT NULL AND end_time < ?", AnnouncementMessageType, MessageStatusOnStart, now). Find(&expiredMessages).Error if err != nil { logger.Errorf("Failed to query expired messages: %v", err) return } if len(expiredMessages) == 0 { logger.Info("No expired announcements found.") return } // 遍历并更新状态 for _, msg := range expiredMessages { err := orm.Eloquent. Table("system_message_config"). Where("id = ?", msg.ID). Update("status", MessageStatusOnExpire).Error if err != nil { logger.Errorf("Failed to update status for message ID %d: %v", msg.ID, err) } else { logger.Infof("Message ID %d marked as expired.", msg.ID) } } }