golang第49天访问日志

Golang第49天：访问日志

在Golang开发中，有时我们需要记录用户对我们应用的访问情况，以便分析用户行为、优化性能和保护安全。访问日志是一种重要的工具，它可以记录每个请求的详细信息，包括请求方法、URL、响应状态码等。在本文中，我们将探讨如何在Golang中实现访问日志功能。

1. 初始化日志记录器

第一步是初始化一个日志记录器，用于记录访问日志。我们可以使用`log`包提供的`New()`函数来创建一个新的日志记录器。通过设置日志输出的目标（例如文件），我们可以灵活地记录日志信息。

```go import ( "log" "os" ) func main() { file, err := os.OpenFile("access.log", os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0666) if err != nil { log.Fatal(err) } defer file.Close() logger := log.New(file, "", log.LstdFlags) // 使用logger记录访问日志 } ```

2. 记录访问日志

一旦我们初始化了日志记录器，就可以开始记录访问日志了。在Golang中，可以通过HTTP中间件的方式来捕获每个请求，并将请求的详细信息记录下来。

```go func AccessLogger(handler http.Handler) http.Handler { return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { logger.Printf("Request from %s: %s %s", r.RemoteAddr, r.Method, r.RequestURI) handler.ServeHTTP(w, r) }) } ```

上述代码中的`AccessLogger`函数是一个HTTP中间件，它将记录每个请求的详细信息，并调用`handler.ServeHTTP()`来处理请求。我们可以将这个中间件应用到我们的HTTP处理器中：

```go func main() { file, err := os.OpenFile("access.log", os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0666) if err != nil { log.Fatal(err) } defer file.Close() logger := log.New(file, "", log.LstdFlags) // 创建HTTP处理器 handler := http.NewServeMux() handler.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) { fmt.Fprintf(w, "Hello, World!") }) // 应用AccessLogger中间件 http.ListenAndServe(":8080", AccessLogger(handler)) } ```

3. 记录更多信息

除了记录请求方法和URL，我们还可以记录更多有用的信息，例如响应状态码、请求耗时等。Golang的`net/http`包中提供了一些工具函数和结构体，可以方便地获取这些信息。

```go func AccessLogger(handler http.Handler) http.Handler { return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { start := time.Now() method := r.Method url := r.URL.String() logger.Printf("Request from %s: %s %s", r.RemoteAddr, method, url) // 调用handler处理请求 handler.ServeHTTP(w, r) elapsed := time.Since(start) status := w.(interface { Status() int }).Status() logger.Printf("Response to %s: %d (%s)", r.RemoteAddr, status, elapsed) }) } ```

上述代码中，我们在中间件中记录了每个请求的耗时和响应状态码。通过调用`time.Now()`来获取请求开始的时间，以及调用`w.(interface{ Status() int }).Status()`来获取响应状态码。

4. 日志切割

随着时间的推移，访问日志文件可能会变得非常庞大，为了方便管理和保持日志文件的可读性，我们可以定期对日志文件进行切割。

```go func main() { file, err := os.OpenFile("access.log", os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0666) if err != nil { log.Fatal(err) } defer file.Close() logger := log.New(file, "", log.LstdFlags) // 创建HTTP处理器 handler := http.NewServeMux() handler.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) { fmt.Fprintf(w, "Hello, World!") }) // 应用AccessLogger中间件 http.ListenAndServe(":8080", AccessLoggerWithRotation(logger, handler)) } func AccessLoggerWithRotation(logger *log.Logger, handler http.Handler) http.Handler { go func() { for range time.Tick(24 * time.Hour) { file.Close() // 创建新的日志文件 file, err := os.OpenFile("access.log", os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0666) if err != nil { log.Fatal(err) } // 更新logger的输出目标 logger.SetOutput(file) } }() return AccessLogger(logger, handler) } ```

上述代码中，我们定义了一个`AccessLoggerWithRotation`函数，它会在每天固定的时间间隔内关闭当前的日志文件，并创建一个新的日志文件。新的日志文件通过更新`logger`的输出目标来实现。

结论

访问日志是开发和运维中必不可少的一部分，它能帮助我们了解用户的行为和优化应用的性能。在本文中，我们学习了如何在Golang中记录访问日志，并添加了额外的功能，如记录更多信息和定期切割日志文件。通过合理地使用访问日志，我们可以提高应用的可观察性和可维护性。

关键字：Golang，访问日志，HTTP中间件