Added code to load database
parent
c60fc1b739
commit
b366e0e1bc
@ -0,0 +1,5 @@
|
|||||||
|
module git.metznet.ca/MetzNet/stampede_2025
|
||||||
|
|
||||||
|
go 1.24.4
|
||||||
|
|
||||||
|
require github.com/mattn/go-sqlite3 v1.14.28
|
@ -0,0 +1,2 @@
|
|||||||
|
github.com/mattn/go-sqlite3 v1.14.28 h1:ThEiQrnbtumT+QMknw63Befp/ce/nUPgBPMlRFEum7A=
|
||||||
|
github.com/mattn/go-sqlite3 v1.14.28/go.mod h1:Uh1q+B4BYcTPb+yiD3kU8Ct7aC0hY9fxUwlHK0RXw+Y=
|
@ -0,0 +1,80 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"os/signal"
|
||||||
|
"syscall"
|
||||||
|
"context"
|
||||||
|
"os"
|
||||||
|
"log/slog"
|
||||||
|
"database/sql"
|
||||||
|
_ "github.com/mattn/go-sqlite3"
|
||||||
|
)
|
||||||
|
|
||||||
|
const SQLITE_TABLES string = ``
|
||||||
|
|
||||||
|
func server(db_path string) error {
|
||||||
|
sigs := make(chan os.Signal, 1)
|
||||||
|
signal.Notify(sigs, syscall.SIGINT, syscall.SIGTERM)
|
||||||
|
|
||||||
|
ctx := context.Background()
|
||||||
|
|
||||||
|
seed_db := false
|
||||||
|
_, err := os.Stat(db_path)
|
||||||
|
if errors.Is(err, os.ErrNotExist) {
|
||||||
|
seed_db = true
|
||||||
|
} else if err != nil {
|
||||||
|
slog.Error("Failed to stat db file", "path", db_path, "error", err)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
db, err := sql.Open("sqlite3", db_path)
|
||||||
|
if err != nil {
|
||||||
|
slog.Error("Failed to open db file", "path", db_path, "error", err)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
if seed_db {
|
||||||
|
slog.Info("Seeding database", "path", db_path)
|
||||||
|
tx, err := db.BeginTx(ctx, nil)
|
||||||
|
if err != nil {
|
||||||
|
slog.Error("Failed to start tx for db init", "error", err)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
_, err = tx.Exec(SQLITE_TABLES)
|
||||||
|
if err != nil {
|
||||||
|
slog.Error("Failed to exec tables for db init", "error", err)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
err = tx.Commit()
|
||||||
|
if err != nil {
|
||||||
|
slog.Error("Failed to commit tx for db init", "error", err)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
slog.Info("Starting server", "path", db_path)
|
||||||
|
run := true
|
||||||
|
for run {
|
||||||
|
select {
|
||||||
|
case <-sigs:
|
||||||
|
run = false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
err = db.Close()
|
||||||
|
if err != nil {
|
||||||
|
slog.Error("Failed to close db", "path", db_path, "error", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
err := server(os.Args[1])
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in New Issue