protected void closeCurrentStream() { if (closableRegistry.unregisterCloseable(currentStream)) { IOUtils.closeQuietly(currentStream); } currentStream = null; }
private static void closeFromRegistry(Closeable closeable, CloseableRegistry registry) { if (registry.unregisterCloseable(closeable)) { IOUtils.closeQuietly(closeable); } }
@Override public boolean hasNext() { if (null != offsets && offPos < offsets.length) { return true; } closeCurrentStream(); while (stateHandleIterator.hasNext()) { currentStateHandle = stateHandleIterator.next(); OperatorStateHandle.StateMetaInfo metaInfo = currentStateHandle.getStateNameToPartitionOffsets().get(stateName); if (null != metaInfo) { long[] metaOffsets = metaInfo.getOffsets(); if (null != metaOffsets && metaOffsets.length > 0) { this.offsets = metaOffsets; this.offPos = 0; if (closableRegistry.unregisterCloseable(currentStream)) { IOUtils.closeQuietly(currentStream); currentStream = null; } return true; } } } return false; }
private static void checkCloseablesRegistered(CloseableRegistry closeableRegistry, Closeable... closeables) { for (Closeable closeable : closeables) { Assert.assertTrue(closeableRegistry.unregisterCloseable(closeable)); } }
if (streamTaskCloseableRegistry.unregisterCloseable(keyedStatedBackend)) { IOUtils.closeQuietly(keyedStatedBackend); if (streamTaskCloseableRegistry.unregisterCloseable(operatorStateBackend)) { IOUtils.closeQuietly(operatorStateBackend); if (streamTaskCloseableRegistry.unregisterCloseable(rawKeyedStateInputs)) { IOUtils.closeQuietly(rawKeyedStateInputs); if (streamTaskCloseableRegistry.unregisterCloseable(rawOperatorStateInputs)) { IOUtils.closeQuietly(rawOperatorStateInputs);
if (closeableRegistry.unregisterCloseable(outputStream)) { result = outputStream.closeAndGetHandle(); outputStream = null; if (closeableRegistry.unregisterCloseable(inputStream)) { IOUtils.closeQuietly(inputStream); if (closeableRegistry.unregisterCloseable(outputStream)) { IOUtils.closeQuietly(outputStream);
if (backendCloseableRegistry.unregisterCloseable(backendInstance)) { try { backendInstance.close();
if (closeableRegistry.unregisterCloseable(inputStream)) { inputStream.close(); if (closeableRegistry.unregisterCloseable(outputStream)) { outputStream.close();
/** * Restore one key groups state handle. */ private void restoreKeyGroupsInStateHandle() throws IOException, StateMigrationException, RocksDBException { try { currentStateHandleInStream = currentKeyGroupsStateHandle.openInputStream(); rocksDBKeyedStateBackend.cancelStreamRegistry.registerCloseable(currentStateHandleInStream); currentStateHandleInView = new DataInputViewStreamWrapper(currentStateHandleInStream); restoreKVStateMetaData(); restoreKVStateData(); } finally { if (rocksDBKeyedStateBackend.cancelStreamRegistry.unregisterCloseable(currentStateHandleInStream)) { IOUtils.closeQuietly(currentStateHandleInStream); } } }
taskCloseableRegistry.unregisterCloseable(operatorStateBackend)) { operatorStateBackend.close(); taskCloseableRegistry.unregisterCloseable(keyedStateBackend)) { keyedStateBackend.close();
if (stateBackend.cancelStreamRegistry.unregisterCloseable(inputStream)) { inputStream.close();
protected void closeCurrentStream() { if (closableRegistry.unregisterCloseable(currentStream)) { IOUtils.closeQuietly(currentStream); } currentStream = null; }
private void closeOutStream() { if (closeStreamOnCancelRegistry.unregisterCloseable(out)) { IOUtils.closeQuietly(out); } }
/** * Unregisters the given argument from the snapshot's {@link CloseableRegistry} and returns <code>true</code> iff * the argument was registered before the call. */ protected boolean unregisterCloseableFromCancellation(@Nullable Closeable toUnregister) { return snapshotCloseableRegistry.unregisterCloseable(toUnregister); }
/** * Unregisters the given argument from the snapshot's {@link CloseableRegistry} and returns <code>true</code> iff * the argument was registered before the call. */ protected boolean unregisterCloseableFromCancellation(@Nullable Closeable toUnregister) { return snapshotCloseableRegistry.unregisterCloseable(toUnregister); }
protected void closeCurrentStream() { if (closableRegistry.unregisterCloseable(currentStream)) { IOUtils.closeQuietly(currentStream); } currentStream = null; }
private static void closeFromRegistry(Closeable closeable, CloseableRegistry registry) { if (registry.unregisterCloseable(closeable)) { IOUtils.closeQuietly(closeable); } }
private void unregisterAndCloseStreamAndResultExtractor() { if (cancelStreamRegistry.unregisterCloseable(streamAndResultExtractor)) { IOUtils.closeQuietly(streamAndResultExtractor); streamAndResultExtractor = null; } }
private <T extends StreamStateHandle> void closeAndUnregisterStream( NonClosingCheckpointOutputStream<? extends T> stream) throws IOException { Preconditions.checkNotNull(stream); CheckpointStreamFactory.CheckpointStateOutputStream delegate = stream.getDelegate(); if (closableRegistry.unregisterCloseable(delegate)) { delegate.close(); } }
private <T extends StreamStateHandle> void closeAndUnregisterStream( NonClosingCheckpointOutputStream<? extends T> stream) throws IOException { Preconditions.checkNotNull(stream); CheckpointStreamFactory.CheckpointStateOutputStream delegate = stream.getDelegate(); if (closableRegistry.unregisterCloseable(delegate)) { delegate.close(); } }