diff options
Diffstat (limited to 'api/internal/controllers/stats.go')
-rw-r--r-- | api/internal/controllers/stats.go | 168 |
1 files changed, 168 insertions, 0 deletions
diff --git a/api/internal/controllers/stats.go b/api/internal/controllers/stats.go new file mode 100644 index 0000000..2234787 --- /dev/null +++ b/api/internal/controllers/stats.go | |||
@@ -0,0 +1,168 @@ | |||
1 | package controllers | ||
2 | |||
3 | import ( | ||
4 | "database/sql" | ||
5 | "github.com/gin-gonic/gin" | ||
6 | "net/http" | ||
7 | "water/api/internal/database" | ||
8 | "water/api/internal/models" | ||
9 | ) | ||
10 | |||
11 | // TODO: add comments to all exported members of package. | ||
12 | |||
13 | // GetAllStatistics connects to the database and queries for all statistics in the database. | ||
14 | // If none have been found it will return an error, otherwise a 200 code is sent along with the list of statistics. | ||
15 | func GetAllStatistics(c *gin.Context) { | ||
16 | db := database.EstablishDBConnection() | ||
17 | defer func(db *sql.DB) { | ||
18 | err := db.Close() | ||
19 | if err != nil { | ||
20 | c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()}) | ||
21 | return | ||
22 | } | ||
23 | }(db) | ||
24 | |||
25 | rows, err := db.Query("SELECT s.date, s.quantity, u.uuid, u.name FROM Statistics s INNER JOIN Users u ON u.id = s.user_id") | ||
26 | if err != nil { | ||
27 | c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()}) | ||
28 | return | ||
29 | } | ||
30 | defer func(rows *sql.Rows) { | ||
31 | err := rows.Close() | ||
32 | if err != nil { | ||
33 | c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()}) | ||
34 | return | ||
35 | } | ||
36 | }(rows) | ||
37 | |||
38 | var data []models.Statistic | ||
39 | |||
40 | for rows.Next() { | ||
41 | var stat models.Statistic | ||
42 | var user models.User | ||
43 | if err := rows.Scan(&stat.Date, &stat.Quantity, &user.UUID, &user.Name); err != nil { | ||
44 | c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()}) | ||
45 | return | ||
46 | } | ||
47 | stat.User = user | ||
48 | data = append(data, stat) | ||
49 | } | ||
50 | |||
51 | c.JSON(http.StatusOK, data) | ||
52 | } | ||
53 | |||
54 | func PostNewStatistic(c *gin.Context) { | ||
55 | var stat models.StatisticPost | ||
56 | |||
57 | if err := c.BindJSON(&stat); err != nil { | ||
58 | c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()}) | ||
59 | return | ||
60 | } | ||
61 | |||
62 | db := database.EstablishDBConnection() | ||
63 | defer func(db *sql.DB) { | ||
64 | err := db.Close() | ||
65 | if err != nil { | ||
66 | c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()}) | ||
67 | return | ||
68 | } | ||
69 | }(db) | ||
70 | |||
71 | result, err := db.Exec("INSERT INTO statistics (date, user_id, quantity) values (?, ?, ?)", stat.Date, stat.UserID, stat.Quantity) | ||
72 | |||
73 | if err != nil { | ||
74 | c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()}) | ||
75 | } | ||
76 | |||
77 | id, err := result.LastInsertId() | ||
78 | if err != nil { | ||
79 | c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()}) | ||
80 | } | ||
81 | |||
82 | c.JSON(http.StatusCreated, gin.H{"status": "created", "id": id}) | ||
83 | } | ||
84 | |||
85 | func GetWeeklyStatistics(c *gin.Context) { | ||
86 | db := database.EstablishDBConnection() | ||
87 | defer func(db *sql.DB) { | ||
88 | err := db.Close() | ||
89 | if err != nil { | ||
90 | c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()}) | ||
91 | return | ||
92 | } | ||
93 | }(db) | ||
94 | |||
95 | rows, err := db.Query("SELECT date, total FROM `WeeklyStatisticsView`") | ||
96 | if err != nil { | ||
97 | c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()}) | ||
98 | return | ||
99 | } | ||
100 | defer func(rows *sql.Rows) { | ||
101 | err := rows.Close() | ||
102 | if err != nil { | ||
103 | c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()}) | ||
104 | return | ||
105 | } | ||
106 | }(rows) | ||
107 | |||
108 | var data []models.WeeklyStatistic | ||
109 | for rows.Next() { | ||
110 | var weeklyStat models.WeeklyStatistic | ||
111 | if err := rows.Scan(&weeklyStat.Date, &weeklyStat.Total); err != nil { | ||
112 | c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()}) | ||
113 | } | ||
114 | data = append(data, weeklyStat) | ||
115 | } | ||
116 | |||
117 | c.JSON(http.StatusOK, data) | ||
118 | } | ||
119 | |||
120 | func GetDailyUserStatistics(c *gin.Context) { | ||
121 | db := database.EstablishDBConnection() | ||
122 | defer func(db *sql.DB) { | ||
123 | err := db.Close() | ||
124 | if err != nil { | ||
125 | c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()}) | ||
126 | return | ||
127 | } | ||
128 | }(db) | ||
129 | |||
130 | rows, err := db.Query("SELECT name, total FROM DailyUserStatistics") | ||
131 | |||
132 | if err != nil { | ||
133 | c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()}) | ||
134 | return | ||
135 | } | ||
136 | defer func(rows *sql.Rows) { | ||
137 | err := rows.Close() | ||
138 | if err != nil { | ||
139 | c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()}) | ||
140 | return | ||
141 | } | ||
142 | }(rows) | ||
143 | |||
144 | var data []models.DailyUserTotals | ||
145 | for rows.Next() { | ||
146 | var stat models.DailyUserTotals | ||
147 | if err := rows.Scan(&stat.Name, &stat.Total); err != nil { | ||
148 | c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()}) | ||
149 | return | ||
150 | } | ||
151 | data = append(data, stat) | ||
152 | } | ||
153 | |||
154 | c.JSON(http.StatusOK, data) | ||
155 | |||
156 | } | ||
157 | |||
158 | func GetUserStatistics(c *gin.Context) { | ||
159 | c.JSON(http.StatusOK, gin.H{"status": "ok", "uuid": c.Param("uuid")}) | ||
160 | } | ||
161 | |||
162 | func UpdateUserStatistic(c *gin.Context) { | ||
163 | c.JSON(http.StatusNoContent, gin.H{"status": "No Content"}) | ||
164 | } | ||
165 | |||
166 | func DeleteUserStatistic(c *gin.Context) { | ||
167 | c.JSON(http.StatusNoContent, gin.H{"status": "No Content"}) | ||
168 | } | ||