Examples with ErrorCode com.facebook.presto.spi.ErrorCode used on opensource projects

Example 6 with ErrorCode

use of com.facebook.presto.spi.ErrorCode in project presto by prestodb.

the class TestHttpRemoteTask method runTest.

private void runTest(FailureScenario failureScenario, boolean useThriftEncoding) throws Exception {
    AtomicLong lastActivityNanos = new AtomicLong(System.nanoTime());
    TestingTaskResource testingTaskResource = new TestingTaskResource(lastActivityNanos, failureScenario);
    HttpRemoteTaskFactory httpRemoteTaskFactory = createHttpRemoteTaskFactory(testingTaskResource, useThriftEncoding);
    RemoteTask remoteTask = createRemoteTask(httpRemoteTaskFactory);
    testingTaskResource.setInitialTaskInfo(remoteTask.getTaskInfo());
    remoteTask.start();
    waitUntilIdle(lastActivityNanos);
    httpRemoteTaskFactory.stop();
    assertTrue(remoteTask.getTaskStatus().getState().isDone(), format("TaskStatus is not in a done state: %s", remoteTask.getTaskStatus()));
    ErrorCode actualErrorCode = getOnlyElement(remoteTask.getTaskStatus().getFailures()).getErrorCode();
    switch(failureScenario) {
        case TASK_MISMATCH:
        case TASK_MISMATCH_WHEN_VERSION_IS_HIGH:
            assertTrue(remoteTask.getTaskInfo().getTaskStatus().getState().isDone(), format("TaskInfo is not in a done state: %s", remoteTask.getTaskInfo()));
            assertEquals(actualErrorCode, REMOTE_TASK_MISMATCH.toErrorCode());
            break;
        case REJECTED_EXECUTION:
            // for a rejection to occur, the http client must be shutdown, which means we will not be able to ge the final task info
            assertEquals(actualErrorCode, REMOTE_TASK_ERROR.toErrorCode());
            break;
        default:
            throw new UnsupportedOperationException();
    }
}

Example 7 with ErrorCode

use of com.facebook.presto.spi.ErrorCode in project presto by prestodb.

the class PrestoSparkExecutionExceptionFactory method isRetryable.

private static boolean isRetryable(ExecutionFailureInfo executionFailureInfo) {
    ErrorCode errorCode = executionFailureInfo.getErrorCode();
    if (errorCode == null) {
        return true;
    }
    ErrorType type = errorCode.getType();
    return type == INTERNAL_ERROR || type == EXTERNAL;
}