Java Code Examples for org.apache.flink.runtime.io.network.api.TaskEventHandler#subscribe()

The following examples show how to use org.apache.flink.runtime.io.network.api.TaskEventHandler#subscribe() . You can vote up the ones you like or vote down the ones you don't like, and go to the original project or source file by following the links above each example. You may check out the related API usage on the sidebar.
Example 1
Source File: TaskEventDispatcher.java    From Flink-CEPplus with Apache License 2.0 6 votes vote down vote up 
/**
 * Subscribes a listener to this dispatcher for events on a partition.
 *
 * @param partitionId
 * 		ID of the partition to subscribe for (must be registered via {@link
 * 		#registerPartition(ResultPartitionID)} first!)
 * @param eventListener
 * 		the event listener to subscribe
 * @param eventType
 * 		event type to subscribe to
 */
public void subscribeToEvent(
		ResultPartitionID partitionId,
		EventListener<TaskEvent> eventListener,
		Class<? extends TaskEvent> eventType) {
	checkNotNull(partitionId);
	checkNotNull(eventListener);
	checkNotNull(eventType);

	TaskEventHandler taskEventHandler;
	synchronized (registeredHandlers) {
		taskEventHandler = registeredHandlers.get(partitionId);
	}
	if (taskEventHandler == null) {
		throw new IllegalStateException(
			"Partition " + partitionId + " not registered at task event dispatcher.");
	}
	taskEventHandler.subscribe(eventListener, eventType);
}
Example 2
Source File: TaskEventHandlerTest.java    From Flink-CEPplus with Apache License 2.0 6 votes vote down vote up 
/**
 * Tests the publish/subscribe mechanisms implemented in the {@link TaskEventHandler}.
 */
@Test
public void testEventNotificationManager() {

	final TaskEventHandler evm = new TaskEventHandler();
	final TestEventListener listener = new TestEventListener();

	evm.subscribe(listener, StringTaskEvent.class);

	final StringTaskEvent stringTaskEvent1 = new StringTaskEvent("Test 1");

	evm.publish(stringTaskEvent1);
	evm.publish(new IntegerTaskEvent(5));

	assertNotNull(listener.getLastReceivedEvent());
	StringTaskEvent receivedStringEvent = (StringTaskEvent) listener.getLastReceivedEvent();
	assertEquals(stringTaskEvent1, receivedStringEvent);
}
Example 3
Source File: TaskEventDispatcher.java    From flink with Apache License 2.0 6 votes vote down vote up 
/**
 * Subscribes a listener to this dispatcher for events on a partition.
 *
 * @param partitionId
 * 		ID of the partition to subscribe for (must be registered via {@link
 * 		#registerPartition(ResultPartitionID)} first!)
 * @param eventListener
 * 		the event listener to subscribe
 * @param eventType
 * 		event type to subscribe to
 */
public void subscribeToEvent(
		ResultPartitionID partitionId,
		EventListener<TaskEvent> eventListener,
		Class<? extends TaskEvent> eventType) {
	checkNotNull(partitionId);
	checkNotNull(eventListener);
	checkNotNull(eventType);

	TaskEventHandler taskEventHandler;
	synchronized (registeredHandlers) {
		taskEventHandler = registeredHandlers.get(partitionId);
	}
	if (taskEventHandler == null) {
		throw new IllegalStateException(
			"Partition " + partitionId + " not registered at task event dispatcher.");
	}
	taskEventHandler.subscribe(eventListener, eventType);
}
Example 4
Source File: TaskEventHandlerTest.java    From flink with Apache License 2.0 6 votes vote down vote up 
/**
 * Tests the publish/subscribe mechanisms implemented in the {@link TaskEventHandler}.
 */
@Test
public void testEventNotificationManager() {

	final TaskEventHandler evm = new TaskEventHandler();
	final TestEventListener listener = new TestEventListener();

	evm.subscribe(listener, StringTaskEvent.class);

	final StringTaskEvent stringTaskEvent1 = new StringTaskEvent("Test 1");

	evm.publish(stringTaskEvent1);
	evm.publish(new IntegerTaskEvent(5));

	assertNotNull(listener.getLastReceivedEvent());
	StringTaskEvent receivedStringEvent = (StringTaskEvent) listener.getLastReceivedEvent();
	assertEquals(stringTaskEvent1, receivedStringEvent);
}
Example 5
Source File: TaskEventDispatcher.java    From flink with Apache License 2.0 6 votes vote down vote up 
/**
 * Subscribes a listener to this dispatcher for events on a partition.
 *
 * @param partitionId
 * 		ID of the partition to subscribe for (must be registered via {@link
 * 		#registerPartition(ResultPartitionID)} first!)
 * @param eventListener
 * 		the event listener to subscribe
 * @param eventType
 * 		event type to subscribe to
 */
public void subscribeToEvent(
		ResultPartitionID partitionId,
		EventListener<TaskEvent> eventListener,
		Class<? extends TaskEvent> eventType) {
	checkNotNull(partitionId);
	checkNotNull(eventListener);
	checkNotNull(eventType);

	TaskEventHandler taskEventHandler;
	synchronized (registeredHandlers) {
		taskEventHandler = registeredHandlers.get(partitionId);
	}
	if (taskEventHandler == null) {
		throw new IllegalStateException(
			"Partition " + partitionId + " not registered at task event dispatcher.");
	}
	taskEventHandler.subscribe(eventListener, eventType);
}
Example 6
Source File: TaskEventHandlerTest.java    From flink with Apache License 2.0 6 votes vote down vote up 
/**
 * Tests the publish/subscribe mechanisms implemented in the {@link TaskEventHandler}.
 */
@Test
public void testEventNotificationManager() {

	final TaskEventHandler evm = new TaskEventHandler();
	final TestEventListener listener = new TestEventListener();

	evm.subscribe(listener, StringTaskEvent.class);

	final StringTaskEvent stringTaskEvent1 = new StringTaskEvent("Test 1");

	evm.publish(stringTaskEvent1);
	evm.publish(new IntegerTaskEvent(5));

	assertNotNull(listener.getLastReceivedEvent());
	StringTaskEvent receivedStringEvent = (StringTaskEvent) listener.getLastReceivedEvent();
	assertEquals(stringTaskEvent1, receivedStringEvent);
}