scout_ante/dashboard-v2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

heavier ratelimiting

Browse Source
This commit is contained in:
Dot-Rar 2020-05-29 16:57:40 +01:00
parent bdaae58c71
commit e2e107d8c9
1 changed files with 9 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
app/http/server.go
View File

@ -38,7 +38,7 @@ func StartServer() {
router.Use(static.Serve("/assets/", static.LocalFile("./public/static", false)))
router.Use(gin.Recovery())
router.Use(createLimiter())
router.Use(createLimiter(600, time.Minute * 10))
// Register templates
router.HTMLRender = createRenderer()
@ -111,8 +111,9 @@ func StartServer() {
userGroup.GET("/guilds", api.GetGuilds)
userGroup.GET("/whitelabel", api.WhitelabelGet)
userGroup.POST("/whitelabel", api.WhitelabelPost)
userGroup.POST("/whitelabel/status", api.WhitelabelStatusPost)
userGroup.Group("/").Use(createLimiter(10, time.Minute)).POST("/whitelabel", api.WhitelabelPost)
userGroup.Group("/").Use(createLimiter(1, time.Second * 5)).POST("/whitelabel/status", api.WhitelabelStatusPost)
}
if err := router.Run(config.Conf.Server.Host); err != nil {
@ -159,11 +160,11 @@ func addManageTemplate(renderer multitemplate.Renderer, name string) multitempla
return renderer
}
func createLimiter() func(*gin.Context) {
func createLimiter(limit int, period time.Duration) func(*gin.Context) {
store := memory.NewStore()
rate := limiter.Rate{
Period: time.Minute * 10,
Limit: 600,
Period: period,
Limit: limit,
}
return mgin.NewMiddleware(limiter.New(store, rate))