Struct DBBatcher

enterprise/coderd/connectionlog/connectionlog.go:120–150

DBBatcher batches connection log upserts and periodically flushes them to the database to reduce per-event write pressure.

// DBBatcher batches connection log upserts and periodically flushes
// them to the database to reduce per-event write pressure.
type DBBatcher struct {
	store database.Store
	log   slog.Logger

	itemCh chan database.UpsertConnectionLogParams

	// dedupedBatch holds entries keyed by connection ID so that
	// PostgreSQL never sees the same row twice in one INSERT …
	// ON CONFLICT DO UPDATE. Connection IDs are globally unique
	// (each new session gets a fresh UUID). Entries with a NULL
	// connection_id (web events) go into nullConnIDBatch instead
	// because NULL != NULL in SQL unique constraints.
	dedupedBatch    map[uuid.UUID]batchEntry
	nullConnIDBatch []batchEntry
	maxBatchSize    int

	// retryCh is a bounded channel of failed batches awaiting
	// retry. A single retry worker goroutine processes this
	// channel, retrying each batch up to maxRetries times before
	// dropping it. If the channel is full, new failures are
	// dropped immediately.
	retryCh chan database.BatchUpsertConnectionLogsParams

	clock    quartz.Clock
	timer    *quartz.Timer
	interval time.Duration

	ctx    context.Context
	cancel context.CancelFunc
	wg     sync.WaitGroup
}

// NewDBBatcher creates a DBBatcher that batches writes to the database
// and starts its background processing loop. Close must be called to