219 lines
8.8 KiB
Python
219 lines
8.8 KiB
Python
import asyncio
|
||
import json
|
||
import logging
|
||
import os
|
||
|
||
from opensearchpy import OpenSearch
|
||
|
||
from services.redis import redis
|
||
from utils.encoders import CustomJSONEncoder
|
||
|
||
# Set redis logging level to suppress DEBUG messages
|
||
logger = logging.getLogger("search")
|
||
logger.setLevel(logging.WARNING)
|
||
|
||
ELASTIC_HOST = os.environ.get("ELASTIC_HOST", "").replace("https://", "")
|
||
ELASTIC_USER = os.environ.get("ELASTIC_USER", "")
|
||
ELASTIC_PASSWORD = os.environ.get("ELASTIC_PASSWORD", "")
|
||
ELASTIC_PORT = os.environ.get("ELASTIC_PORT", 9200)
|
||
ELASTIC_URL = os.environ.get(
|
||
"ELASTIC_URL",
|
||
f"https://{ELASTIC_USER}:{ELASTIC_PASSWORD}@{ELASTIC_HOST}:{ELASTIC_PORT}",
|
||
)
|
||
REDIS_TTL = 86400 # 1 день в секундах
|
||
|
||
index_settings = {
|
||
"settings": {
|
||
"index": {"number_of_shards": 1, "auto_expand_replicas": "0-all"},
|
||
"analysis": {
|
||
"analyzer": {
|
||
"ru": {
|
||
"tokenizer": "standard",
|
||
"filter": ["lowercase", "ru_stop", "ru_stemmer"],
|
||
}
|
||
},
|
||
"filter": {
|
||
"ru_stemmer": {"type": "stemmer", "language": "russian"},
|
||
"ru_stop": {"type": "stop", "stopwords": "_russian_"},
|
||
},
|
||
},
|
||
},
|
||
"mappings": {
|
||
"properties": {
|
||
"body": {"type": "text", "analyzer": "ru"},
|
||
"title": {"type": "text", "analyzer": "ru"},
|
||
"subtitle": {"type": "text", "analyzer": "ru"},
|
||
"lead": {"type": "text", "analyzer": "ru"},
|
||
"media": {"type": "text", "analyzer": "ru"},
|
||
}
|
||
},
|
||
}
|
||
|
||
expected_mapping = index_settings["mappings"]
|
||
|
||
# Создание цикла событий
|
||
search_loop = asyncio.get_event_loop()
|
||
|
||
|
||
def get_indices_stats():
|
||
indices_stats = search_service.client.cat.indices(format="json")
|
||
for index_info in indices_stats:
|
||
index_name = index_info["index"]
|
||
if not index_name.startswith("."):
|
||
index_health = index_info["health"]
|
||
index_status = index_info["status"]
|
||
pri_shards = index_info["pri"]
|
||
rep_shards = index_info["rep"]
|
||
docs_count = index_info["docs.count"]
|
||
docs_deleted = index_info["docs.deleted"]
|
||
store_size = index_info["store.size"]
|
||
pri_store_size = index_info["pri.store.size"]
|
||
|
||
logger.info(f"Index: {index_name}")
|
||
logger.info(f"Health: {index_health}")
|
||
logger.info(f"Status: {index_status}")
|
||
logger.info(f"Primary Shards: {pri_shards}")
|
||
logger.info(f"Replica Shards: {rep_shards}")
|
||
logger.info(f"Documents Count: {docs_count}")
|
||
logger.info(f"Deleted Documents: {docs_deleted}")
|
||
logger.info(f"Store Size: {store_size}")
|
||
logger.info(f"Primary Store Size: {pri_store_size}")
|
||
|
||
|
||
class SearchService:
|
||
def __init__(self, index_name="search_index"):
|
||
logger.info("Инициализируем поиск...")
|
||
self.index_name = index_name
|
||
self.client = None
|
||
self.lock = asyncio.Lock()
|
||
|
||
# Инициализация клиента OpenSearch
|
||
if ELASTIC_HOST:
|
||
try:
|
||
self.client = OpenSearch(
|
||
hosts=[{"host": ELASTIC_HOST, "port": ELASTIC_PORT}],
|
||
http_compress=True,
|
||
http_auth=(ELASTIC_USER, ELASTIC_PASSWORD),
|
||
use_ssl=True,
|
||
verify_certs=False,
|
||
ssl_assert_hostname=False,
|
||
ssl_show_warn=False,
|
||
# ca_certs = ca_certs_path
|
||
)
|
||
logger.info("Клиент OpenSearch.org подключен")
|
||
|
||
# Создание задачи и запуск в цикле событий
|
||
search_loop.create_task(self.check_index())
|
||
except Exception as exc:
|
||
logger.error(f"Ошибка подключения к OpenSearch: {exc}")
|
||
self.client = None
|
||
else:
|
||
logger.warning("env var ELASTIC_HOST is not set")
|
||
|
||
async def info(self):
|
||
if isinstance(self.client, OpenSearch):
|
||
logger.info("Поиск подключен")
|
||
get_indices_stats()
|
||
|
||
def delete_index(self):
|
||
if self.client:
|
||
logger.warning(f"[!!!] Удаляем индекс {self.index_name}")
|
||
self.client.indices.delete(index=self.index_name, ignore_unavailable=True)
|
||
|
||
def create_index(self):
|
||
if self.client:
|
||
logger.info(f"Создается индекс: {self.index_name}")
|
||
self.client.indices.create(index=self.index_name, body=index_settings)
|
||
logger.info(f"Индекс {self.index_name} создан")
|
||
|
||
async def check_index(self):
|
||
if self.client:
|
||
logger.info(f"Проверяем индекс {self.index_name}...")
|
||
if not self.client.indices.exists(index=self.index_name):
|
||
self.create_index()
|
||
self.client.indices.put_mapping(index=self.index_name, body=expected_mapping)
|
||
else:
|
||
logger.info(f"Найден существующий индекс {self.index_name}")
|
||
# Проверка и обновление структуры индекса, если необходимо
|
||
result = self.client.indices.get_mapping(index=self.index_name)
|
||
if isinstance(result, str):
|
||
result = json.loads(result)
|
||
if isinstance(result, dict):
|
||
mapping = result.get(self.index_name, {}).get("mappings")
|
||
logger.info(f"Найдена структура индексации: {mapping['properties'].keys()}")
|
||
expected_keys = expected_mapping["properties"].keys()
|
||
if mapping and mapping["properties"].keys() != expected_keys:
|
||
logger.info(f"Ожидаемая структура индексации: {expected_mapping}")
|
||
logger.warning("[!!!] Требуется переиндексация всех данных")
|
||
self.delete_index()
|
||
self.client = None
|
||
else:
|
||
logger.error("клиент не инициализован, невозможно проверить индекс")
|
||
|
||
def index(self, shout):
|
||
if self.client:
|
||
logger.info(f"Индексируем пост {shout.id}")
|
||
index_body = {
|
||
"body": shout.body,
|
||
"title": shout.title,
|
||
"subtitle": shout.subtitle,
|
||
"lead": shout.lead,
|
||
"media": shout.media,
|
||
}
|
||
asyncio.create_task(self.perform_index(shout, index_body))
|
||
else:
|
||
logger.error("клиент не инициализован, невозможно проидексировать")
|
||
|
||
async def perform_index(self, shout, index_body):
|
||
if self.client:
|
||
try:
|
||
await asyncio.wait_for(
|
||
self.client.index(index=self.index_name, id=str(shout.id), body=index_body), timeout=40.0
|
||
)
|
||
except asyncio.TimeoutError:
|
||
logger.error(f"Indexing timeout for shout {shout.id}")
|
||
except Exception as e:
|
||
logger.error(f"Indexing error for shout {shout.id}: {e}")
|
||
|
||
async def search(self, text, limit, offset):
|
||
logger.info(f"Ищем: {text} {offset}+{limit}")
|
||
search_body = {
|
||
"query": {"multi_match": {"query": text, "fields": ["title", "lead", "subtitle", "body", "media"]}}
|
||
}
|
||
|
||
if self.client:
|
||
search_response = self.client.search(
|
||
index=self.index_name,
|
||
body=search_body,
|
||
size=limit,
|
||
from_=offset,
|
||
_source=False,
|
||
_source_excludes=["title", "body", "subtitle", "media", "lead", "_index"],
|
||
)
|
||
hits = search_response["hits"]["hits"]
|
||
results = [{"id": hit["_id"], "score": hit["_score"]} for hit in hits]
|
||
|
||
# если результаты не пустые
|
||
if results:
|
||
# Кэширование в Redis с TTL
|
||
redis_key = f"search:{text}:{offset}+{limit}"
|
||
await redis.execute(
|
||
"SETEX",
|
||
redis_key,
|
||
REDIS_TTL,
|
||
json.dumps(results, cls=CustomJSONEncoder),
|
||
)
|
||
return results
|
||
return []
|
||
|
||
|
||
search_service = SearchService()
|
||
|
||
|
||
async def search_text(text: str, limit: int = 50, offset: int = 0):
|
||
payload = []
|
||
if search_service.client:
|
||
# Использование метода search_post из OpenSearchService
|
||
payload = await search_service.search(text, limit, offset)
|
||
return payload
|