support select star

bigtableql/__init__.py

@@ -3,3 +3,5 @@
 RESERVED_ROWKEY = "_row_key"
 RESERVED_TIMESTAMP = "_timestamp"
 DEFAULT_SEPARATOR = "#"
+
+SELECT_STAR = "Wildcard"

bigtableql/client.py

@@ -2,7 +2,12 @@
 import pyarrow
 from typing import List
 from bigtableql import parser, composer, scanner, executor
-from bigtableql import RESERVED_ROWKEY, RESERVED_TIMESTAMP, DEFAULT_SEPARATOR
+from bigtableql import (
+    RESERVED_ROWKEY,
+    RESERVED_TIMESTAMP,
+    DEFAULT_SEPARATOR,
+    SELECT_STAR,
+)
 
 
 class Client:
@@ -52,13 +57,15 @@ def query(self, column_family_id: str, sql: str) -> List[pyarrow.RecordBatch]:
 
         row_key_identifiers = table_catalog["row_key_identifiers"]
         non_qualifiers = set(row_key_identifiers) | {RESERVED_TIMESTAMP}
-        qualifiers = (projection | selection) - non_qualifiers
+
+        columns = table_catalog["column_families"][column_family_id]["columns"].keys()
+        if SELECT_STAR in projection:
+            qualifiers = set(columns)
+        else:
+            qualifiers = (projection | selection) - non_qualifiers
 
         for qualifier in qualifiers:
-            if (
-                qualifier
-                not in table_catalog["column_families"][column_family_id]["columns"]
-            ):
+            if qualifier not in columns:
                 raise Exception(
                     f"table {table_name}, column_family {column_family_id}: {qualifier} not found"
                 )

bigtableql/parser.py

@@ -1,6 +1,7 @@
 import sqloxide
 import functools
 from typing import Tuple
+from bigtableql import SELECT_STAR
 
 BINARY_OPERATION = "BinaryOp"
 FUNCTION_OPERATION = "Function"
@@ -38,7 +39,10 @@ def _parse_table_name(select_from) -> str:
     return select_from[0]["relation"]["Table"]["name"][0]["value"]
 
 
-def _parse_projection(select_projection) -> set:
+def _parse_projection(select_projection: list) -> set:
+    if SELECT_STAR in select_projection:
+        return {SELECT_STAR}
+
     projection = [
         [
             _parse_identifier_key(arg["Unnamed"])

tests/sql.yaml

@@ -22,6 +22,14 @@
     - {"_row_key": ["abc"]}
   compose_success:
     - "abc"
+- sql: select *, age from users where "_row_key" = 'abc'
+  parse_success:
+    - "users"
+    - ["Wildcard"]
+    - ["_row_key"]
+    - {"_row_key": ["abc"]}
+  compose_success:
+    - "abc"
 - sql: select age from users where "_row_key" = 'abc' and gender = 'M'
   parse_success:
     - "users"