Saving timestampsHEADv1.5.0master

3 files changed, 8 insertions, 6 deletions

diff --git a/cmd/dump.go b/cmd/dump.go
index ec8976b..ddc26e8 100644
--- a/cmd/dump.go
+++ b/cmd/dump.go
@@ -44,7 +44,7 @@ var dumpCmd = &cobra.Command{
 		conn := db.Connect()
 		defer conn.Close()
 
-		rows, err := conn.Query(fmt.Sprintf("SELECT name, website, message FROM %s", guestbookName))
+		rows, err := conn.Query(fmt.Sprintf("SELECT name, website, message, time FROM %s", guestbookName))
 		if err != nil {
 			log.Fatal(err)
 		}
@@ -54,7 +54,7 @@ var dumpCmd = &cobra.Command{
 
 		for rows.Next() {
 			var entry db.GuestbookEntry
-			if err := rows.Scan(&entry.Name, &entry.Website, &entry.Message); err != nil {
+			if err := rows.Scan(&entry.Name, &entry.Website, &entry.Message, &entry.Time); err != nil {
 				log.Fatal(err)
 			}
 			entries = append(entries, entry)
diff --git a/cmd/root.go b/cmd/root.go
index a150426..5ba7ae6 100644
--- a/cmd/root.go
+++ b/cmd/root.go
@@ -27,7 +27,7 @@ import (
 
 var rootCmd = &cobra.Command{
 	Use:   "guestbook",
-	Version: "v1.4.0",
+	Version: "v1.5.0",
 	Short: "Standalone guestbook server for static websites",
 	Long: "Standalone guestbook server for static websites",
 }
diff --git a/cmd/serve.go b/cmd/serve.go
index a5727df..7d0b8c1 100644
--- a/cmd/serve.go
+++ b/cmd/serve.go
@@ -20,6 +20,7 @@ package cmd
 import (
 	"fmt"
 	"log"
+	"time"
 	"strings"
 	"net/http"
 
@@ -67,7 +68,8 @@ var serveCmd = &cobra.Command{
 				"id INT AUTO_INCREMENT PRIMARY KEY," +
 				"name TEXT," +
 				"website TEXT," +
-				"message TEXT" +
+				"message TEXT," +
+				"time VARCHAR(30)" +
 				");",
 				guestbookName,
 			),
@@ -102,7 +104,7 @@ var serveCmd = &cobra.Command{
 				return
 			}
 
-			stmt, err := conn.Prepare(fmt.Sprintf("INSERT INTO %s (name, website, message) VALUES (?, ?, ?)", guestbookName))
+			stmt, err := conn.Prepare(fmt.Sprintf("INSERT INTO %s (name, website, message, time) VALUES (?, ?, ?, ?)", guestbookName))
 			if err != nil {
 				log.Printf("Error occurred while preparing statement data: %v\n")
 			    http.Redirect(w, r, errorRedirect, http.StatusSeeOther)
@@ -110,7 +112,7 @@ var serveCmd = &cobra.Command{
 			}
 			defer stmt.Close()
 
-			_, err = stmt.Exec(data.Name, data.Website, data.Message)
+			_, err = stmt.Exec(data.Name, data.Website, data.Message, time.Now().Format(time.RFC3339))
 			if err != nil {
 				log.Printf("Error occurred while saving data: %v\n")
 			    http.Redirect(w, r, errorRedirect, http.StatusSeeOther)