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())