From dbd75fe2287d86c36892439b83b58288c34e700d Mon Sep 17 00:00:00 2001
From: Julien Dessaux
Date: Wed, 8 Sep 2021 16:22:16 +0200
Subject: Added database function to get a train stop name from its id

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

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

diff --git a/pkg/database/stop.go b/pkg/database/stop.go
index 519b9aa..47f9ce1 100644
--- a/pkg/database/stop.go
+++ b/pkg/database/stop.go
@@ -13,6 +13,21 @@ func (env *DBEnv) CountStops() (i int, err error) {
 	return
 }
 
+func (env *DBEnv) GetStop(id string) (*model.Stop, error) {
+	query := `SELECT name FROM stops WHERE id = $1;`
+	stop := model.Stop{Id: id}
+	err := env.db.QueryRow(
+		query,
+		id,
+	).Scan(
+		&stop.Name,
+	)
+	if err != nil {
+		return nil, newQueryError("Could not run database query", err)
+	}
+	return &stop, nil
+}
+
 func (env *DBEnv) ReplaceAndImportStops(trainStops []model.Stop) error {
 	pre_query := `DELETE FROM stops;`
 	query := `
-- 
cgit v1.2.3