Package org.apache.drill.common.expression

Examples of org.apache.drill.common.expression.PathSegment

211
212
213
214
215
216
217
218
219
220
221
  }

  /** hack to make ref and full work together... need to figure out if this is still necessary. **/
  private FieldReference getRef(NamedExpression e){
    FieldReference ref = e.getRef();
    PathSegment seg = ref.getRootSegment();

//    if(seg.isNamed() && "output".contentEquals(seg.getNameSegment().getPath())){
//      return new FieldReference(ref.getPath().toString().subSequence(7, ref.getPath().length()), ref.getPosition());
//    }
    return ref;
View Full Code Here
100
101
102
103
104
105
106
107
108
109
110
111
112
        }
        rowKeyOnly = false;
        NameSegment root = column.getRootSegment();
        byte[] family = root.getPath().getBytes();
        transformed.add(SchemaPath.getSimplePath(root.getPath()));
        PathSegment child = root.getChild();
        if (child != null && child.isNamed()) {
          byte[] qualifier = child.getNameSegment().getPath().getBytes();
          hbaseScan.addColumn(family, qualifier);
        } else {
          hbaseScan.addFamily(family);
        }
      }
View Full Code Here
99
100
101
102
103
104
105
106
107
108
109
110
111
        }
        rowKeyOnly = false;
        NameSegment root = column.getRootSegment();
        byte[] family = root.getPath().getBytes();
        transformed.add(SchemaPath.getSimplePath(root.getPath()));
        PathSegment child = root.getChild();
        if (child != null && child.isNamed()) {
          byte[] qualifier = child.getNameSegment().getPath().getBytes();
          hbaseScan.addColumn(family, qualifier);
        } else {
          hbaseScan.addFamily(family);
        }
      }
View Full Code Here
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
  public static MessageType getProjection(MessageType schema, Collection<SchemaPath> columns) {
    MessageType projection = null;
    for (SchemaPath path : columns) {
      List<String> segments = Lists.newArrayList();
      PathSegment rootSegment = path.getRootSegment();
      PathSegment seg = rootSegment;
      String messageName = schema.getName();
      while(seg != null){
        if(seg.isNamed()) {
          segments.add(seg.getNameSegment().getPath());
        }
        seg = seg.getChild();
      }
      String[] pathSegments = new String[segments.size()];
      segments.toArray(pathSegments);
      Type type = null;
      try {
View Full Code Here
239
240
241
242
243
244
245
246
247
    }
  }

  private FieldReference getRef(NamedExpression e) {
    FieldReference ref = e.getRef();
    PathSegment seg = ref.getRootSegment();

    return ref;
  }
View Full Code Here
124
125
126
127
128
129
130
131
132
133
134
      return null;
    }

    RefFieldsVisitor v = new RefFieldsVisitor(rowType);
    for (RexNode exp : projects) {
      PathSegment segment = exp.accept(v);
      v.addColumn(segment);
    }

    return v.getInfo();
View Full Code Here
268
269
270
271
272
273
274
275
276
277
278
279
280
281
    }

    @Override
    public PathSegment visitCall(RexCall call) {
      if ("ITEM".equals(call.getOperator().getName())) {
        PathSegment mapOrArray = call.operands.get(0).accept(this);
        if (mapOrArray != null) {
          if (call.operands.get(1) instanceof RexLiteral) {
            return mapOrArray.cloneWithNewChild(convertLiteral((RexLiteral) call.operands.get(1)));
          }
          return mapOrArray;
        }
      } else {
        for (RexNode operand : call.operands) {
View Full Code Here
236
237
238
239
240
241
242
243
244
245
246
  }

  /** hack to make ref and full work together... need to figure out if this is still necessary. **/
  private FieldReference getRef(NamedExpression e) {
    FieldReference ref = e.getRef();
    PathSegment seg = ref.getRootSegment();

//    if (seg.isNamed() && "output".contentEquals(seg.getNameSegment().getPath())) {
//      return new FieldReference(ref.getPath().toString().subSequence(7, ref.getPath().length()), ref.getPosition());
//    }
    return ref;
View Full Code Here
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
          break done;
        }
      }

      for (SchemaPath sp : jsonReaderWithState.getNullColumns()) {
        PathSegment root = sp.getRootSegment();
        BaseWriter.MapWriter fieldWriter = writer.rootAsMap();
        if (root.getChild() != null && !root.getChild().isArray()) {
          fieldWriter = fieldWriter.map(root.getNameSegment().getPath());
          while (root.getChild().getChild() != null
              && !root.getChild().isArray()) {
            fieldWriter = fieldWriter.map(root.getChild().getNameSegment()
                .getPath());
            root = root.getChild();
          }
          fieldWriter.integer(root.getChild().getNameSegment().getPath());
        } else {
          fieldWriter.integer(root.getNameSegment().getPath());
        }
      }

      writer.setValueCount(docCount);
      logger.debug("Took {} ms to get {} records",
View Full Code Here
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
        case WRITE_FAILED:
          break outside;
        };
      }
      for (SchemaPath sp :jsonReader.getNullColumns() ) {
        PathSegment root = sp.getRootSegment();
        BaseWriter.MapWriter fieldWriter = writer.rootAsMap();
        while (root.getChild() != null && ! root.getChild().isArray()) {
          fieldWriter = fieldWriter.map(root.getNameSegment().getPath());
          root = root.getChild();
        }
        fieldWriter.integer(root.getNameSegment().getPath());
      }

      writer.setValueCount(recordCount);
      return recordCount;
View Full Code Here
TOP

Related Classes of org.apache.drill.common.expression.PathSegment

  • org.apache.drill.common.logical.data.Project
  • org.apache.drill.exec.expr.EvaluationVisitor$EvalVisitor
  • org.apache.drill.exec.physical.impl.flatten.FlattenRecordBatch
  • org.apache.drill.exec.physical.impl.project.ProjectRecordBatch
  • org.apache.drill.exec.planner.physical.PrelUtil
  • org.apache.drill.exec.planner.physical.PrelUtil$RefFieldsVisitor
  • org.apache.drill.exec.record.MaterializedField
  • org.apache.drill.exec.record.SimpleVectorWrapper
  • org.apache.drill.exec.record.TypedFieldId
  • org.apache.drill.exec.store.easy.json.JSONRecordReader2
    • Copyright © 2018 www.massapicom. All rights reserved.
    All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.