增加字符编码/HTTP Header管理

internal/db/models/http_header_dao.go

@@ -1,9 +1,13 @@
 package models
 
 import (
+	"encoding/json"
+	"errors"
+	"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs/shared"
 	_ "github.com/go-sql-driver/mysql"
 	"github.com/iwind/TeaGo/Tea"
 	"github.com/iwind/TeaGo/dbs"
+	"github.com/iwind/TeaGo/types"
 )
 
 const (
@@ -27,7 +31,7 @@ func NewHTTPHeaderDAO() *HTTPHeaderDAO {
 var SharedHTTPHeaderDAO = NewHTTPHeaderDAO()
 
 // 启用条目
-func (this *HTTPHeaderDAO) EnableHTTPHeader(id uint32) error {
+func (this *HTTPHeaderDAO) EnableHTTPHeader(id int64) error {
 	_, err := this.Query().
 		Pk(id).
 		Set("state", HTTPHeaderStateEnabled).
@@ -45,7 +49,7 @@ func (this *HTTPHeaderDAO) DisableHTTPHeader(id uint32) error {
 }
 
 // 查找启用中的条目
-func (this *HTTPHeaderDAO) FindEnabledHTTPHeader(id uint32) (*HTTPHeader, error) {
+func (this *HTTPHeaderDAO) FindEnabledHTTPHeader(id int64) (*HTTPHeader, error) {
 	result, err := this.Query().
 		Pk(id).
 		Attr("state", HTTPHeaderStateEnabled).
@@ -57,9 +61,68 @@ func (this *HTTPHeaderDAO) FindEnabledHTTPHeader(id uint32) (*HTTPHeader, error)
 }
 
 // 根据主键查找名称
-func (this *HTTPHeaderDAO) FindHTTPHeaderName(id uint32) (string, error) {
+func (this *HTTPHeaderDAO) FindHTTPHeaderName(id int64) (string, error) {
 	return this.Query().
 		Pk(id).
 		Result("name").
 		FindStringCol("")
 }
+
+// 创建Header
+func (this *HTTPHeaderDAO) CreateHeader(name string, value string) (int64, error) {
+	op := NewHTTPHeaderOperator()
+	op.State = HTTPHeaderStateEnabled
+	op.IsOn = true
+	op.Name = name
+	op.Value = value
+	_, err := this.Save(op)
+	if err != nil {
+		return 0, err
+	}
+	return types.Int64(op.Id), nil
+}
+
+// 修改Header
+func (this *HTTPHeaderDAO) UpdateHeader(headerId int64, name string, value string) error {
+	if headerId <= 0 {
+		return errors.New("invalid headerId")
+	}
+
+	op := NewHTTPHeaderOperator()
+	op.Id = headerId
+	op.Name = name
+	op.Value = value
+	_, err := this.Save(op)
+
+	// TODO 更新相关配置
+
+	return err
+}
+
+// 组合Header配置
+func (this *HTTPHeaderDAO) ComposeHeaderConfig(headerId int64) (*shared.HTTPHeaderConfig, error) {
+	header, err := this.FindEnabledHTTPHeader(headerId)
+	if err != nil {
+		return nil, err
+	}
+	if header == nil {
+		return nil, nil
+	}
+
+	config := &shared.HTTPHeaderConfig{}
+	config.Id = int64(header.Id)
+	config.IsOn = header.IsOn == 1
+	config.Name = header.Name
+	config.Value = header.Value
+
+	if len(header.Status) > 0 {
+		status := &shared.HTTPStatusConfig{}
+		err = json.Unmarshal([]byte(header.Status), status)
+		if err != nil {
+			return nil, err
+		}
+		config.Status = status
+	}
+
+	return config, nil
+}