68 lines
1.4 KiB
Go
68 lines
1.4 KiB
Go
package sqlite3
|
|
|
|
import (
|
|
"github.com/fiatjaf/eventstore"
|
|
"github.com/jmoiron/sqlx"
|
|
"github.com/jmoiron/sqlx/reflectx"
|
|
_ "github.com/mattn/go-sqlite3"
|
|
)
|
|
|
|
const (
|
|
queryLimit = 100
|
|
queryIDsLimit = 500
|
|
queryAuthorsLimit = 500
|
|
queryKindsLimit = 10
|
|
queryTagsLimit = 10
|
|
)
|
|
|
|
var _ eventstore.Store = (*SQLite3Backend)(nil)
|
|
|
|
var ddls = []string{
|
|
`CREATE TABLE IF NOT EXISTS event (
|
|
id text NOT NULL,
|
|
pubkey text NOT NULL,
|
|
created_at integer NOT NULL,
|
|
kind integer NOT NULL,
|
|
tags jsonb NOT NULL,
|
|
content text NOT NULL,
|
|
sig text NOT NULL);`,
|
|
`CREATE UNIQUE INDEX IF NOT EXISTS ididx ON event(id)`,
|
|
`CREATE INDEX IF NOT EXISTS pubkeyprefix ON event(pubkey)`,
|
|
`CREATE INDEX IF NOT EXISTS timeidx ON event(created_at DESC)`,
|
|
`CREATE INDEX IF NOT EXISTS kindidx ON event(kind)`,
|
|
}
|
|
|
|
func (b *SQLite3Backend) Init() error {
|
|
db, err := sqlx.Connect("sqlite3", b.DatabaseURL)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
db.Mapper = reflectx.NewMapperFunc("json", sqlx.NameMapper)
|
|
b.DB = db
|
|
|
|
for _, ddl := range ddls {
|
|
_, err = b.DB.Exec(ddl)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
if b.QueryLimit == 0 {
|
|
b.QueryLimit = queryLimit
|
|
}
|
|
if b.QueryIDsLimit == 0 {
|
|
b.QueryIDsLimit = queryIDsLimit
|
|
}
|
|
if b.QueryAuthorsLimit == 0 {
|
|
b.QueryAuthorsLimit = queryAuthorsLimit
|
|
}
|
|
if b.QueryKindsLimit == 0 {
|
|
b.QueryKindsLimit = queryKindsLimit
|
|
}
|
|
if b.QueryTagsLimit == 0 {
|
|
b.QueryTagsLimit = queryTagsLimit
|
|
}
|
|
return nil
|
|
}
|