async def update_entry(_id: str, ctx: object, db: AioRedis):
"""
Updates a rate limiter entry.
@param ctx: (object) data to use for update
@param db: (object) db connection
"""
await asyncio.gather(
RateLimiter._set_indexes(pydash.merge(ctx, {'_id': _id}), db),
db.hmset_dict(_id, ctx)
)
async def create_rule(ctx: object, db: AioRedis):
"""
Creates a rate limiter rule.
@param ctx: (object) data to be inserted
@param db: (object) db connection
"""
ctx['_id'] = str(bson.ObjectId())
await asyncio.gather(
RateLimiter._set_indexes(ctx, db),
db.hmset_dict(ctx['_id'], ctx),
db.sadd(rules_set, ctx['_id']),
)
async def create_entry(ctx, db: AioRedis):
"""
Creates a rate limiter entry.
@param ctx: (object) data to be inserted
@param db: (object) db connection
"""
ctx['_id'] = str(bson.ObjectId())
await asyncio.gather(
RateLimiter._set_indexes(ctx, db),
db.hmset_dict(ctx['_id'], ctx),
db.sadd(entry_set, ctx['_id']),
db.expire(ctx['_id'], int(ctx['timeout']))
)
async def create(ctx: object, endpoint_cacher_db: AioRedis, service_db):
"""
creates an endpoint cache
@param ctx: (object) data to be inserted
@param endpoint_cacher_db: (object) db connection
@param service_db: (object) db connection
"""
ctx['_id'] = str(bson.ObjectId())
if 'service_id' in ctx:
await Service.check_exists(ctx['service_id'], service_db)
if 'response_codes' in ctx:
response_codes = ctx['response_codes']
response_codes_id = str(bson.ObjectId())
for response_code in response_codes:
await endpoint_cacher_db.sadd(response_codes_id, response_code)
ctx['response_codes'] = response_codes_id
await asyncio.gather(
EndpointCacher._set_indexes(ctx, endpoint_cacher_db),
endpoint_cacher_db.hmset_dict(ctx['_id'], ctx),
endpoint_cacher_db.sadd(endpoint_cache_set, ctx['_id']),
)
