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