feat: adding impala random function (#483)

* adding impala RAND function * docs yo * Update README.md accepting changes Co-authored-by: Neha Nene <[email protected]> Co-authored-by: Neha Nene <[email protected]>
GoogleCloudPlatform · May 23, 2022 · 93d2072 · 93d2072
1 parent 1fb7e8b
commit 93d2072
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -203,6 +203,10 @@ data-validation (--verbose or -v) validate row
                         Comma-separated key value pair labels for the run.
   [--format or -fmt]    Format for stdout output. Supported formats are (text, csv, json, table).
                         Defaults to table.
+  [--use-random-row or -rr]
+                        Finds a set of random rows of the first primary key supplied.
+  [--random-row-batch-size or -rbs]
+                        Row batch size used for random row filters (default 10,000).
 ```
 
 #### Schema Validations

diff --git a/data_validation/query_builder/random_row_builder.py b/data_validation/query_builder/random_row_builder.py
@@ -20,6 +20,7 @@
 import ibis.expr.rules as rlz
 import ibis.backends.base_sqlalchemy.compiler as sql_compiler
 from ibis_bigquery import BigQueryClient
+from ibis.backends.impala.client import ImpalaClient
 from ibis.backends.pandas.client import PandasClient
 import ibis.backends.pandas.execution.util as pandas_util
 
@@ -43,6 +44,7 @@
 RANDOM_SORT_SUPPORTS = {
     PandasClient: "NA",
     BigQueryClient: "RAND()",
+    ImpalaClient: "RAND()",
 }