From 9cae9c1d2a0b4f7fa72f3075541b9ffafe1a7275 Mon Sep 17 00:00:00 2001
From: Zach Berwaldt <zberwaldt@tutamail.com>
Date: Fri, 15 Mar 2024 18:49:43 -0400
Subject: Add routes for preference, clean up and add types

---
 api/internal/controllers/preferences.go | 45 +++++++++++++++++++++++++++++++++
 1 file changed, 45 insertions(+)
 create mode 100644 api/internal/controllers/preferences.go

(limited to 'api/internal/controllers/preferences.go')

diff --git a/api/internal/controllers/preferences.go b/api/internal/controllers/preferences.go
new file mode 100644
index 0000000..a1bcf4f
--- /dev/null
+++ b/api/internal/controllers/preferences.go
@@ -0,0 +1,45 @@
+package controllers
+
+import (
+	"github.com/gin-gonic/gin"
+	"net/http"
+	"database/sql"
+	"water/api/internal/database"
+	"water/api/internal/models"
+)
+
+func GetSizes(c *gin.Context) {
+		db := database.EstablishDBConnection()
+	defer func(db *sql.DB) {
+		err := db.Close()
+		if err != nil {
+			c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+		}
+	}(db)
+	
+	rows, err := db.Query("SELECT id, size, unit  FROM Sizes")
+	if err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+		return
+	}
+	defer func(rows *sql.Rows) {
+		err := rows.Close()
+		if err != nil {
+			c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+			return
+		}
+	}(rows)
+
+	var data []models.Size
+
+	for rows.Next() {
+		var size models.Size
+		if err := rows.Scan(&size.ID, &size.Size, &size.Unit); err != nil {
+			c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+			return
+		}
+		data = append(data, size)
+	}
+
+	c.JSON(http.StatusOK, data)
+}
-- 
cgit v1.1