This commit is contained in:
parent
5aa8258f16
commit
d14f0c2f95
|
@ -16,25 +16,25 @@ class SearchService:
|
|||
async def init(session):
|
||||
async with SearchService.lock:
|
||||
logging.info("[services.search] Initializing SearchService")
|
||||
SearchService.cache = {}
|
||||
|
||||
@staticmethod
|
||||
async def search(text, limit: int = 50, offset: int = 0) -> List[Shout]:
|
||||
cached = await redis.execute("GET", text)
|
||||
if not cached:
|
||||
async def search(text: str, limit: int = 50, offset: int = 0) -> List[Shout]:
|
||||
try:
|
||||
payload = await redis.execute("GET", text)
|
||||
|
||||
if not payload:
|
||||
async with SearchService.lock:
|
||||
# Use aiohttp to send a request to ElasticSearch
|
||||
async with aiohttp.ClientSession() as session:
|
||||
search_url = f"https://search.discours.io/search?q={text}"
|
||||
try:
|
||||
async with session.get(search_url) as response:
|
||||
if response.status == 200:
|
||||
payload = await response.json()
|
||||
await redis.execute("SET", text, json.dumps(payload)) # use redis as cache
|
||||
return payload[offset : offset + limit]
|
||||
else:
|
||||
logging.error(f"[services.search] response: {response.status} {await response.text()}")
|
||||
except Exception as e:
|
||||
logging.error(f"[services.search] error: {e}")
|
||||
else:
|
||||
return json.loads(cached)[offset : offset + limit]
|
||||
logging.error(f"[services.search] Error during search: {e}")
|
||||
payload = []
|
||||
|
||||
return payload[offset : offset + limit]
|
||||
|
|
Loading…
Reference in New Issue
Block a user