Begin switching away from Redis
This commit is contained in:
10
db.py
10
db.py
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user