refactor: Streamline executor paths (#1708)

Author: TrevorBergeron

bigframes/core/array_value.py

@@ -172,33 +172,6 @@ def order_ambiguous(self) -> bool:
     def supports_fast_peek(self) -> bool:
         return bigframes.core.tree_properties.can_fast_peek(self.node)
 
-    def as_cached(
-        self: ArrayValue,
-        cache_table: google.cloud.bigquery.Table,
-        ordering: Optional[orderings.RowOrdering],
-    ) -> ArrayValue:
-        """
-        Replace the node with an equivalent one that references a table where the value has been materialized to.
-        """
-        table = nodes.GbqTable.from_table(cache_table)
-        source = nodes.BigqueryDataSource(
-            table, ordering=ordering, n_rows=cache_table.num_rows
-        )
-        # Assumption: GBQ cached table uses field name as bq column name
-        scan_list = nodes.ScanList(
-            tuple(
-                nodes.ScanItem(field.id, field.dtype, field.id.name)
-                for field in self.node.fields
-            )
-        )
-        node = nodes.CachedTableNode(
-            original_node=self.node,
-            source=source,
-            table_session=self.session,
-            scan_list=scan_list,
-        )
-        return ArrayValue(node)
-
     def get_column_type(self, key: str) -> bigframes.dtypes.Dtype:
         return self.schema.get_type(key)
 

bigframes/core/compile/__init__.py

@@ -13,9 +13,13 @@
 # limitations under the License.
 from __future__ import annotations
 
-from bigframes.core.compile.api import SQLCompiler, test_only_ibis_inferred_schema
+from bigframes.core.compile.api import test_only_ibis_inferred_schema
+from bigframes.core.compile.compiler import compile_sql
+from bigframes.core.compile.configs import CompileRequest, CompileResult
 
 __all__ = [
-    "SQLCompiler",
     "test_only_ibis_inferred_schema",
+    "compile_sql",
+    "CompileRequest",
+    "CompileResult",
 ]

bigframes/core/compile/api.py

@@ -13,43 +13,13 @@
 # limitations under the License.
 from __future__ import annotations
 
-from typing import Optional, Sequence, Tuple, TYPE_CHECKING
-
-import google.cloud.bigquery as bigquery
+from typing import TYPE_CHECKING
 
 from bigframes.core import rewrite
-from bigframes.core.compile import compiler, configs
+from bigframes.core.compile import compiler
 
 if TYPE_CHECKING:
     import bigframes.core.nodes
-    import bigframes.core.ordering
-
-
-class SQLCompiler:
-    def compile(
-        self,
-        node: bigframes.core.nodes.BigFrameNode,
-        *,
-        ordered: bool = True,
-        limit: Optional[int] = None,
-    ) -> str:
-        """Compile node into sql where rows are sorted with ORDER BY."""
-        request = configs.CompileRequest(node, sort_rows=ordered, peek_count=limit)
-        return compiler.compile_sql(request).sql
-
-    def compile_raw(
-        self,
-        node: bigframes.core.nodes.BigFrameNode,
-    ) -> Tuple[
-        str, Sequence[bigquery.SchemaField], bigframes.core.ordering.RowOrdering
-    ]:
-        """Compile node into sql that exposes all columns, including hidden ordering-only columns."""
-        request = configs.CompileRequest(
-            node, sort_rows=False, materialize_all_order_keys=True
-        )
-        result = compiler.compile_sql(request)
-        assert result.row_order is not None
-        return result.sql, result.sql_schema, result.row_order
 
 
 def test_only_ibis_inferred_schema(node: bigframes.core.nodes.BigFrameNode):