package middlewares

import (
	"furtuna-be/internal/common/request"
	"furtuna-be/internal/constants"
	"furtuna-be/internal/utils/generator"
	"github.com/gin-gonic/gin"
)

func Trace() gin.HandlerFunc {
	return func(c *gin.Context) {
		traceId := c.Request.Header.Get("Trace-Id")
		if traceId == "" {
			traceId = generator.GenerateUUID()
		}

		request.SetTraceId(c, traceId)
		c.Set(constants.ContextRequestID, traceId)
	}
}