1 files changed, 47 insertions, 0 deletions
diff --git a/pkg/database/stop.go b/pkg/database/stop.go
new file mode 100644
index 0000000..519b9aa
--- /dev/null
+++ b/pkg/database/stop.go
@@ -0,0 +1,47 @@
+package database
+
+import (
+	"git.adyxax.org/adyxax/trains/pkg/model"
+)
+
+func (env *DBEnv) CountStops() (i int, err error) {
+	query := `SELECT count(*) from stops;`
+	err = env.db.QueryRow(query).Scan(&i)
+	if err != nil {
+		return 0, newQueryError("Could not run database query: most likely the schema is corrupted", err)
+	}
+	return
+}
+
+func (env *DBEnv) ReplaceAndImportStops(trainStops []model.Stop) error {
+	pre_query := `DELETE FROM stops;`
+	query := `
+		INSERT INTO stops
+			(id, name)
+		VALUES
+			($1, $2);`
+	tx, err := env.db.Begin()
+	if err != nil {
+		return newTransactionError("Could not Begin()", err)
+	}
+	_, err = tx.Exec(pre_query)
+	if err != nil {
+		tx.Rollback()
+		return newQueryError("Could not run database query: most likely the schema is corrupted", err)
+	}
+	for i := 0; i < len(trainStops); i++ {
+		_, err = tx.Exec(
+			query,
+			trainStops[i].Id,
+			trainStops[i].Name,
+		)
+		if err != nil {
+			tx.Rollback()
+			return newQueryError("Could not run database query", err)
+		}
+	}
+	if err := tx.Commit(); err != nil {
+		return newTransactionError("Could not commit transaction", err)
+	}
+	return nil
+}