-
Notifications
You must be signed in to change notification settings - Fork 2
/
db.go
95 lines (74 loc) · 1.75 KB
/
db.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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
package sqlite
import (
"database/sql"
"log"
"fmt"
"github.com/bennyz/example-finder/util"
"github.com/bennyz/example-finder/persistence"
_ "github.com/mattn/go-sqlite3"
)
const (
createTable = `CREATE TABLE IF NOT EXISTS repo_data(repo_id INTEGER PRIMARY KEY, data JSON1)`
truncateTable = `DELETE FROM repo_data`
insertKey = `INSERT INTO repo_data(repo_id, data) values(?, ?)`
getKeys = `SELECT data FROM repo_data WHERE repo_id IN (%s)`
)
type sqlite struct {
*sql.DB
}
// New initializes sqlite database
func New(path string) (persistence.Storage, error) {
db, err := sql.Open("sqlite3", path)
if err != nil {
log.Fatal(err)
}
_, err = db.Exec(createTable)
if err != nil {
log.Fatal(err)
}
return &sqlite{db}, nil
}
func (s *sqlite) Save(key int64, value persistence.JSONValue) (int64, error) {
tx, err := s.Begin()
if err != nil {
log.Fatal(err)
}
stmt, err := tx.Prepare(insertKey)
if err != nil {
log.Fatal(err)
}
_, err = stmt.Exec(key, value)
if err != nil {
log.Fatal(err)
}
tx.Commit()
defer stmt.Close()
return key, nil
}
func (s *sqlite) Get(keys []int64) ([]persistence.JSONValue, error) {
var keysRaw []int
for key := range keys {
keysRaw = append(keysRaw, key)
}
query := fmt.Sprintf(getKeys, util.SliceToString(keys))
rows, err := s.Query(query)
if err != nil {
log.Fatal("Failed querying rows. Query:", query, "Keys", keys, "error:", err)
}
var result []persistence.JSONValue
for rows.Next() {
var value persistence.JSONValue
err := rows.Scan(&value)
if err != nil {
log.Fatal("Failed scanning row. Rows:", rows, "err", err)
}
result = append(result, value)
}
return result, err
}
func (s *sqlite) Clear() {
_, err := s.Exec(truncateTable)
if err != nil {
log.Fatal(err)
}
}