74 lines
1.2 KiB
Go
74 lines
1.2 KiB
Go
package main
|
|
|
|
import (
|
|
"database/sql"
|
|
|
|
_ "modernc.org/sqlite" // Import the SQLite driver
|
|
)
|
|
|
|
func initDb() *sql.DB {
|
|
|
|
db, err := sql.Open("sqlite", "./data/data.db?_pragma=journal_mode(WAL)&_pragma=synchronous(NORMAL)&_pragma=foreign_keys(ON)")
|
|
if err != nil {
|
|
e.Logger.Fatal(err)
|
|
}
|
|
|
|
// Create a table
|
|
sqlStmt := `
|
|
CREATE TABLE IF NOT EXISTS boxes (id INTEGER PRIMARY KEY, value BOOLEAN);
|
|
`
|
|
|
|
_, err = db.Exec(sqlStmt)
|
|
if err != nil {
|
|
e.Logger.Fatalf("%q: %s\n", err, sqlStmt)
|
|
}
|
|
|
|
return db
|
|
}
|
|
|
|
func getBoxes() map[int]bool {
|
|
|
|
rows, err := db.Query("SELECT id, value FROM boxes")
|
|
if err != nil {
|
|
e.Logger.Fatal(err)
|
|
}
|
|
|
|
boxes := make(map[int]bool, 1000)
|
|
for i := 0; i < 1000; i++ {
|
|
boxes[i] = false
|
|
}
|
|
for rows.Next() {
|
|
var id int
|
|
var value bool
|
|
err = rows.Scan(&id, &value)
|
|
if err != nil {
|
|
e.Logger.Fatal(err)
|
|
}
|
|
boxes[id] = value
|
|
}
|
|
defer rows.Close()
|
|
|
|
return boxes
|
|
}
|
|
|
|
func setBox(id int, value bool) {
|
|
e.Logger.Info("START INSERT")
|
|
|
|
stmt, err := db.Prepare("INSERT OR REPLACE INTO boxes(id, value) VALUES(?, ?)")
|
|
if err != nil {
|
|
e.Logger.Fatal(err)
|
|
}
|
|
|
|
e.Logger.Info("PREPARED")
|
|
|
|
res, err := stmt.Exec(id, value)
|
|
if err != nil {
|
|
e.Logger.Fatal(err)
|
|
}
|
|
stmt.Close()
|
|
|
|
e.Logger.Info("INSERTED")
|
|
|
|
e.Logger.Info(res)
|
|
}
|