// Package gen // @Link https://github.com/xiujiecn/xiu-admin // @Copyright Copyright (c) 2026 LiXiujie // @Author Lxj // @License https://github.com/xiujiecn/xiu-admin/blob/master/LICENSE // @AutoGenerate Version package common import ( "context" "fmt" "git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/dao" "git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/library/contexts" "git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/library/exportcenter" "git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/library/websocket" "git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/library/xgorm/handler" "git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/library/xgorm/hook" "git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/model" "git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/model/entity" "git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/service" "git.5iot.com/iotlcm-public/iotlcm-admin-svr/utility/convert" "strings" "github.com/gogf/gf/v2/database/gdb" "github.com/gogf/gf/v2/errors/gerror" "github.com/gogf/gf/v2/frame/g" "github.com/gogf/gf/v2/os/gtime" "github.com/gogf/gf/v2/util/gconv" ) type sCommonTNotifyLog struct{} func NewCommonTNotifyLog() *sCommonTNotifyLog { return &sCommonTNotifyLog{} } func init() { service.RegisterCommonTNotifyLog(NewCommonTNotifyLog()) } // Model 通知日志ORM模型 func (s *sCommonTNotifyLog) Model(ctx context.Context, option ...*handler.Option) *gdb.Model { if len(option) == 0 { option = append(option, &handler.Option{ FilterTenant: true, FilterAuth: true, }) } return handler.Model(dao.TNotifyLog.Ctx(ctx), option...) } // List 获取通知日志列表 func (s *sCommonTNotifyLog) List(ctx context.Context, in *model.TNotifyLogListParam) (list []*model.TNotifyLogListModel, totalCount int, err error) { mod := dao.TNotifyLog.Ctx(ctx) // 字段过滤 mod = mod.Fields(model.TNotifyLogListModel{}) if title := strings.TrimSpace(in.NotifyTitle); title != "" { mod = mod.WhereLike(dao.TNotifyLog.Columns().NotifyTitle, "%"+title+"%") } // 查询发送用户类型 if in.SendUserType != nil && *in.SendUserType > 0 { mod = mod.Where(dao.TNotifyLog.Columns().SendUserType, *in.SendUserType) } // 查询发送时间范围 if in.SendTimeStart != nil { mod = mod.WhereGTE(dao.TNotifyLog.Columns().SendTime, in.SendTimeStart) } if in.SendTimeEnd != nil { mod = mod.WhereLTE(dao.TNotifyLog.Columns().SendTime, in.SendTimeEnd.EndOfDay()) } // 兼容旧的单日发送时间查询 if in.SendTime != nil { mod = mod.Where(fmt.Sprintf("DATE(%s) = ?", dao.TNotifyLog.Columns().SendTime), in.SendTime.Format("Y-m-d")) } userId := contexts.GetUserId(ctx) var sysUser *entity.SysUser dao.SysUser.Ctx(ctx).WherePri(userId).Scan(&sysUser) userType := sysUser.UserType // 1. 查找顶级部门(ParentId = 0) var topDept entity.SysDept err = dao.SysDept.Ctx(ctx). Where(dao.SysDept.Columns().ParentId, 0). Scan(&topDept) if err != nil { return } // 获取当前用户部门ID deptId := contexts.GetDeptId(ctx) // 只有非顶级用户才根据用户类型过滤 if deptId != topDept.DeptId { if userType == "sys_user" { mod = mod.Where(dao.TNotifyLog.Columns().SendUserType, 1) } else if userType == "app_user" { mod = mod.Where(dao.TNotifyLog.Columns().SendUserType, 2) } } // 分页 mod = mod.Page(in.Page, in.PageSize) // 排序 mod = mod.OrderDesc(dao.TNotifyLog.Columns().NotifyId) // 操作人摘要信息 mod = mod.Hook(hook.MemberSummary) // 查询数据 if err = mod.ScanAndCount(&list, &totalCount, false); err != nil { err = gerror.Wrap(err, "获取通知日志列表失败,请稍后重试!") return } return } // Export 导出通知日志(异步:提交任务后到下载中心取文件) func (s *sCommonTNotifyLog) Export(ctx context.Context, in *model.TNotifyLogListParam) (err error) { _, err = exportcenter.Submit(ctx, &exportcenter.Request{ BizKey: "tNotifyLog", BizName: "通知日志", PermCode: "cpm:common:tNotifyLog:export", QueryParam: in, Tags: func() ([]string, error) { return convert.GetEntityDescTags(model.TNotifyLogListModel{}) }, Fetch: func(ctx context.Context, page, pageSize int) (rows []interface{}, total int, err error) { query := *in query.Page = page query.PageSize = pageSize list, count, err := s.List(ctx, &query) if err != nil { return nil, 0, err } rows = make([]interface{}, 0, len(list)) for _, item := range list { rows = append(rows, item) } return rows, count, nil }, }) return err } // Edit 修改/新增通知日志 func (s *sCommonTNotifyLog) Edit(ctx context.Context, in *model.TNotifyLogEditParam) (err error) { var ( pushPayload g.Map pushUserIds []uint64 ) err = g.DB().Transaction(ctx, func(ctx context.Context, tx gdb.TX) (err error) { // 修改 if in.NotifyId > 0 { tenantId := contexts.GetTenantId(ctx) in.TenantId = &tenantId updatedBy := contexts.GetUserId(ctx) in.UpdatedBy = &updatedBy in.UpdatedAt = gtime.Now() if _, err = s.Model(ctx). Fields(model.TNotifyLogUpdateFields{}). WherePri(in.NotifyId).Data(in).Update(); err != nil { err = gerror.Wrap(err, "修改通知日志失败,请稍后重试!") } return } // 新增 tenantId := contexts.GetTenantId(ctx) in.TenantId = &tenantId createdDept := contexts.GetDeptId(ctx) in.CreatedDept = &createdDept createdBy := contexts.GetUserId(ctx) in.CreatedBy = &createdBy in.CreatedAt = gtime.Now() in.SendTime = gtime.Now() result, err := s.Model(ctx, &handler.Option{FilterAuth: false}). Fields(model.TNotifyLogInsertFields{}). Data(in).Insert() if err != nil { err = gerror.Wrap(err, "新增通知日志失败,请稍后重试!") return } notifyId, err := result.LastInsertId() if err != nil { err = gerror.Wrap(err, "获取通知ID失败") return } pushPayload = g.Map{ "notifyId": notifyId, "notifyTitle": gconv.String(in.NotifyTitle), "notifyContent": gconv.String(in.NotifyContent), "sendUserType": gconv.Int(in.SendUserType), "sendTime": in.SendTime, } var userType string if in.SendUserType != nil { switch *in.SendUserType { case 1: userType = "sys_user" case 2: userType = "app_user" } } if userType != "" { var users []*entity.SysUser if err = tx.Ctx(ctx).Model(dao.SysUser.Table()). Fields(dao.SysUser.Columns().UserId). Where(dao.SysUser.Columns().TenantId, tenantId). Where(dao.SysUser.Columns().UserType, userType). Where(dao.SysUser.Columns().Status, "0"). Scan(&users); err != nil { err = gerror.Wrap(err, "获取通知推送用户失败") return } for _, user := range users { if user.UserId > 0 { pushUserIds = append(pushUserIds, uint64(user.UserId)) } } } return }) if err != nil { return } for _, userId := range pushUserIds { g.Log().Info(ctx, "推送通知到用户 %v", userId) websocket.SendToUser(userId, &websocket.WResponse{ Event: "notify:new", Data: pushPayload, }) } return } // Delete 删除通知日志 func (s *sCommonTNotifyLog) Delete(ctx context.Context, in *model.TNotifyLogDeleteParam) (err error) { if _, err = s.Model(ctx).WherePri(in.NotifyId).Unscoped().Delete(); err != nil { err = gerror.Wrap(err, "删除通知日志失败,请稍后重试!") return } return } // View 获取通知日志指定信息 func (s *sCommonTNotifyLog) View(ctx context.Context, in *model.TNotifyLogViewParam) (res *model.TNotifyLogViewModel, err error) { if err = dao.TNotifyLog.Ctx(ctx).WherePri(in.NotifyId).Hook(hook.MemberSummary).Scan(&res); err != nil { err = gerror.Wrap(err, "获取通知日志信息,请稍后重试!") return } return }