forked from tangled.org/core
Something went wrong. Try again.
1.5 kB · 83 lines
Go
at master
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384package cursor
import ( "database/sql" "fmt"
_ "github.com/mattn/go-sqlite3")
type SqliteStore struct { db *sql.DB tableName string}
type SqliteStoreOpt func(*SqliteStore)
func WithTableName(name string) SqliteStoreOpt { return func(s *SqliteStore) { s.tableName = name }}
func NewSQLiteStore(dbPath string, opts ...SqliteStoreOpt) (*SqliteStore, error) { db, err := sql.Open("sqlite3", dbPath+"?_foreign_keys=1") if err != nil { return nil, fmt.Errorf("failed to open sqlite database: %w", err) }
store := &SqliteStore{ db: db, tableName: "cursors", }
for _, o := range opts { o(store) }
if err := store.init(); err != nil { return nil, err }
return store, nil}
func (s *SqliteStore) init() error { createTable := fmt.Sprintf(` create table if not exists %s ( knot text primary key, cursor text );`, s.tableName) _, err := s.db.Exec(createTable) return err}
func (s *SqliteStore) Set(knot string, cursor int64) { query := fmt.Sprintf(` insert into %s (knot, cursor) values (?, ?) on conflict(knot) do update set cursor=excluded.cursor; `, s.tableName)
_, err := s.db.Exec(query, knot, cursor)
if err != nil { // TODO: log here }}
func (s *SqliteStore) Get(knot string) (cursor int64) { query := fmt.Sprintf(` select cursor from %s where knot = ?; `, s.tableName) err := s.db.QueryRow(query, knot).Scan(&cursor)
if err != nil { if err != sql.ErrNoRows { // TODO: log here } return 0 }
return cursor}