用户界面设置中增加流量、带宽相关设置

internal/setup/sql_upgrade.go

@@ -78,6 +78,9 @@ var upgradeFuncs = []*upgradeVersion{
 	{
 		"0.4.8", upgradeV0_4_8,
 	},
+	{
+		"0.4.9", upgradeV0_4_9,
+	},
 }
 
 // UpgradeSQLData 升级SQL数据
@@ -676,7 +679,7 @@ func upgradeV0_4_7(db *dbs.DB) error {
 	return nil
 }
 
-// v0.4.7
+// v0.4.8
 func upgradeV0_4_8(db *dbs.DB) error {
 	// 设置edgeIPLists中的serverId
 	{
@@ -727,3 +730,37 @@ func upgradeV0_4_8(db *dbs.DB) error {
 
 	return nil
 }
+
+// v0.4.9
+func upgradeV0_4_9(db *dbs.DB) error {
+	// 升级用户UI配置
+	{
+		one, err := db.FindOne("SELECT value FROM edgeSysSettings WHERE code=?", systemconfigs.SettingCodeUserUIConfig)
+		if err != nil {
+			return err
+		}
+		if one != nil {
+			var valueJSON = one.GetBytes("value")
+			if len(valueJSON) > 0 {
+				var config = &systemconfigs.UserUIConfig{}
+				err = json.Unmarshal(valueJSON, config)
+				if err == nil {
+					config.ShowTrafficCharts = true
+					config.ShowBandwidthCharts = true
+					config.BandwidthUnit = systemconfigs.BandwidthUnitBit
+					configJSON, err := json.Marshal(config)
+					if err != nil {
+						return errors.New("encode UserUIConfig failed: " + err.Error())
+					} else {
+						_, err := db.Exec("UPDATE edgeSysSettings SET value=? WHERE code=?", configJSON, systemconfigs.SettingCodeUserUIConfig)
+						if err != nil {
+							return err
+						}
+					}
+				}
+			}
+		}
+	}
+
+	return nil
+}

internal/setup/sql_upgrade_test.go

@@ -194,3 +194,23 @@ 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",
+		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_9(db)
+	if err != nil {
+		t.Fatal(err)
+	}
+	t.Log("ok")
+}