/
users.go
102 lines (89 loc) · 1.89 KB
/
users.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 handlers
import (
"net/http"
"strconv"
"database/sql"
"github.com/labstack/echo"
"github.com/Team-Fruit/SignPicDB/web/models"
)
type (
count struct {
Count uint `json:"count"`
}
response struct {
Message string `json:"message"`
}
)
func (h *handler) GetList(c echo.Context) (err error) {
var page, pagesize uint64
if pagestr := c.QueryParam("page"); pagestr != "" {
if page, err = strconv.ParseUint(pagestr, 10, 32); err != nil {
return
}
if page < 1 {
page = 1
}
} else {
page = 1
}
if pagesizestr := c.QueryParam("pagesize"); pagesizestr != "" {
if pagesize, err = strconv.ParseUint(pagesizestr, 10, 32); err != nil {
return
}
if pagesize < 1 {
pagesize = 1
}
if pagesize > 100 {
pagesize = 100
}
} else {
pagesize = 100
}
w := models.UserWhere{}
if err = c.Bind(&w); err != nil {
return
}
if err = c.Validate(&w); err != nil {
return
}
var l []models.User
if l, err = h.Model.FindUsers(w, uint(pagesize*(page-1)), uint(pagesize)); err != nil {
return
}
if len(l) == 0 {
l = make([]models.User, 0)
}
return c.JSON(http.StatusOK, l)
}
func (h *handler) GetUser(c echo.Context) (err error) {
id := c.Param("id")
var data models.UserData
if data, err = h.Model.GetUserData(id); err != nil {
if err == sql.ErrNoRows {
return c.JSON(http.StatusNotFound, response{"Not Found"})
}
return
}
return c.JSON(http.StatusOK, data)
}
func (h *handler) GetUniqueUserCount(c echo.Context) (err error) {
w := models.UserWhere{}
if err = c.Bind(&w); err != nil {
return
}
if err = c.Validate(&w); err != nil {
return
}
var cnt uint
if cnt, err = h.Model.CountUniqueUser(w); err != nil {
return
}
return c.JSON(http.StatusOK, count{cnt})
}
func (h *handler) GetPlayCount(c echo.Context) (err error) {
var cnt uint
if cnt, err = h.Model.SumPlayCount(); err != nil {
return
}
return c.JSON(http.StatusOK, count{cnt})
}