EdgeCommon/pkg/serverconfigs/http_access_log_storage_policy.go

package serverconfigs

import (
	"github.com/TeaOSLab/EdgeCommon/pkg/configutils"
	"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs/shared"
	"github.com/iwind/TeaGo/Tea"
	"gopkg.in/yaml.v3"
	"io/ioutil"
	"os"
	"strconv"
)

// 日志存储策略
// 存储在configs/accesslog.storage.$id.conf
type HTTPAccessLogStoragePolicy struct {
	Id      int64                          `yaml:"id" json:"id"`
	Name    string                         `yaml:"name" json:"name"`
	IsOn    bool                           `yaml:"isOn" json:"isOn"`
	Type    string                         `yaml:"type" json:"type"`       // 存储类型
	Options map[string]interface{}         `yaml:"options" json:"options"` // 存储选项
	Conds   *shared.HTTPRequestCondsConfig `yaml:"conds" json:"conds"`     // 请求条件
}

// 校验
func (this *HTTPAccessLogStoragePolicy) Init() error {
	// cond
	if this.Conds != nil {
		err := this.Conds.Init()
		if err != nil {
			return err
		}
	}

	return nil
}

// 保存
func (this *HTTPAccessLogStoragePolicy) Save() error {
	shared.Locker.Lock()
	defer shared.Locker.Unlock()

	data, err := yaml.Marshal(this)
	if err != nil {
		return err
	}

	filename := "accesslog.storage." + this.IdString() + ".conf"
	return ioutil.WriteFile(Tea.ConfigFile(filename), data, 0666)
}

// 删除
func (this *HTTPAccessLogStoragePolicy) Delete() error {
	filename := "accesslog.storage." + this.IdString() + ".conf"
	return os.Remove(Tea.ConfigFile(filename))
}

// 匹配关键词
func (this *HTTPAccessLogStoragePolicy) MatchKeyword(keyword string) (matched bool, name string, tags []string) {
	if configutils.MatchKeyword(this.Name, keyword) || configutils.MatchKeyword(this.Type, keyword) {
		matched = true
		name = this.Name
		if len(this.Type) > 0 {
			tags = []string{"类型：" + this.Type}
		}
	}
	return
}

// 将ID转换为字符串
func (this *HTTPAccessLogStoragePolicy) IdString() string {
	return strconv.FormatInt(this.Id, 10)
}