Jon Staab
114 lines
2.9 KiB
Go
114 lines
2.9 KiB
Go
package sqlite
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
"sync"
|
|
|
|
"fiatjaf.com/nostr/eventstore"
|
|
_ "github.com/mattn/go-sqlite3"
|
|
)
|
|
|
|
var _ eventstore.Store = (*SqliteBackend)(nil)
|
|
|
|
type SqliteBackend struct {
|
|
sync.RWMutex
|
|
// Path is where the database will be saved
|
|
Path string
|
|
|
|
db *sql.DB
|
|
FTSAvailable bool
|
|
}
|
|
|
|
func (s *SqliteBackend) Close() {
|
|
if s.db != nil {
|
|
s.db.Close()
|
|
}
|
|
}
|
|
|
|
func (s *SqliteBackend) Init() error {
|
|
if s.Path == "" {
|
|
return fmt.Errorf("missing Path")
|
|
}
|
|
|
|
var err error
|
|
s.db, err = sql.Open("sqlite3", s.Path+"?_journal_mode=WAL&_sync=NORMAL&_cache_size=1000&_foreign_keys=true")
|
|
if err != nil {
|
|
return fmt.Errorf("failed to open database: %w", err)
|
|
}
|
|
|
|
// Create tables and indexes
|
|
if err := s.createSchema(); err != nil {
|
|
return fmt.Errorf("failed to create schema: %w", err)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (s *SqliteBackend) createSchema() error {
|
|
// Create basic schema first
|
|
basicSchema := `
|
|
CREATE TABLE IF NOT EXISTS events (
|
|
id TEXT PRIMARY KEY,
|
|
created_at INTEGER NOT NULL,
|
|
kind INTEGER NOT NULL,
|
|
pubkey TEXT NOT NULL,
|
|
content TEXT NOT NULL,
|
|
tags TEXT NOT NULL,
|
|
sig TEXT NOT NULL
|
|
);
|
|
|
|
CREATE INDEX IF NOT EXISTS idx_events_created_at ON events(created_at);
|
|
CREATE INDEX IF NOT EXISTS idx_events_kind ON events(kind);
|
|
CREATE INDEX IF NOT EXISTS idx_events_pubkey ON events(pubkey);
|
|
CREATE INDEX IF NOT EXISTS idx_events_kind_pubkey ON events(kind, pubkey);
|
|
CREATE INDEX IF NOT EXISTS idx_events_kind_pubkey_created_at ON events(kind, pubkey, created_at DESC);
|
|
|
|
CREATE TABLE IF NOT EXISTS event_tags (
|
|
event_id TEXT NOT NULL,
|
|
key TEXT NOT NULL,
|
|
value TEXT NOT NULL,
|
|
FOREIGN KEY (event_id) REFERENCES events(id) ON DELETE CASCADE
|
|
);
|
|
|
|
CREATE INDEX IF NOT EXISTS idx_event_tags_event_id ON event_tags(event_id);
|
|
CREATE INDEX IF NOT EXISTS idx_event_tags_key ON event_tags(key);
|
|
CREATE INDEX IF NOT EXISTS idx_event_tags_key_value ON event_tags(key, value);
|
|
`
|
|
|
|
if _, err := s.db.Exec(basicSchema); err != nil {
|
|
return fmt.Errorf("failed to create schema: %w", err)
|
|
}
|
|
|
|
// Try to create FTS5 schema - if it fails, continue without it
|
|
ftsSchema := `
|
|
CREATE VIRTUAL TABLE IF NOT EXISTS events_fts USING fts5(
|
|
content,
|
|
content='events',
|
|
content_rowid='rowid'
|
|
);
|
|
|
|
CREATE TRIGGER IF NOT EXISTS events_ai AFTER INSERT ON events BEGIN
|
|
INSERT INTO events_fts(rowid, content) VALUES (new.rowid, new.content);
|
|
END;
|
|
|
|
CREATE TRIGGER IF NOT EXISTS events_ad AFTER DELETE ON events BEGIN
|
|
INSERT INTO events_fts(events_fts, rowid, content) VALUES('delete', old.rowid, old.content);
|
|
END;
|
|
|
|
CREATE TRIGGER IF NOT EXISTS events_au AFTER UPDATE ON events BEGIN
|
|
INSERT INTO events_fts(events_fts, rowid, content) VALUES('delete', old.rowid, old.content);
|
|
INSERT INTO events_fts(rowid, content) VALUES (new.rowid, new.content);
|
|
END;
|
|
`
|
|
|
|
if _, err := s.db.Exec(ftsSchema); err != nil {
|
|
// FTS5 not available, continue without full-text search
|
|
s.FTSAvailable = false
|
|
} else {
|
|
s.FTSAvailable = true
|
|
}
|
|
|
|
return nil
|
|
}
|