Do some more cleanup
This commit is contained in:
9
db.go
9
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()
|
||||
|
||||
|
||||
Reference in New Issue
Block a user