sing-box-extended/log/id.go

package log

import (
	"context"
	"math/rand"
	"time"

	"github.com/sagernet/sing/common/random"
)

func init() {
	random.InitializeSeed()
}

type idKey struct{}
type muxIdKey struct{}
type hwidKey struct{}

type ID struct {
	ID        uint32
	CreatedAt time.Time
}

func ContextWithNewID(ctx context.Context) context.Context {
	return ContextWithID(ctx, ID{
		ID:        rand.Uint32(),
		CreatedAt: time.Now(),
	})
}

func ContextWithID(ctx context.Context, id ID) context.Context {
	return context.WithValue(ctx, (*idKey)(nil), id)
}

func IDFromContext(ctx context.Context) (ID, bool) {
	id, loaded := ctx.Value((*idKey)(nil)).(ID)
	return id, loaded
}

func ContextWithNewMuxID(ctx context.Context) context.Context {
	return ContextWithMuxID(ctx, ID{
		ID:        rand.Uint32(),
		CreatedAt: time.Now(),
	})
}

func ContextWithMuxID(ctx context.Context, id ID) context.Context {
	return context.WithValue(ctx, (*muxIdKey)(nil), id)
}

func MuxIDFromContext(ctx context.Context) (ID, bool) {
	id, loaded := ctx.Value((*muxIdKey)(nil)).(ID)
	return id, loaded
}

func ContextWithHWID(ctx context.Context, id ID) context.Context {
	return context.WithValue(ctx, (*hwidKey)(nil), id)
}

func HWIDFromContext(ctx context.Context) (ID, bool) {
	id, loaded := ctx.Value((*hwidKey)(nil)).(ID)
	return id, loaded
}