Java Code Examples for org.apache.flink.util.InstantiationUtil#writeObjectToConfig()

public void setOutputPartitioner(Partitioner<?> partitioner, int outputNum) {
	try {
		InstantiationUtil.writeObjectToConfig(partitioner, config, OUTPUT_PARTITIONER + outputNum);
	}
	catch (Throwable t) {
		throw new RuntimeException("Could not serialize custom partitioner.", t);
	}
}
 

public void setStateKeySerializer(TypeSerializer<?> serializer) {
	try {
		InstantiationUtil.writeObjectToConfig(serializer, this.config, STATE_KEY_SERIALIZER);
	} catch (IOException e) {
		throw new StreamTaskException("Could not serialize state key serializer.", e);
	}
}
 

public void setOutEdgesInOrder(List<StreamEdge> outEdgeList) {
	try {
		InstantiationUtil.writeObjectToConfig(outEdgeList, this.config, EDGES_IN_ORDER);
	} catch (IOException e) {
		throw new StreamTaskException("Could not serialize outputs in order.", e);
	}
}
 

public void setOutputPartitioner(Partitioner<?> partitioner, int outputNum) {
	try {
		InstantiationUtil.writeObjectToConfig(partitioner, config, OUTPUT_PARTITIONER + outputNum);
	}
	catch (Throwable t) {
		throw new RuntimeException("Could not serialize custom partitioner.", t);
	}
}
 

public void setTransitiveChainedTaskConfigs(Map<Integer, StreamConfig> chainedTaskConfigs) {

		try {
			InstantiationUtil.writeObjectToConfig(chainedTaskConfigs, this.config, CHAINED_TASK_CONFIG);
		} catch (IOException e) {
			throw new StreamTaskException("Could not serialize configuration.", e);
		}
	}
 

public void setOutEdgesInOrder(List<StreamEdge> outEdgeList) {
	try {
		InstantiationUtil.writeObjectToConfig(outEdgeList, this.config, EDGES_IN_ORDER);
	} catch (IOException e) {
		throw new StreamTaskException("Could not serialize outputs in order.", e);
	}
}
 

public void setInPhysicalEdges(List<StreamEdge> inEdges) {
	try {
		InstantiationUtil.writeObjectToConfig(inEdges, this.config, IN_STREAM_EDGES);
	} catch (IOException e) {
		throw new StreamTaskException("Cannot serialize inward edges.", e);
	}
}
 

private void setTypeSerializer(String key, TypeSerializer<?> typeWrapper) {
	try {
		InstantiationUtil.writeObjectToConfig(typeWrapper, this.config, key);
	} catch (IOException e) {
		throw new StreamTaskException("Could not serialize type serializer.", e);
	}
}
 

public void setInPhysicalEdges(List<StreamEdge> inEdges) {
	try {
		InstantiationUtil.writeObjectToConfig(inEdges, this.config, IN_STREAM_EDGES);
	} catch (IOException e) {
		throw new StreamTaskException("Cannot serialize inward edges.", e);
	}
}
 

public void addIterationAggregator(String name, Aggregator<?> aggregator) {
	int num = this.config.getInteger(ITERATION_NUM_AGGREGATORS, 0);
	this.config.setString(ITERATION_AGGREGATOR_NAME_PREFIX + num, name);
	try {
			InstantiationUtil.writeObjectToConfig(aggregator, this.config, ITERATION_AGGREGATOR_PREFIX + num);
	} catch (IOException e) {
			throw new RuntimeException("Error while writing the aggregator object to the task configuration.");
	}
	this.config.setInteger(ITERATION_NUM_AGGREGATORS, num + 1);
}
 

public void setChainedOutputs(List<StreamEdge> chainedOutputs) {
	try {
		InstantiationUtil.writeObjectToConfig(chainedOutputs, this.config, CHAINED_OUTPUTS);
	} catch (IOException e) {
		throw new StreamTaskException("Cannot serialize chained outputs.", e);
	}
}
 

public void setStreamOperator(StreamOperator<?> operator) {
	if (operator != null) {
		config.setClass(USER_FUNCTION, operator.getClass());

		try {
			InstantiationUtil.writeObjectToConfig(operator, this.config, SERIALIZEDUDF);
		} catch (IOException e) {
			throw new StreamTaskException("Cannot serialize operator object "
					+ operator.getClass() + ".", e);
		}
	}
}
 

private void setTypeSerializer(String key, TypeSerializer<?> typeWrapper) {
	try {
		InstantiationUtil.writeObjectToConfig(typeWrapper, this.config, key);
	} catch (IOException e) {
		throw new StreamTaskException("Could not serialize type serializer.", e);
	}
}
 

/**
 * Sets the default convergence criterion of a {@link DeltaIteration}
 *
 * @param aggregatorName
 * @param convCriterion
 */
public void setImplicitConvergenceCriterion(String aggregatorName, ConvergenceCriterion<?> convCriterion) {
	try {
		InstantiationUtil.writeObjectToConfig(convCriterion, this.config, ITERATION_IMPLICIT_CONVERGENCE_CRITERION);
	} catch (IOException e) {
		throw new RuntimeException("Error while writing the implicit convergence criterion object to the task configuration.");
	}
	this.config.setString(ITERATION_IMPLICIT_CONVERGENCE_CRITERION_AGG_NAME, aggregatorName);
}
 

public void setStatePartitioner(int input, KeySelector<?, ?> partitioner) {
	try {
		InstantiationUtil.writeObjectToConfig(partitioner, this.config, STATE_PARTITIONER + input);
	} catch (IOException e) {
		throw new StreamTaskException("Could not serialize state partitioner.", e);
	}
}
 

public void setStatePartitioner(int input, KeySelector<?, ?> partitioner) {
	try {
		InstantiationUtil.writeObjectToConfig(partitioner, this.config, STATE_PARTITIONER + input);
	} catch (IOException e) {
		throw new StreamTaskException("Could not serialize state partitioner.", e);
	}
}
 

public void addIterationAggregators(Collection<AggregatorWithName<?>> aggregators) {
	int num = this.config.getInteger(ITERATION_NUM_AGGREGATORS, 0);
	for (AggregatorWithName<?> awn : aggregators) {
		this.config.setString(ITERATION_AGGREGATOR_NAME_PREFIX + num, awn.getName());
		try {
			InstantiationUtil.writeObjectToConfig(awn.getAggregator(), this.config, ITERATION_AGGREGATOR_PREFIX + num);
		} catch (IOException e) {
			throw new RuntimeException("Error while writing the aggregator object to the task configuration.");
		}
		num++;
	}
	this.config.setInteger(ITERATION_NUM_AGGREGATORS, num);
}
 

public void addIterationAggregators(Collection<AggregatorWithName<?>> aggregators) {
	int num = this.config.getInteger(ITERATION_NUM_AGGREGATORS, 0);
	for (AggregatorWithName<?> awn : aggregators) {
		this.config.setString(ITERATION_AGGREGATOR_NAME_PREFIX + num, awn.getName());
		try {
			InstantiationUtil.writeObjectToConfig(awn.getAggregator(), this.config, ITERATION_AGGREGATOR_PREFIX + num);
		} catch (IOException e) {
			throw new RuntimeException("Error while writing the aggregator object to the task configuration.");
		}
		num++;
	}
	this.config.setInteger(ITERATION_NUM_AGGREGATORS, num);
}
 

public void setOutEdges(List<StreamEdge> outEdges) {
	try {
		InstantiationUtil.writeObjectToConfig(outEdges, this.config, OUT_STREAM_EDGES);
	} catch (IOException e) {
		throw new StreamTaskException("Cannot serialize outward edges.", e);
	}
}
 

@Override
public void writeParametersToConfig(Configuration config) {
	try {
		InstantiationUtil.writeObjectToConfig(comparator, config, CONFIG_KEY);
	}
	catch (Exception e) {
		throw new RuntimeException("Could not serialize comparator into the configuration.", e);
	}
}