feat(queue): integrate redis queue and 2-minute paced ai consumer worker
This commit is contained in:
@@ -0,0 +1,42 @@
|
||||
import os
|
||||
import logging
|
||||
from typing import Optional
|
||||
import redis.asyncio as redis
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
REDIS_URL = os.getenv("REDIS_URL", "redis://copykar_redis:6379/0" if os.path.exists("/app") else "redis://localhost:6379/0")
|
||||
|
||||
class RedisQueue:
|
||||
def __init__(self, redis_url: Optional[str] = None, queue_key: str = "copykar:queue:incoming"):
|
||||
self.redis_url = redis_url or REDIS_URL
|
||||
self.queue_key = queue_key
|
||||
self.client: Optional[redis.Redis] = None
|
||||
|
||||
async def connect(self):
|
||||
if not self.client:
|
||||
self.client = redis.from_url(self.redis_url, decode_responses=True)
|
||||
await self.client.ping()
|
||||
logger.info(f"Connected to Redis at {self.redis_url}")
|
||||
|
||||
async def push(self, post_id: int):
|
||||
if not self.client:
|
||||
await self.connect()
|
||||
await self.client.rpush(self.queue_key, str(post_id))
|
||||
logger.info(f"Enqueued post ID {post_id} to Redis queue [{self.queue_key}]")
|
||||
|
||||
async def pop(self) -> Optional[int]:
|
||||
if not self.client:
|
||||
await self.connect()
|
||||
val = await self.client.lpop(self.queue_key)
|
||||
return int(val) if val else None
|
||||
|
||||
async def qsize(self) -> int:
|
||||
if not self.client:
|
||||
await self.connect()
|
||||
return await self.client.llen(self.queue_key)
|
||||
|
||||
async def close(self):
|
||||
if self.client:
|
||||
await self.client.aclose()
|
||||
logger.info("Redis connection closed.")
|
||||
Reference in New Issue
Block a user