Saving timestampsHEADv1.5.0master

1 files changed, 5 insertions, 3 deletions

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)