浏览代码

refactor(repository): 优化 IP黑白名单处理逻辑

-移除了 TCP、UDP 和 Web转发中对 AllowIpList 和 DenyIpList 的条件判断
- 直接将 req.AllowIpList 和 req.DenyIpList 赋值给 updateData- 简化了代码结构,提高了代码可读性和维护性
fusu 1 月之前
父节点
当前提交
7d15b8e20b
共有 3 个文件被更改,包括 11 次插入18 次删除
  1. 4 6
      internal/repository/tcpforwarding.go
  2. 4 6
      internal/repository/udpforwarding.go
  3. 3 6
      internal/repository/webforwarding.go

+ 4 - 6
internal/repository/tcpforwarding.go

@@ -129,13 +129,11 @@ func (r *tcpforwardingRepository) EditTcpforwardingIps(ctx context.Context, req
 		updateData["backend_list"] = req.BackendList
 	}
 
-	if len(req.AllowIpList) > 0 {
-		updateData["allow_ip_list"] = req.AllowIpList
-	}
+	updateData["allow_ip_list"] = req.AllowIpList
+
+
+	updateData["deny_ip_list"] = req.DenyIpList
 
-	if len(req.DenyIpList) > 0 {
-		updateData["deny_ip_list"] = req.DenyIpList
-	}
 
 	// 始终更新更新时间
 	updateData["updated_at"] = time.Now()

+ 4 - 6
internal/repository/udpforwarding.go

@@ -130,13 +130,11 @@ func (r *udpForWardingRepository) EditUdpForwardingIps(ctx context.Context, req
 		updateData["backend_list"] = req.BackendList
 	}
 
-	if len(req.AllowIpList) > 0 {
-		updateData["allow_ip_list"] = req.AllowIpList
-	}
+	updateData["allow_ip_list"] = req.AllowIpList
+
+
+	updateData["deny_ip_list"] = req.DenyIpList
 
-	if len(req.DenyIpList) > 0 {
-		updateData["deny_ip_list"] = req.DenyIpList
-	}
 
 	// 始终更新更新时间
 	updateData["updated_at"] = time.Now()

+ 3 - 6
internal/repository/webforwarding.go

@@ -147,13 +147,10 @@ func (r *webForwardingRepository) EditWebForwardingIps(ctx context.Context, req
 		updateData["backend_list"] = req.BackendList
 	}
 
-	if len(req.AllowIpList) > 0 {
-		updateData["allow_ip_list"] = req.AllowIpList
-	}
+	updateData["allow_ip_list"] = req.AllowIpList
+
+	updateData["deny_ip_list"] = req.DenyIpList
 
-	if len(req.DenyIpList) > 0 {
-		updateData["deny_ip_list"] = req.DenyIpList
-	}
 
 	// 始终更新更新时间
 	updateData["updated_at"] = time.Now()