Change system for hanayo requests
This commit is contained in:
@@ -69,7 +69,7 @@ func initialCaretaker(c *gin.Context, f func(md common.MethodData) common.CodeMe
|
||||
}
|
||||
|
||||
// requests from hanayo should not be rate limited.
|
||||
if ip != "127.0.0.1" || c.Request.UserAgent() != "hanayo" {
|
||||
if !(c.Request.Header.Get("H-Key") == cf.HanayoKey && c.Request.UserAgent() == "hanayo") {
|
||||
perUserRequestLimiter(md.ID(), c.ClientIP())
|
||||
}
|
||||
|
||||
|
@@ -13,11 +13,15 @@ import (
|
||||
"github.com/gin-gonic/gin"
|
||||
)
|
||||
|
||||
var db *sql.DB
|
||||
var (
|
||||
db *sql.DB
|
||||
cf common.Conf
|
||||
)
|
||||
|
||||
// Start begins taking HTTP connections.
|
||||
func Start(conf common.Conf, dbO *sql.DB) *gin.Engine {
|
||||
db = dbO
|
||||
cf = conf
|
||||
|
||||
setUpLimiter()
|
||||
|
||||
|
Reference in New Issue
Block a user