go-base/config/subconfigs/log_config.go

package subconfigs

import (
	"fmt"
	"strings"
)

// LogConfig 日志配置
type LogConfig struct {
	BaseConfig
	Level  string `yaml:"level" json:"level"`   // 日志级别
	Output string `yaml:"output" json:"output"` // 输出目标：console,file,es
	//JSONFormat bool   `yaml:"json_format" json:"json_format"` // 是否JSON格式

	// 文件输出配置
	FilePath   string `yaml:"file_path" json:"file_path"`
	MaxSize    int    `yaml:"max_size" json:"max_size"`
	MaxBackups int    `yaml:"max_backups" json:"max_backups"`
	MaxAge     int    `yaml:"max_age" json:"max_age"`
	Compress   bool   `yaml:"compress" json:"compress"`

	// ES输出配置（当Output包含"es"时生效）
	ESPath     string `yaml:"es_path" json:"es_path"`         // ES地址
	ESUsername string `yaml:"es_username" json:"es_username"` // 用户名（可选）
	ESPassword string `yaml:"es_password" json:"es_password"` // 密码（可选）
	//ESAPIKey   string `yaml:"es_api_key" json:"es_api_key"`     // API Key（可选）
	//ESBuffer   int    `yaml:"es_buffer" json:"es_buffer"`       // 缓冲大小，默认10000
	//ESMaxRetry int    `yaml:"es_max_retry" json:"es_max_retry"` // 最大重试，默认3
}

// NewLogConfig 创建日志配置实例
func NewLogConfig() *LogConfig {
	return &LogConfig{}
}

// SetDefaults 设置默认值
func (c *LogConfig) SetDefaults() {
	c.Level = "info"
	c.Output = "console"
	c.FilePath = "logs/service.log"
	c.MaxSize = 100
	c.MaxBackups = 3
	c.MaxAge = 30
	c.Compress = true
	c.ESPath = "logs/es-service.log"
	//c.JSONFormat = true
}

// Load 从yaml数据加载
func (c *LogConfig) Load(data map[string]interface{}) error {
	return c.LoadFromYAML(data, c)
}

// Validate 验证配置
func (c *LogConfig) Validate() error {
	// 验证日志级别
	validLevels := map[string]bool{
		"debug": true, "info": true, "warn": true, "error": true,
	}
	if !validLevels[strings.ToLower(c.Level)] {
		return fmt.Errorf("invalid log level: %s", c.Level)
	}

	// 验证输出类型
	validOutputs := map[string]bool{
		"console": true, "file": true, "both": true,
	}
	if !validOutputs[c.Output] {
		return fmt.Errorf("invalid log output: %s", c.Output)
	}

	return nil
}

// 自动注册
func init() {
	Register("log", &LogConfig{})
}