-
Notifications
You must be signed in to change notification settings - Fork 2
/
sqlite.go
65 lines (50 loc) · 1.06 KB
/
sqlite.go
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
package sqlite
import (
"database/sql"
"log"
"time"
_ "github.com/mattn/go-sqlite3"
"github.com/mtlynch/screenjournal/v2/screenjournal"
)
const (
timeFormat = time.RFC3339
)
type (
Store struct {
ctx *sql.DB
}
rowScanner interface {
Scan(...interface{}) error
}
)
func New(path string, optimizeForLitestream bool) Store {
log.Printf("reading DB from %s", path)
ctx, err := sql.Open("sqlite3", path)
if err != nil {
log.Fatalln(err)
}
if _, err := ctx.Exec(`
PRAGMA temp_store = FILE;
PRAGMA journal_mode = WAL;
`); err != nil {
log.Fatalf("failed to set pragmas: %v", err)
}
store := Store{ctx: ctx}
if optimizeForLitestream {
store.optimizeForLitestream()
}
store.applyMigrations()
return store
}
func parseDatetime(s string) (time.Time, error) {
return time.Parse(timeFormat, s)
}
func formatTime(t time.Time) string {
return t.Format(timeFormat)
}
func formatWatchDate(w screenjournal.WatchDate) string {
return formatTime(w.Time())
}
func formatReleaseDate(rd screenjournal.ReleaseDate) string {
return formatTime(rd.Time())
}