mirror of
https://github.com/psviderski/uncloud.git
synced 2026-08-26 19:13:34 +00:00
43 lines
1.4 KiB
Go
43 lines
1.4 KiB
Go
package machine
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/jmoiron/sqlx"
|
|
_ "modernc.org/sqlite"
|
|
)
|
|
|
|
const DBFileName = "machine.db"
|
|
|
|
// NewDB creates a new connection to machine SQLite database and runs schema migrations if necessary.
|
|
func NewDB(path string) (*sqlx.DB, error) {
|
|
// - Write-Ahead Logging (WAL) mode for better read/write performance.
|
|
// - Busy timeout (5s) to make concurrent writes wait on each other instead of failing immediately.
|
|
conn := path + "?_pragma=journal_mode=WAL&_pragma=synchronous=NORMAL&_pragma=busy_timeout=5000&_time_format=sqlite"
|
|
db, err := sqlx.Connect("sqlite", conn)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("connect to SQLite database '%s': %w", conn, err)
|
|
}
|
|
|
|
schema := `
|
|
CREATE TABLE IF NOT EXISTS containers (
|
|
id TEXT NOT NULL PRIMARY KEY,
|
|
service_id TEXT NOT NULL,
|
|
service_name TEXT AS (json_extract(service_spec, '$.Name')),
|
|
service_spec TEXT NOT NULL CHECK (json_valid(service_spec)),
|
|
-- 'subsecond' modifier is used to store timestamps with millisecond precision.
|
|
created_at TIMESTAMP NOT NULL DEFAULT (datetime('subsecond')),
|
|
updated_at TIMESTAMP NOT NULL DEFAULT (datetime('subsecond'))
|
|
);
|
|
|
|
CREATE INDEX IF NOT EXISTS idx_containers_service_id ON containers (service_id);
|
|
CREATE INDEX IF NOT EXISTS idx_containers_service_name ON containers (service_name);
|
|
`
|
|
|
|
if _, err = db.Exec(schema); err != nil {
|
|
return nil, fmt.Errorf("create schema: %w", err)
|
|
}
|
|
|
|
return db, nil
|
|
}
|