oarepo-runtime 1.5.64__py3-none-any.whl → 1.5.65__py3-none-any.whl

oarepo_runtime/info/check.py

@@ -0,0 +1,96 @@
+import json, random
+from flask import Response, Blueprint
+from flask import current_app as flask_current_app
+from invenio_db import db # sql alchemy session
+from invenio_cache import current_cache # redis
+import redis
+from invenio_search import current_search_client # opensearch
+from celery import current_app as celery_current_app, shared_task
+
+
+def check_db_connection():
+    if not has_database_connection():
+        return "DB connection failed"
+    else:
+        return "ok"
+
+def has_database_connection():
+    try:
+        db.session.begin()
+        db.session.execute("select * from alembic_version").scalar()
+        return True
+    except:
+        return False
+    finally:
+        db.session.rollback()
+
+def check_cache_connection():
+    try:
+        rand = random.randint(1, 1000)
+        if current_cache.set('mykey', str(rand)):
+            val = current_cache.get('mykey')
+            if val == str(rand):
+                return "ok"
+            else:
+                return "Cache returned unexpected value."
+        else:
+            return "Failed to set cache key."
+    except redis.exceptions.ConnectionError as e:
+        if isinstance(e.__cause__, ConnectionRefusedError):
+            return "Cache connection error"
+        return "Cache exception"
+    except Exception as other:
+        return str(other)
+
+
+
+#  https://github.com/celery/celery/issues/4283
+def check_message_queue():
+    from kombu.exceptions import OperationalError
+    try:
+        #celery_current_app.autodiscover_tasks(['oarepo_runtime.info'])
+        ping_task = celery_current_app.send_task('ping')
+        if ping_task.get(timeout=2) == 'pong':
+            return "ok"
+    except OperationalError as e:
+        if isinstance(e.__cause__, ConnectionRefusedError):
+            return "RabbitMQ connection refused"
+        return "Other RabbitMQ-related error"
+    except Exception as other:
+        return str(other)
+
+
+def check_opensearch():
+    try:
+        is_available = current_search_client.ping()
+        if is_available:
+            return "ok"
+        else:
+            return "OpenSearch is not available"
+    except Exception as e:
+        return str(e)
+
+
+blueprint = Blueprint("repository-check", __name__, url_prefix="/.well-known")
+
+@blueprint.route("/check")
+def check_connections() -> Response:
+    """
+    Function to check all necessary connections (database, redis, rabbitmq, opensearch)
+    Return Response 200 if everything is OK or Response 500 with json and all failed tests
+    """
+    check_list = {}
+    with flask_current_app.app_context():
+        check_list['db'] = check_db_connection()
+        check_list['cache'] = check_cache_connection()
+        check_list['mq'] = check_message_queue()
+        check_list['opensearch'] = check_opensearch()
+
+    failed_checks = {key: value for key, value in check_list.items() if value != 'ok'}
+
+    if failed_checks:
+        return Response(json.dumps(failed_checks), status=500, content_type='application/json')
+
+    return Response(json.dumps(check_list), status=200, content_type='application/json') # good
+
+

oarepo_runtime/tasks.py

@@ -0,0 +1,6 @@
+from celery import shared_task
+
+# TODO priority if something big is in queue already
+@shared_task(name='ping')
+def ping():
+    return 'pong'

{oarepo_runtime-1.5.64.dist-info → oarepo_runtime-1.5.65.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: oarepo-runtime
-Version: 1.5.64
+Version: 1.5.65
 Summary: A set of runtime extensions of Invenio repository
 Description-Content-Type: text/markdown
 License-File: LICENSE

{oarepo_runtime-1.5.64.dist-info → oarepo_runtime-1.5.65.dist-info}/RECORD

@@ -3,6 +3,7 @@ oarepo_runtime/ext.py,sha256=zatf-tujtuP1Flo34p_5GJ38yKxzkAz0ENBjljapPlA,2373
 oarepo_runtime/ext_config.py,sha256=NiXqgag3QJABf06r5NiDYjd9AvoYoRY9UWiplo8Vcg4,2056
 oarepo_runtime/profile.py,sha256=QzrQoZncjoN74ZZnpkEKakNk08KCzBU7m6y42RN8AMY,1637
 oarepo_runtime/proxies.py,sha256=NN_WNj1xuKc-OveoZmzvTFlUonNjSmLIGsv_JUcHGls,285
+oarepo_runtime/tasks.py,sha256=AciXN1fUq6tzfzNSICh1S6XoHGWiuH76bUqXyzTsOPU,140
 oarepo_runtime/uow.py,sha256=iyF3R2oCPSVUu38GXoxZallgRD-619q1fWTb8sSaeyQ,4412
 oarepo_runtime/cli/__init__.py,sha256=daAODgUAB9Nb0O0Kg8vSgJIPKJm92EHMTRuoKwxBCug,373
 oarepo_runtime/cli/assets.py,sha256=Yrg9O2dXmrOGH87_3SNxmgrgyt2w4ZzkvQQJVTmLZjk,3634
@@ -41,6 +42,7 @@ oarepo_runtime/datastreams/writers/validation_errors.py,sha256=wOCXdniR6so_4Expd
 oarepo_runtime/datastreams/writers/yaml.py,sha256=XchUJHQ58E2Mfgs8elImXbL38jFtI8Hfoye6yaR0gKI,1482
 oarepo_runtime/i18n/__init__.py,sha256=h0knW_HwiyIt5TBHfdGqN7_BBYfpz1Fw6zhVy0C28fM,111
 oarepo_runtime/info/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+oarepo_runtime/info/check.py,sha256=WWAqMIBmc7veQKflRBe97_PoDsvJ1DrDrJuTDNeGDaI,3076
 oarepo_runtime/info/views.py,sha256=UkLGzf6Vo8GLutfzJW3IAyq9sdPexICUFAvz903autY,11829
 oarepo_runtime/records/__init__.py,sha256=3vzRsAPxl4d5QOnGyls-vUg4E6PunmR4ACObtacMAIQ,1038
 oarepo_runtime/records/dumpers/__init__.py,sha256=OmzNhLdMNKibmCksnj9eTX9xPBG30dziiK3j3bAAp3k,233
@@ -119,9 +121,9 @@ oarepo_runtime/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hS
 oarepo_runtime/utils/functools.py,sha256=gKS9YZtlIYcDvdNA9cmYO00yjiXBYV1jg8VpcRUyQyg,1324
 oarepo_runtime/utils/path.py,sha256=V1NVyk3m12_YLbj7QHYvUpE1wScO78bYsX1LOLeXDkI,3108
 tests/pkg_data/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-oarepo_runtime-1.5.64.dist-info/LICENSE,sha256=h2uWz0OaB3EN-J1ImdGJZzc7yvfQjvHVYdUhQ-H7ypY,1064
-oarepo_runtime-1.5.64.dist-info/METADATA,sha256=PUOOR2X2Q4nhO6-QdV8-O4b0R7SvNP06YXPP9aMfZdY,4720
-oarepo_runtime-1.5.64.dist-info/WHEEL,sha256=R0nc6qTxuoLk7ShA2_Y-UWkN8ZdfDBG2B6Eqpz2WXbs,91
-oarepo_runtime-1.5.64.dist-info/entry_points.txt,sha256=QrlXAKuPDVBinaSh_v3yO9_Nb9ZNmJCJ0VFcCW-z0Jg,327
-oarepo_runtime-1.5.64.dist-info/top_level.txt,sha256=bHhlkT1_RQC4IkfTQCqA3iN4KCB6cSFQlsXpQMSP-bE,21
-oarepo_runtime-1.5.64.dist-info/RECORD,,
+oarepo_runtime-1.5.65.dist-info/LICENSE,sha256=h2uWz0OaB3EN-J1ImdGJZzc7yvfQjvHVYdUhQ-H7ypY,1064
+oarepo_runtime-1.5.65.dist-info/METADATA,sha256=ZgBG9WKIQEODgFiYua5ZpiwmylRyGKmT2tIXq7mQDGo,4720
+oarepo_runtime-1.5.65.dist-info/WHEEL,sha256=R0nc6qTxuoLk7ShA2_Y-UWkN8ZdfDBG2B6Eqpz2WXbs,91
+oarepo_runtime-1.5.65.dist-info/entry_points.txt,sha256=0cschM0RHc6UJ1uudhu4EP0hrVStPGpgMO-XEDGRtY4,430
+oarepo_runtime-1.5.65.dist-info/top_level.txt,sha256=bHhlkT1_RQC4IkfTQCqA3iN4KCB6cSFQlsXpQMSP-bE,21
+oarepo_runtime-1.5.65.dist-info/RECORD,,

{oarepo_runtime-1.5.64.dist-info → oarepo_runtime-1.5.65.dist-info}/entry_points.txt

@@ -5,7 +5,9 @@ oarepo_runtime = oarepo_runtime.ext:OARepoRuntime
 oarepo_runtime = oarepo_runtime.ext:OARepoRuntime
 
 [invenio_base.blueprints]
+oarepo_runtime_check = oarepo_runtime.info.check:blueprint
 oarepo_runtime_info = oarepo_runtime.info.views:create_wellknown_blueprint
 
 [invenio_celery.tasks]
+oarepo_runtime_check = oarepo_runtime.tasks
 oarepo_runtime_datastreams = oarepo_runtime.datastreams