From dc5d066698bd1dc6a609f773b1333fa5025672fd Mon Sep 17 00:00:00 2001
From: Siegfried Siegert <Siegfried.Siegert@mimbach49.de>
Date: Sun, 20 Sep 2020 17:53:48 +0200
Subject: [PATCH] Read project name from database

---
 models/devices.go | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)

diff --git a/models/devices.go b/models/devices.go
index dfb76b2..417da8e 100644
--- a/models/devices.go
+++ b/models/devices.go
@@ -29,6 +29,7 @@ func (db *DB) AllDevices() ([]*Device, error) {
 		if err != nil {
 			return nil, err
 		}
+		device.ProjectName, err = db.GetProjectName(device.CustomerID)
 		devices = append(devices, device)
 	}
 	if err = rows.Err(); err != nil {
@@ -56,3 +57,26 @@ func (db *DB) InsertDevice(device *Device) error {
 
 	return err
 }
+
+func (db *DB) GetProjectName(projectNumber string) (string, error) {
+
+	projectNameRows, err := db.Query("SELECT projectName FROM projects WHERE projectNumber =  $1", projectNumber)
+	if err != nil {
+		return "", err
+	}
+	defer projectNameRows.Close()
+
+	projectName := ""
+	for projectNameRows.Next() {
+		err := projectNameRows.Scan(&projectName)
+		if err != nil {
+			return "", err
+		}
+	}
+	if err = projectNameRows.Err(); err != nil {
+		return "", err
+	}
+
+	return projectName, err
+
+}