package im

import "context"

func NewWithTraceId(traceId string) context.Context {
	return NewContext(nil, traceId)
}

func NewContext(ctx context.Context, traceId string) context.Context {
	if ctx == nil {
		ctx = context.Background()
	}
	return context.WithValue(ctx, "traceId", traceId)
}