Java Code Examples for org.apache.flink.streaming.api.windowing.assigners.WindowAssigner#WindowAssignerContext

The following examples show how to use org.apache.flink.streaming.api.windowing.assigners.WindowAssigner#WindowAssignerContext . 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: DynamicProcessingTimeSessionWindowsTest.java From flink with Apache License 2.0

6 votes

@Test
public void testWindowAssignment() {

	WindowAssigner.WindowAssignerContext mockContext = mock(WindowAssigner.WindowAssignerContext.class);
	SessionWindowTimeGapExtractor<String> extractor = mock(SessionWindowTimeGapExtractor.class);
	when(extractor.extract(eq("gap5000"))).thenReturn(5000L);
	when(extractor.extract(eq("gap4000"))).thenReturn(4000L);
	when(extractor.extract(eq("gap9000"))).thenReturn(9000L);

	DynamicProcessingTimeSessionWindows<String> assigner = DynamicProcessingTimeSessionWindows.withDynamicGap(extractor);

	when(mockContext.getCurrentProcessingTime()).thenReturn(0L);
	assertThat(assigner.assignWindows("gap5000", Long.MIN_VALUE, mockContext), contains(timeWindow(0, 5000)));

	when(mockContext.getCurrentProcessingTime()).thenReturn(4999L);
	assertThat(assigner.assignWindows("gap4000", Long.MIN_VALUE, mockContext), contains(timeWindow(4999, 8999)));

	when(mockContext.getCurrentProcessingTime()).thenReturn(5000L);
	assertThat(assigner.assignWindows("gap9000", Long.MIN_VALUE, mockContext), contains(timeWindow(5000, 14000)));
}

Example 2

Source File: TumblingProcessingTimeWindowsTest.java From flink with Apache License 2.0

6 votes

@Test
public void testTimeUnits() {
	// sanity check with one other time unit

	WindowAssigner.WindowAssignerContext mockContext =
			mock(WindowAssigner.WindowAssignerContext.class);

	TumblingProcessingTimeWindows assigner = TumblingProcessingTimeWindows.of(Time.seconds(5), Time.seconds(1));

	when(mockContext.getCurrentProcessingTime()).thenReturn(1000L);
	assertThat(assigner.assignWindows("String", Long.MIN_VALUE, mockContext), contains(timeWindow(1000, 6000)));

	when(mockContext.getCurrentProcessingTime()).thenReturn(5999L);
	assertThat(assigner.assignWindows("String", Long.MIN_VALUE, mockContext), contains(timeWindow(1000, 6000)));

	when(mockContext.getCurrentProcessingTime()).thenReturn(6000L);
	assertThat(assigner.assignWindows("String", Long.MIN_VALUE, mockContext), contains(timeWindow(6000, 11000)));
}

Example 3

Source File: TumblingProcessingTimeWindowsTest.java From flink with Apache License 2.0

6 votes

@Test
public void testWindowAssignmentWithNegativeGlobalOffset() {
	WindowAssigner.WindowAssignerContext mockContext =
		mock(WindowAssigner.WindowAssignerContext.class);

	TumblingProcessingTimeWindows assigner = TumblingProcessingTimeWindows.of(Time.milliseconds(5000), Time.milliseconds(-100));

	when(mockContext.getCurrentProcessingTime()).thenReturn(100L);
	assertThat(assigner.assignWindows("String", Long.MIN_VALUE, mockContext), contains(timeWindow(-100, 4900)));

	when(mockContext.getCurrentProcessingTime()).thenReturn(4899L);
	assertThat(assigner.assignWindows("String", Long.MIN_VALUE, mockContext), contains(timeWindow(-100, 4900)));

	when(mockContext.getCurrentProcessingTime()).thenReturn(4900L);
	assertThat(assigner.assignWindows("String", Long.MIN_VALUE, mockContext), contains(timeWindow(4900, 9900)));
}

Example 4

Source File: ProcessingTimeSessionWindowsTest.java From flink with Apache License 2.0

6 votes

@Test
public void testWindowAssignment() {
	WindowAssigner.WindowAssignerContext mockContext =
			mock(WindowAssigner.WindowAssignerContext.class);

	ProcessingTimeSessionWindows assigner = ProcessingTimeSessionWindows.withGap(Time.milliseconds(5000));

	when(mockContext.getCurrentProcessingTime()).thenReturn(0L);
	assertThat(assigner.assignWindows("String", Long.MIN_VALUE, mockContext), contains(timeWindow(0, 5000)));

	when(mockContext.getCurrentProcessingTime()).thenReturn(4999L);
	assertThat(assigner.assignWindows("String", Long.MIN_VALUE, mockContext), contains(timeWindow(4999, 9999)));

	when(mockContext.getCurrentProcessingTime()).thenReturn(5000L);
	assertThat(assigner.assignWindows("String", Long.MIN_VALUE, mockContext), contains(timeWindow(5000, 10000)));
}

Example 5

Source File: ProcessingTimeSessionWindowsTest.java From flink with Apache License 2.0

6 votes

@Test
public void testWindowAssignment() {
	WindowAssigner.WindowAssignerContext mockContext =
			mock(WindowAssigner.WindowAssignerContext.class);

	ProcessingTimeSessionWindows assigner = ProcessingTimeSessionWindows.withGap(Time.milliseconds(5000));

	when(mockContext.getCurrentProcessingTime()).thenReturn(0L);
	assertThat(assigner.assignWindows("String", Long.MIN_VALUE, mockContext), contains(timeWindow(0, 5000)));

	when(mockContext.getCurrentProcessingTime()).thenReturn(4999L);
	assertThat(assigner.assignWindows("String", Long.MIN_VALUE, mockContext), contains(timeWindow(4999, 9999)));

	when(mockContext.getCurrentProcessingTime()).thenReturn(5000L);
	assertThat(assigner.assignWindows("String", Long.MIN_VALUE, mockContext), contains(timeWindow(5000, 10000)));
}

Example 6

Source File: ProcessingTimeSessionWindowsTest.java From flink with Apache License 2.0

6 votes

@Test
public void testTimeUnits() {
	// sanity check with one other time unit

	WindowAssigner.WindowAssignerContext mockContext =
			mock(WindowAssigner.WindowAssignerContext.class);

	ProcessingTimeSessionWindows assigner = ProcessingTimeSessionWindows.withGap(Time.seconds(5));

	when(mockContext.getCurrentProcessingTime()).thenReturn(0L);
	assertThat(assigner.assignWindows("String", Long.MIN_VALUE, mockContext), contains(timeWindow(0, 5000)));

	when(mockContext.getCurrentProcessingTime()).thenReturn(4999L);
	assertThat(assigner.assignWindows("String", Long.MIN_VALUE, mockContext), contains(timeWindow(4999, 9999)));

	when(mockContext.getCurrentProcessingTime()).thenReturn(5000L);
	assertThat(assigner.assignWindows("String", Long.MIN_VALUE, mockContext), contains(timeWindow(5000, 10000)));
}

Example 7

Source File: TumblingProcessingTimeWindowsTest.java From Flink-CEPplus with Apache License 2.0

6 votes

@Test
public void testTimeUnits() {
	// sanity check with one other time unit

	WindowAssigner.WindowAssignerContext mockContext =
			mock(WindowAssigner.WindowAssignerContext.class);

	TumblingProcessingTimeWindows assigner = TumblingProcessingTimeWindows.of(Time.seconds(5), Time.seconds(1));

	when(mockContext.getCurrentProcessingTime()).thenReturn(1000L);
	assertThat(assigner.assignWindows("String", Long.MIN_VALUE, mockContext), contains(timeWindow(1000, 6000)));

	when(mockContext.getCurrentProcessingTime()).thenReturn(5999L);
	assertThat(assigner.assignWindows("String", Long.MIN_VALUE, mockContext), contains(timeWindow(1000, 6000)));

	when(mockContext.getCurrentProcessingTime()).thenReturn(6000L);
	assertThat(assigner.assignWindows("String", Long.MIN_VALUE, mockContext), contains(timeWindow(6000, 11000)));
}

Example 8

Source File: DynamicEventTimeSessionWindowsTest.java From flink with Apache License 2.0

5 votes

@Test
public void testWindowAssignment() {

	WindowAssigner.WindowAssignerContext mockContext = mock(WindowAssigner.WindowAssignerContext.class);
	SessionWindowTimeGapExtractor<String> extractor = mock(SessionWindowTimeGapExtractor.class);
	when(extractor.extract(eq("gap5000"))).thenReturn(5000L);
	when(extractor.extract(eq("gap4000"))).thenReturn(4000L);
	when(extractor.extract(eq("gap9000"))).thenReturn(9000L);

	DynamicEventTimeSessionWindows<String> assigner = DynamicEventTimeSessionWindows.withDynamicGap(extractor);

	assertThat(assigner.assignWindows("gap5000", 0L, mockContext), contains(timeWindow(0, 5000)));
	assertThat(assigner.assignWindows("gap4000", 4999L, mockContext), contains(timeWindow(4999, 8999)));
	assertThat(assigner.assignWindows("gap9000", 5000L, mockContext), contains(timeWindow(5000, 14000)));
}

Example 9

Source File: SlidingEventTimeWindowsTest.java From Flink-CEPplus with Apache License 2.0

5 votes

@Test
public void testWindowAssignmentWithNegativeOffset() {
	WindowAssigner.WindowAssignerContext mockContext =
		mock(WindowAssigner.WindowAssignerContext.class);

	SlidingEventTimeWindows assigner =
		SlidingEventTimeWindows.of(Time.milliseconds(5000), Time.milliseconds(1000), Time.milliseconds(-100));

	assertThat(assigner.assignWindows("String", 0L, mockContext), containsInAnyOrder(
		timeWindow(-4100, 900),
		timeWindow(-3100, 1900),
		timeWindow(-2100, 2900),
		timeWindow(-1100, 3900),
		timeWindow(-100, 4900)));

	assertThat(assigner.assignWindows("String", 4899L, mockContext), containsInAnyOrder(
		timeWindow(-100, 4900),
		timeWindow(900, 5900),
		timeWindow(1900, 6900),
		timeWindow(2900, 7900),
		timeWindow(3900, 8900)));

	assertThat(assigner.assignWindows("String", 4900L, mockContext), containsInAnyOrder(
		timeWindow(900, 5900),
		timeWindow(1900, 6900),
		timeWindow(2900, 7900),
		timeWindow(3900, 8900),
		timeWindow(4900, 9900)));
}

Example 10

Source File: SlidingProcessingTimeWindowsTest.java From flink with Apache License 2.0

5 votes

@Test
public void testWindowAssignmentWithOffset() {
	WindowAssigner.WindowAssignerContext mockContext =
			mock(WindowAssigner.WindowAssignerContext.class);

	SlidingProcessingTimeWindows assigner =
			SlidingProcessingTimeWindows.of(Time.milliseconds(5000), Time.milliseconds(1000), Time.milliseconds(100));

	when(mockContext.getCurrentProcessingTime()).thenReturn(100L);
	assertThat(assigner.assignWindows("String", Long.MIN_VALUE, mockContext), containsInAnyOrder(
			timeWindow(-3900, 1100),
			timeWindow(-2900, 2100),
			timeWindow(-1900, 3100),
			timeWindow(-900, 4100),
			timeWindow(100, 5100)));

	when(mockContext.getCurrentProcessingTime()).thenReturn(5099L);
	assertThat(assigner.assignWindows("String", Long.MIN_VALUE, mockContext), containsInAnyOrder(
			timeWindow(100, 5100),
			timeWindow(1100, 6100),
			timeWindow(2100, 7100),
			timeWindow(3100, 8100),
			timeWindow(4100, 9100)));

	when(mockContext.getCurrentProcessingTime()).thenReturn(5100L);
	assertThat(assigner.assignWindows("String", Long.MIN_VALUE, mockContext), containsInAnyOrder(
			timeWindow(1100, 6100),
			timeWindow(2100, 7100),
			timeWindow(3100, 8100),
			timeWindow(4100, 9100),
			timeWindow(5100, 10100)));
}

Example 11

Source File: SlidingProcessingTimeWindowsTest.java From Flink-CEPplus with Apache License 2.0

5 votes

@Test
public void testWindowAssignmentWithNegativeOffset() {
	WindowAssigner.WindowAssignerContext mockContext =
		mock(WindowAssigner.WindowAssignerContext.class);

	SlidingProcessingTimeWindows assigner =
		SlidingProcessingTimeWindows.of(Time.milliseconds(5000), Time.milliseconds(1000), Time.milliseconds(-100));

	when(mockContext.getCurrentProcessingTime()).thenReturn(0L);
	assertThat(assigner.assignWindows("String", Long.MIN_VALUE, mockContext), containsInAnyOrder(
		timeWindow(-4100, 900),
		timeWindow(-3100, 1900),
		timeWindow(-2100, 2900),
		timeWindow(-1100, 3900),
		timeWindow(-100, 4900)));

	when(mockContext.getCurrentProcessingTime()).thenReturn(4899L);
	assertThat(assigner.assignWindows("String", Long.MIN_VALUE, mockContext), containsInAnyOrder(
		timeWindow(-100, 4900),
		timeWindow(900, 5900),
		timeWindow(1900, 6900),
		timeWindow(2900, 7900),
		timeWindow(3900, 8900)));

	when(mockContext.getCurrentProcessingTime()).thenReturn(4900L);
	assertThat(assigner.assignWindows("String", Long.MIN_VALUE, mockContext), containsInAnyOrder(
		timeWindow(900, 5900),
		timeWindow(1900, 6900),
		timeWindow(2900, 7900),
		timeWindow(3900, 8900),
		timeWindow(4900, 9900)));
}

Example 12

Source File: TumblingEventTimeWindowsTest.java From flink with Apache License 2.0

5 votes

@Test
public void testWindowAssignmentWithNegativeOffset() {
	WindowAssigner.WindowAssignerContext mockContext =
		mock(WindowAssigner.WindowAssignerContext.class);

	TumblingEventTimeWindows assigner = TumblingEventTimeWindows.of(Time.milliseconds(5000), Time.milliseconds(-100));

	assertThat(assigner.assignWindows("String", 0L, mockContext), contains(timeWindow(-100, 4900)));
	assertThat(assigner.assignWindows("String", 4899L, mockContext), contains(timeWindow(-100, 4900)));
	assertThat(assigner.assignWindows("String", 4900L, mockContext), contains(timeWindow(4900, 9900)));
}

Example 13

Source File: SlidingProcessingTimeWindowsTest.java From Flink-CEPplus with Apache License 2.0

5 votes

@Test
public void testWindowAssignment() {
	WindowAssigner.WindowAssignerContext mockContext =
			mock(WindowAssigner.WindowAssignerContext.class);

	SlidingProcessingTimeWindows assigner =
			SlidingProcessingTimeWindows.of(Time.milliseconds(5000), Time.milliseconds(1000));

	when(mockContext.getCurrentProcessingTime()).thenReturn(0L);
	assertThat(assigner.assignWindows("String", Long.MIN_VALUE, mockContext), containsInAnyOrder(
			timeWindow(-4000, 1000),
			timeWindow(-3000, 2000),
			timeWindow(-2000, 3000),
			timeWindow(-1000, 4000),
			timeWindow(0, 5000)));

	when(mockContext.getCurrentProcessingTime()).thenReturn(4999L);
	assertThat(assigner.assignWindows("String", Long.MIN_VALUE, mockContext), containsInAnyOrder(
			timeWindow(0, 5000),
			timeWindow(1000, 6000),
			timeWindow(2000, 7000),
			timeWindow(3000, 8000),
			timeWindow(4000, 9000)));

	when(mockContext.getCurrentProcessingTime()).thenReturn(5000L);
	assertThat(assigner.assignWindows("String", Long.MIN_VALUE, mockContext), containsInAnyOrder(
			timeWindow(1000, 6000),
			timeWindow(2000, 7000),
			timeWindow(3000, 8000),
			timeWindow(4000, 9000),
			timeWindow(5000, 10000)));
}

Example 14

Source File: EventTimeSessionWindowsTest.java From flink with Apache License 2.0

5 votes

@Test
public void testTimeUnits() {
	// sanity check with one other time unit

	final int sessionGap = 5000;

	WindowAssigner.WindowAssignerContext mockContext =
			mock(WindowAssigner.WindowAssignerContext.class);

	EventTimeSessionWindows assigner = EventTimeSessionWindows.withGap(Time.seconds(sessionGap / 1000));

	assertThat(assigner.assignWindows("String", 0L, mockContext), contains(timeWindow(0, 0 + sessionGap)));
	assertThat(assigner.assignWindows("String", 4999L, mockContext), contains(timeWindow(4999, 4999 + sessionGap)));
	assertThat(assigner.assignWindows("String", 5000L, mockContext), contains(timeWindow(5000, 5000 + sessionGap)));
}

Example 15

Source File: TumblingEventTimeWindowsTest.java From flink with Apache License 2.0

5 votes

@Test
public void testWindowAssignmentWithOffset() {
	WindowAssigner.WindowAssignerContext mockContext =
			mock(WindowAssigner.WindowAssignerContext.class);

	TumblingEventTimeWindows assigner = TumblingEventTimeWindows.of(Time.milliseconds(5000), Time.milliseconds(100));

	assertThat(assigner.assignWindows("String", 100L, mockContext), contains(timeWindow(100, 5100)));
	assertThat(assigner.assignWindows("String", 5099L, mockContext), contains(timeWindow(100, 5100)));
	assertThat(assigner.assignWindows("String", 5100L, mockContext), contains(timeWindow(5100, 10100)));
}

Example 16

Source File: EventTimeSessionWindowsTest.java From flink with Apache License 2.0

5 votes

@Test
public void testWindowAssignment() {
	final int sessionGap = 5000;

	WindowAssigner.WindowAssignerContext mockContext =
			mock(WindowAssigner.WindowAssignerContext.class);

	EventTimeSessionWindows assigner = EventTimeSessionWindows.withGap(Time.milliseconds(sessionGap));

	assertThat(assigner.assignWindows("String", 0L, mockContext), contains(timeWindow(0, 0 + sessionGap)));
	assertThat(assigner.assignWindows("String", 4999L, mockContext), contains(timeWindow(4999, 4999 + sessionGap)));
	assertThat(assigner.assignWindows("String", 5000L, mockContext), contains(timeWindow(5000, 5000 + sessionGap)));
}

Example 17

Source File: TumblingEventTimeWindowsTest.java From Flink-CEPplus with Apache License 2.0

5 votes

@Test
public void testTimeUnits() {
	// sanity check with one other time unit

	WindowAssigner.WindowAssignerContext mockContext =
			mock(WindowAssigner.WindowAssignerContext.class);

	TumblingEventTimeWindows assigner = TumblingEventTimeWindows.of(Time.seconds(5), Time.seconds(1));

	assertThat(assigner.assignWindows("String", 1000L, mockContext), contains(timeWindow(1000, 6000)));
	assertThat(assigner.assignWindows("String", 5999L, mockContext), contains(timeWindow(1000, 6000)));
	assertThat(assigner.assignWindows("String", 6000L, mockContext), contains(timeWindow(6000, 11000)));
}

Example 18

Source File: SlidingEventTimeWindowsTest.java From flink with Apache License 2.0

5 votes

@Test
public void testWindowAssignment() {
	WindowAssigner.WindowAssignerContext mockContext =
			mock(WindowAssigner.WindowAssignerContext.class);

	SlidingEventTimeWindows assigner =
			SlidingEventTimeWindows.of(Time.milliseconds(5000), Time.milliseconds(1000));

	assertThat(assigner.assignWindows("String", 0L, mockContext), containsInAnyOrder(
			timeWindow(-4000, 1000),
			timeWindow(-3000, 2000),
			timeWindow(-2000, 3000),
			timeWindow(-1000, 4000),
			timeWindow(0, 5000)));

	assertThat(assigner.assignWindows("String", 4999L, mockContext), containsInAnyOrder(
			timeWindow(0, 5000),
			timeWindow(1000, 6000),
			timeWindow(2000, 7000),
			timeWindow(3000, 8000),
			timeWindow(4000, 9000)));

	assertThat(assigner.assignWindows("String", 5000L, mockContext), containsInAnyOrder(
			timeWindow(1000, 6000),
			timeWindow(2000, 7000),
			timeWindow(3000, 8000),
			timeWindow(4000, 9000),
			timeWindow(5000, 10000)));
}

Example 19

Source File: SlidingProcessingTimeWindowsTest.java From flink with Apache License 2.0

5 votes

@Test
public void testWindowAssignment() {
	WindowAssigner.WindowAssignerContext mockContext =
			mock(WindowAssigner.WindowAssignerContext.class);

	SlidingProcessingTimeWindows assigner =
			SlidingProcessingTimeWindows.of(Time.milliseconds(5000), Time.milliseconds(1000));

	when(mockContext.getCurrentProcessingTime()).thenReturn(0L);
	assertThat(assigner.assignWindows("String", Long.MIN_VALUE, mockContext), containsInAnyOrder(
			timeWindow(-4000, 1000),
			timeWindow(-3000, 2000),
			timeWindow(-2000, 3000),
			timeWindow(-1000, 4000),
			timeWindow(0, 5000)));

	when(mockContext.getCurrentProcessingTime()).thenReturn(4999L);
	assertThat(assigner.assignWindows("String", Long.MIN_VALUE, mockContext), containsInAnyOrder(
			timeWindow(0, 5000),
			timeWindow(1000, 6000),
			timeWindow(2000, 7000),
			timeWindow(3000, 8000),
			timeWindow(4000, 9000)));

	when(mockContext.getCurrentProcessingTime()).thenReturn(5000L);
	assertThat(assigner.assignWindows("String", Long.MIN_VALUE, mockContext), containsInAnyOrder(
			timeWindow(1000, 6000),
			timeWindow(2000, 7000),
			timeWindow(3000, 8000),
			timeWindow(4000, 9000),
			timeWindow(5000, 10000)));
}

Example 20

Source File: WindowOperator.java From flink with Apache License 2.0

4 votes

@Override
	public void open() throws Exception {
		super.open();

		this.numLateRecordsDropped = metrics.counter(LATE_ELEMENTS_DROPPED_METRIC_NAME);
		timestampedCollector = new TimestampedCollector<>(output);

		internalTimerService =
				getInternalTimerService("window-timers", windowSerializer, this);

		triggerContext = new Context(null, null);
		processContext = new WindowContext(null);

		windowAssignerContext = new WindowAssigner.WindowAssignerContext() {
			@Override
			public long getCurrentProcessingTime() {
				return internalTimerService.currentProcessingTime();
			}
		};

		// create (or restore) the state that hold the actual window contents
		// NOTE - the state may be null in the case of the overriding evicting window operator
		if (windowStateDescriptor != null) {
			windowState = (InternalAppendingState<K, W, IN, ACC, ACC>) getOrCreateKeyedState(windowSerializer, windowStateDescriptor);
		}

		// create the typed and helper states for merging windows
		if (windowAssigner instanceof MergingWindowAssigner) {

			// store a typed reference for the state of merging windows - sanity check
			if (windowState instanceof InternalMergingState) {
				windowMergingState = (InternalMergingState<K, W, IN, ACC, ACC>) windowState;
			}
			// TODO this sanity check should be here, but is prevented by an incorrect test (pending validation)
			// TODO see WindowOperatorTest.testCleanupTimerWithEmptyFoldingStateForSessionWindows()
			// TODO activate the sanity check once resolved
//			else if (windowState != null) {
//				throw new IllegalStateException(
//						"The window uses a merging assigner, but the window state is not mergeable.");
//			}

			@SuppressWarnings("unchecked")
			final Class<Tuple2<W, W>> typedTuple = (Class<Tuple2<W, W>>) (Class<?>) Tuple2.class;

			final TupleSerializer<Tuple2<W, W>> tupleSerializer = new TupleSerializer<>(
					typedTuple,
					new TypeSerializer[] {windowSerializer, windowSerializer});

			final ListStateDescriptor<Tuple2<W, W>> mergingSetsStateDescriptor =
					new ListStateDescriptor<>("merging-window-set", tupleSerializer);

			// get the state that stores the merging sets
			mergingSetsState = (InternalListState<K, VoidNamespace, Tuple2<W, W>>)
					getOrCreateKeyedState(VoidNamespaceSerializer.INSTANCE, mergingSetsStateDescriptor);
			mergingSetsState.setCurrentNamespace(VoidNamespace.INSTANCE);
		}
	}