This repository has been archived by the owner on Sep 16, 2021. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 4
/
search.go
101 lines (81 loc) · 2.36 KB
/
search.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
96
97
98
99
100
101
package models
import (
"context"
"database/sql"
)
// Search type is a struct for search queries.
type Search struct {
Query string `json:"query" validate:"required"`
Type int `json:"type"`
}
// SearchRecommendation search for recommendations.
func (c *Conn) SearchRecommendation(ctx context.Context, offset, limit int, search string) (*[]Recommendation, error) {
var recommendations []Recommendation
err := c.db.SelectContext(
ctx,
&recommendations,
querySearchRecommendations,
"%"+search+"%",
"%"+search+"%",
"%"+search+"%",
offset,
limit,
)
if err != nil && err != sql.ErrNoRows {
return nil, err
}
return &recommendations, nil
}
// SearchUser search for users.
func (c *Conn) SearchUser(ctx context.Context, search string) (*UserResult, error) {
var users []User
err := c.db.SelectContext(ctx, &users, querySearchUsers, "%"+search+"%", 20)
if err != nil && err != sql.ErrNoRows {
return nil, err
}
return &UserResult{&users}, nil
}
// SearchGenre search for genres.
func (c *Conn) SearchGenre(ctx context.Context, search string) (*GenreResult, error) {
var genres []Genre
err := c.db.SelectContext(ctx, &genres, querySearchGenres, "%"+search+"%", 20)
if err != nil && err != sql.ErrNoRows {
return nil, err
}
return &GenreResult{&genres}, nil
}
// SearchKeyword search for keywords.
func (c *Conn) SearchKeyword(ctx context.Context, search string) (*KeywordResult, error) {
var keywords []Keyword
err := c.db.SelectContext(ctx, &keywords, querySearchKeywords, "%"+search+"%", 20)
if err != nil && err != sql.ErrNoRows {
return nil, err
}
return &KeywordResult{&keywords}, nil
}
// SearchSource search for sources.
func (c *Conn) SearchSource(ctx context.Context, search string) (*SourceResult, error) {
var sources []Source
err := c.db.SelectContext(ctx, &sources, querySearchSources, "%"+search+"%", 20)
if err != nil && err != sql.ErrNoRows {
return nil, err
}
return &SourceResult{&sources}, nil
}
// GetSearchRecommendationTotalRows retrieves the total
// rows of recommendations table.
func (c *Conn) GetSearchRecommendationTotalRows(ctx context.Context, search string) (int, error) {
var totalRows int
err := c.db.GetContext(
ctx,
&totalRows,
querySearchRecommendationsTotalRows,
"%"+search+"%",
"%"+search+"%",
"%"+search+"%",
)
if err != nil && err != sql.ErrNoRows {
return 0, err
}
return totalRows, nil
}