diff --git a/internal/storage/writer.go b/internal/storage/writer.go index f135feb..d822eb3 100644 --- a/internal/storage/writer.go +++ b/internal/storage/writer.go @@ -94,3 +94,13 @@ func WriteReq(db *RequestWriter, resultCh <-chan *LogEntry) { } } } + +func (w *RequestWriter) GetRequestCount() (int, error) { + var count int + err := w.db.QueryRow("SELECT COUNT(*) FROM requests").Scan(&count) + return count, err +} + +func (w *RequestWriter) Close() error { + return w.db.Close() +}