Java ValueStateDescriptor类代码示例

import org.apache.flink.api.common.state.ValueStateDescriptor; //导入依赖的package包/类
@Override
public void open(Configuration config) {
  ValueStateDescriptor<AbstractStatisticsWrapper<AisMessage>> descriptor =
      new ValueStateDescriptor<AbstractStatisticsWrapper<AisMessage>>("trajectoryStatistics",
          TypeInformation.of(new TypeHint<AbstractStatisticsWrapper<AisMessage>>() {}));

  statisticsOfTrajectory = getRuntimeContext().getState(descriptor);

}
 

import org.apache.flink.api.common.state.ValueStateDescriptor; //导入依赖的package包/类
/**
 * Override open from RichFunctions set
 * This method allow the retrieval of the state and set it as queryable
 */
@Override
@SuppressWarnings("unchecked")
public void open(Configuration config) {
    ValueStateDescriptor descriptor =
            new ValueStateDescriptor(
                    "consumption",
                    LampConsumption.class); // default value of the state, if nothing was set
    this.consumption = getRuntimeContext().getState(descriptor);
    descriptor.setQueryable("consumption-list-api");
}
 

import org.apache.flink.api.common.state.ValueStateDescriptor; //导入依赖的package包/类
FlinkCombiningStateWithContext(
    KeyedStateBackend<ByteBuffer> flinkStateBackend,
    StateTag<CombiningState<InputT, AccumT, OutputT>> address,
    CombineWithContext.CombineFnWithContext<InputT, AccumT, OutputT> combineFn,
    StateNamespace namespace,
    Coder<AccumT> accumCoder,
    FlinkStateInternals<K> flinkStateInternals,
    CombineWithContext.Context context) {

  this.namespace = namespace;
  this.address = address;
  this.combineFn = combineFn;
  this.flinkStateBackend = flinkStateBackend;
  this.flinkStateInternals = flinkStateInternals;
  this.context = context;

  flinkStateDescriptor = new ValueStateDescriptor<>(
      address.getId(), new CoderTypeSerializer<>(accumCoder));
}
 

import org.apache.flink.api.common.state.ValueStateDescriptor; //导入依赖的package包/类
@Override
public void initializeState(StateInitializationContext context) throws Exception {
	super.initializeState(context);

	if (nfaOperatorState == null) {
		nfaOperatorState = getRuntimeContext().getState(
			new ValueStateDescriptor<>(
					NFA_OPERATOR_STATE_NAME,
					new NFA.NFASerializer<>(inputSerializer)));
	}

	if (elementQueueState == null) {
		elementQueueState = getRuntimeContext().getMapState(
				new MapStateDescriptor<>(
						EVENT_QUEUE_STATE_NAME,
						LongSerializer.INSTANCE,
						new ListSerializer<>(inputSerializer)
				)
		);
	}
}
 

import org.apache.flink.api.common.state.ValueStateDescriptor; //导入依赖的package包/类
/**
 * Verify that {@link ValueStateDescriptor} allows {@code null} as default.
 */
@Test
public void testValueStateNullAsDefaultValue() throws Exception {
	AbstractKeyedStateBackend<Integer> backend = createKeyedBackend(IntSerializer.INSTANCE);

	ValueStateDescriptor<String> kvId = new ValueStateDescriptor<>("id", String.class, null);

	ValueState<String> state = backend.getPartitionedState(VoidNamespace.INSTANCE, VoidNamespaceSerializer.INSTANCE, kvId);

	backend.setCurrentKey(1);
	assertNull(state.value());

	state.update("Ciao");
	assertEquals("Ciao", state.value());

	state.clear();
	assertNull(state.value());

	backend.dispose();
}
 

import org.apache.flink.api.common.state.ValueStateDescriptor; //导入依赖的package包/类
/**
 * Verify that an empty {@code ValueState} will yield the default value.
 */
@Test
public void testValueStateDefaultValue() throws Exception {
	AbstractKeyedStateBackend<Integer> backend = createKeyedBackend(IntSerializer.INSTANCE);

	ValueStateDescriptor<String> kvId = new ValueStateDescriptor<>("id", String.class, "Hello");

	ValueState<String> state = backend.getPartitionedState(VoidNamespace.INSTANCE, VoidNamespaceSerializer.INSTANCE, kvId);

	backend.setCurrentKey(1);
	assertEquals("Hello", state.value());

	state.update("Ciao");
	assertEquals("Ciao", state.value());

	state.clear();
	assertEquals("Hello", state.value());

	backend.dispose();
}
 

import org.apache.flink.api.common.state.ValueStateDescriptor; //导入依赖的package包/类
@Test
public void testCopyDefaultValue() throws Exception {
	final AbstractKeyedStateBackend<Integer> backend = createKeyedBackend(IntSerializer.INSTANCE);

	ValueStateDescriptor<IntValue> kvId = new ValueStateDescriptor<>("id", IntValue.class, new IntValue(-1));

	ValueState<IntValue> state = backend.getPartitionedState(VoidNamespace.INSTANCE, VoidNamespaceSerializer.INSTANCE, kvId);

	backend.setCurrentKey(1);
	IntValue default1 = state.value();

	backend.setCurrentKey(2);
	IntValue default2 = state.value();

	assertNotNull(default1);
	assertNotNull(default2);
	assertEquals(default1, default2);
	assertFalse(default1 == default2);

	backend.dispose();
}
 

import org.apache.flink.api.common.state.ValueStateDescriptor; //导入依赖的package包/类
/**
 * Publishes the keyed stream as a queryable ValueState instance.
 *
 * @param queryableStateName Name under which to the publish the queryable state instance
 * @param stateDescriptor State descriptor to create state instance from
 * @return Queryable state instance
 */
@PublicEvolving
public QueryableStateStream<KEY, T> asQueryableState(
		String queryableStateName,
		ValueStateDescriptor<T> stateDescriptor) {

	transform("Queryable state: " + queryableStateName,
			getType(),
			new QueryableValueStateOperator<>(queryableStateName, stateDescriptor));

	stateDescriptor.initializeSerializerUnlessSet(getExecutionConfig());

	return new QueryableStateStream<>(
			queryableStateName,
			stateDescriptor,
			getKeyType().createSerializer(getExecutionConfig()));
}
 

import org.apache.flink.api.common.state.ValueStateDescriptor; //导入依赖的package包/类
@Override
public void open() throws Exception {
	super.open();

	if (serializedInitialValue == null) {
		throw new RuntimeException("No initial value was serialized for the fold " +
				"operator. Probably the setOutputType method was not called.");
	}

	try (ByteArrayInputStream bais = new ByteArrayInputStream(serializedInitialValue);
		DataInputViewStreamWrapper in = new DataInputViewStreamWrapper(bais)) {
		initialValue = outTypeSerializer.deserialize(in);
	}

	ValueStateDescriptor<OUT> stateId = new ValueStateDescriptor<>(STATE_NAME, outTypeSerializer);
	values = getPartitionedState(stateId);
}
 

import org.apache.flink.api.common.state.ValueStateDescriptor; //导入依赖的package包/类
@Test
public void testValueStateInstantiation() throws Exception {

	final ExecutionConfig config = new ExecutionConfig();
	config.registerKryoType(Path.class);

	final AtomicReference<Object> descriptorCapture = new AtomicReference<>();

	StreamingRuntimeContext context = new StreamingRuntimeContext(
			createDescriptorCapturingMockOp(descriptorCapture, config),
			createMockEnvironment(),
			Collections.<String, Accumulator<?, ?>>emptyMap());

	ValueStateDescriptor<TaskInfo> descr = new ValueStateDescriptor<>("name", TaskInfo.class);
	context.getState(descr);

	StateDescriptor<?, ?> descrIntercepted = (StateDescriptor<?, ?>) descriptorCapture.get();
	TypeSerializer<?> serializer = descrIntercepted.getSerializer();

	// check that the Path class is really registered, i.e., the execution config was applied
	assertTrue(serializer instanceof KryoSerializer);
	assertTrue(((KryoSerializer<?>) serializer).getKryo().getRegistration(Path.class).getId() > 0);
}
 

import org.apache.flink.api.common.state.ValueStateDescriptor; //导入依赖的package包/类
@Override
public void initializeState(StateInitializationContext context) throws Exception {
    super.initializeState(context);

    if (this.engineState == null) {
        this.engineState = getRuntimeContext().getState(new ValueStateDescriptor<>(ESPER_SERVICE_PROVIDER_STATE, new EsperEngineSerializer()));
    }
}
 

import org.apache.flink.api.common.state.ValueStateDescriptor; //导入依赖的package包/类
@Override
public void open(Configuration config) {
  ValueStateDescriptor<PriorityQueue<Tuple3<String, Long, String>>> descriptor =
      new ValueStateDescriptor<>(
      // state name
          "sorted-raw-messages",
          // type information of state
          TypeInformation.of(new TypeHint<PriorityQueue<Tuple3<String, Long, String>>>() {}));
  queueState = getRuntimeContext().getState(descriptor);
}
 

import org.apache.flink.api.common.state.ValueStateDescriptor; //导入依赖的package包/类
@Override
public void open(Configuration config) {
  ValueStateDescriptor<Long> descriptor =
      new ValueStateDescriptor<Long>("lastTimestamp", TypeInformation.of(new TypeHint<Long>() {}));
  lastTimestamp = getRuntimeContext().getState(descriptor);

}
 

import org.apache.flink.api.common.state.ValueStateDescriptor; //导入依赖的package包/类
@Override
public void open(Configuration config) {
  ValueStateDescriptor<PriorityQueue<AisMessage>> descriptor = new ValueStateDescriptor<>(
  // state name
      "sorted-ais-messages",
      // type information of state
      TypeInformation.of(new TypeHint<PriorityQueue<AisMessage>>() {}));
  queueState = getRuntimeContext().getState(descriptor);
}
 

import org.apache.flink.api.common.state.ValueStateDescriptor; //导入依赖的package包/类
@Override
public void open(Configuration parameters) throws Exception {
	ValueStateDescriptor<EventStateMachine.State> descriptor = new ValueStateDescriptor<>(
			"state", // the state name
			TypeInformation.of(EventStateMachine.State.class)); // type information
	state = getRuntimeContext().getState(descriptor);
}