|
|
| from __future__ import annotations |
| import time |
| from .query_builder import build_search_query,build_autocomplete_query |
| async def search(es,config,q,filters=None,page=1,size=10,com_facets=True): |
| body=build_search_query(q,filters,page,size,com_facets,config); start=time.monotonic() |
| resp=await es.search(index=config.es_index,body=body) |
| return {"raw":resp,"took_ms":int((time.monotonic()-start)*1000)} |
| async def autocomplete(es,config,q,size=8): |
| resp=await es.search(index=config.es_index,body=build_autocomplete_query(q,size,config)) |
| seen=set(); r=[] |
| for hit in resp["hits"]["hits"]: |
| for f in config.suggest_fields: |
| val=hit["_source"].get(f) |
| if val and val not in seen: seen.add(val); r.append(val) |
| return r[:size] |
| async def get_doc(es,config,doc_id): |
| try: return (await es.get(index=config.es_index,id=doc_id))["_source"] |
| except: return None |
| async def get_stats(es,config): |
| c=await es.count(index=config.es_index); h=await es.cluster.health() |
| return {"index":config.es_index,"total_docs":c.get("count",0),"es_status":h.get("status","unknown")} |
|
|
