重新实现套餐相关功能

internal/setup/sql_upgrade.go

@@ -95,6 +95,9 @@ var upgradeFuncs = []*upgradeVersion{
 	{
 		"1.2.1", upgradeV1_2_1,
 	},
+	{
+		"1.2.9", upgradeV1_2_9,
+	},
 }
 
 // UpgradeSQLData 升级SQL数据
@@ -746,3 +749,24 @@ func upgradeV1_2_1(db *dbs.DB) error {
 	}
 	return nil
 }
+
+// v1.2.9
+func upgradeV1_2_9(db *dbs.DB) error {
+	// 升级套餐网站数限制
+	{
+		_, err := db.Exec("UPDATE edgePlans SET totalServers=1 WHERE totalServers=0")
+		if err != nil {
+			return err
+		}
+	}
+
+	// 升级网站流量限制状态
+	{
+		_, err := db.Exec("UPDATE edgeServers SET trafficLimitStatus=NULL WHERE trafficLimitStatus IS NOT NULL")
+		if err != nil {
+			return err
+		}
+	}
+
+	return nil
+}

internal/setup/sql_upgrade_test.go

@@ -271,4 +271,21 @@ func TestUpgradeSQLData_v1_2_1(t *testing.T) {
 	t.Log("ok")
 }
 
-
+func TestUpgradeSQLData_v1_2_9(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 = upgradeV1_2_9(db)
+	if err != nil {
+		t.Fatal(err)
+	}
+	t.Log("ok")
+}