1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
|
package repository
import (
"context"
"database/sql"
"fmt"
// Register the pure-Go SQLite driver with database/sql.
_ "modernc.org/sqlite"
)
// SQLite is a concrete Store implementation backed by SQLite.
type SQLite struct {
db *sql.DB
}
const sqliteBusyTimeoutMS = 5000
// New creates a SQLite store from an existing *sql.DB after initializing the schema.
func New(db *sql.DB) (*SQLite, error) {
if err := initializeSchema(db); err != nil {
return nil, fmt.Errorf("initialize schema: %w", err)
}
return &SQLite{db: db}, nil
}
// Open opens a SQLite database at the given DSN and returns a connected Store.
func Open(dsn string) (*SQLite, error) {
db, err := sql.Open("sqlite", dsn)
if err != nil {
return nil, err
}
db.SetMaxOpenConns(1)
db.SetMaxIdleConns(1)
if err := db.Ping(); err != nil {
_ = db.Close()
return nil, err
}
if _, err := db.Exec(fmt.Sprintf(`PRAGMA busy_timeout = %d;`, sqliteBusyTimeoutMS)); err != nil {
_ = db.Close()
return nil, fmt.Errorf("set busy timeout: %w", err)
}
s, err := New(db)
if err != nil {
_ = db.Close()
return nil, err
}
return s, nil
}
// Close closes the underlying database connection.
func (s *SQLite) Close() error {
return s.db.Close()
}
// Ping checks the database connection health.
func (s *SQLite) Ping(ctx context.Context) error {
return s.db.PingContext(ctx)
}
var _ Store = (*SQLite)(nil)
var _ APITokenRepo = (*SQLite)(nil)
var _ PodcastRepo = (*SQLite)(nil)
var _ ProgressTransactionStore = (*SQLite)(nil)
type sqlScanner interface {
Scan(dest ...any) error
}
type sqlExecer interface {
ExecContext(ctx context.Context, query string, args ...any) (sql.Result, error)
}
type sqlQueryRower interface {
QueryRowContext(ctx context.Context, query string, args ...any) *sql.Row
}
func boolToInt(b bool) int {
if b {
return 1
}
return 0
}
func intToBool(i int) bool {
return i != 0
}
|
