1、增加钱包白名单
This commit is contained in:
@ -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
|
||||
}
|
||||
}
|
||||
|
||||
@ -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)
|
||||
}
|
||||
}
|
||||
|
||||
@ -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:"创建时间"`
|
||||
}
|
||||
|
||||
@ -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
|
||||
}
|
||||
|
||||
@ -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
|
||||
|
||||
Reference in New Issue
Block a user