Add `users` table

This commit is contained in:
Netscape Navigator 2020-09-11 07:44:10 -05:00
parent 83a6194b7c
commit bf6ef1d07c
3 changed files with 53 additions and 34 deletions

View File

@ -8,23 +8,6 @@ import (
"modernc.org/ql"
)
type migration struct {
up string
down string
}
var migrations = []migration{
migration{
up: `CREATE TABLE IF NOT EXISTS configs (
key string NOT NULL,
value string NOT NULL
);
CREATE UNIQUE INDEX IF NOT EXISTS unique_configs_key ON configs (key);
`,
down: `DROP TABLE IF EXISTS configs`,
},
}
func openDB() *sql.DB {
ql.RegisterDriver()
pigeonPath := maybeSetupPigeonDir()
@ -41,22 +24,7 @@ func openDB() *sql.DB {
log.Fatalf("failed to ping db: %s", err1)
}
tx, err := db.Begin()
if err != nil {
log.Fatalf("Failed to start transaction: %s", err)
}
for _, migration := range migrations {
_, err := tx.Exec(migration.up)
if err != nil {
log.Fatalf("Migration failure: %s", err)
}
}
if tx.Commit() != nil {
log.Fatal(err)
}
migrateUp(db)
return db
}

51
project/migrations.go Normal file
View File

@ -0,0 +1,51 @@
package main
import (
"database/sql"
"log"
)
type migration struct {
up string
down string
}
var migrations = []migration{
migration{
up: `CREATE TABLE IF NOT EXISTS configs (
key string NOT NULL,
value string NOT NULL
);
CREATE UNIQUE INDEX IF NOT EXISTS unique_configs_key ON configs (key);
`,
down: `DROP TABLE IF EXISTS configs`,
},
migration{
up: `CREATE TABLE IF NOT EXISTS users (
mhash string NOT NULL,
status string NOT NULL
);
CREATE UNIQUE INDEX IF NOT EXISTS unique_users_mhash ON users (mhash);
`,
down: `DROP TABLE IF EXISTS users`,
},
}
func migrateUp(db *sql.DB) {
tx, err := db.Begin()
if err != nil {
log.Fatalf("Failed to start transaction: %s", err)
}
for i, migration := range migrations {
_, err := tx.Exec(migration.up)
if err != nil {
log.Fatalf("Migration failure(%d): %s", i, err)
}
}
if tx.Commit() != nil {
log.Fatal(err)
}
}

View File

@ -6,7 +6,7 @@ import (
)
func showIdentity() string {
existingKey := GetConfig("private_key")
existingKey := GetConfig("public_key")
if len(existingKey) == 0 {
return "NONE"
}