From d14605d1388aaa7cc9ef1c230eae5ba14c9cef44 Mon Sep 17 00:00:00 2001 From: Elizabeth Hunt Date: Sun, 21 Apr 2024 18:46:40 -0700 Subject: initial commit --- database/migrate.go | 42 ++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 42 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..f2087cc --- /dev/null +++ b/database/migrate.go @@ -0,0 +1,42 @@ +package database + +import ( + "log" + + "database/sql" + + _ "github.com/mattn/go-sqlite3" +) + +type Migrator func(*sql.DB) (*sql.DB, error) + +func MigrateBackups(dbConn *sql.DB) (*sql.DB, error) { + log.Println("migrating backups table") + + _, err := dbConn.Exec(`CREATE TABLE IF NOT EXISTS backups ( + hostname TEXT NOT NULL, + received_on TIMESTAMP DEFAULT CURRENT_TIMESTAMP + );`) + if err != nil { + return dbConn, err + } + + return dbConn, nil +} + +func Migrate(dbConn *sql.DB) (*sql.DB, error) { + log.Println("migrating database") + + migrations := []Migrator{ + MigrateBackups, + } + + for _, migration := range migrations { + dbConn, err := migration(dbConn) + if err != nil { + return dbConn, err + } + } + + return dbConn, nil +} -- cgit v1.2.3-70-g09d2