优化代码

2025-11-06 01:50:25 +08:00 · 2022-09-23 15:25:51 +08:00
parent 5655f89ba6
commit c94b3c26c1
18 changed files with 358 additions and 1457 deletions
--- a/internal/setup/sql_upgrade_ext.go
+++ b/internal/setup/sql_upgrade_ext.go
@@ -0,0 +1,105 @@
+// Copyright 2022 Liuxiangchao iwind.liu@gmail.com. All rights reserved. Official site: https://goedge.cn .
+//go:build !plus
+
+package setup
+
+import (
+	"encoding/json"
+	"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs"
+	"github.com/iwind/TeaGo/dbs"
+	"github.com/iwind/TeaGo/types"
+	"regexp"
+)
+
+// v0.2.8.1
+func upgradeV0_2_8_1(db *dbs.DB) error {
+	// 升级EdgeDNS线路
+	ones, _, err := db.FindOnes("SELECT id, dnsRoutes FROM edgeNodes WHERE dnsRoutes IS NOT NULL")
+	if err != nil {
+		return err
+	}
+	for _, one := range ones {
+		var nodeId = one.GetInt64("id")
+		var dnsRoutes = one.GetString("dnsRoutes")
+		if len(dnsRoutes) == 0 {
+			continue
+		}
+		var m = map[string][]string{}
+		err = json.Unmarshal([]byte(dnsRoutes), &m)
+		if err != nil {
+			continue
+		}
+		var isChanged = false
+		var reg = regexp.MustCompile(`^\d+$`)
+		for k, routes := range m {
+			for index, route := range routes {
+				if reg.MatchString(route) {
+					route = "id:" + route
+					isChanged = true
+				}
+				routes[index] = route
+			}
+			m[k] = routes
+		}
+
+		if isChanged {
+			mJSON, err := json.Marshal(m)
+			if err != nil {
+				return err
+			}
+			_, err = db.Exec("UPDATE edgeNodes SET dnsRoutes=? WHERE id=? LIMIT 1", string(mJSON), nodeId)
+			if err != nil {
+				return err
+			}
+		}
+	}
+
+	return nil
+}
+
+// v0.5.3
+func upgradeV0_5_3(db *dbs.DB) error {
+	// 升级集群服务配置
+	{
+		type oldGlobalConfig struct {
+			// HTTP & HTTPS相关配置
+			HTTPAll struct {
+				MatchDomainStrictly  bool                                `yaml:"matchDomainStrictly" json:"matchDomainStrictly"`   // 是否严格匹配域名
+				AllowMismatchDomains []string                            `yaml:"allowMismatchDomains" json:"allowMismatchDomains"` // 允许的不匹配的域名
+				DefaultDomain        string                              `yaml:"defaultDomain" json:"defaultDomain"`               // 默认的域名
+				DomainMismatchAction *serverconfigs.DomainMismatchAction `yaml:"domainMismatchAction" json:"domainMismatchAction"` // 不匹配时采取的动作
+			} `yaml:"httpAll" json:"httpAll"`
+		}
+
+		value, err := db.FindCol(0, "SELECT value FROM edgeSysSettings WHERE code='serverGlobalConfig'")
+		if err != nil {
+			return err
+		}
+		if value != nil {
+			var valueJSON = []byte(types.String(value))
+			var oldConfig = &oldGlobalConfig{}
+			err = json.Unmarshal(valueJSON, oldConfig)
+			if err == nil {
+				var newConfig = &serverconfigs.GlobalServerConfig{}
+				newConfig.HTTPAll.MatchDomainStrictly = oldConfig.HTTPAll.MatchDomainStrictly
+				newConfig.HTTPAll.AllowMismatchDomains = oldConfig.HTTPAll.AllowMismatchDomains
+				newConfig.HTTPAll.DefaultDomain = oldConfig.HTTPAll.DefaultDomain
+				if oldConfig.HTTPAll.DomainMismatchAction != nil {
+					newConfig.HTTPAll.DomainMismatchAction = oldConfig.HTTPAll.DomainMismatchAction
+				}
+				newConfig.HTTPAll.AllowNodeIP = true
+
+				newConfig.Log.RecordServerError = false
+				newConfigJSON, err := json.Marshal(newConfig)
+				if err == nil {
+					_, err = db.Exec("UPDATE edgeNodeClusters SET globalServerConfig=?", newConfigJSON)
+					if err != nil {
+						return err
+					}
+				}
+			}
+		}
+	}
+
+	return nil
+}