Queries: Added Param mechanism, to help speed up query construction.

Author: erezsh

data_diff/queries/ast_classes.py

@@ -6,7 +6,7 @@
 
 from data_diff.utils import ArithString, join_iter
 
-from .compiler import Compilable, Compiler
+from .compiler import Compilable, Compiler, cv_params
 from .base import SKIP, CompileError, DbPath, Schema, args_as_tuple
 
 
@@ -691,3 +691,18 @@ def compile(self, c: Compiler) -> str:
 class Commit(Statement):
     def compile(self, c: Compiler) -> str:
         return "COMMIT" if not c.database.is_autocommit else SKIP
+
+@dataclass
+class Param(ExprNode, ITable):
+    """A value placeholder, to be specified at compilation time using the `cv_params` context variable."""
+
+    name: str
+
+    @property
+    def source_table(self):
+        return self
+
+    def compile(self, c: Compiler) -> str:
+        params = cv_params.get()
+        return c._compile(params[self.name])
+

data_diff/queries/compiler.py

@@ -8,10 +8,15 @@
 from data_diff.utils import ArithString
 from data_diff.databases.database_types import AbstractDialect, DbPath
 
+import contextvars
+
+cv_params = contextvars.ContextVar("params")
+
 
 @dataclass
 class Compiler:
     database: AbstractDialect
+    params: dict = {}
     in_select: bool = False  # Compilation runtime flag
     in_join: bool = False  # Compilation runtime flag
 
@@ -21,7 +26,10 @@ class Compiler:
 
     _counter: List = [0]
 
-    def compile(self, elem) -> str:
+    def compile(self, elem, params=None) -> str:
+        if params:
+            cv_params.set(params)
+
         res = self._compile(elem)
         if self.root and self._subqueries:
             subq = ", ".join(f"\n  {k} AS ({v})" for k, v in self._subqueries.items())