diff options
Diffstat (limited to 'modules/caddyhttp')
-rw-r--r-- | modules/caddyhttp/caddylog/log.go | 22 |
1 files changed, 20 insertions, 2 deletions
diff --git a/modules/caddyhttp/caddylog/log.go b/modules/caddyhttp/caddylog/log.go index 5bbab2a..f7bc9fd 100644 --- a/modules/caddyhttp/caddylog/log.go +++ b/modules/caddyhttp/caddylog/log.go @@ -12,13 +12,31 @@ import ( func init() { caddy2.RegisterModule(caddy2.Module{ Name: "http.middleware.log", - New: func() (interface{}, error) { return &Log{}, nil }, + New: func() (interface{}, error) { return new(Log), nil }, + OnLoad: func(instances []interface{}, priorState interface{}) (interface{}, error) { + var counter int + if priorState != nil { + counter = priorState.(int) + } + counter++ + for _, inst := range instances { + logInst := inst.(*Log) + logInst.counter = counter + } + log.Println("State is now:", counter) + return counter, nil + }, + OnUnload: func(state interface{}) error { + log.Println("Closing log files, state:", state) + return nil + }, }) } // Log implements a simple logging middleware. type Log struct { Filename string + counter int } func (l *Log) ServeHTTP(w http.ResponseWriter, r *http.Request, next caddyhttp.Handler) error { @@ -28,7 +46,7 @@ func (l *Log) ServeHTTP(w http.ResponseWriter, r *http.Request, next caddyhttp.H return err } - log.Println("latency:", time.Now().Sub(start)) + log.Println("latency:", time.Now().Sub(start), l.counter) return nil } |