From 65fe409c0445da51552fbabf23a0b69a87f220f6 Mon Sep 17 00:00:00 2001
From: TsMask <340112800@qq.com>
Date: Thu, 15 Aug 2024 09:44:37 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E5=AF=B9=E7=99=BB=E5=BD=95=EF=BC=8C?=
 =?UTF-8?q?=E7=BD=91=E5=85=83=E4=BF=A1=E6=81=AF=E6=8E=A5=E5=8F=A3=E5=8A=A0?=
 =?UTF-8?q?=E8=A7=A3=E5=AF=86?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/modules/common/common.go                   | 2 ++
 src/modules/network_element/network_element.go | 4 ++++
 2 files changed, 6 insertions(+)

diff --git a/src/modules/common/common.go b/src/modules/common/common.go
index bb6c68b0..3bbbd97a 100644
--- a/src/modules/common/common.go
+++ b/src/modules/common/common.go
@@ -52,6 +52,7 @@ func Setup(router *gin.Engine) {
 			// 	Count: 10,
 			// 	Type:  middleware.LIMIT_IP,
 			// }),
+			middleware.CryptoApi(true, true),
 			controller.NewAccount.Login,
 		)
 		indexGroup.GET("/getInfo", middleware.PreAuthorize(nil), controller.NewAccount.Info)
@@ -74,6 +75,7 @@ func Setup(router *gin.Engine) {
 			// 	Count: 10,
 			// 	Type:  middleware.LIMIT_IP,
 			// }),
+			middleware.CryptoApi(true, true),
 			controller.NewRegister.Register,
 		)
 	}
diff --git a/src/modules/network_element/network_element.go b/src/modules/network_element/network_element.go
index bc068e48..89cd4624 100644
--- a/src/modules/network_element/network_element.go
+++ b/src/modules/network_element/network_element.go
@@ -80,15 +80,18 @@ func Setup(router *gin.Engine) {
 			controller.NewNeInfo.List,
 		)
 		neInfoGroup.GET("/:infoId",
+			middleware.CryptoApi(false, true),
 			middleware.PreAuthorize(nil),
 			controller.NewNeInfo.Info,
 		)
 		neInfoGroup.POST("",
+			middleware.CryptoApi(true, true),
 			middleware.PreAuthorize(nil),
 			collectlogs.OperateLog(collectlogs.OptionNew("log.operate.title.neInfo", collectlogs.BUSINESS_TYPE_INSERT)),
 			controller.NewNeInfo.Add,
 		)
 		neInfoGroup.PUT("",
+			middleware.CryptoApi(true, true),
 			middleware.PreAuthorize(nil),
 			collectlogs.OperateLog(collectlogs.OptionNew("log.operate.title.neInfo", collectlogs.BUSINESS_TYPE_UPDATE)),
 			controller.NewNeInfo.Edit,
@@ -108,6 +111,7 @@ func Setup(router *gin.Engine) {
 			controller.NewNeHost.List,
 		)
 		neHostGroup.GET("/:hostId",
+			middleware.CryptoApi(false, true),
 			middleware.PreAuthorize(nil),
 			controller.NewNeHost.Info,
 		)