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

Rate Limiting - added In-Memory mode

Browse Source
This commit is contained in:
Marko Milić 2019-02-15 14:52:43 +01:00
parent 155e0333d1
commit ab57441e8b
2 changed files with 22 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/main/resources/props/sample.props.template
View File

@ -517,6 +517,8 @@ featured_apis=elasticSearchWarehouseV300
# Define how many calls per time a Customer can make
# In case isn't defined default value is "false"
# use_consumer_limits=false
# In case isn't defined default value is "false"
# use_consumer_limits_in_memory_mode=false
# redis_address=127.0.0.1
# redis_port=6379
# -----------------------------------------------------

28
src/main/scala/code/api/util/RateLimitUtil.scala
View File

@ -51,18 +51,30 @@ object RateLimitPeriod extends Enumeration {
object RateLimitUtil extends MdcLoggable {
val useConsumerLimits = APIUtil.getPropsAsBoolValue("use_consumer_limits", false)
val inMemoryMode = APIUtil.getPropsAsBoolValue("use_consumer_limits_in_memory_mode", false)
lazy val jedis = Props.mode match {
case Props.RunModes.Test =>
import ai.grakn.redismock.RedisServer
import redis.clients.jedis.Jedis
val server = RedisServer.newRedisServer // bind to a random port
server.start()
new Jedis(server.getHost, server.getBindPort)
startMockedRedis(mode="Test")
case _ =>
val port = APIUtil.getPropsAsIntValue("redis_port", 6379)
val url = APIUtil.getPropsValue("redis_address", "127.0.0.1")
new Jedis(url, port)
if(inMemoryMode == true) {
startMockedRedis(mode="In-Memory")
} else {
val port = APIUtil.getPropsAsIntValue("redis_port", 6379)
val url = APIUtil.getPropsValue("redis_address", "127.0.0.1")
new Jedis(url, port)
}
}
private def startMockedRedis(mode: String): Jedis = {
import ai.grakn.redismock.RedisServer
import redis.clients.jedis.Jedis
val server = RedisServer.newRedisServer // bind to a random port
server.start()
logger.info(msg = "-------------| Mocked Redis instance has been run in " + mode + " mode")
logger.info(msg = "-------------| at host: " + server.getHost)
logger.info(msg = "-------------| at port: " + server.getBindPort)
new Jedis(server.getHost, server.getBindPort)
}
def isRedisAvailable() = {