org.apache.flink.util.LambdaUtil Java Examples

@Nullable
@Override
public CompletableFuture<T> triggerCheckpoint(long checkpointId, long timestamp, final Executor executor) throws Exception {
	final Executor wrappedExecutor = new Executor() {
		@Override
		public void execute(Runnable command) {
			executor.execute(new WrappedCommand(userClassLoader, command));
		}
	};

	return LambdaUtil.withContextClassLoader(
			userClassLoader,
			() -> hook.triggerCheckpoint(checkpointId, timestamp, wrappedExecutor));
}
 

@Nullable
@Override
public CompletableFuture<T> triggerCheckpoint(long checkpointId, long timestamp, final Executor executor) throws Exception {
	final Executor wrappedExecutor = command -> executor.execute(new WrappedCommand(userClassLoader, command));

	return LambdaUtil.withContextClassLoader(
			userClassLoader,
			() -> hook.triggerCheckpoint(checkpointId, timestamp, wrappedExecutor));
}
 

@Override
public void writeRecord(Savepoint savepoint) throws IOException {
	String path = LambdaUtil.withContextClassLoader(getRuntimeContext().getUserCodeClassLoader(), () -> {
			try (CheckpointMetadataOutputStream out = targetLocation.createMetadataOutputStream()) {
				Checkpoints.storeCheckpointMetadata(savepoint, out);
				CompletedCheckpointStorageLocation finalizedLocation = out.closeAndFinalizeCheckpoint();
				return finalizedLocation.getExternalPointer();
			}
	});

	LOG.info("Savepoint written to " + path);
}
 

@Override
public void writeRecord(CheckpointMetadata metadata) throws IOException {
	String path = LambdaUtil.withContextClassLoader(getRuntimeContext().getUserCodeClassLoader(), () -> {
			try (CheckpointMetadataOutputStream out = targetLocation.createMetadataOutputStream()) {
				Checkpoints.storeCheckpointMetadata(metadata, out);
				CompletedCheckpointStorageLocation finalizedLocation = out.closeAndFinalizeCheckpoint();
				return finalizedLocation.getExternalPointer();
			}
	});

	LOG.info("Savepoint written to " + path);
}
 

@Nullable
@Override
public CompletableFuture<T> triggerCheckpoint(long checkpointId, long timestamp, final Executor executor) throws Exception {
	final Executor wrappedExecutor = new Executor() {
		@Override
		public void execute(Runnable command) {
			executor.execute(new WrappedCommand(userClassLoader, command));
		}
	};

	return LambdaUtil.withContextClassLoader(
			userClassLoader,
			() -> hook.triggerCheckpoint(checkpointId, timestamp, wrappedExecutor));
}
 

@Override
public void close() throws Exception {
	LambdaUtil.withContextClassLoader(userClassLoader, hook::close);
}
 

@Override
public void run() {
	LambdaUtil.withContextClassLoader(userClassLoader, command::run);
}
 

@Nullable
@Override
public SimpleVersionedSerializer<T> createCheckpointDataSerializer() {
	return LambdaUtil.withContextClassLoader(userClassLoader, hook::createCheckpointDataSerializer);
}
 

@Override
public void restoreCheckpoint(long checkpointId, @Nullable T checkpointData) throws Exception {
	LambdaUtil.withContextClassLoader(
			userClassLoader,
			() -> hook.restoreCheckpoint(checkpointId, checkpointData));
}
 

@Override
public String getIdentifier() {
	return LambdaUtil.withContextClassLoader(userClassLoader, hook::getIdentifier);
}
 

@Override
public void close() throws Exception {
	LambdaUtil.withContextClassLoader(userClassLoader, hook::close);
}
 

@Override
public void reset() throws Exception {
	LambdaUtil.withContextClassLoader(userClassLoader, hook::reset);
}
 

@Override
public void run() {
	LambdaUtil.withContextClassLoader(userClassLoader, command::run);
}
 

@Nullable
@Override
public SimpleVersionedSerializer<T> createCheckpointDataSerializer() {
	return LambdaUtil.withContextClassLoader(userClassLoader, hook::createCheckpointDataSerializer);
}
 

@Override
public void restoreCheckpoint(long checkpointId, @Nullable T checkpointData) throws Exception {
	LambdaUtil.withContextClassLoader(
			userClassLoader,
			() -> hook.restoreCheckpoint(checkpointId, checkpointData));
}
 

@Override
public String getIdentifier() {
	return LambdaUtil.withContextClassLoader(userClassLoader, hook::getIdentifier);
}
 

@Override
public void close() throws Exception {
	LambdaUtil.withContextClassLoader(userClassLoader, hook::close);
}
 

@Override
public void reset() throws Exception {
	LambdaUtil.withContextClassLoader(userClassLoader, hook::reset);
}
 

@Override
public void run() {
	LambdaUtil.withContextClassLoader(userClassLoader, command::run);
}
 

@Nullable
@Override
public SimpleVersionedSerializer<T> createCheckpointDataSerializer() {
	return LambdaUtil.withContextClassLoader(userClassLoader, hook::createCheckpointDataSerializer);
}
 

@Override
public void restoreCheckpoint(long checkpointId, @Nullable T checkpointData) throws Exception {
	LambdaUtil.withContextClassLoader(
			userClassLoader,
			() -> hook.restoreCheckpoint(checkpointId, checkpointData));
}
 

@Override
public String getIdentifier() {
	return LambdaUtil.withContextClassLoader(userClassLoader, hook::getIdentifier);
}
 

@Override
public void reset() throws Exception {
	LambdaUtil.withContextClassLoader(userClassLoader, hook::reset);
}
 

/**
 * Iterates through the passed state handles and calls discardState() on each handle that is not null. All
 * occurring exceptions are suppressed and collected until the iteration is over and emitted as a single exception.
 *
 * @param handlesToDiscard State handles to discard. Passed iterable is allowed to deliver null values.
 * @throws Exception exception that is a collection of all suppressed exceptions that were caught during iteration
 */
public static void bestEffortDiscardAllStateObjects(
	Iterable<? extends StateObject> handlesToDiscard) throws Exception {
	LambdaUtil.applyToAllWhileSuppressingExceptions(handlesToDiscard, StateObject::discardState);
}
 

/**
 * Iterates through the passed state handles and calls discardState() on each handle that is not null. All
 * occurring exceptions are suppressed and collected until the iteration is over and emitted as a single exception.
 *
 * @param handlesToDiscard State handles to discard. Passed iterable is allowed to deliver null values.
 * @throws Exception exception that is a collection of all suppressed exceptions that were caught during iteration
 */
public static void bestEffortDiscardAllStateObjects(
	Iterable<? extends StateObject> handlesToDiscard) throws Exception {
	LambdaUtil.applyToAllWhileSuppressingExceptions(handlesToDiscard, StateObject::discardState);
}
 

/**
 * Iterates through the passed state handles and calls discardState() on each handle that is not null. All
 * occurring exceptions are suppressed and collected until the iteration is over and emitted as a single exception.
 *
 * @param handlesToDiscard State handles to discard. Passed iterable is allowed to deliver null values.
 * @throws Exception exception that is a collection of all suppressed exceptions that were caught during iteration
 */
public static void bestEffortDiscardAllStateObjects(
	Iterable<? extends StateObject> handlesToDiscard) throws Exception {
	LambdaUtil.applyToAllWhileSuppressingExceptions(handlesToDiscard, StateObject::discardState);
}