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.executor.executor_utils import apply_predicate
from evadb.models.storage.batch import Batch
from evadb.plan_nodes.nested_loop_join_plan import NestedLoopJoinPlan


class NestedLoopJoinExecutor(AbstractExecutor):
    def __init__(self, db: EvaDBDatabase, node: NestedLoopJoinPlan):
        super().__init__(db, node)
        self.predicate = node.join_predicate

    def exec(self, *args, **kwargs) -> Iterator[Batch]:
        outer = self.children[0]
        inner = self.children[1]
        for row1 in outer.exec(**kwargs):
            for row2 in inner.exec(**kwargs):
                result_batch = Batch.join(row1, row2)
                result_batch.reset_index()
                result_batch = apply_predicate(
                    result_batch, self.predicate, self.catalog()
                )
                if not result_batch.empty():
                    yield result_batch

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.executor.executor_utils import apply_predicate	1✔
20	from evadb.models.storage.batch import Batch	1✔
21	from evadb.plan_nodes.nested_loop_join_plan import NestedLoopJoinPlan	1✔
22
23
24	class NestedLoopJoinExecutor(AbstractExecutor):	1✔
25	def __init__(self, db: EvaDBDatabase, node: NestedLoopJoinPlan):	1✔
26	super().__init__(db, node)	1✔
27	self.predicate = node.join_predicate	1✔
28
29	def exec(self, args, *kwargs) -> Iterator[Batch]:	1✔
30	outer = self.children[0]	×
31	inner = self.children[1]	×
32	for row1 in outer.exec(**kwargs):	×
33	for row2 in inner.exec(**kwargs):	×
34	result_batch = Batch.join(row1, row2)	×
35	result_batch.reset_index()	×
36	result_batch = apply_predicate(	×
37	result_batch, self.predicate, self.catalog()
38	)
39	if not result_batch.empty():	×
40	yield result_batch	×

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