private List<RestrictedMapRDBSubScanSpec> getEndPointFragmentMapping(int minorFragmentId) { List<RestrictedMapRDBSubScanSpec> restrictedSubScanSpecList = Lists.newArrayList(); List<MapRDBSubScanSpec> subScanSpecList = endpointFragmentMapping.get(minorFragmentId); for (MapRDBSubScanSpec s : subScanSpecList) { restrictedSubScanSpecList.add((RestrictedMapRDBSubScanSpec) s); } return restrictedSubScanSpecList; }
@Override public List<EndpointAffinity> getOperatorAffinity() { if (affinities == null) { affinities = AffinityCreator.getAffinityMap(Lists.newArrayList(partitionWorkMap.values())); } return affinities; }
@Override public Object readObject() { List<Object> valList = Lists.newArrayList(); for (int i = repeatedHolder.start; i < repeatedHolder.end; i++) { valList.add(repeatedHolder.vector.getAccessor().getObject(i)); } return valList; }
@Override public Object readObject() { List<Object> valList = Lists.newArrayList(); for (int i = repeatedHolder.start; i < repeatedHolder.end; i++) { valList.add(repeatedHolder.vector.getAccessor().getObject(i)); } return valList; }
@Override public Object readObject() { List<Object> valList = Lists.newArrayList(); for (int i = repeatedHolder.start; i < repeatedHolder.end; i++) { valList.add(repeatedHolder.vector.getAccessor().getObject(i)); } return valList; }
@Override public Object readObject() { List<Object> valList = Lists.newArrayList(); for (int i = repeatedHolder.start; i < repeatedHolder.end; i++) { valList.add(repeatedHolder.vector.getAccessor().getObject(i)); } return valList; }
@Override public Object readObject() { List<Object> valList = Lists.newArrayList(); for (int i = repeatedHolder.start; i < repeatedHolder.end; i++) { valList.add(repeatedHolder.vector.getAccessor().getObject(i)); } return valList; }
@Override public Object readObject() { List<Object> valList = Lists.newArrayList(); for (int i = repeatedHolder.start; i < repeatedHolder.end; i++) { valList.add(repeatedHolder.vector.getAccessor().getObject(i)); } return valList; }
@Override public Object readObject() { List<Object> valList = Lists.newArrayList(); for (int i = repeatedHolder.start; i < repeatedHolder.end; i++) { valList.add(repeatedHolder.vector.getAccessor().getObject(i)); } return valList; }
@Override public Object readObject() { List<Object> valList = Lists.newArrayList(); for (int i = repeatedHolder.start; i < repeatedHolder.end; i++) { valList.add(repeatedHolder.vector.getAccessor().getObject(i)); } return valList; }
@Override public Object readObject() { List<Object> valList = Lists.newArrayList(); for (int i = repeatedHolder.start; i < repeatedHolder.end; i++) { valList.add(repeatedHolder.vector.getAccessor().getObject(i)); } return valList; }
@Override public Object readObject() { List<Object> valList = Lists.newArrayList(); for (int i = repeatedHolder.start; i < repeatedHolder.end; i++) { valList.add(repeatedHolder.vector.getAccessor().getObject(i)); } return valList; }
@Override public LogicalExpression visitBooleanOperator(BooleanOperator op, Void value) throws RuntimeException { List<LogicalExpression> newArgs = Lists.newArrayList(); newArgs.addAll(op.args); Collections.sort(newArgs, costComparator); return new BooleanOperator(op.getName(), newArgs, op.getPosition()); }
@Override public KafkaSubScan getSpecificScan(int minorFragmentId) { List<PartitionScanWork> workList = assignments.get(minorFragmentId); List<KafkaPartitionScanSpec> scanSpecList = Lists.newArrayList(); for (PartitionScanWork work : workList) { scanSpecList.add(work.partitionScanSpec); } return new KafkaSubScan(getUserName(), kafkaStoragePlugin, columns, scanSpecList); }
@Override public LogicalExpression visitFunctionHolderExpression(FunctionHolderExpression holder, Void value) throws RuntimeException { List<LogicalExpression> args = Lists.newArrayList(); for (int i = 0; i < holder.args.size(); ++i) { LogicalExpression newExpr = holder.args.get(i).accept(this, value); assert newExpr != null; args.add(newExpr); } //replace with a new function call, since its argument could be changed. return holder.copy(args); }
@Test public void user2_db_u0_only_infoSchema() throws Exception { updateClient(org1Users[2]); fromInfoSchemaHelper(db_u0_only, newArrayList(u0_vw_voter_all_755), newArrayList(TableType.VIEW)); }
@Override public KuduSubScan getSpecificScan(int minorFragmentId) { List<KuduWork> workList = assignments.get(minorFragmentId); List<KuduSubScanSpec> scanSpecList = Lists.newArrayList(); for (KuduWork work : workList) { scanSpecList.add(new KuduSubScanSpec(getTableName(), work.getPartitionKeyStart(), work.getPartitionKeyEnd())); } return new KuduSubScan(kuduStoragePlugin, scanSpecList, this.columns); }
@Override public void setup(OperatorContext context, OutputMutator output) throws ExecutionSetupException { this.writer = new VectorContainerWriter(output, unionEnabled); messageReader = MessageReaderFactory.getMessageReader(kafkaMsgReader); messageReader.init(context.getManagedBuffer(), Lists.newArrayList(getColumns()), this.writer, this.enableAllTextMode, this.readNumbersAsDouble); msgItr = new MessageIterator(messageReader.getConsumer(plugin), subScanSpec, kafkaPollTimeOut); }
@Test public void testTableColumnsIndex() { ColumnListsCache cache = new ColumnListsCache(); List<FieldSchema> columns = Lists.newArrayList(); columns.add(new FieldSchema("f1", "int", null)); columns.add(new FieldSchema("f2", "int", null)); assertEquals(0, cache.addOrGet(columns)); }
@Test public void testPartitionColumnsIndex() { ColumnListsCache cache = new ColumnListsCache(); List<FieldSchema> columns = Lists.newArrayList(); columns.add(new FieldSchema("f1", "int", null)); columns.add(new FieldSchema("f2", "int", null)); cache.addOrGet(columns); columns.add(new FieldSchema("f3", "int", null)); assertEquals(1, cache.addOrGet(columns)); }