georgia-tech-db / eva / #725

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

52.63
/eva/executor/nested_loop_join_executor.py
1 
# coding=utf-8
2 
# Copyright 2018-2022 EVA
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 eva.executor.abstract_executor import AbstractExecutor




1✔






18



from eva.executor.executor_utils import apply_predicate




1✔





19



from eva.models.storage.batch import Batch




1✔





20



from eva.plan_nodes.nested_loop_join_plan import NestedLoopJoinPlan




1✔





21












22












23



class NestedLoopJoinExecutor(AbstractExecutor):




1✔





24



    def __init__(self, node: NestedLoopJoinPlan):




1✔





25



        super().__init__(node)




1✔





26



        self.predicate = node.join_predicate




1✔





27












28



    def exec(self, *args, **kwargs) -> Iterator[Batch]:




1✔





29



        outer = self.children[0]




×







30



        inner = self.children[1]




×








31



        for row1 in outer.exec():




×








32



            for row2 in inner.exec():




×








33



                result_batch = Batch.join(row1, row2)




×








34



                result_batch.reset_index()




×








35



                result_batch = apply_predicate(result_batch, self.predicate)




×








36



                if not result_batch.empty():




×








37



                    yield result_batch




×





















    

  • 

    •