1、增加钱包白名单

This commit is contained in:
2025-07-26 09:08:29 +08:00
parent 0a2b24e5a3
commit f5ade5b632
5 changed files with 90 additions and 23 deletions

View File

@ -234,3 +234,26 @@ func (e MmWalletWhiteList) Delete(c *gin.Context) {
}
e.OK(req.GetId(), "删除成功")
}
// Export 导出钱包白名单
func (e MmWalletWhiteList) Export(c *gin.Context) {
req := dto.MmWalletWhiteListGetPageReq{}
s := service.MmWalletWhiteList{}
err := e.MakeContext(c).
MakeOrm().
Bind(&req).
MakeService(&s.Service).
Errors
if err != nil {
e.Logger.Error(err)
e.Error(500, err, err.Error())
return
}
// p := actions.GetPermissionFromContext(c)
if err := s.ExportList(c, &req); err != nil {
e.Error(500, err, fmt.Sprintf("导出钱包白名单失败,\r\n失败信息 %s", err.Error()))
return
}
}

View File

@ -25,5 +25,6 @@ func registerMmWalletWhiteListRouter(v1 *gin.RouterGroup, authMiddleware *jwt.Gi
r.DELETE("", api.Delete)
r.POST("batch", api.BatchInsert)
r.GET("export", actions.PermissionAction(), api.Export)
}
}

View File

@ -111,3 +111,10 @@ type MmWalletWhiteListDeleteReq struct {
func (s *MmWalletWhiteListDeleteReq) GetId() interface{} {
return s.Ids
}
// 导出excel对象
type MmWalletWhiteListExportResp struct {
Address string `json:"address" excel:"钱包地址"`
StatusName string `json:"status_name" excel:"状态"`
CreateTime string `json:"create_time" excel:"创建时间"`
}

View File

@ -2,10 +2,8 @@ package service
import (
"errors"
"strconv"
"strings"
"github.com/bytedance/sonic"
"github.com/go-admin-team/go-admin-core/sdk/service"
"gorm.io/gorm"
@ -13,7 +11,6 @@ import (
"go-admin/app/admin/service/dto"
"go-admin/common/actions"
cDto "go-admin/common/dto"
"go-admin/common/global"
"go-admin/common/helper"
)
@ -202,36 +199,36 @@ func (e *MmRiskLog) SendMsg(datas []models.MmRiskLog) error {
return nil
}
userService := SysUser{Service: e.Service}
userIds, err := userService.GetByPermission("admin:mmAlarmLog:notice")
// userService := SysUser{Service: e.Service}
// userIds, err := userService.GetByPermission("admin:mmAlarmLog:notice")
if err != nil {
e.Log.Errorf("获取管理员失败:%s", err)
}
// if err != nil {
// e.Log.Errorf("获取管理员失败:%s", err)
// }
userIdStr := []string{}
// userIdStr := []string{}
for _, item := range userIds {
userIdStr = append(userIdStr, strconv.Itoa(item))
}
// for _, item := range userIds {
// userIdStr = append(userIdStr, strconv.Itoa(item))
// }
if err := e.Orm.Save(&alarmLogs).Error; err != nil {
e.Log.Errorf("保存告警日志失败:%s", err)
return err
}
for _, item := range alarmLogs {
notice := global.AlarmLogNotice{
MachineId: item.MachineId,
BiosId: item.BiosId,
Content: item.Content,
}
vals, _ := sonic.Marshal(notice)
// for _, item := range alarmLogs {
// notice := global.AlarmLogNotice{
// MachineId: item.MachineId,
// BiosId: item.BiosId,
// Content: item.Content,
// }
// vals, _ := sonic.Marshal(notice)
if vals != nil {
global.WebSocketHub.SendToAll(userIdStr, vals)
}
}
// if vals != nil {
// global.WebSocketHub.SendToAll(userIdStr, vals)
// }
// }
return nil
}

View File

@ -4,6 +4,7 @@ import (
"errors"
"strings"
"github.com/gin-gonic/gin"
"github.com/go-admin-team/go-admin-core/sdk/service"
"gorm.io/gorm"
@ -19,6 +20,44 @@ type MmWalletWhiteList struct {
service.Service
}
// 导出excel
func (e MmWalletWhiteList) ExportList(c *gin.Context, req *dto.MmWalletWhiteListGetPageReq) error {
var err error
var data models.MmWalletWhiteList
var list []models.MmWalletWhiteList
err = e.Orm.Model(&data).
Scopes(
cDto.MakeCondition(req.GetNeedSearch()),
).
Find(&list).Error
if err != nil {
return err
}
excelData := make([]dto.MmWalletWhiteListExportResp, 0)
for _, item := range list {
statusName := ""
if item.Status == 1 {
statusName = "启用"
} else {
statusName = "禁用"
}
excelData = append(excelData, dto.MmWalletWhiteListExportResp{
Address: item.Address,
StatusName: statusName,
CreateTime: item.CreatedAt.Format("2006-01-02 15:04:05"),
})
}
if err := helper.ExportExcel(c, "钱包白名单导出", excelData, []string{}); err != nil {
return err
}
return nil
}
// GetPage 获取MmWalletWhiteList列表
func (e *MmWalletWhiteList) GetPage(c *dto.MmWalletWhiteListGetPageReq, p *actions.DataPermission, list *[]models.MmWalletWhiteList, count *int64) error {
var err error