yujunjun/haystack
1
0
Fork 0
mirror of https://github.com/deepset-ai/haystack.git synced 2025-09-02 04:46:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

Test ES connection only for the default user (#1028)

Browse Source
This commit is contained in:
oryx1729 2021-05-04 15:03:19 +02:00 committed by GitHub
parent c41101ff74
commit 9bec8859f2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
haystack/document_store/elasticsearch.py
View File

@ -180,11 +180,17 @@ class ElasticsearchDocumentStore(BaseDocumentStore):
scheme=scheme, ca_certs=ca_certs, verify_certs=verify_certs,
timeout=timeout)
# Test connection
# Test connection
try:
status = client.ping()
if not status:
raise ConnectionError(f"Initial connection to Elasticsearch failed. Make sure you run an Elasticsearch instance at `{hosts}` and that it has finished the initial ramp up (can take > 30s).")
# ping uses a HEAD request on the root URI. In some cases, the user might not have permissions for that,
# resulting in a HTTP Forbidden 403 response.
if username in ["", "elastic"]:
status = client.ping()
if not status:
raise ConnectionError(
f"Initial connection to Elasticsearch failed. Make sure you run an Elasticsearch instance "
f"at `{hosts}` and that it has finished the initial ramp up (can take > 30s)."
)
except Exception:
raise ConnectionError(
f"Initial connection to Elasticsearch failed. Make sure you run an Elasticsearch instance at `{hosts}` and that it has finished the initial ramp up (can take > 30s).")