From 19aff26259edab2be636c2d0ea52a9e3b4f82324 Mon Sep 17 00:00:00 2001
From: Egor Matveev <e.matveev@hh.ru>
Date: Wed, 1 Jan 2025 13:53:05 +0300
Subject: [PATCH] fix

---
 main.go | 17 ++++++++++++++---
 1 file changed, 14 insertions(+), 3 deletions(-)

diff --git a/main.go b/main.go
index 8d7344a..8fabc37 100644
--- a/main.go
+++ b/main.go
@@ -1,15 +1,26 @@
 package main
 
 import (
+	"log"
 	"net/http"
 	"queues-go/app/routers"
 	client "queues-go/app/storage/mongo"
+	"time"
 )
 
+func handlerWrapper(f func(http.ResponseWriter, *http.Request)) func(http.ResponseWriter, *http.Request) {
+	return func(w http.ResponseWriter, r *http.Request) {
+		start := time.Now()
+		f(w, r)
+		log.Printf("%s %s", r.URL, time.Since(start))
+	}
+}
+
 func main() {
 	client.Connect()
-	http.HandleFunc("/api/v1/take", routers.Take)
-	http.HandleFunc("/api/v1/finish", routers.Finish)
-	http.HandleFunc("/api/v1/put", routers.Put)
+	http.HandleFunc("/api/v1/take", handlerWrapper(routers.Take))
+	http.HandleFunc("/api/v1/finish", handlerWrapper(routers.Finish))
+	http.HandleFunc("/api/v1/put", handlerWrapper(routers.Put))
+	log.Printf("Server started")
 	http.ListenAndServe(":1239", nil)
 }