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
|
package db
type SearchPage struct {
Path string
Title string
Content string
}
type SearchResult struct {
Path string
Title string
Snippet string
}
func (d *DB) IndexPage(p SearchPage) error {
tx, err := d.db.Begin()
if err != nil {
return err
}
defer tx.Rollback()
if _, err = tx.Exec(`DELETE FROM pages_fts WHERE path = ?`, p.Path); err != nil {
return err
}
if _, err = tx.Exec(
`INSERT INTO pages_fts (path, title, content) VALUES (?, ?, ?)`,
p.Path, p.Title, p.Content,
); err != nil {
return err
}
return tx.Commit()
}
func (d *DB) UnindexPage(path string) error {
_, err := d.db.Exec(`DELETE FROM pages_fts WHERE path = ?`, path)
return err
}
// Search runs a full-text query and returns up to 20 results with snippets.
func (d *DB) Search(query string) ([]SearchResult, error) {
rows, err := d.db.Query(`
SELECT path, title,
snippet(pages_fts, 2, '<mark>', '</mark>', '...', 20)
FROM pages_fts
WHERE pages_fts MATCH ?
ORDER BY rank
LIMIT 20
`, query)
if err != nil {
return nil, err
}
defer rows.Close()
var results []SearchResult
for rows.Next() {
var r SearchResult
if err := rows.Scan(&r.Path, &r.Title, &r.Snippet); err != nil {
return nil, err
}
results = append(results, r)
}
return results, rows.Err()
}
|
