diff --git a/db.go b/db.go index b4081f3..fe5c5c5 100644 --- a/db.go +++ b/db.go @@ -5,6 +5,7 @@ import ( "gorm.io/driver/sqlite" "gorm.io/gorm" + "gorm.io/gorm/clause" ) type DB interface { @@ -39,7 +40,6 @@ func GetDB() (DB, error) { return db, nil } - func (db *DBWrapper) Raw(sql string, args ...any) *gorm.DB { return db.db.Raw(sql, args...) } @@ -49,5 +49,7 @@ func (db *DBWrapper) DB() *gorm.DB { } func (db *DBWrapper) SaveKillmails(killmails []Killmail) error { - return db.db.CreateInBatches(killmails, 10).Error -} \ No newline at end of file + return db.db.Session(&gorm.Session{FullSaveAssociations: true}). + Clauses(clause.OnConflict{DoNothing: true}). + CreateInBatches(killmails, 10).Error +}