Java SchemaPath.getSimplePath方法代码示例

import org.apache.drill.common.expression.SchemaPath; //导入方法依赖的package包/类
@Override
public void populatePartitionVectors(ValueVector[] vectors, List<PartitionLocation> partitions,
                                     BitSet partitionColumnBitSet, Map<Integer, String> fieldNameMap) {
  int record = 0;
  for (PartitionLocation partitionLocation: partitions) {
    for (int partitionColumnIndex : BitSets.toIter(partitionColumnBitSet)) {
      SchemaPath column = SchemaPath.getSimplePath(fieldNameMap.get(partitionColumnIndex));
      ((ParquetGroupScan) scanRel.getGroupScan()).populatePruningVector(vectors[partitionColumnIndex], record, column,
          partitionLocation.getEntirePartitionLocation());
    }
    record++;
  }

  for (ValueVector v : vectors) {
    if (v == null) {
      continue;
    }
    v.getMutator().setValueCount(partitions.size());
  }

}
 

import org.apache.drill.common.expression.SchemaPath; //导入方法依赖的package包/类
private MapVector getOrCreateFamilyVector(String familyName, boolean allocateOnCreate) {
  try {
    MapVector v = familyVectorMap.get(familyName);
    if(v == null) {
      SchemaPath column = SchemaPath.getSimplePath(familyName);
      MaterializedField field = MaterializedField.create(column, COLUMN_FAMILY_TYPE);
      v = outputMutator.addField(field, MapVector.class);
      if (allocateOnCreate) {
        v.allocateNew();
      }
      getColumns().add(column);
      familyVectorMap.put(familyName, v);
    }
    return v;
  } catch (SchemaChangeException e) {
    throw new DrillRuntimeException(e);
  }
}
 

import org.apache.drill.common.expression.SchemaPath; //导入方法依赖的package包/类
@Override
public Integer getIdIfValid(String name) {
  SchemaPath schemaPath = SchemaPath.getSimplePath(name);
  int id = partitionColumns.indexOf(schemaPath);
  if (id == -1) {
    return null;
  }
  return id;
}
 

import org.apache.drill.common.expression.SchemaPath; //导入方法依赖的package包/类
/**
 * Creates a copier that does a project for every Nth record from a VectorContainer incoming into VectorContainer
 * outgoing. Each Ordering in orderings generates a column, and evaluation of the expression associated with each
 * Ordering determines the value of each column. These records will later be sorted based on the values in each
 * column, in the same order as the orderings.
 *
 * @param sv4
 * @param incoming
 * @param outgoing
 * @param orderings
 * @return
 * @throws SchemaChangeException
 */
private SampleCopier getCopier(SelectionVector4 sv4, VectorContainer incoming, VectorContainer outgoing,
    List<Ordering> orderings, List<ValueVector> localAllocationVectors) throws SchemaChangeException {
  final ErrorCollector collector = new ErrorCollectorImpl();
  final ClassGenerator<SampleCopier> cg = CodeGenerator.getRoot(SampleCopier.TEMPLATE_DEFINITION,
      context.getFunctionRegistry());

  int i = 0;
  for (Ordering od : orderings) {
    final LogicalExpression expr = ExpressionTreeMaterializer.materialize(od.getExpr(), incoming, collector, context.getFunctionRegistry());
    SchemaPath schemaPath = SchemaPath.getSimplePath("f" + i++);
    TypeProtos.MajorType.Builder builder = TypeProtos.MajorType.newBuilder().mergeFrom(expr.getMajorType())
        .clearMode().setMode(TypeProtos.DataMode.REQUIRED);
    TypeProtos.MajorType newType = builder.build();
    MaterializedField outputField = MaterializedField.create(schemaPath, newType);
    if (collector.hasErrors()) {
      throw new SchemaChangeException(String.format(
          "Failure while trying to materialize incoming schema.  Errors:\n %s.", collector.toErrorString()));
    }

    ValueVector vector = TypeHelper.getNewVector(outputField, oContext.getAllocator());
    localAllocationVectors.add(vector);
    TypedFieldId fid = outgoing.add(vector);
    ValueVectorWriteExpression write = new ValueVectorWriteExpression(fid, expr, true);
    HoldingContainer hc = cg.addExpr(write);
    cg.getEvalBlock()._if(hc.getValue().eq(JExpr.lit(0)))._then()._return(JExpr.FALSE);
  }
  cg.rotateBlock();
  cg.getEvalBlock()._return(JExpr.TRUE);
  outgoing.buildSchema(BatchSchema.SelectionVectorMode.NONE);
  try {
    SampleCopier sampleCopier = context.getImplementationClass(cg);
    sampleCopier.setupCopier(context, sv4, incoming, outgoing);
    return sampleCopier;
  } catch (ClassTransformationException | IOException e) {
    throw new SchemaChangeException(e);
  }
}
 

import org.apache.drill.common.expression.SchemaPath; //导入方法依赖的package包/类
public static MaterializedField create(String path, MajorType type){
  SchemaPath p = SchemaPath.getSimplePath(path);
  return create(p, type);
}