Begin switching away from Redis

This commit is contained in:
2022-10-21 11:14:51 +01:00
parent 8c06ec44cb
commit 73e2d24e0a
4 changed files with 24 additions and 49 deletions

10
db.py
View File

@@ -5,7 +5,7 @@ import orjson
# Kafka
from aiokafka import AIOKafkaProducer
from redis import StrictRedis
import redis
import util
@@ -14,10 +14,10 @@ import util
log = util.get_logger("db")
# Redis (legacy)
r = StrictRedis(unix_socket_path="/var/run/redis/redis.sock", db=0)
r = redis.from_url("redis://ssdb:1289", db=0)
# AIORedis
ar = aioredis.from_url("unix:///var/run/redis/redis.sock", db=0)
ar = aioredis.from_url("redis://ssdb:1289", db=0)
TYPES_MAIN = [
"msg",
@@ -116,7 +116,7 @@ async def queue_message(msg):
Queue a message on the Redis buffer.
"""
message = orjson.dumps(msg)
await ar.sadd(KEYNAME, message)
await ar.zadd(KEYNAME, message)
async def queue_message_bulk(data):
@@ -125,4 +125,4 @@ async def queue_message_bulk(data):
"""
for msg in data:
message = orjson.dumps(msg)
await ar.sadd(KEYNAME, message)
await ar.zadd(KEYNAME, message)