logqueue

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

type DatabaseConnector interface {
	Open(context.Context) (*gorm.DB, error)
}

type DatabaseConnectorFunc func(context.Context) (*gorm.DB, error)

type DeliveryMetrics interface {
	Observe(deliveryqueue.Stats, bool)
	BacklogCleared(uint64, uint64)
	WriteFailure()
	SnapshotFailure()
}

type DeliverySettingDefinition struct {
	Default string
	Min     int64
	Max     int64
}

type DeliverySettings struct {
	QueueMaxEntries   int
	QueueMaxBytes     int64
	DeliveryBatchSize int
	BackoffMaxSeconds int
}

type DeliveryStatus struct {
	Queue       deliveryqueue.Stats
	SchemaReady bool
	LastError   string
}

type HistoricalAggregateBatch struct {
	Hourly   []models.UsageHourlyBucket
	Duration []models.UsageDurationHistogram
	TTFT     []models.UsageTTFTHistogram
	TPS      []models.UsageTPSHistogram
	UserTTFT []models.UsageUserTTFTHistogram
	UserTPS  []models.UsageUserTPSHistogram
}

type HistoricalRequestBatchWriter struct {
	DBFinder func() *gorm.DB
}

type LogBatch struct {
	Request      models.RequestLog               `json:"request"`
	Traces       []models.RequestTrace           `json:"traces,omitempty"`
	TokenDaily   []models.TokenDailyBilling      `json:"token_daily,omitempty"`
	ChannelDaily []models.ChannelDailyBilling    `json:"channel_daily,omitempty"`
	Hourly       []models.UsageHourlyBucket      `json:"hourly,omitempty"`
	Duration     []models.UsageDurationHistogram `json:"duration,omitempty"`
	TTFT         []models.UsageTTFTHistogram     `json:"ttft,omitempty"`
	TPS          []models.UsageTPSHistogram      `json:"tps,omitempty"`
	UserTTFT     []models.UsageUserTTFTHistogram `json:"user_ttft,omitempty"`
	UserTPS      []models.UsageUserTPSHistogram  `json:"user_tps,omitempty"`
	APIRequest   *models.APIRequestLog           `json:"api_request,omitempty"`
	APITraces    []models.APIRequestTrace        `json:"api_traces,omitempty"`
}

type LogBatchWriter struct {
	DBFinder func() *gorm.DB
}

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

type SQLiteLogConnector struct {
	Path      string
	Connector *masterdatabase.Connector
}

type SettingsFinder interface {
	Find(context.Context) DeliverySettings
}

type WorkerOptions struct {
	Queue        *deliveryqueue.Queue[LogBatch]
	Writer       *LogBatchWriter
	Settings     SettingsFinder
	Connector    DatabaseConnector
	Handoff      func(*gorm.DB) *gorm.DB
	Metrics      DeliveryMetrics
	SnapshotPath string

	PollInterval     time.Duration
	SettingsInterval time.Duration
	SnapshotInterval time.Duration
	FlushTimeout     time.Duration
	RetryBase        time.Duration
	RetryMax         time.Duration
	OnError          func(error)
}