From 6568b6d72355974eacec1d43e638c9bd6f53722a Mon Sep 17 00:00:00 2001
From: Jan Eitzinger <jan@moebiusband.org>
Date: Tue, 3 Sep 2024 13:40:11 +0200
Subject: [PATCH] Prepare transaction API for general usage

---
 internal/importer/initDB.go        |  2 +-
 internal/repository/transaction.go | 13 ++++++-------
 2 files changed, 7 insertions(+), 8 deletions(-)

diff --git a/internal/importer/initDB.go b/internal/importer/initDB.go
index fe49e94..7e9fed5 100644
--- a/internal/importer/initDB.go
+++ b/internal/importer/initDB.go
@@ -27,7 +27,7 @@ func InitDB() error {
 	starttime := time.Now()
 	log.Print("Building job table...")
 
-	t, err := r.TransactionInit()
+	t, err := r.TransactionInit(repository.NamedJobInsert)
 	if err != nil {
 		log.Warn("Error while initializing SQL transactions")
 		return err
diff --git a/internal/repository/transaction.go b/internal/repository/transaction.go
index 9398354..992b423 100644
--- a/internal/repository/transaction.go
+++ b/internal/repository/transaction.go
@@ -15,20 +15,19 @@ type Transaction struct {
 	stmt *sqlx.NamedStmt
 }
 
-func (r *JobRepository) TransactionInit() (*Transaction, error) {
+func (r *JobRepository) TransactionInit(sqlStmt string) (*Transaction, error) {
 	var err error
 	t := new(Transaction)
-	// Inserts are bundled into transactions because in sqlite,
-	// that speeds up inserts A LOT.
+
 	t.tx, err = r.DB.Beginx()
 	if err != nil {
 		log.Warn("Error while bundling transactions")
 		return nil, err
 	}
 
-	t.stmt, err = t.tx.PrepareNamed(NamedJobInsert)
+	t.stmt, err = t.tx.PrepareNamed(sqlStmt)
 	if err != nil {
-		log.Warn("Error while preparing namedJobInsert")
+		log.Warn("Error while preparing SQL statement in transaction")
 		return nil, err
 	}
 
@@ -63,8 +62,8 @@ func (r *JobRepository) TransactionEnd(t *Transaction) error {
 	return nil
 }
 
-func (r *JobRepository) TransactionAdd(t *Transaction, job schema.Job) (int64, error) {
-	res, err := t.stmt.Exec(job)
+func (r *JobRepository) TransactionAdd(t *Transaction, obj interface{}) (int64, error) {
+	res, err := t.stmt.Exec(obj)
 	if err != nil {
 		log.Errorf("repository initDB(): %v", err)
 		return 0, err