From 8257e6c4aa5abc2098b2169d90dc007fe9d762eb Mon Sep 17 00:00:00 2001
From: Pavel <pashamullin202@gmail.com>
Date: Fri, 7 Jun 2024 17:25:02 +0300
Subject: [PATCH] add hlog midleware

---
 go.mod            |  3 +++
 go.sum            |  8 ++++++++
 log_mw/hlog_mw.go | 26 ++++++++++++++++++++++++++
 3 files changed, 37 insertions(+)
 create mode 100644 log_mw/hlog_mw.go

diff --git a/go.mod b/go.mod
index 812785b..3ce9649 100644
--- a/go.mod
+++ b/go.mod
@@ -8,6 +8,7 @@ require (
 	github.com/dgrijalva/jwt-go v3.2.0+incompatible
 	github.com/gofiber/fiber/v2 v2.51.0
 	github.com/stretchr/testify v1.8.4
+	github.com/themakers/hlog v0.0.0-20191205140925-235e0e4baddf
 	go.mongodb.org/mongo-driver v1.13.1
 )
 
@@ -30,6 +31,8 @@ require (
 	github.com/xdg-go/scram v1.1.2 // indirect
 	github.com/xdg-go/stringprep v1.0.4 // indirect
 	github.com/youmark/pkcs8 v0.0.0-20181117223130-1be2e3e5546d // indirect
+	go.uber.org/multierr v1.10.0 // indirect
+	go.uber.org/zap v1.27.0 // indirect
 	golang.org/x/crypto v0.7.0 // indirect
 	golang.org/x/sync v0.0.0-20220722155255-886fb9371eb4 // indirect
 	golang.org/x/sys v0.14.0 // indirect
diff --git a/go.sum b/go.sum
index 965fa43..87ccabf 100644
--- a/go.sum
+++ b/go.sum
@@ -30,6 +30,8 @@ github.com/rivo/uniseg v0.2.0 h1:S1pD9weZBuJdFmowNwbpi7BJ8TNftyUImj/0WQi72jY=
 github.com/rivo/uniseg v0.2.0/go.mod h1:J6wj4VEh+S6ZtnVlnTBMWIodfgj8LQOQFoIToxlJtxc=
 github.com/stretchr/testify v1.8.4 h1:CcVxjf3Q8PM0mHUKJCdn+eZZtm5yQwehR5yeSVQQcUk=
 github.com/stretchr/testify v1.8.4/go.mod h1:sz/lmYIOXD/1dqDmKjjqLyZ2RngseejIcXlSw2iwfAo=
+github.com/themakers/hlog v0.0.0-20191205140925-235e0e4baddf h1:TJJm6KcBssmbWzplF5lzixXl1RBAi/ViPs1GaSOkhwo=
+github.com/themakers/hlog v0.0.0-20191205140925-235e0e4baddf/go.mod h1:1FsorU3vnXO9xS9SrhUp8fRb/6H/Zfll0rPt1i4GWaA=
 github.com/valyala/bytebufferpool v1.0.0 h1:GqA5TC/0021Y/b9FG4Oi9Mr3q7XYx6KllzawFIhcdPw=
 github.com/valyala/bytebufferpool v1.0.0/go.mod h1:6bBcMArwyJ5K/AmCkWv1jt77kVWyCJ6HpOuEn7z0Csc=
 github.com/valyala/fasthttp v1.50.0 h1:H7fweIlBm0rXLs2q0XbalvJ6r0CUPFWK3/bB4N13e9M=
@@ -47,6 +49,12 @@ github.com/youmark/pkcs8 v0.0.0-20181117223130-1be2e3e5546d/go.mod h1:rHwXgn7Jul
 github.com/yuin/goldmark v1.4.13/go.mod h1:6yULJ656Px+3vBD8DxQVa3kxgyrAnzto9xy5taEt/CY=
 go.mongodb.org/mongo-driver v1.13.1 h1:YIc7HTYsKndGK4RFzJ3covLz1byri52x0IoMB0Pt/vk=
 go.mongodb.org/mongo-driver v1.13.1/go.mod h1:wcDf1JBCXy2mOW0bWHwO/IOYqdca1MPCwDtFu/Z9+eo=
+go.uber.org/goleak v1.3.0 h1:2K3zAYmnTNqV73imy9J1T3WC+gmCePx2hEGkimedGto=
+go.uber.org/goleak v1.3.0/go.mod h1:CoHD4mav9JJNrW/WLlf7HGZPjdw8EucARQHekz1X6bE=
+go.uber.org/multierr v1.10.0 h1:S0h4aNzvfcFsC3dRF1jLoaov7oRaKqRGC/pUEJ2yvPQ=
+go.uber.org/multierr v1.10.0/go.mod h1:20+QtiLqy0Nd6FdQB9TLXag12DsQkrbs3htMFfDN80Y=
+go.uber.org/zap v1.27.0 h1:aJMhYGrd5QSmlpLMr2MftRKl7t8J8PTZPA732ud/XR8=
+go.uber.org/zap v1.27.0/go.mod h1:GB2qFLM7cTU87MWRP2mPIjqfIDnGu+VIO4V/SdhGo2E=
 golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
 golang.org/x/crypto v0.0.0-20210921155107-089bfa567519/go.mod h1:GvvjBRRGRdwPK5ydBHafDWAxML/pGHZbMvKqRZ5+Abc=
 golang.org/x/crypto v0.0.0-20220622213112-05595931fe9d/go.mod h1:IxCIyHEi3zRg3s0A5j5BB6A9Jmi73HwBIUl50j+osU4=
diff --git a/log_mw/hlog_mw.go b/log_mw/hlog_mw.go
new file mode 100644
index 0000000..c639618
--- /dev/null
+++ b/log_mw/hlog_mw.go
@@ -0,0 +1,26 @@
+package log_mw
+
+import (
+	"github.com/gofiber/fiber/v2"
+	"github.com/themakers/hlog"
+	"strings"
+)
+
+const HlogCtxKey string = "logger"
+
+func ContextLogger(logger hlog.Logger) fiber.Handler {
+	return func(c *fiber.Ctx) error {
+		c.Locals(HlogCtxKey, logger.With(map[string]string{
+			"ctxuserip":   c.IP(),
+			"ctxuserport": c.Port(),
+			"keydomain":   strings.Join(c.Subdomains(), "/"),
+			"keypath":     c.Path(),
+		}))
+		return c.Next()
+	}
+}
+
+func ExtractLogger(ctx *fiber.Ctx) hlog.Logger {
+	logger := ctx.Context().UserValue(HlogCtxKey).(hlog.Logger)
+	return logger
+}