From 747f4fb70ff61d3f8a7bb1bed896a913b0999450 Mon Sep 17 00:00:00 2001 From: Ben Morrison Date: Sun, 19 May 2019 23:45:48 -0400 Subject: middleware func to attach remote ip to context --- main.go | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) (limited to 'main.go') diff --git a/main.go b/main.go index 4b62f27..4508edf 100644 --- a/main.go +++ b/main.go @@ -66,7 +66,7 @@ func main() { // handlers.CompressHandler gzips all responses. // Write/Read timeouts are self explanatory. server := &http.Server{ - Handler: handlers.CompressHandler(index), + Handler: handlers.CompressHandler(ipMiddleware(index)), Addr: portnum, WriteTimeout: 15 * time.Second, ReadTimeout: 15 * time.Second, @@ -77,6 +77,12 @@ func main() { if err != nil { log.Printf("%v\n", err) } + defer func() { + err := server.Close() + if err != nil { + log.Printf("%v\n", err) + } + }() closelog <- true } -- cgit 1.4.1-2-gfad0