智能DNS支持自定义端口

internal/setup/sql_upgrade.go

@@ -81,6 +81,9 @@ var upgradeFuncs = []*upgradeVersion{
 	{
 		"0.4.9", upgradeV0_4_9,
 	},
+	{
+		"0.4.11", upgradeV0_4_11,
+	},
 }
 
 // UpgradeSQLData 升级SQL数据
@@ -792,3 +795,51 @@ func upgradeV0_4_9(db *dbs.DB) error {
 
 	return nil
 }
+
+// v0.4.11
+func upgradeV0_4_11(db *dbs.DB) error {
+	// 升级ns端口
+	{
+		// TCP
+		{
+			var config = &serverconfigs.TCPProtocolConfig{}
+			config.IsOn = true
+			config.Listen = []*serverconfigs.NetworkAddressConfig{
+				{
+					Protocol:  serverconfigs.ProtocolTCP,
+					PortRange: "53",
+				},
+			}
+			configJSON, err := json.Marshal(config)
+			if err != nil {
+				return err
+			}
+			_, err = db.Exec("UPDATE edgeNSClusters SET tcp=? WHERE tcp IS NULL", configJSON)
+			if err != nil {
+				return err
+			}
+		}
+
+		// UDP
+		{
+			var config = &serverconfigs.UDPProtocolConfig{}
+			config.IsOn = true
+			config.Listen = []*serverconfigs.NetworkAddressConfig{
+				{
+					Protocol:  serverconfigs.ProtocolUDP,
+					PortRange: "53",
+				},
+			}
+			configJSON, err := json.Marshal(config)
+			if err != nil {
+				return err
+			}
+			_, err = db.Exec("UPDATE edgeNSClusters SET udp=? WHERE udp IS NULL", configJSON)
+			if err != nil {
+				return err
+			}
+		}
+	}
+
+	return nil
+}

internal/setup/sql_upgrade_test.go

@@ -195,7 +195,6 @@ func TestUpgradeSQLData_v0_4_8(t *testing.T) {
 	t.Log("ok")
 }
 
-
 func TestUpgradeSQLData_v0_4_9(t *testing.T) {
 	db, err := dbs.NewInstanceFromConfig(&dbs.DBConfig{
 		Driver: "mysql",
@@ -214,3 +213,22 @@ func TestUpgradeSQLData_v0_4_9(t *testing.T) {
 	}
 	t.Log("ok")
 }
+
+func TestUpgradeSQLData_v0_4_11(t *testing.T) {
+	db, err := dbs.NewInstanceFromConfig(&dbs.DBConfig{
+		Driver: "mysql",
+		Dsn:    "root:123456@tcp(127.0.0.1:3306)/db_edge?charset=utf8mb4&timeout=30s",
+		Prefix: "edge",
+	})
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer func() {
+		_ = db.Close()
+	}()
+	err = upgradeV0_4_11(db)
+	if err != nil {
+		t.Fatal(err)
+	}
+	t.Log("ok")
+}