瀏覽代碼

skip ns ip if contains network cidr

abhishek9686 4 天之前
父節點
當前提交
34d4499e4e
共有 1 個文件被更改,包括 6 次插入2 次删除
  1. 6 2
      migrate/migrate.go

+ 6 - 2
migrate/migrate.go

@@ -5,6 +5,7 @@ import (
 	"encoding/json"
 	"fmt"
 	"log"
+	"net"
 	"os"
 	"time"
 
@@ -78,8 +79,11 @@ func migrateNameservers() {
 				Status:    true,
 				CreatedBy: user.UserName,
 			}
-			for _, ip := range netI.NameServers {
-				ns.Servers = append(ns.Servers, ip)
+			for _, nsIP := range netI.NameServers {
+				_, cidr, _ := net.ParseCIDR(netI.AddressRange)
+				if !cidr.Contains(net.ParseIP(nsIP)) {
+					ns.Servers = append(ns.Servers, nsIP)
+				}
 			}
 			ns.Create(db.WithContext(context.TODO()))
 			netI.NameServers = []string{}