From 96440e242b3c8a63e9bc03158bd238a79b50fb26 Mon Sep 17 00:00:00 2001
From: PhatPhuckDave <david.majdandzic@hotmail.com>
Date: Tue, 6 Jan 2026 13:57:09 +0100
Subject: [PATCH] Do some more cleanup

---
 db.go   | 9 +++++++--
 main.go | 5 -----
 2 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/db.go b/db.go
index 798a521..035a1c5 100644
--- a/db.go
+++ b/db.go
@@ -19,6 +19,7 @@ import (
 
 type DB interface {
 	Init() error
+	Get() *gorm.DB
 
 	SaveKillmails(killmails []Killmail) error
 	QueryFits(params QueryParams) (*FitStatistics, error)
@@ -63,8 +64,8 @@ func GetDB() (DB, error) {
 		ch: conn,
 		db: sdb,
 	}
-
-	return db, nil
+	err = db.Init()
+	return db, err
 }
 
 func GetDBSqlite() (*gorm.DB, error) {
@@ -91,6 +92,10 @@ func GetDBClickhouse() (driver.Conn, error) {
 	return clickhouse.Open(options)
 }
 
+func (db *DBWrapper) Get() *gorm.DB {
+	return db.db
+}
+
 func (db *DBWrapper) Init() error {
 	ctx := context.Background()
 
diff --git a/main.go b/main.go
index b769754..44be407 100644
--- a/main.go
+++ b/main.go
@@ -25,11 +25,6 @@ func main() {
 		return
 	}
 
-	if err := db.InitTables(); err != nil {
-		logger.Error("Failed to initialize tables: %v", err)
-		return
-	}
-
 	if *ingest {
 		DoIngest()
 		return