artin/OBP-API
1
0
Fork 0
mirror of https://github.com/OpenBankProject/OBP-API.git synced 2026-02-06 17:37:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

bugfix/Fix consumer rate limiting state function

Browse Source
This commit is contained in:
Marko Milić 2025-09-05 08:29:51 +02:00
parent 280c3ee4df
commit f2a1eacaec
1 changed files with 9 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
obp-api/src/main/scala/code/api/util/RateLimitingUtil.scala
View File

@ -153,13 +153,14 @@ object RateLimitingUtil extends MdcLoggable {
def getInfo(consumerKey: String, period: LimitCallPeriod): ((Option[Long], Option[Long]), LimitCallPeriod) = {
val key = createUniqueKey(consumerKey, period)
val ttl = Redis.use(JedisMethod.TTL, key).get.toLong
ttl match {
case -2 =>
((None, None), period)
case _ =>
((Redis.use(JedisMethod.TTL, key).map(_.toLong), Some(ttl)), period)
}
// get TTL
val ttlOpt: Option[Long] = Redis.use(JedisMethod.TTL, key).map(_.toLong)
// get value (assuming string storage)
val valueOpt: Option[Long] = Redis.use(JedisMethod.GET, key).map(_.toLong)
((valueOpt, ttlOpt), period)
}
getInfo(consumerKey, RateLimitingPeriod.PER_SECOND) ::
@ -167,7 +168,7 @@ object RateLimitingUtil extends MdcLoggable {
getInfo(consumerKey, RateLimitingPeriod.PER_HOUR) ::
getInfo(consumerKey, RateLimitingPeriod.PER_DAY) ::
getInfo(consumerKey, RateLimitingPeriod.PER_WEEK) ::
getInfo(consumerKey, RateLimitingPeriod.PER_MONTH) ::
getInfo(consumerKey, RateLimitingPeriod.PER_MONTH) ::
Nil
}