[proxy]添加简单的分组管理

Author: iwind

teaconfigs/server_group.go

@@ -0,0 +1,41 @@
+package teaconfigs
+
+import (
+	"github.com/iwind/TeaGo/lists"
+	stringutil "github.com/iwind/TeaGo/utils/string"
+)
+
+type ServerGroup struct {
+	Id        string   `yaml:"id" json:"id"`
+	IsOn      bool     `yaml:"isOn" json:"isOn"`
+	Name      string   `yaml:"name" json:"name"`
+	ServerIds []string `yaml:"serverIds" json:"serverIds"`
+}
+
+func NewServerGroup() *ServerGroup {
+	return &ServerGroup{
+		Id:        stringutil.Rand(16),
+		IsOn:      true,
+		ServerIds: []string{},
+	}
+}
+
+func (this *ServerGroup) Add(serverId ...string) {
+	for _, id := range serverId {
+		if lists.ContainsString(this.ServerIds, id) {
+			continue
+		}
+		this.ServerIds = append(this.ServerIds, id)
+	}
+}
+
+func (this *ServerGroup) Remove(serverId string) {
+	result := []string{}
+	for _, id := range this.ServerIds {
+		if id == serverId {
+			continue
+		}
+		result = append(result, id)
+	}
+	this.ServerIds = result
+}

teaconfigs/server_group_list.go

@@ -0,0 +1,87 @@
+package teaconfigs
+
+import (
+	"github.com/TeaWeb/code/teaconfigs/shared"
+	"github.com/go-yaml/yaml"
+	"github.com/iwind/TeaGo/Tea"
+	"github.com/iwind/TeaGo/logs"
+	"io/ioutil"
+	"os"
+)
+
+type ServerGroupList struct {
+	Groups []*ServerGroup `yaml:"groups" json:"groups"`
+}
+
+func SharedServerGroupList() *ServerGroupList {
+	list := &ServerGroupList{
+		Groups: []*ServerGroup{},
+	}
+	filename := "servergrouplist.conf"
+	data, err := ioutil.ReadFile(Tea.ConfigFile(filename))
+	if err != nil {
+		if !os.IsNotExist(err) {
+			logs.Error(err)
+		}
+		return list
+	}
+
+	err = yaml.Unmarshal(data, list)
+	if err != nil {
+		logs.Error(err)
+		return list
+	}
+
+	return list
+}
+
+func (this *ServerGroupList) Add(group *ServerGroup) {
+	this.Groups = append(this.Groups, group)
+}
+
+func (this *ServerGroupList) Remove(id string) {
+	result := []*ServerGroup{}
+	for _, group := range this.Groups {
+		if group.Id == id {
+			continue
+		}
+		result = append(result, group)
+	}
+	this.Groups = result
+}
+
+// 查找分组
+func (this *ServerGroupList) Find(id string) *ServerGroup {
+	for _, group := range this.Groups {
+		if group.Id == id {
+			return group
+		}
+	}
+	return nil
+}
+
+// 检查是否包含某个代理服务ID
+func (this *ServerGroupList) ContainsServer(serverId string) bool {
+	for _, group := range this.Groups {
+		for _, id := range group.ServerIds {
+			if id == serverId {
+				return true
+			}
+		}
+	}
+	return false
+}
+
+// 保存
+func (this *ServerGroupList) Save() error {
+	data, err := yaml.Marshal(this)
+	if err != nil {
+		return err
+	}
+
+	shared.Locker.Lock()
+	defer shared.Locker.WriteUnlock()
+
+	filename := "servergrouplist.conf"
+	return ioutil.WriteFile(Tea.ConfigFile(filename), data, 0666)
+}

teaconfigs/server_group_list_test.go

@@ -0,0 +1,10 @@
+package teaconfigs
+
+import (
+	"github.com/iwind/TeaGo/logs"
+	"testing"
+)
+
+func TestSharedServerGroupList(t *testing.T) {
+	logs.PrintAsJSON(SharedServerGroupList(), t)
+}

teaweb/actions/default/cache/helper.go

@@ -12,6 +12,6 @@ type Helper struct {
 // 缓存相关Helper
 func (this *Helper) BeforeAction(action *actions.ActionObject) {
 	if action.Request.Method == http.MethodGet {
-		proxyutils.AddServerMenu(action)
+		proxyutils.AddServerMenu(action, false)
 	}
 }

teaweb/actions/default/proxy/certs/helper.go

@@ -13,5 +13,5 @@ func (this *Helper) BeforeAction(action *actions.ActionObject) {
 	if action.Request.Method != http.MethodGet {
 		return
 	}
-	proxyutils.AddServerMenu(action)
+	proxyutils.AddServerMenu(action, false)
 }

teaweb/actions/default/proxy/helper.go

@@ -11,6 +11,10 @@ type Helper struct {
 
 func (this *Helper) BeforeAction(action *actions.ActionObject) {
 	if action.Request.Method == http.MethodGet && !action.HasPrefix("/proxy/status") {
-		proxyutils.AddServerMenu(action)
+		if action.HasPrefix("/proxy/add") {
+			proxyutils.AddServerMenu(action, false)
+		} else {
+			proxyutils.AddServerMenu(action, true)
+		}
 	}
 }

teaweb/actions/default/proxy/import.go

@@ -0,0 +1,5 @@
+package proxy
+
+import (
+	_ "github.com/TeaWeb/code/teaweb/actions/default/proxy/servergroups"
+)

teaweb/actions/default/proxy/locations/waf.go

@@ -22,7 +22,7 @@ func (this *WafAction) RunGet(params struct {
 
 	_, location := locationutils.SetCommonInfo(this, params.ServerId, params.LocationId, "waf")
 
-	proxyutils.AddServerMenu(this)
+	proxyutils.AddServerMenu(this, true)
 
 	wafList := []maps.Map{}
 	for _, waf := range teaconfigs.SharedWAFList().FindAllConfigs() {

teaweb/actions/default/proxy/log/day.go

@@ -46,7 +46,7 @@ func (this *DayAction) Run(params struct {
 	}
 	this.Data["searchIP"] = params.SearchIP
 
-	proxyutils.AddServerMenu(this)
+	proxyutils.AddServerMenu(this, true)
 
 	// 检查数据库连接
 	this.Data["mongoError"] = ""

teaweb/actions/default/proxy/log/history.go

@@ -26,7 +26,7 @@ func (this *HistoryAction) Run(params struct {
 		"id": server.Id,
 	}
 
-	proxyutils.AddServerMenu(this)
+	proxyutils.AddServerMenu(this, true)
 
 	// 检查MongoDB连接
 	this.Data["mongoError"] = ""