Fixed ip key

Author: vladyslav2

context/keys/keys.go

@@ -18,6 +18,7 @@ const (
 	RequestLogID
 
 	RequestIDStr = "X-Request-Id"
+	IPAddressStr = "IP-Address"
 )
 
 func GetAsString(ctx context.Context, key ContextKey) string {

server/middleware/ctx.go

@@ -10,9 +10,9 @@ import (
 func SetIPAddress(next echo.HandlerFunc) echo.HandlerFunc {
 	return func(c echo.Context) error {
 		ip := keys.GetIPAddress(c.Request().Header)
-		c.Set(keys.IPAddress, ip)
+		c.Set(keys.IPAddressStr, ip)
 
-		ctx := context.WithValue(c.Request().Context(), keys.IPAddress, ip)
+		ctx := context.WithValue(c.Request().Context(), keys.IPAddressStr, ip)
 		c.SetRequest(c.Request().WithContext(ctx))
 		return next(c)
 	}

server/middleware/ip_address.go

@@ -12,10 +12,10 @@ func SetLogger(next echo.HandlerFunc) echo.HandlerFunc {
 	return func(c echo.Context) error {
 		// get request's context
 		ctx := c.Request().Context()
-		ipAddress, _ := c.Get(keys.IPAddress).(string)
-		identityID, _ := keys.GetCtxValue(ctx, keys.IdentityID).(string)
-		requestID, _ := keys.GetCtxValue(ctx, keys.RequestID).(string)
-		msgID, _ := keys.GetCtxValue(ctx, keys.MSGID).(string)
+		ipAddress, _ := keys.GetAsString(ctx, keys.IPAddressStr)
+		identityID, _ := keys.GetAsString(ctx, keys.IdentityID)
+		requestID, _ := keys.GetAsString(ctx, keys.RequestID)
+		msgID, _ := keys.GetAsString(ctx, keys.MSGID)
 
 		logInfo := logger.ServiceContext{
 			Service: verser.GetService(),