package main

import (
    "context"
    "github.com/lifezq/log"
    "go.uber.org/zap"
    "go.uber.org/zap/zapcore"
)

func main() {
    // 初始化日志
    log.Init(log.Options{
        Filename:     "app",           // 日志文件名称
        MaxCount:     7,               // 最大保存7个日志文件
        CallerEnable: true,            // 开启调用栈信息
        LogLevel:     zapcore.InfoLevel, // 日志级别
        CloseConsole: false,           // 不关闭控制台输出
        Fields: []zap.Field{           // 全局固定字段
            zap.String("service", "example"),
        },
    })

    // 创建上下文
    ctx := context.Background()
    ctx = context.WithValue(ctx, "request_id", "123456")

    // 记录日志
    log.Infof(ctx, "Hello, world!")
    log.Errorf(ctx, "Error occurred: %v", "something went wrong")

    // 刷新日志缓冲区
    log.Sync()
}

// 记录Info级别日志
log.Info(ctx, "This is an info message")

// 记录Error级别日志
log.Error(ctx, "This is an error message", zap.String("detail", "error details"))

// 使用格式化字符串
log.Infof(ctx, "Hello %s!", "world")

// 使用键值对形式
log.Infow(ctx, "User login", "user_id", 123, "ip", "192.168.1.1")

// 在上下文中设置值
ctx := context.Background()
ctx = context.WithValue(ctx, "request_id", "123456")
ctx = context.WithValue(ctx, "user_id", 789)

// 初始化时配置要从上下文中提取的字段
    log.Init(log.Options{
        // ... 其他配置
        CtxFields: []string{"request_id", "user_id"}, // 从上下文中提取这些字段
        Fields: []zap.Field{
            zap.String("service", "api"),
        },
    })

// 记录日志，会自动包含上下文中的字段
log.Info(ctx, "API request received")
// 输出会包含 request_id=123456 和 user_id=789

import (
    "github.com/lifezq/log"
    "gorm.io/gorm"
)

func setupGORM() *gorm.DB {
    // 初始化GORM日志记录器
    gormLogger := log.GormLogger{
        LogLevel:                  gl.Info,           // 日志级别
        IgnoreRecordNotFoundError: true,              // 忽略记录未找到错误
        SlowThreshold:             200 * time.Millisecond, // 慢SQL阈值
    }

    // 连接数据库时使用自定义日志记录器
    db, err := gorm.Open(mysql.Open("dsn"), &gorm.Config{
        Logger: gormLogger.LogMode(gl.Info),
    })
    
    return db
}

import (
    "github.com/lifezq/log"
    "github.com/elastic/go-elasticsearch/v7"
)

func setupES() (*elasticsearch.Client, error) {
    // 初始化Elasticsearch客户端
    cfg := elasticsearch.Config{
        Addresses: []string{"http://localhost:9200"},
        Logger: log.EsLogger{
            RequestEnabled:  true,  // 开启请求日志
            ResponseEnabled: false, // 关闭响应日志
        },
    }

    client, err := elasticsearch.NewClient(cfg)
    return client, err
}

import (
    "context"
    "fmt"
    "github.com/lifezq/log"
)

func main() {
    ctx := context.Background()
    
    // 获取一个io.Writer，写入的内容会被记录为Info级别日志
    writer := log.SafeWriter(ctx)
    
    // 使用fmt.Fprintf写入内容
    fmt.Fprintf(writer, "This will be logged as info level")
    fmt.Fprintf(writer, "\nAnother line to log")
}

log.Init(log.Options{
    Filename:     "/var/log/app/app",  // 日志文件路径
    MaxCount:     30,                  // 保存30天日志
    CallerEnable: false,               // 生产环境关闭调用栈信息
    LogLevel:     zapcore.WarnLevel,   // 只记录Warn及以上级别
    CloseConsole: true,                // 关闭控制台输出
    Fields: []zap.Field{
        zap.String("env", "production"),
    },
})

log.Init(log.Options{
    Filename:     "app",              // 日志文件名称
    MaxCount:     7,                   // 保存7天日志
    CallerEnable: true,                // 开启调用栈信息
    LogLevel:     zapcore.DebugLevel,  // 记录所有级别日志
    CloseConsole: false,               // 开启控制台输出
    Fields: []zap.Field{
        zap.String("env", "development"),
    },
})