budget

type Lease interface {
	Context() context.Context
	Limits() Limits
	Complete(Usage) error
}

type Limits struct {
	MaxConcurrent           int
	MaxExecution            time.Duration
	MaxEstimatedScannedRows int64
	MaxScannedRows          int64 // deprecated alias
	MaxReturnedRows         int64
	MaxReturnedBytes        int64
	MaxSessionCost          int64
}

type Manager interface {
	Acquire(context.Context, Scope) (Lease, error)
}

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

type Reservation struct {
	Cost                 int64
	EstimatedScannedRows int64
}

type ReservingManager interface {
	Manager
	AcquireWithReservation(context.Context, Scope, Reservation) (Lease, error)
}

type Scope struct {
	Role    string
	Tenant  string
	Session string
}

type SessionManager interface {
	Manager
	CloseSession(string)
}

type Usage struct {
	// ScannedRows is an actual observed scan count. Zero means unknown or none;
	// callers must not substitute ReturnedRows for it.
	ScannedRows int64
	// EstimatedScannedRows is a planner estimate, not an observed scan count.
	EstimatedScannedRows int64
	ReturnedRows         int64
	ReturnedBytes        int64
	Duration             time.Duration
	Cost                 int64
}