def get_query_engine_status(engine_id): engine_checker = None # Security check with DBSession() as session: verify_query_engine_permission(engine_id, session=session) engine = admin_logic.get_query_engine_by_id(engine_id, session=session) engine_checker = get_engine_checker_class( engine.get_feature_params().get("status_checker", "NullChecker")) api_assert(engine_checker is not None, "Invalid engine checker") return engine_checker.check(engine_id=engine_id, uid=current_user.id)
def test_query_engine_connection( name, language, executor, executor_params, feature_params, ): status_checker = get_engine_checker_class(feature_params["status_checker"]) executor_class = get_executor_class(language, executor) pseudo_engine_dict = { "name": name, "language": language, "executor": executor, "executor_params": executor_params, "feature_params": feature_params, } return status_checker.perform_check_with_executor(executor_class, executor_params, pseudo_engine_dict)
def poll_engine_status(self, checker_name, engine_id): from lib.engine_status_checker import get_engine_checker_class get_engine_checker_class(checker_name).get_server_status_task(engine_id)
def poll_engine_status(self, checker_name, engine_id): from lib.engine_status_checker import get_engine_checker_class get_engine_checker_class(checker_name).check_and_store_engine_status( engine_id)