转到：将日志记录添加到每个路由器

Question

Go : add logging to each router

I want to log all my network request in Go web app.

Something like negroni:

// https://github.com/codegangsta/negroni/blob/master/logger.go
// NewLogger returns a new Logger instance
func NewLogger() *Logger {
  return &Logger{log.New(os.Stdout, "[negroni] ", 0)}
}

func (l *Logger) ServeHTTP(rw http.ResponseWriter, r *http.Request, next http.HandlerFunc) {
  start := time.Now()
  l.Printf("Started %s %s", r.Method, r.URL.Path)

  next(rw, r)

  res := rw.(ResponseWriter)
  l.Printf("Completed %v %s in %v", res.Status(), http.StatusText(res.Status()), time.Since(start))
}

So here's my code:

router := httprouter.New()

handler := func(w http.ResponseWriter, r *http.Request, _ httprouter.Params) {
  type Page struct {
    Title string
  }
  tp := template.Must(template.ParseFiles("templates/main.html", "templates/base.html"))
  err := tp.ExecuteTemplate(w, "base", &Page{Title: "AAA"})
  if err != nil {
    log.Fatal(err)
}

router.Handle("GET", "/", handler)

l := log.New(os.Stdout, "[AAA] ", 0)
l.Printf("Listening 0.0.0.0%s", PORT)

l.Fatal(http.ListenAndServe(PORT, router))

If I want to do this, I have to add start := time.Now() and time.Since(start) manually to each router in my code.package main

I think I should wrap it and use interface but don't know how to get started.

How do I implement one simple logging interface and apply all the routed handlers so that I can debug with all the loggings...

Negroni does like:

router := mux.NewRouter()
router.HandleFunc("/", HomeHandler)

n := negroni.New(Middleware1, Middleware2)
// Or use a middleware with the Use() function
n.Use(Middleware3)
// router goes last
n.UseHandler(router)

n.Run(":3000")

Answer 1

Wrap the root handler with a handler that logs and delegates to another handler:

 type RequestLogger struct {
     h http.Handler
     l *Logger
 }

 func (rl RequestLogger) ServeHTTP(w http.ResponseWriter, r *http.Request) {
     start := time.Now()
     rl.l.Printf("Started %s %s", r.Method, r.URL.Path)
     rl.h.ServeHTTP(w, r)
     rl.l.Printf("Completed %s %s in %v", r.Method, r.URL.Path, time.Since(start))
 }

 ---

 l := log.New(os.Stdout, "[AAA] ", 0)
 l.Printf("Listening 0.0.0.0%s", PORT)
 l.Fatal(http.ListenAndServe(PORT, RequestLogger{h:router, l:l}))