init project

internal/common/request/context.go

@@ -0,0 +1,53 @@
+package request
+
+import (
+	"context"
+	"github.com/gin-gonic/gin"
+)
+
+const (
+	ReqInfoKey reqInfoKeyType = "request-info"
+)
+
+func SetTraceId(c *gin.Context, traceId string) {
+	info, exists := c.Get(ReqInfoKey)
+	if exists {
+		parsedInfo := info.(RequestInfo)
+		parsedInfo.TraceId = traceId
+
+		c.Set(ReqInfoKey, parsedInfo)
+
+		return
+	}
+	c.Set(ReqInfoKey, RequestInfo{TraceId: traceId})
+}
+
+func SetUserId(c *gin.Context, userId int64) {
+	info, exists := c.Get(ReqInfoKey)
+	if exists {
+		parsedInfo := info.(RequestInfo)
+		parsedInfo.UserId = userId
+
+		c.Set(ReqInfoKey, parsedInfo)
+
+		return
+	}
+
+	c.Set(ReqInfoKey, RequestInfo{UserId: userId})
+}
+
+func SetUserContext(c *gin.Context, payload map[string]interface{}) {
+	c.Set(ReqInfoKey, RequestInfo{
+		UserId: int64(payload["userId"].(float64)),
+		Role:   payload["role"].(string),
+	})
+}
+
+func ContextWithReqInfo(c *gin.Context) context.Context {
+	info, ok := c.Get(ReqInfoKey)
+	if ok {
+		return WithRequestInfo(c, info.(RequestInfo))
+	}
+
+	return WithRequestInfo(c, RequestInfo{})
+}