Merge pull request #7421 from opsmill/pog-warn-graph-version

Warn if the database graph version is incorrect

Author: ogenstad

backend/infrahub/database/graph.py

@@ -0,0 +1,21 @@
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+from infrahub.core.graph import GRAPH_VERSION
+from infrahub.core.initialization import get_root_node
+from infrahub.log import get_logger
+
+if TYPE_CHECKING:
+    from infrahub.database import InfrahubDatabase
+
+
+log = get_logger()
+
+
+async def validate_graph_version(db: InfrahubDatabase) -> None:
+    root = await get_root_node(db=db)
+    if root.graph_version != GRAPH_VERSION:
+        log.warning(
+            f"Expected database graph version {GRAPH_VERSION} but got {root.graph_version}, possibly 'infrahub upgrade' has not been executed"
+        )

backend/infrahub/server.py

@@ -24,6 +24,7 @@
 from infrahub.components import ComponentType
 from infrahub.constants.environment import INSTALLATION_TYPE
 from infrahub.core.initialization import initialization
+from infrahub.database.graph import validate_graph_version
 from infrahub.dependencies.registry import build_component_registry
 from infrahub.exceptions import Error, ValidationError
 from infrahub.graphql.api.endpoints import router as graphql_router
@@ -85,6 +86,9 @@ async def app_initialization(application: FastAPI, enable_scheduler: bool = True
     async with application.state.db.start_session() as db:
         await initialization(db=db, add_database_indexes=True)
 
+    async with database.start_session() as dbs:
+        await validate_graph_version(db=dbs)
+
     # Initialize the workflow after the registry has been setup
     await service.initialize_workflow()
 

backend/infrahub/workers/infrahub_async.py

@@ -18,6 +18,7 @@
 from infrahub.components import ComponentType
 from infrahub.core import registry
 from infrahub.core.initialization import initialization
+from infrahub.database.graph import validate_graph_version
 from infrahub.dependencies.registry import build_component_registry
 from infrahub.git import initialize_repositories_directory
 from infrahub.lock import initialize_lock
@@ -129,6 +130,9 @@ async def setup(
 
             await self.service.component.refresh_schema_hash()
 
+        async with self.service.database.start_session() as dbs:
+            await validate_graph_version(db=dbs)
+
         initialize_repositories_directory()
         build_component_registry()
         await self.service.scheduler.start_schedule()