private static <T extends StateObject> RunnableFuture<SnapshotResult<T>> createSnapshotResult(Class<T> clazz) { return DoneFuture.of(SnapshotResult.withLocalState(mock(clazz), mock(clazz))); } }
StateHandleDummyUtil.createNewOperatorStateHandle(2, random); SnapshotResult<KeyedStateHandle> snapKeyMan = SnapshotResult.withLocalState( StateHandleDummyUtil.deepDummyCopy(keyedTemplate), StateHandleDummyUtil.deepDummyCopy(keyedTemplate)); SnapshotResult<KeyedStateHandle> snapKeyRaw = SnapshotResult.withLocalState( StateHandleDummyUtil.deepDummyCopy(keyedTemplate), StateHandleDummyUtil.deepDummyCopy(keyedTemplate)); SnapshotResult<OperatorStateHandle> snapOpMan = SnapshotResult.withLocalState( StateHandleDummyUtil.deepDummyCopy(operatorTemplate), StateHandleDummyUtil.deepDummyCopy(operatorTemplate)); SnapshotResult<OperatorStateHandle> snapOpRaw = SnapshotResult.withLocalState( StateHandleDummyUtil.deepDummyCopy(operatorTemplate), StateHandleDummyUtil.deepDummyCopy(operatorTemplate));
return SnapshotResult.withLocalState(jmKeyedState, localKeyedState); } else {
return SnapshotResult.withLocalState(jmKeyedState, localKeyedState); } else {
return SnapshotResult.withLocalState(jmKeyedState, localKeyedState); } else {
groups, metaInfos, localStreamStateHandle); return SnapshotResult.withLocalState(snapshot, localSnapshot); } else { return SnapshotResult.of(snapshot);
@Nonnull @Override public SnapshotResult<StreamStateHandle> closeAndFinalizeCheckpointStreamResult() throws IOException { final StreamStateHandle primaryStreamStateHandle; try { primaryStreamStateHandle = outputStream.closeAndGetPrimaryHandle(); } catch (IOException primaryEx) { try { outputStream.close(); } catch (IOException closeEx) { primaryEx = ExceptionUtils.firstOrSuppressed(closeEx, primaryEx); } throw primaryEx; } StreamStateHandle secondaryStreamStateHandle = null; try { secondaryStreamStateHandle = outputStream.closeAndGetSecondaryHandle(); } catch (IOException secondaryEx) { LOG.warn("Exception from secondary/local checkpoint stream.", secondaryEx); } if (primaryStreamStateHandle != null) { if (secondaryStreamStateHandle != null) { return SnapshotResult.withLocalState(primaryStreamStateHandle, secondaryStreamStateHandle); } else { return SnapshotResult.of(primaryStreamStateHandle); } } else { return SnapshotResult.empty(); } }
@Nonnull @Override public SnapshotResult<StreamStateHandle> closeAndFinalizeCheckpointStreamResult() throws IOException { final StreamStateHandle primaryStreamStateHandle; try { primaryStreamStateHandle = outputStream.closeAndGetPrimaryHandle(); } catch (IOException primaryEx) { try { outputStream.close(); } catch (IOException closeEx) { primaryEx = ExceptionUtils.firstOrSuppressed(closeEx, primaryEx); } throw primaryEx; } StreamStateHandle secondaryStreamStateHandle = null; try { secondaryStreamStateHandle = outputStream.closeAndGetSecondaryHandle(); } catch (IOException secondaryEx) { LOG.warn("Exception from secondary/local checkpoint stream.", secondaryEx); } if (primaryStreamStateHandle != null) { if (secondaryStreamStateHandle != null) { return SnapshotResult.withLocalState(primaryStreamStateHandle, secondaryStreamStateHandle); } else { return SnapshotResult.of(primaryStreamStateHandle); } } else { return SnapshotResult.empty(); } }
@Nonnull @Override public SnapshotResult<StreamStateHandle> closeAndFinalizeCheckpointStreamResult() throws IOException { final StreamStateHandle primaryStreamStateHandle; try { primaryStreamStateHandle = outputStream.closeAndGetPrimaryHandle(); } catch (IOException primaryEx) { try { outputStream.close(); } catch (IOException closeEx) { primaryEx = ExceptionUtils.firstOrSuppressed(closeEx, primaryEx); } throw primaryEx; } StreamStateHandle secondaryStreamStateHandle = null; try { secondaryStreamStateHandle = outputStream.closeAndGetSecondaryHandle(); } catch (IOException secondaryEx) { LOG.warn("Exception from secondary/local checkpoint stream.", secondaryEx); } if (primaryStreamStateHandle != null) { if (secondaryStreamStateHandle != null) { return SnapshotResult.withLocalState(primaryStreamStateHandle, secondaryStreamStateHandle); } else { return SnapshotResult.of(primaryStreamStateHandle); } } else { return SnapshotResult.empty(); } }