status-go/services/ext/context.go

package ext

import (
	"context"
	"time"

	"github.com/status-im/status-go/db"
)

// ContextKey is a type used for keys in ext Context.
type ContextKey struct {
	Name string
}

// NewContextKey returns new ContextKey instance.
func NewContextKey(name string) ContextKey {
	return ContextKey{Name: name}
}

var (
	historyDBKey       = NewContextKey("history_db")
	requestRegistryKey = NewContextKey("request_registry")
	timeKey            = NewContextKey("time")
)

// NewContext creates Context with all required fields.
func NewContext(ctx context.Context, source TimeSource, registry *RequestsRegistry, storage db.Storage) Context {
	ctx = context.WithValue(ctx, historyDBKey, db.NewHistoryStore(storage))
	ctx = context.WithValue(ctx, timeKey, source)
	ctx = context.WithValue(ctx, requestRegistryKey, registry)
	return Context{ctx}
}

// TimeSource is a type used for current time.
type TimeSource func() time.Time

// Context provides access to request-scoped values.
type Context struct {
	context.Context
}

// HistoryStore returns db.HistoryStore instance associated with this request.
func (c Context) HistoryStore() db.HistoryStore {
	return c.Value(historyDBKey).(db.HistoryStore)
}

// Time returns current time using time function associated with this request.
func (c Context) Time() time.Time {
	return c.Value(timeKey).(TimeSource)()
}

// RequestRegistry returns RequestRegistry that tracks each request life-span.
func (c Context) RequestRegistry() *RequestsRegistry {
	return c.Value(requestRegistryKey).(*RequestsRegistry)
}