package logger

import (
	"context"
	"fmt"
	"github.com/gin-gonic/gin"
	"github.com/google/uuid"
)

const (
	TraceKey = "traceId"
	ReqIn    = "req_in"
	ReqOut   = "req_out"
)

type Trace struct {
	TraceId string
}

func GetTraceFromHeader(c *gin.Context) *Trace {
	if traceId := c.Request.Header.Get(TraceKey); traceId != "" {
		return &Trace{TraceId: traceId}
	}
	trace := new(Trace)
	trace.TraceId = fmt.Sprint(uuid.New())
	return trace
}

func GetTraceFromContext(c context.Context) *Trace {
	if value := c.Value(TraceKey); value != nil {
		return value.(*Trace)
	}
	Log.Warnln("从请求context中获取trace为空!")
	trace := new(Trace)
	trace.TraceId = fmt.Sprint(uuid.New())
	return trace
}