chore: 并入 iotlcm-admin-svr 历史到 admin-svr/（原提交 49818c6）
This commit is contained in:
MiFaZhan
@@ -0,0 +1,264 @@
|
||||
// Package gen
|
||||
// @Link https://github.com/xiujiecn/xiu-admin
|
||||
// @Copyright Copyright (c) 2026 LiXiujie
|
||||
// @Author Lxj <xxx@example.com>
|
||||
// @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
|
||||
}
|
||||
Reference in New Issue
Block a user