34cde48c-a13e-4064-b6d4-2dcd2780f450

Build # 34cde48c-a13e-4064-b6d4-2dcd2780f450

Build Type

circle-ci

Committed by jarulraj

Commit Message

checkpoint

Pull Request Pull Request #562: feat: benchmark and server tests

Run Details

26 of 26 new or added lines in 8 files covered. (100.0%)

8173 of 8721 relevant lines covered (93.72%)

0.94 hits per line

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

63.33

/eva/executor/lateral_join_executor.py

# coding=utf-8
# Copyright 2018-2022 EVA
#
# 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 Generator, Iterator

from eva.executor.abstract_executor import AbstractExecutor
from eva.executor.executor_utils import ExecutorError, apply_predicate, apply_project
from eva.models.storage.batch import Batch
from eva.plan_nodes.lateral_join_plan import LateralJoinPlan
from eva.utils.logging_manager import logger


class LateralJoinExecutor(AbstractExecutor):
    def __init__(self, node: LateralJoinPlan):
        super().__init__(node)
        self.predicate = node.join_predicate
        self.join_project = node.join_project

    def validate(self):
        pass

    def exec(self, *args, **kwargs) -> Iterator[Batch]:
        outer = self.children[0]
        inner = self.children[1]
        try:
            for outer_batch in outer.exec(**kwargs):
                for result_batch in inner.exec(lateral_input=outer_batch):
                    result_batch = Batch.join(outer_batch, result_batch)
                    result_batch.reset_index()
                    result_batch = apply_predicate(result_batch, self.predicate)
                    result_batch = apply_project(result_batch, self.join_project)
                    if not result_batch.empty():
                        yield result_batch
        except Exception as e:
            logger.error(e)
            raise ExecutorError(e)

    def __call__(self, *args, **kwargs) -> Generator[Batch, None, None]:
        yield from self.exec(*args, **kwargs)

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 Generator, Iterator	1✔
16
17	from eva.executor.abstract_executor import AbstractExecutor	1✔
18	from eva.executor.executor_utils import ExecutorError, apply_predicate, apply_project	1✔
19	from eva.models.storage.batch import Batch	1✔
20	from eva.plan_nodes.lateral_join_plan import LateralJoinPlan	1✔
21	from eva.utils.logging_manager import logger	1✔
22
23
24	class LateralJoinExecutor(AbstractExecutor):	1✔
25	def __init__(self, node: LateralJoinPlan):	1✔
26	super().__init__(node)	1✔
27	self.predicate = node.join_predicate	1✔
28	self.join_project = node.join_project	1✔
29
30	def validate(self):	1✔
31	pass	×
32
33	def exec(self, args, *kwargs) -> Iterator[Batch]:	1✔
34	outer = self.children[0]	1✔
35	inner = self.children[1]	1✔
36	try:	1✔
37	for outer_batch in outer.exec(**kwargs):	1✔
38	for result_batch in inner.exec(lateral_input=outer_batch):	1✔
39	result_batch = Batch.join(outer_batch, result_batch)	×
40	result_batch.reset_index()	×
41	result_batch = apply_predicate(result_batch, self.predicate)	×
42	result_batch = apply_project(result_batch, self.join_project)	×
43	if not result_batch.empty():	×
44	yield result_batch	×
45	except Exception as e:	×
46	logger.error(e)	×
47	raise ExecutorError(e)	×
48
49	def __call__(self, args, *kwargs) -> Generator[Batch, None, None]:	1✔
50	yield from self.exec(args, *kwargs)	×

georgia-tech-db / eva / 34cde48c-a13e-4064-b6d4-2dcd2780f450

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