#755

Committed 04 Sep 2023 09:49PM UTC coverage: 69.69% (-10.6%) from 80.336%

Build # #755

Build Type

push

circle-ci

Committed by

jiashenC

Commit Message

update case

Run Details

8130 of 11666 relevant lines covered (69.69%)

0.7 hits per line

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

71.43

/evadb/executor/use_executor.py

# 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 ExecutorError
from evadb.models.storage.batch import Batch
from evadb.parser.use_statement import UseStatement
from evadb.third_party.databases.interface import get_database_handler


class UseExecutor(AbstractExecutor):
    def __init__(self, db: EvaDBDatabase, node: UseStatement):
        super().__init__(db, node)
        self._database_name = node.database_name
        self._query_string = node.query_string

    def exec(self, *args, **kwargs) -> Iterator[Batch]:
        db_catalog_entry = self.db.catalog().get_database_catalog_entry(
            self._database_name
        )

        if db_catalog_entry is None:
            raise ExecutorError(
                f"{self._database_name} data source does not exist. Use CREATE DATABASE to add a new data source."
            )

        handler = get_database_handler(
            db_catalog_entry.engine,
            **db_catalog_entry.params,
        )

        handler.connect()
        resp = handler.execute_native_query(self._query_string)
        handler.disconnect()

        if resp.error is None:
            return Batch(resp.data)
        else:
            raise ExecutorError(resp.error)

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 ExecutorError	1✔
20	from evadb.models.storage.batch import Batch	1✔
21	from evadb.parser.use_statement import UseStatement	1✔
22	from evadb.third_party.databases.interface import get_database_handler	1✔
23
24
25	class UseExecutor(AbstractExecutor):	1✔
26	def __init__(self, db: EvaDBDatabase, node: UseStatement):	1✔
27	super().__init__(db, node)	1✔
28	self._database_name = node.database_name	1✔
29	self._query_string = node.query_string	1✔
30
31	def exec(self, args, *kwargs) -> Iterator[Batch]:	1✔
32	db_catalog_entry = self.db.catalog().get_database_catalog_entry(	1✔
33	self._database_name
34	)
35
36	if db_catalog_entry is None:	1✔
37	raise ExecutorError(
38	f"{self._database_name} data source does not exist. Use CREATE DATABASE to add a new data source."
39	)
40
41	handler = get_database_handler(	×
42	db_catalog_entry.engine,
43	**db_catalog_entry.params,
44	)
45
46	handler.connect()	×
47	resp = handler.execute_native_query(self._query_string)	×
48	handler.disconnect()	×
49
50	if resp.error is None:	×
51	return Batch(resp.data)	×
52	else:
53	raise ExecutorError(resp.error)

georgia-tech-db / eva / #755

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