a89ccb4a-7f6e-4be4-b286-750d999fbf3a

# 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 abc import ABC, abstractmethod
from collections import deque
from typing import TYPE_CHECKING, Any, Generator, Iterable, List, TypeVar

if TYPE_CHECKING:
    from evadb.catalog.catalog_manager import CatalogManager
from evadb.configuration.configuration_manager import ConfigurationManager
from evadb.database import EvaDBDatabase
from evadb.models.storage.batch import Batch
from evadb.plan_nodes.abstract_plan import AbstractPlan

AbstractExecutor = TypeVar("AbstractExecutor")


class AbstractExecutor(ABC):
    """
    An abstract class for the executor engine
    Arguments:
        node (AbstractPlan): Plan node corresponding to this executor
    """

    def __init__(self, db: EvaDBDatabase, node: AbstractPlan):
        self._db = db
        self._node = node
        self._config: ConfigurationManager = db.config if db else None
        self._children = []

    def catalog(self) -> "CatalogManager":
        """The object is intentionally generated on demand to prevent serialization issues. Having a SQLAlchemy object as a member variable can cause problems with multiprocessing. See get_catalog_instance()"""
        return self._db.catalog() if self._db else None

    def append_child(self, child: AbstractExecutor):
        """
        appends a child executor node

        Arguments:
            child {AbstractExecutor} -- child node
        """
        self._children.append(child)

    @property
    def children(self) -> List[AbstractExecutor]:
        """
        Returns the list of child executor
        Returns:
            [] -- list of children
        """
        return self._children

    @children.setter
    def children(self, children):
        self._children = children

    @property
    def node(self) -> AbstractPlan:
        return self._node

    @property
    def db(self) -> EvaDBDatabase:
        return self._db

    @property
    def config(self) -> ConfigurationManager:
        return self._config

    @abstractmethod
    def exec(self, *args, **kwargs) -> Iterable[Batch]:
        """
        This method is implemented by every executor.
        Contains logic for that executor;
        For retrieval based executor : It fetches frame batches from
        child nodes and emits it to parent node.
        """

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

    def bfs(self):
        """Returns a generator which visits all nodes in execution tree in
        breadth-first search (BFS) traversal order.

        Returns:
            the generator object.
        """
        queue = deque([self])
        while queue:
            node = queue.popleft()
            yield node
            for child in node.children:
                queue.append(child)

    def find_all(self, execution_type: Any):
        """Returns a generator which visits all the nodes in execution tree and yields one that matches the passed `execution_type`.

        Args:
            execution_type (Any): execution type to match with

        Returns:
            the generator object.
        """

        for node in self.bfs():
            if isinstance(node, execution_type):
                yield node

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 abc import ABC, abstractmethod	1✔
16	from collections import deque	1✔
17	from typing import TYPE_CHECKING, Any, Generator, Iterable, List, TypeVar	1✔
18
19	if TYPE_CHECKING:
20	from evadb.catalog.catalog_manager import CatalogManager
21	from evadb.configuration.configuration_manager import ConfigurationManager	1✔
22	from evadb.database import EvaDBDatabase	1✔
23	from evadb.models.storage.batch import Batch	1✔
24	from evadb.plan_nodes.abstract_plan import AbstractPlan	1✔
25
26	AbstractExecutor = TypeVar("AbstractExecutor")	1✔
27
28
29	class AbstractExecutor(ABC):	1✔
30	"""
31	An abstract class for the executor engine
32	Arguments:
33	node (AbstractPlan): Plan node corresponding to this executor
34	"""
35
36	def __init__(self, db: EvaDBDatabase, node: AbstractPlan):	1✔
37	self._db = db	1✔
38	self._node = node	1✔
39	self._config: ConfigurationManager = db.config if db else None	1✔
40	self._children = []	1✔
41
42	def catalog(self) -> "CatalogManager":	1✔
43	"""The object is intentionally generated on demand to prevent serialization issues. Having a SQLAlchemy object as a member variable can cause problems with multiprocessing. See get_catalog_instance()"""
44	return self._db.catalog() if self._db else None	1✔
45
46	def append_child(self, child: AbstractExecutor):	1✔
47	"""
48	appends a child executor node
49
50	Arguments:
51	child {AbstractExecutor} -- child node
52	"""
53	self._children.append(child)	1✔
54
55	@property	1✔
56	def children(self) -> List[AbstractExecutor]:	1✔
57	"""
58	Returns the list of child executor
59	Returns:
60	[] -- list of children
61	"""
62	return self._children	1✔
63
64	@children.setter	1✔
65	def children(self, children):	1✔
66	self._children = children	×
67
68	@property	1✔
69	def node(self) -> AbstractPlan:	1✔
70	return self._node	1✔
71
72	@property	1✔
73	def db(self) -> EvaDBDatabase:	1✔
74	return self._db	1✔
75
76	@property	1✔
77	def config(self) -> ConfigurationManager:	1✔
78	return self._config	1✔
79
80	@abstractmethod	1✔
81	def exec(self, args, *kwargs) -> Iterable[Batch]:	1✔
82	"""
83	This method is implemented by every executor.
84	Contains logic for that executor;
85	For retrieval based executor : It fetches frame batches from
86	child nodes and emits it to parent node.
87	"""
88
89	def __call__(self, args, *kwargs) -> Generator[Batch, None, None]:	1✔
90	yield from self.exec(args, *kwargs)	×
91
92	def bfs(self):	1✔
93	"""Returns a generator which visits all nodes in execution tree in
94	breadth-first search (BFS) traversal order.
95
96	Returns:
97	the generator object.
98	"""
99	queue = deque([self])	×
100	while queue:	×
101	node = queue.popleft()	×
102	yield node	×
103	for child in node.children:	×
104	queue.append(child)	×
105
106	def find_all(self, execution_type: Any):	1✔
107	"""Returns a generator which visits all the nodes in execution tree and yields one that matches the passed `execution_type`.
108
109	Args:
110	execution_type (Any): execution type to match with
111
112	Returns:
113	the generator object.
114	"""
115
116	for node in self.bfs():	×
117	if isinstance(node, execution_type):	×
118	yield node	×

georgia-tech-db / eva / a89ccb4a-7f6e-4be4-b286-750d999fbf3a

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