new HashMap<CombinePathInputFormat, CombineFilter>(); Set<Path> poolSet = new HashSet<Path>(); LockedDriverState lDrvStat = LockedDriverState.getLockedDriverState(); if (lDrvStat != null && lDrvStat.isAborted()) { throw new IOException("Operation is Canceled. ");
lDrvState.stateLock.unlock(); if (lDrvState.isAborted()) { LOG.info("Executing command(queryId=" + queryId + ") has been interrupted after " + duration + " seconds"); } else {
LockedDriverState.setLockedDriverState(lDrvState); queryDisplay.setHmsTimings(QueryDisplay.Phase.COMPILATION, compileHMSTimings); boolean isInterrupted = lDrvState.isAborted(); if (isInterrupted && !deferClose) { closeInProcess(true);
SQLState = null; downstreamError = null; LockedDriverState.setLockedDriverState(lDrvState); if (lDrvState.isAborted()) { closeInProcess(true); } else {
LockedDriverState lDrvStat = LockedDriverState.getLockedDriverState(); if (lDrvStat != null && lDrvStat.isAborted()) { throw new IOException("Operation is Canceled."); } else { for (final Path path : pathsToAdd) { if (lDrvStat != null && lDrvStat.isAborted()) { throw new IOException("Operation is Canceled.");
if (lDrvState != null) { lDrvState.stateLock.lock(); if (lDrvState.isAborted()) { isInterrupted = true;
expectedLocks.add(new ZooKeeperHiveLock("default", new HiveLockObject(), HiveLockMode.SHARED)); expectedLocks.add(new ZooKeeperHiveLock("default.table1", new HiveLockObject(), HiveLockMode.SHARED)); LockedDriverState lDrvState = new LockedDriverState(); LockedDriverState lDrvInp = new LockedDriverState(); lDrvInp.abort(); LockException lEx = new LockException(ErrorMsg.LOCK_ACQUIRE_CANCELLED.getMsg()); when(mockLockManager.lock(anyListOf(HiveLockObj.class), eq(false), eq(lDrvState))).thenReturn(expectedLocks);
@VisibleForTesting static List<Path> getInputPathsWithPool(JobConf job, MapWork work, Path hiveScratchDir, Context ctx, boolean skipDummy, List<Path> pathsToAdd, ExecutorService pool) throws IOException, ExecutionException, InterruptedException { LockedDriverState lDrvStat = LockedDriverState.getLockedDriverState(); List<Path> finalPathsToAdd = new ArrayList<>(); try { Map<GetInputPathsCallable, Future<Path>> getPathsCallableToFuture = new LinkedHashMap<>(); for (final Path path : pathsToAdd) { if (lDrvStat != null && lDrvStat.isAborted()) { throw new IOException("Operation is Canceled."); } GetInputPathsCallable callable = new GetInputPathsCallable(path, job, work, hiveScratchDir, ctx, skipDummy); getPathsCallableToFuture.put(callable, pool.submit(callable)); } pool.shutdown(); for (Map.Entry<GetInputPathsCallable, Future<Path>> future : getPathsCallableToFuture.entrySet()) { if (lDrvStat != null && lDrvStat.isAborted()) { throw new IOException("Operation is Canceled."); } Path newPath = future.getValue().get(); updatePathForMapWork(newPath, work, future.getKey().path); finalPathsToAdd.add(newPath); } } finally { pool.shutdownNow(); } return finalPathsToAdd; }
@Override public void close() { lDrvState.stateLock.lock(); try { releaseDriverContext(); if (lDrvState.driverState == DriverState.COMPILING || lDrvState.driverState == DriverState.EXECUTING) { lDrvState.abort(); } releasePlan(); releaseCachedResult(); releaseFetchTask(); releaseResStream(); releaseContext(); lDrvState.driverState = DriverState.CLOSED; } finally { lDrvState.stateLock.unlock(); LockedDriverState.removeLockedDriverState(); } destroy(); }
private void checkInterrupted(String msg, HookContext hookContext, PerfLogger perfLogger) throws CommandProcessorResponse { if (lDrvState.isAborted()) { throw createProcessorResponse(handleInterruptionWithHook(msg, hookContext, perfLogger)); } }
@Override protected LockedDriverState initialValue() { return new LockedDriverState(); } };