Skip to content

Commit c789251

Browse files
committed
added changes
1 parent 0fb2c85 commit c789251

File tree

4 files changed

+20
-8
lines changed

4 files changed

+20
-8
lines changed

leetstudy_backend/cmd/cache/rate_limiter.go

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -41,12 +41,13 @@ func RateLimitByIP(limit int, window time.Duration) echo.MiddlewareFunc {
4141
func RateLimitByUser(limit int, window time.Duration) echo.MiddlewareFunc {
4242
return func(next echo.HandlerFunc) echo.HandlerFunc {
4343
return func(c echo.Context) error {
44-
userID := c.Request().Header.Get("X-User-ID") // Or get from JWT context
45-
46-
if userID == "" {
44+
userID:= c.Get("user_id")
45+
if userID == nil {
4746
return c.String(http.StatusUnauthorized, "Missing user ID")
4847
}
4948

49+
50+
5051
db := database.GetDB()
5152
var isPremium bool
5253
err := db.QueryRow("SELECT is_premium FROM users WHERE id = $1", userID).Scan(&isPremium)

leetstudy_backend/main.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -88,7 +88,7 @@ func main() {
8888
//c.AddFunc("0 8 * * *", cronjobs.SendDailyDigests) // daily 8AM
8989
//c.AddFunc("0 9 * * 1", cronjobs.SendWeeklyDigests) // weekly (Monday) 9AM
9090
c.AddFunc("* * * * *", cronjobs.SendDailyDigests)
91-
//c.AddFunc("* * * * *", cronjobs.SendWeeklyDigests)
91+
c.AddFunc("* * * * *", cronjobs.SendWeeklyDigests)
9292
c.Start()
9393

9494
// testing
@@ -164,7 +164,7 @@ func main() {
164164
e.GET("/sessions/:sessionId/messages", handlers.GetChatMessagesHandler)
165165
e.DELETE("/sessions/:sessionId", handlers.DeleteChatSessionHandler)
166166
// Feedback
167-
e.POST("/feedback", handlers.HandleFeedback, cache.RateLimitByIP(2, time.Hour));
167+
e.POST("/feedback", handlers.HandleFeedback, handlers.Authorize, cache.RateLimitByUser(2, time.Minute));
168168

169169
//Section for ai hints //TODO
170170
// add rate limiting here

leetstudy_backend/readme.md

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -81,4 +81,9 @@ brew services stop redis
8181
8282
```
8383
stripe listen --forward-to http://localhost:8080/stripe-webhook
84-
```
84+
```
85+
86+
```
87+
redis-cli FLUSHDB
88+
```
89+

log_filtered.csv

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,10 @@
1-
04/20,338c6d5d,updated to add redis
2-
04/20,2c2c5bfc,updated front end to display messages
31
04/20,b738ce2e,update read me and added cache support for more routes
42
04/20,930cb39a,added caching support for reset password
3+
04/24,2e9c09c9,added problem completion
4+
04/24,6600141b,Delete leetstudy_backend/leetstudy_backend
5+
04/24,6fe46440,updated
6+
04/24,aee0c8a3,merged
7+
04/24,1f931174,fixed merge conflicts
8+
04/24,c937c78d,added email digest option
9+
04/24,24d6ce42,added email digest option
10+
04/25,0fb2c85d,finished rate limiting for prem

0 commit comments

Comments
 (0)