use tracing::{info, warn, error, instrument};
use tracing_subscriber::{registry, prelude::*, fmt, EnvFilter};
use std::path::Path;

fn main() {
    // ============================================================
    // 1. 配置非阻塞文件写入器 (Non-blocking File Appender)
    // ============================================================
    // 每天轮转一个新文件，文件名如: app.log.2024-01-01
    let file_appender = tracing_appender::rolling::daily("./logs", "app.log");

    // 关键：使用 non_blocking 包装，避免文件 I/O 阻塞主线程（特别是 Tokio 运行时）
    // _guard 必须持有到程序结束，否则日志可能丢失
    let (non_blocking, _guard) = tracing_appender::non_blocking(file_appender);

    // ============================================================
    // 2. 定义 JSON 文件层 (给机器看)
    // ============================================================
    let json_file_layer = fmt::layer()
        .json()                       // 开启 JSON 格式
        .with_writer(non_blocking)    // 写入到文件
        .with_file(true)              // 记录文件名
        .with_line_number(true)       // 记录行号
        .with_thread_ids(true)        // 记录线程 ID
        .with_target(false)           // JSON 中通常不需要 target (模块路径)，因为有 file
        .flatten_event(true)          // 建议：将字段平铺在根对象中，而不是嵌套在 "fields" 对象里
        .with_filter(EnvFilter::new("info")); // 文件日志级别：通常是 INFO 或 WARN

    // ============================================================
    // 3. 定义控制台层 (给人看)
    // ============================================================
    let console_layer = fmt::layer()
        .pretty()                     // 漂亮的格式 (多行，带颜色)
        .with_writer(std::io::stdout) // 写入标准输出
        .with_target(false)           // 保持清爽，隐藏详细模块路径
        .with_filter(EnvFilter::from_default_env()); // 读取 RUST_LOG 环境变量

    // ============================================================
    // 4. 注册所有层
    // ============================================================
    registry()
        .with(json_file_layer)
        .with(console_layer)
        .init();

    // ============================================================
    // 5. 测试日志
    // ============================================================
    perform_task("user_123", 500);
}

// 使用 instrument 自动记录函数进入/退出和参数
#[instrument(fields(request_id = "req-abc-123"))]
fn perform_task(user: &str, amount: u32) {
    info!(action = "payment", status = "processing", "开始处理支付任务");

    // 模拟一些工作
    std::thread::sleep(std::time::Duration::from_millis(100));

    if amount > 1000 {
        warn!(amount, "金额较大，需要人工审核");
    }

    error!(error_code = 5001, "连接数据库失败");
}