在Go语言HTTP服务器中实现请求日志文件输出

func LogMiddleware(handler http.Handler) http.Handler {
  return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
    // 错误示例：这只会输出到终端
    fmt.Printf("%s %s %s\n", r.RemoteAddr, r.Method, r.URL)
    handler.ServeHTTP(w, r)
  })
}

// config.json
{
  "Path": "./public",
  "Port": "8080",
  "LogFile": "request.log"
}

package main

import (
    "encoding/json"
    "fmt"
    "io/ioutil"
    "log"
    "net/http"
    "os" // 导入os包用于文件操作
)

// Options 结构体用于加载配置
type Options struct {
    Path    string `json:"Path"`
    Port    string `json:"Port"`
    LogFile string `json:"LogFile"` // 新增日志文件路径配置
}

// requestLogFile 是一个全局变量，用于存储日志文件的句柄
var requestLogFile *os.File

// LogMiddleware 是一个HTTP中间件，用于记录请求信息
func LogMiddleware(handler http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        // 使用 fmt.Fprintf 将日志写入到 requestLogFile
        // 注意：这里需要确保 requestLogFile 已经被正确初始化
        if requestLogFile != nil {
            fmt.Fprintf(requestLogFile, "%s %s %s\n", r.RemoteAddr, r.Method, r.URL)
        } else {
            // 如果日志文件未初始化，则退回到标准输出，并打印警告
            log.Printf("警告: 日志文件未初始化，请求信息将输出到终端: %s %s %s\n", r.RemoteAddr, r.Method, r.URL)
        }
        handler.ServeHTTP(w, r)
    })
}

func main() {
    // 1. 初始化配置，设置默认值
    op := &Options{Path: "./", Port: "8001", LogFile: "request.log"}

    // 尝试从 config.json 文件加载配置
    data, err := ioutil.ReadFile("./config.json")
    if err == nil {
        // 如果读取成功，则解析JSON数据
        if unmarshalErr := json.Unmarshal(data, op); unmarshalErr != nil {
            log.Printf("警告: 解析 config.json 失败，将使用默认或部分配置。错误: %v", unmarshalErr)
        }
    } else {
        log.Printf("警告: config.json 文件未找到或无法读取，将使用默认配置。错误: %v", err)
    }

    // 2. 打开或创建日志文件
    var openFileErr error
    requestLogFile, openFileErr = os.Create(op.LogFile)
    if openFileErr != nil {
        // 如果无法创建日志文件，则这是一个严重错误，程序应终止
        log.Fatalf("无法创建日志文件 %s: %v", op.LogFile, openFileErr)
    }
    // 使用 defer 确保程序退出时日志文件句柄被关闭
    defer requestLogFile.Close()

    // 3. 设置HTTP文件服务器
    // 创建一个文件服务器，服务于 op.Path 指定的目录
    http.Handle("/", http.FileServer(http.Dir(op.Path)))

    // 4. 启动HTTP服务器，并应用日志中间件
    log.Printf("服务器将在端口 %s 上运行，提供文件服务路径 %s，请求日志将写入 %s", op.Port, op.Path, op.LogFile)
    // 使用 LogMiddleware 包装 http.DefaultServeMux
    serverErr := http.ListenAndServe(":"+op.Port, LogMiddleware(http.DefaultServeMux))
    if serverErr != nil {
        // 如果服务器启动失败，则是一个严重错误
        log.Fatalf("ListenAndServe 失败: %v", serverErr)
    }
}