Examples with AlgebricksAbsolutePartitionConstraint org.apache.hyracks.algebricks.common.constraints.AlgebricksAbsolutePartitionConstraint used on opensource projects

Example 16 with AlgebricksAbsolutePartitionConstraint

use of org.apache.hyracks.algebricks.common.constraints.AlgebricksAbsolutePartitionConstraint in project asterixdb by apache.

the class MetadataProvider method getWriteFileRuntime.

@Override
public Pair<IPushRuntimeFactory, AlgebricksPartitionConstraint> getWriteFileRuntime(IDataSink sink, int[] printColumns, IPrinterFactory[] printerFactories, RecordDescriptor inputDesc) {
    FileSplitDataSink fsds = (FileSplitDataSink) sink;
    FileSplitSinkId fssi = fsds.getId();
    FileSplit fs = fssi.getFileSplit();
    File outFile = new File(fs.getPath());
    String nodeId = fs.getNodeName();
    SinkWriterRuntimeFactory runtime = new SinkWriterRuntimeFactory(printColumns, printerFactories, outFile, getWriterFactory(), inputDesc);
    AlgebricksPartitionConstraint apc = new AlgebricksAbsolutePartitionConstraint(new String[] { nodeId });
    return new Pair<>(runtime, apc);
}

Example 17 with AlgebricksAbsolutePartitionConstraint

use of org.apache.hyracks.algebricks.common.constraints.AlgebricksAbsolutePartitionConstraint in project asterixdb by apache.

the class StoragePathUtil method splitProviderAndPartitionConstraints.

public static Pair<IFileSplitProvider, AlgebricksPartitionConstraint> splitProviderAndPartitionConstraints(FileSplit[] splits) {
    IFileSplitProvider splitProvider = new ConstantFileSplitProvider(splits);
    String[] loc = new String[splits.length];
    for (int p = 0; p < splits.length; p++) {
        loc[p] = splits[p].getNodeName();
    }
    AlgebricksPartitionConstraint pc = new AlgebricksAbsolutePartitionConstraint(loc);
    return new Pair<>(splitProvider, pc);
}

Example 18 with AlgebricksAbsolutePartitionConstraint

use of org.apache.hyracks.algebricks.common.constraints.AlgebricksAbsolutePartitionConstraint in project asterixdb by apache.

the class LocalFSInputStreamFactory method configurePartitionConstraint.

private void configurePartitionConstraint() throws AsterixException {
    Set<String> locs = new TreeSet<>();
    for (int i = 0; i < inputFileSplits.length; i++) {
        locs.add(inputFileSplits[i].getNodeName());
    }
    constraints = new AlgebricksAbsolutePartitionConstraint(locs.toArray(new String[locs.size()]));
}