Java UserCodeWrapper类代码示例

import org.apache.flink.api.common.operators.util.UserCodeWrapper; //导入依赖的package包/类
@Override
public void initializeOnMaster(ClassLoader loader) throws Exception {
	if (this.outputFormat == null) {
		TaskConfig cfg = new TaskConfig(getConfiguration());
		UserCodeWrapper<OutputFormat<?>> wrapper = cfg.<OutputFormat<?>>getStubWrapper(loader);
	
		if (wrapper == null) {
			throw new Exception("No output format present in OutputFormatVertex's task configuration.");
		}

		this.outputFormat = wrapper.getUserCodeObject(OutputFormat.class, loader);
		this.outputFormat.configure(cfg.getStubParameters());
	}
	
	if (this.outputFormat instanceof InitializeOnMaster) {
		((InitializeOnMaster) this.outputFormat).initializeGlobal(getParallelism());
	}
}
 

import org.apache.flink.api.common.operators.util.UserCodeWrapper; //导入依赖的package包/类
@Override
public void initializeOnMaster(ClassLoader loader) throws Exception {
	if (inputFormat == null) {
		TaskConfig cfg = new TaskConfig(getConfiguration());
		UserCodeWrapper<InputFormat<?, ?>> wrapper = cfg.<InputFormat<?, ?>>getStubWrapper(loader);
		
		if (wrapper == null) {
			throw new Exception("No input format present in InputFormatVertex's task configuration.");
		}
		
		inputFormat = wrapper.getUserCodeObject(InputFormat.class, loader);
		inputFormat.configure(cfg.getStubParameters());
	}
	
	setInputSplitSource(inputFormat);
}
 

import org.apache.flink.api.common.operators.util.UserCodeWrapper; //导入依赖的package包/类
public CrossOperatorBase(UserCodeWrapper<FT> udf, BinaryOperatorInformation<IN1, IN2, OUT> operatorInfo, String name) {
	super(udf, operatorInfo, name);
	
	if (this instanceof CrossWithSmall) {
		setCrossHint(CrossHint.SECOND_IS_SMALL);
	}
	else if (this instanceof CrossWithLarge) {
		setCrossHint(CrossHint.FIRST_IS_SMALL);
	}
}
 

import org.apache.flink.api.common.operators.util.UserCodeWrapper; //导入依赖的package包/类
public void setStubWrapper(UserCodeWrapper<?> wrapper) {
	try {
		InstantiationUtil.writeObjectToConfig(wrapper, this.config, STUB_OBJECT);
	} catch (IOException e) {
		throw new CorruptConfigurationException("Could not write the user code wrapper " + wrapper.getClass() + " : " + e.toString(), e);
	}
}
 

import org.apache.flink.api.common.operators.util.UserCodeWrapper; //导入依赖的package包/类
@SuppressWarnings("unchecked")
public <T> UserCodeWrapper<T> getStubWrapper(ClassLoader cl) {
	try {
		return (UserCodeWrapper<T>) InstantiationUtil.readObjectFromConfig(this.config, STUB_OBJECT, cl);
	} catch (ClassNotFoundException | IOException e) {
		throw new CorruptConfigurationException("Could not read the user code wrapper: " + e.getMessage(), e);
	}
}
 

import org.apache.flink.api.common.operators.util.UserCodeWrapper; //导入依赖的package包/类
private static <T extends InputFormat<?,?>> InputFormatVertex createInput(UserCodeWrapper<T> stub, String name, JobGraph graph,
		int degreeOfParallelism)
{
	InputFormatVertex inputVertex = new InputFormatVertex(name);
	graph.addVertex(inputVertex);
	
	inputVertex.setInvokableClass(DataSourceTask.class);
	inputVertex.setParallelism(degreeOfParallelism);

	TaskConfig inputConfig = new TaskConfig(inputVertex.getConfiguration());
	inputConfig.setStubWrapper(stub);
	
	return inputVertex;
}
 

import org.apache.flink.api.common.operators.util.UserCodeWrapper; //导入依赖的package包/类
/**
 * Creates a Builder with the provided {@link JoinFunction} implementation
 * 
 * @param udf The {@link JoinFunction} implementation for this Match operator.
 * @param keyClass The class of the key data type.
 * @param keyColumn1 The position of the key in the first input's records.
 * @param keyColumn2 The position of the key in the second input's records.
 */
protected Builder(UserCodeWrapper<JoinFunction> udf, Class<? extends Key<?>> keyClass, int keyColumn1, int keyColumn2) {
	this.udf = udf;
	this.keyClasses = new ArrayList<Class<? extends Key<?>>>();
	this.keyClasses.add(keyClass);
	this.keyColumns1 = new ArrayList<Integer>();
	this.keyColumns1.add(keyColumn1);
	this.keyColumns2 = new ArrayList<Integer>();
	this.keyColumns2.add(keyColumn2);
	this.inputs1 = new ArrayList<Operator<Record>>();
	this.inputs2 = new ArrayList<Operator<Record>>();
	this.broadcastInputs = new HashMap<String, Operator<Record>>();
}
 

import org.apache.flink.api.common.operators.util.UserCodeWrapper; //导入依赖的package包/类
/**
 * Creates a Builder with the provided {@link CoGroupFunction} implementation.
 * 
 * @param udf The {@link CoGroupFunction} implementation for this CoGroup operator.
 * @param keyClass The class of the key data type.
 * @param keyColumn1 The position of the key in the first input's records.
 * @param keyColumn2 The position of the key in the second input's records.
 */
protected Builder(UserCodeWrapper<org.apache.flink.api.common.functions.CoGroupFunction<Record, Record, Record>> udf, Class<? extends Key<?>> keyClass,
		int keyColumn1, int keyColumn2)
{
	this.udf = udf;
	this.keyClasses = new ArrayList<Class<? extends Key<?>>>();
	this.keyClasses.add(keyClass);
	this.keyColumns1 = new ArrayList<Integer>();
	this.keyColumns1.add(keyColumn1);
	this.keyColumns2 = new ArrayList<Integer>();
	this.keyColumns2.add(keyColumn2);
	this.inputs1 = new ArrayList<Operator<Record>>();
	this.inputs2 = new ArrayList<Operator<Record>>();
	this.broadcastInputs = new HashMap<String, Operator<Record>>();
}
 

import org.apache.flink.api.common.operators.util.UserCodeWrapper; //导入依赖的package包/类
/**
 * Creates a Builder with the provided {@link ReduceFunction} implementation.
 * 
 * @param udf The {@link ReduceFunction} implementation for this Reduce contract.
 */
public static Builder builder(ReduceFunction udf) {
	UserCodeWrapper<ReduceFunction> original = new UserCodeObjectWrapper<ReduceFunction>(udf);
	UserCodeWrapper<GroupReduceFunction<Record, Record>> wrapped =
			new UserCodeObjectWrapper<GroupReduceFunction<Record, Record>>(new WrappingReduceFunction(udf));
	
	return new Builder(original, wrapped);
}
 

import org.apache.flink.api.common.operators.util.UserCodeWrapper; //导入依赖的package包/类
/**
 * Creates a Builder with the provided {@link ReduceFunction} implementation.
 * 
 * @param wrappedUdf The {@link ReduceFunction} implementation for this Reduce contract.
 */
private Builder(UserCodeWrapper<ReduceFunction> originalUdf, UserCodeWrapper<GroupReduceFunction<Record, Record>> wrappedUdf) {
	this.originalUdf = originalUdf;
	this.udfWrapper = wrappedUdf;
	this.keyClasses = new ArrayList<Class<? extends Key<?>>>();
	this.keyColumns = new ArrayList<Integer>();
	this.inputs = new ArrayList<Operator<Record>>();
	this.broadcastInputs = new HashMap<String, Operator<Record>>();
}
 

import org.apache.flink.api.common.operators.util.UserCodeWrapper; //导入依赖的package包/类
/**
 * Creates a Builder with the provided {@link CrossFunction} implementation.
 * 
 * @param udf The {@link CrossFunction} implementation for this Cross operator.
 */
protected Builder(UserCodeWrapper<CrossFunction> udf) {
	this.udf = udf;
	this.inputs1 = new ArrayList<Operator<Record>>();
	this.inputs2 = new ArrayList<Operator<Record>>();
	this.broadcastInputs = new HashMap<String, Operator<Record>>();
}
 

import org.apache.flink.api.common.operators.util.UserCodeWrapper; //导入依赖的package包/类
public InnerJoinOperatorBase(UserCodeWrapper<FT> udf, BinaryOperatorInformation<IN1, IN2, OUT> operatorInfo,
		int[] keyPositions1, int[] keyPositions2, String name) {
	super(udf, operatorInfo, keyPositions1, keyPositions2, name);
}
 

import org.apache.flink.api.common.operators.util.UserCodeWrapper; //导入依赖的package包/类
@Override
public UserCodeWrapper<?> getUserCodeWrapper() {
	return null;
}
 

import org.apache.flink.api.common.operators.util.UserCodeWrapper; //导入依赖的package包/类
public CoGroupRawOperatorBase(UserCodeWrapper<FT> udf, BinaryOperatorInformation<IN1, IN2, OUT> operatorInfo, int[] keyPositions1, int[] keyPositions2, String name) {
	super(udf, operatorInfo, keyPositions1, keyPositions2, name);
	this.combinableFirst = false;
	this.combinableSecond = false;
}
 

import org.apache.flink.api.common.operators.util.UserCodeWrapper; //导入依赖的package包/类
public MapOperatorBase(UserCodeWrapper<FT> udf, UnaryOperatorInformation<IN, OUT> operatorInfo, String name) {
	super(udf, operatorInfo, name);
}