示例#1
0
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)
示例#2
0
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)
示例#4
0
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)