orm

type AfterCreateHook interface {
	AfterCreate(context.Context) error
}

type AfterDeleteHook interface {
	AfterDelete(context.Context) error
}

type AfterFindHook interface {
	AfterFind(context.Context) error
}

type AfterUpdateHook interface {
	AfterUpdate(context.Context) error
}

type Attribute struct {
	Key   string
	Value any
}

type BeforeCreateHook interface {
	BeforeCreate(context.Context) error
}

type BeforeDeleteHook interface {
	BeforeDelete(context.Context) error
}

type BeforeFindHook interface {
	BeforeFind(context.Context) error
}

type BeforeUpdateHook interface {
	BeforeUpdate(context.Context) error
}

type Config struct {
	DB                  *sql.DB
	Tx                  *sql.Tx
	Dialect             dialect.Dialect
	DriverName          string
	Schema              *schema.Schema
	Logger              Logger
	Observability       ObservabilityConfig
	Access              access.Engine
	PrepareStatements   bool
	SoftDeleteByDefault bool
}

type Counter interface {
	Add(context.Context, float64, ...Attribute)
}

type DB struct {
	// contains filtered or unexported fields
}

type Histogram interface {
	Record(context.Context, float64, ...Attribute)
}

type Logger interface {
	Printf(string, ...any)
}

type Meter interface {
	Counter(name string) Counter
	Histogram(name string) Histogram
}

type MeterProvider interface {
	Meter(name string) Meter
}

type Observability = ObservabilityConfig

type ObservabilityConfig struct {
	Tracing            bool
	Metrics            bool
	TraceSQL           TraceSQLMode
	SQLLogging         SQLLogMode
	SlowQueryThreshold time.Duration
	MaskParameters     bool
	MaskedFields       []string
	Logger             ObservabilityLogger
	TracerProvider     TracerProvider
	MeterProvider      MeterProvider
}

type ObservabilityLogger interface {
	LogSQL(context.Context, SQLLogEntry)
	LogError(context.Context, error)
}

type QueryOption func(*queryState)

type SQLLogEntry struct {
	SQL          string
	Args         []any
	Duration     time.Duration
	AffectedRows int64
	Timestamp    time.Time
	Err          error
	Slow         bool
	Visibility   TraceSQLMode
	Driver       string
	Dialect      string
	Operation    string
	Table        string
}

type SQLLogMode string

type Session struct {
	// contains filtered or unexported fields
}

type Span interface {
	End()
	RecordError(error)
	SetAttributes(...Attribute)
}

type SpanOption interface {
	// contains filtered or unexported methods
}

type TraceSQLMode string

type Tracer interface {
	Start(context.Context, string, ...SpanOption) (context.Context, Span)
}

type TracerProvider interface {
	Tracer(name string) Tracer
}