From 0abcdd11eda39792f325bbfa0301e55f30457640 Mon Sep 17 00:00:00 2001
From: Julien Dessaux
Date: Thu, 9 Sep 2021 21:55:10 +0200
Subject: Added database function to get all train stops

---
 pkg/database/stop.go | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

(limited to 'pkg/database/stop.go')

diff --git a/pkg/database/stop.go b/pkg/database/stop.go
index 47f9ce1..0c2547c 100644
--- a/pkg/database/stop.go
+++ b/pkg/database/stop.go
@@ -28,6 +28,23 @@ func (env *DBEnv) GetStop(id string) (*model.Stop, error) {
 	return &stop, nil
 }
 
+func (env *DBEnv) GetStops() (stops []model.Stop, err error) {
+	query := `SELECT id, name FROM stops;`
+	rows, err := env.db.Query(query)
+	if err != nil {
+		return nil, newQueryError("Could not run database query", err)
+	}
+	defer rows.Close()
+	for rows.Next() {
+		var stop model.Stop
+		if err := rows.Scan(&stop.Id, &stop.Name); err != nil {
+			return nil, newQueryError("Could not run database query", err)
+		}
+		stops = append(stops, stop)
+	}
+	return
+}
+
 func (env *DBEnv) ReplaceAndImportStops(trainStops []model.Stop) error {
 	pre_query := `DELETE FROM stops;`
 	query := `
-- 
cgit v1.2.3