Add listing 7.15

mcalavera81 · May 8, 2023 · c3bd278 · c3bd278
1 parent 05fed73
commit c3bd278
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/ch07/cmd/shortd/shortd.go b/ch07/cmd/shortd/shortd.go
@@ -8,6 +8,7 @@ import (
 	"os"
 	"time"
 
+	"github.com/inancgumus/effective-go/ch07/httpio"
 	"github.com/inancgumus/effective-go/ch07/shortener"
 )
 
@@ -28,6 +29,9 @@ func main() {
 		Handler:     http.TimeoutHandler(shortenerServer, *timeout, "timeout"),
 		ReadTimeout: *timeout,
 	}
+	if os.Getenv("BITE_DEBUG") == "1" {
+		server.Handler = httpio.LoggingMiddleware(server.Handler)
+	}
 	err := server.ListenAndServe()
 	if !errors.Is(err, http.ErrServerClosed) {
 		fmt.Fprintln(os.Stderr, "server closed unexpectedly:", err)