package util

import (
	"context"
	"net/http"
)

const (
	RequestIdHttpHeader = "X-Request-ID"
	RequestIDKey        = "x-request-id"
)

func GetRequestID(ctx context.Context) string {
	if ctx == nil {
		return ""
	}
	id, _ := ctx.Value(RequestIDKey).(string)
	return id
}

func WithRequestID(ctx context.Context, id string) context.Context {
	return context.WithValue(ctx, RequestIDKey, id)
}

func ReqWithRequestId(req *http.Request, ctx context.Context) {
	req.Header.Set(RequestIdHttpHeader, GetRequestID(ctx))
}