add new 'weekday' column to database

data.go

@@ -3,6 +3,7 @@ package main
 import (
 	"database/sql"
 	"fmt"
+	"strings"
 
 	_ "github.com/tursodatabase/libsql-client-go/libsql"
 	_ "modernc.org/sqlite"
@@ -21,6 +22,7 @@ type Dog struct {
 	Service   string
 	Quantity  int
 	Price     float64
+	Weekday   int
 }
 
 func Init() error {
@@ -32,6 +34,11 @@ func Init() error {
 	// No need to check for error here, if the connection can be made, the tables will be created
 	_ = createTables()
 
+	err = updateTables()
+	if err != nil {
+		return fmt.Errorf("error updating tables: %v", err)
+	}
+
 	return nil
 }
 
@@ -58,6 +65,7 @@ func createTables() error {
             service TEXT,
             quantity INTEGER,
             price INTEGER
+            weekday INTEGER
         );
     `)
 	if err != nil {
@@ -67,6 +75,17 @@ func createTables() error {
 	return nil
 }
 
+func updateTables() error {
+	_, err := db.Exec(`
+        ALTER TABLE dogs ADD COLUMN weekday INTEGER;
+    `)
+	if err == nil || strings.Contains(err.Error(), "duplicate column name") {
+		return nil
+	}
+
+	return fmt.Errorf("error updating dogs table: %v", err)
+}
+
 func getDogs() ([]Dog, error) {
 	rows, err := db.Query("SELECT * FROM dogs")
 	if err != nil {
@@ -86,6 +105,7 @@ func getDogs() ([]Dog, error) {
 			&dog.Service,
 			&dog.Quantity,
 			&dog.Price,
+			&dog.Weekday,
 		)
 		if err != nil {
 			rows.Close()
@@ -109,6 +129,7 @@ func getDog(id int) (Dog, error) {
 		&dog.Service,
 		&dog.Quantity,
 		&dog.Price,
+		&dog.Weekday,
 	)
 	if err != nil {
 		return Dog{}, fmt.Errorf("error getting dog: %v", err)
@@ -127,9 +148,10 @@ func addDog(dog Dog) error {
             email,
             service,
             quantity,
-            price
-        ) VALUES (?, ?, ?, ?, ?, ?, ?, ?)
-    `, dog.Name, dog.OwnerName, dog.Address, dog.City, dog.Email, dog.Service, dog.Quantity, dog.Price)
+            price,
+            weekday
+        ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)
+    `, dog.Name, dog.OwnerName, dog.Address, dog.City, dog.Email, dog.Service, dog.Quantity, dog.Price, dog.Weekday)
 	if err != nil {
 		return fmt.Errorf("error adding dog: %v", err)
 	}
@@ -147,9 +169,10 @@ func updateDog(dog Dog) error {
             email = ?,
             service = ?,
             quantity = ?,
-            price = ?
+            price = ?,
+            weekday = ?
         WHERE id = ?
-    `, dog.Name, dog.OwnerName, dog.Address, dog.City, dog.Email, dog.Service, dog.Quantity, dog.Price, dog.ID)
+    `, dog.Name, dog.OwnerName, dog.Address, dog.City, dog.Email, dog.Service, dog.Quantity, dog.Price, dog.Weekday, dog.ID)
 	if err != nil {
 		return fmt.Errorf("error updating dog: %v", err)
 	}

data_test.go

@@ -15,6 +15,7 @@ var testDog Dog = Dog{
 	Service:   "walk",
 	Quantity:  1,
 	Price:     25,
+	Weekday:   2,
 }
 
 func TestInit(t *testing.T) {
@@ -80,9 +81,10 @@ func TestCreateTables(t *testing.T) {
             email,
             service,
             quantity,
-            price
-        ) VALUES (?, ?, ?, ?, ?, ?, ?, ?)
-    `, testDog.Name, testDog.OwnerName, testDog.Address, testDog.City, testDog.Email, testDog.Service, testDog.Quantity, testDog.Price)
+            price,
+            weekday
+        ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)
+    `, testDog.Name, testDog.OwnerName, testDog.Address, testDog.City, testDog.Email, testDog.Service, testDog.Quantity, testDog.Price, testDog.Weekday)
 
 	if err != nil {
 		t.Errorf("createTables() error = %q", err)
@@ -105,6 +107,27 @@ func TestCreateTables(t *testing.T) {
 	})
 }
 
+func TestUpdateTables(t *testing.T) {
+	err := updateTables()
+	if err != nil {
+		t.Errorf("updateTables() error = %q", err)
+	}
+
+	// Force error
+	oldDbUrl := dbUrl
+	dbUrl = "someBadUrl"
+	_ = connect()
+	err = updateTables()
+	if err == nil {
+		t.Errorf("no error returned when expected")
+	}
+
+	t.Cleanup(func() {
+		dbUrl = oldDbUrl
+		_ = connect()
+	})
+}
+
 func TestAddDog(t *testing.T) {
 	err := addDog(testDog)
 	if err != nil {