diff options
| author | Julien Dessaux | 2021-09-08 16:22:16 +0200 |
|---|---|---|
| committer | Julien Dessaux | 2021-09-08 16:22:16 +0200 |
| commit | dbd75fe2287d86c36892439b83b58288c34e700d (patch) | |
| tree | 9a752a3cc48f199c6188550eaafb2bdbdfba568f /pkg/database/stop.go | |
| parent | Renamed TrainStop to simply Stop (diff) | |
| download | trains-dbd75fe2287d86c36892439b83b58288c34e700d.tar.gz trains-dbd75fe2287d86c36892439b83b58288c34e700d.tar.bz2 trains-dbd75fe2287d86c36892439b83b58288c34e700d.zip | |
Added database function to get a train stop name from its id
Diffstat (limited to '')
| -rw-r--r-- | pkg/database/stop.go | 15 |
1 files changed, 15 insertions, 0 deletions
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 := ` |