Switch to SSDB for message queueing

This commit is contained in:
2022-10-21 07:20:30 +01:00
parent 8c596ec516
commit e32b330ef4
4 changed files with 8 additions and 18 deletions

6
db.py
View File

@@ -44,8 +44,6 @@ KEYNAME = "queue"
async def store_kafka_batch(data):
print("FAKE STORE KAFKA BATCH")
return
# log.debug(f"Storing Kafka batch of {len(data)} messages")
producer = AIOKafkaProducer(bootstrap_servers="kafka:9092")
await producer.start()
@@ -116,7 +114,7 @@ async def queue_message(msg):
Queue a message on the Redis buffer.
"""
message = orjson.dumps(msg)
await ar.zadd(KEYNAME, message)
await ar.lpush(KEYNAME, message)
async def queue_message_bulk(data):
@@ -125,4 +123,4 @@ async def queue_message_bulk(data):
"""
for msg in data:
message = orjson.dumps(msg)
await ar.zadd(KEYNAME, message)
await ar.lpush(KEYNAME, message)