Fix: INF-254 retry method defined on the higher level makes primary mapper defined...

[pti/o2.git] / o2ims / adapter / orm.py

diff --git a/o2ims/adapter/orm.py b/o2ims/adapter/orm.py
index 6c4a678..43b48b0 100644 (file)
--- a/o2ims/adapter/orm.py
+++ b/o2ims/adapter/orm.py
@@ -13,6 +13,7 @@
 #  limitations under the License.\r
 \r
 # from typing_extensions import Required\r
+from retry import retry\r
 from sqlalchemy import (\r
     Table,\r
     MetaData,\r
@@ -27,6 +28,7 @@ from sqlalchemy import (
     # Boolean,\r
     # engine,\r
     # event,\r
+    exc,\r
 )\r
 \r
 from sqlalchemy.orm import mapper, relationship\r
@@ -161,6 +163,13 @@ configuration = Table(
 )\r
 \r
 \r
+@retry((exc.IntegrityError), tries=3, delay=2)\r
+def wait_for_metadata_ready(engine):\r
+    # wait for mapper ready\r
+    metadata.create_all(engine, checkfirst=True)\r
+    logger.info("metadata is ready")\r
+\r
+\r
 def start_o2ims_mappers(engine=None):\r
     logger.info("Starting O2 IMS mappers")\r
 \r
@@ -187,4 +196,4 @@ def start_o2ims_mappers(engine=None):
     mapper(confModel.Configuration, configuration)\r
 \r
     if engine is not None:\r
-        metadata.create_all(engine, checkfirst=True)\r
+        wait_for_metadata_ready(engine)\r