taskContext.registerObject(EndOfStreamStates.class.getName(), new EndOfStreamStates(taskContext.getTaskModel().getSystemStreamPartitions(), producerTaskCounts)); taskContext.registerObject(WatermarkStates.class.getName(), new WatermarkStates(taskContext.getTaskModel().getSystemStreamPartitions(), producerTaskCounts, context.getContainerContext().getContainerMetricsRegistry()));
taskContext.registerObject(EndOfStreamStates.class.getName(), new EndOfStreamStates(taskContext.getTaskModel().getSystemStreamPartitions(), producerTaskCounts)); taskContext.registerObject(WatermarkStates.class.getName(), new WatermarkStates(taskContext.getTaskModel().getSystemStreamPartitions(), producerTaskCounts, context.getContainerContext().getContainerMetricsRegistry()));
taskContext.registerObject(EndOfStreamStates.class.getName(), new EndOfStreamStates(taskContext.getTaskModel().getSystemStreamPartitions(), producerTaskCounts)); taskContext.registerObject(WatermarkStates.class.getName(), new WatermarkStates(taskContext.getTaskModel().getSystemStreamPartitions(), producerTaskCounts, context.getContainerContext().getContainerMetricsRegistry()));
taskContext.registerObject(EndOfStreamStates.class.getName(), new EndOfStreamStates(taskContext.getTaskModel().getSystemStreamPartitions(), producerTaskCounts)); taskContext.registerObject(WatermarkStates.class.getName(), new WatermarkStates(taskContext.getTaskModel().getSystemStreamPartitions(), producerTaskCounts, context.getContainerContext().getContainerMetricsRegistry()));
taskContext.registerObject(EndOfStreamStates.class.getName(), new EndOfStreamStates(taskContext.getTaskModel().getSystemStreamPartitions(), producerTaskCounts)); taskContext.registerObject(WatermarkStates.class.getName(), new WatermarkStates(taskContext.getTaskModel().getSystemStreamPartitions(), producerTaskCounts, context.getContainerContext().getContainerMetricsRegistry()));
/** * Given a registered object, fetchObject should get it. If an object is not registered at a key, then fetchObject * should return null. */ @Test public void testRegisterAndFetchObject() { String value = "hello world"; taskContext.registerObject("key", value); assertEquals(value, taskContext.fetchObject("key")); assertNull(taskContext.fetchObject("not a key")); } }