b7e09b55-9333-4c49-b273-87a69e5c463f

# coding=utf-8
# Copyright 2018-2023 EvaDB
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
from typing import Iterator

from evadb.database import EvaDBDatabase
from evadb.executor.abstract_executor import AbstractExecutor
from evadb.models.storage.batch import Batch
from evadb.plan_nodes.function_scan_plan import FunctionScanPlan


class FunctionScanExecutor(AbstractExecutor):
    """
    Executes functional expression which yields a table of rows
    Arguments:
        node (AbstractPlan): FunctionScanPlan

    """

    def __init__(self, db: EvaDBDatabase, node: FunctionScanPlan):
        super().__init__(db, node)
        self.func_expr = node.func_expr
        self.do_unnest = node.do_unnest

    def exec(self, *args, **kwargs) -> Iterator[Batch]:
        assert (
            "lateral_input" in kwargs
        ), "Key lateral_input not passed to the FunctionScan"
        lateral_input = kwargs.get("lateral_input")
        if not lateral_input.empty():
            res = self.func_expr.evaluate(lateral_input)

            # persist stats of function expression
            if self.func_expr.function_obj and self.func_expr._stats:
                function_id = self.func_expr.function_obj.row_id
                self.catalog().upsert_function_cost_catalog_entry(
                    function_id,
                    self.func_expr.function_obj.name,
                    self.func_expr._stats.prev_cost,
                )

            if not res.empty():
                if self.do_unnest:
                    res.unnest(res.columns)

                yield res

1	# coding=utf-8
2	# Copyright 2018-2023 EvaDB
3	#
4	# Licensed under the Apache License, Version 2.0 (the "License");
5	# you may not use this file except in compliance with the License.
6	# You may obtain a copy of the License at
7	#
8	# http://www.apache.org/licenses/LICENSE-2.0
9	#
10	# Unless required by applicable law or agreed to in writing, software
11	# distributed under the License is distributed on an "AS IS" BASIS,
12	# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13	# See the License for the specific language governing permissions and
14	# limitations under the License.
15	from typing import Iterator	1✔
16
17	from evadb.database import EvaDBDatabase	1✔
18	from evadb.executor.abstract_executor import AbstractExecutor	1✔
19	from evadb.models.storage.batch import Batch	1✔
20	from evadb.plan_nodes.function_scan_plan import FunctionScanPlan	1✔
21
22
23	class FunctionScanExecutor(AbstractExecutor):	1✔
24	"""
25	Executes functional expression which yields a table of rows
26	Arguments:
27	node (AbstractPlan): FunctionScanPlan
28
29	"""
30
31	def __init__(self, db: EvaDBDatabase, node: FunctionScanPlan):	1✔
32	super().__init__(db, node)	1✔
33	self.func_expr = node.func_expr	1✔
34	self.do_unnest = node.do_unnest	1✔
35
36	def exec(self, args, *kwargs) -> Iterator[Batch]:	1✔
37	assert (	×
38	"lateral_input" in kwargs
39	), "Key lateral_input not passed to the FunctionScan"
40	lateral_input = kwargs.get("lateral_input")	×
41	if not lateral_input.empty():	×
42	res = self.func_expr.evaluate(lateral_input)	×
43
44	# persist stats of function expression
45	if self.func_expr.function_obj and self.func_expr._stats:	×
46	function_id = self.func_expr.function_obj.row_id	×
47	self.catalog().upsert_function_cost_catalog_entry(	×
48	function_id,
49	self.func_expr.function_obj.name,
50	self.func_expr._stats.prev_cost,
51	)
52
53	if not res.empty():	×
54	if self.do_unnest:	×
55	res.unnest(res.columns)	×
56
57	yield res	×

georgia-tech-db / eva / b7e09b55-9333-4c49-b273-87a69e5c463f

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous