guestbook!

1 files changed, 19 insertions, 0 deletions

diff --git a/database/migrate.go b/database/migrate.go
index de1db4c..1609bc2 100644
--- a/database/migrate.go
+++ b/database/migrate.go
@@ -102,6 +102,24 @@ func MigrateUserSessions(dbConn *sql.DB) (*sql.DB, error) {
 	return dbConn, nil
 }
 
+func MigrateGuestBook(dbConn *sql.DB) (*sql.DB, error) {
+	log.Println("migrating guest_book table")
+
+	_, err := dbConn.Exec(`CREATE TABLE IF NOT EXISTS guest_book (
+		id TEXT PRIMARY KEY,
+		name TEXT NOT NULL,
+		message TEXT NOT NULL,
+		created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
+	);`)
+	if err != nil {
+		return dbConn, err
+	}
+
+	_, err = dbConn.Exec(`CREATE INDEX IF NOT EXISTS idx_guest_book_created_at ON guest_book (created_at);`)
+
+	return dbConn, nil
+}
+
 func Migrate(dbConn *sql.DB) (*sql.DB, error) {
 	log.Println("migrating database")
 
@@ -111,6 +129,7 @@ func Migrate(dbConn *sql.DB) (*sql.DB, error) {
 		MigrateApiKeys,
 		MigrateDomainOwners,
 		MigrateDNSRecords,
+		MigrateGuestBook,
 	}
 
 	for _, migration := range migrations {