From d25ec27fb1c3df175e1b98af1fdc26881d68a1ab Mon Sep 17 00:00:00 2001
From: Elizabeth Hunt <elizabeth.hunt@simponic.xyz>
Date: Sun, 5 Jan 2025 16:39:13 -0800
Subject: initial commit by simponic-infra

---
 database/migrate.go | 46 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 46 insertions(+)
 create mode 100644 database/migrate.go

(limited to 'database/migrate.go')

diff --git a/database/migrate.go b/database/migrate.go
new file mode 100644
index 0000000..06213dd
--- /dev/null
+++ b/database/migrate.go
@@ -0,0 +1,46 @@
+package database
+
+import (
+	"log"
+
+	"database/sql"
+
+	_ "github.com/mattn/go-sqlite3"
+)
+
+type Migrator func(*sql.DB) (*sql.DB, error)
+
+func AddWhoisUpdates(dbConn *sql.DB) (*sql.DB, error) {
+	_, err := dbConn.Exec(`CREATE TABLE IF NOT EXISTS updates (
+           name TEXT,
+           time TIMESTAMP
+        );`)
+	if err != nil {
+		return dbConn, err
+	}
+
+	log.Println("creating time index on updates table")
+	_, err = dbConn.Exec(`CREATE INDEX IF NOT EXISTS idx_update_timestamp ON updates (time);`)
+	if err != nil {
+		return dbConn, err
+	}
+
+	return dbConn, nil
+}
+
+func Migrate(dbConn *sql.DB) (*sql.DB, error) {
+	log.Println("migrating database")
+
+	migrations := []Migrator{
+		AddWhoisUpdates,
+	}
+
+	for _, migration := range migrations {
+		dbConn, err := migration(dbConn)
+		if err != nil {
+			return dbConn, err
+		}
+	}
+
+	return dbConn, nil
+}
-- 
cgit v1.2.3-70-g09d2