create a method for execution changes

voltrondata · Sep 20, 2024 · 6f6ad9d · 6f6ad9d
1 parent b4927d2
commit 6f6ad9d
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/src/gateway/server.py b/src/gateway/server.py
@@ -198,6 +198,7 @@ def reset(self):
         self.reattach_requests = 0
         self.release_requests = 0
         self.database_resets = 0
+        self.engine_changes = 0
 
         self.requests = []
         self.plans = []
@@ -234,6 +235,11 @@ def _ReinitializeExecution(self) -> None:
         self._sql_backend.reset_connection()
         return None
 
+    def _ChangeExecution(self) -> None:
+        self._statistics.engine_changes += 1
+        self._backend = None
+        self._sql_backend = None
+
     def ExecutePlan(
             self, request: pb2.ExecutePlanRequest, context: grpc.RpcContext
     ) -> Generator[pb2.ExecutePlanResponse, None, None]:
@@ -401,8 +407,7 @@ def Config(self, request, context):
                             case _:
                                 raise ValueError(f"Unknown backend: {pair.value}")
                         if need_reset:
-                            self._backend = None
-                            self._sql_backend = None
+                            self._ChangeExecution()
                     elif pair.key == "spark-substrait-gateway.reset_statistics":
                         self._statistics.reset()
                 response.pairs.extend(request.operation.set.pairs)