protected PrestoThriftSplitBatch getIndexSplitsSync( PrestoThriftSchemaTableName schemaTableName, List<String> indexColumnNames, PrestoThriftPageResult keys, int maxSplitCount, PrestoThriftNullableToken nextToken) throws PrestoThriftServiceException { throw new PrestoThriftServiceException("Index join is not supported", false); }
public static PrestoThriftBlock doubleData(PrestoThriftDouble doubleData) { return new PrestoThriftBlock(null, null, doubleData, null, null, null, null, null, null, null); }
public ThriftTableMetadata(PrestoThriftTableMetadata thriftTableMetadata, TypeManager typeManager) { this(thriftTableMetadata.getSchemaTableName().toSchemaTableName(), columnMetadata(thriftTableMetadata.getColumns(), typeManager), Optional.ofNullable(thriftTableMetadata.getComment()), thriftTableMetadata.getIndexableKeys() != null ? thriftTableMetadata.getIndexableKeys() : ImmutableList.of()); }
private static PrestoThriftPageResult pageResult(int value, PrestoThriftId nextToken) { return new PrestoThriftPageResult(ImmutableList.of(integerData(new PrestoThriftInteger(null, new int[] {value}))), 1, nextToken); } }
@Override public ListenableFuture<PrestoThriftPageResult> getRows(PrestoThriftId splitId, List<String> columns, long maxBytes, PrestoThriftNullableToken nextToken) { if (rowsPerSplit == 0) { return immediateFuture(new PrestoThriftPageResult(ImmutableList.of(), 0, null)); } int key = Ints.fromByteArray(splitId.getId()); int offset = nextToken.getToken() != null ? Ints.fromByteArray(nextToken.getToken().getId()) : 0; PrestoThriftId newNextToken = offset + 1 < rowsPerSplit ? new PrestoThriftId(Ints.toByteArray(offset + 1)) : null; return immediateFuture(pageResult(key * 10 + offset, newNextToken)); }
private void testPageReadWrite(Random random, int records) { testReadWrite(random, records, blocks -> { List<PrestoThriftBlock> columnBlocks = new ArrayList<>(columns.size()); for (int i = 0; i < columns.size(); i++) { columnBlocks.add(fromBlock(blocks.get(i), columns.get(i).getType())); } return new PrestoThriftPageResult(columnBlocks, records, null); }); }
private PrestoThriftNullableTableMetadata getTableMetadata(SchemaTableName schemaTableName) { // treat invalid names as not found PrestoThriftSchemaTableName name; try { name = new PrestoThriftSchemaTableName(schemaTableName); } catch (IllegalArgumentException e) { return new PrestoThriftNullableTableMetadata(null); } try { return client.get().getTableMetadata(name); } catch (PrestoThriftServiceException | TException e) { throw toPrestoException(e); } } }
public SchemaTableName toSchemaTableName() { return new SchemaTableName(getSchemaName(), getTableName()); } }
@Test public void testSummarize() { assertEquals(summarize(bytes()), ""); assertEquals(summarize(bytes(1)), "01"); assertEquals(summarize(bytes(255, 254, 253, 252, 251, 250, 249)), "FFFEFDFCFBFAF9"); assertEquals(summarize(bytes(0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 249, 250, 251, 252, 253, 254, 255)), "00010203040506070809F9FAFBFCFDFEFF"); assertEquals(summarize(bytes(0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 247, 248, 249, 250, 251, 252, 253, 254, 255)), "0001020304050607..F8F9FAFBFCFDFEFF"); }
@ThriftConstructor public PrestoThriftSchemaTableName(String schemaName, String tableName) { this.schemaName = checkValidName(schemaName); this.tableName = checkValidName(tableName); }
private static String nextString(Random random) { return nextString(random, MAX_VARCHAR_GENERATED_LENGTH); }
public static PrestoThriftBlock jsonData(PrestoThriftJson jsonData) { return new PrestoThriftBlock(null, null, null, null, null, null, null, jsonData, null, null); }
public static PrestoThriftBlock integerData(PrestoThriftInteger integerData) { return new PrestoThriftBlock(integerData, null, null, null, null, null, null, null, null, null); }
public static PrestoThriftBlock varcharData(PrestoThriftVarchar varcharData) { return new PrestoThriftBlock(null, null, null, varcharData, null, null, null, null, null, null); }
public static PrestoThriftBlock bigintArrayData(PrestoThriftBigintArray bigintArrayData) { return new PrestoThriftBlock(null, null, null, null, null, null, null, null, null, bigintArrayData); }
public static PrestoThriftBlock booleanData(PrestoThriftBoolean booleanData) { return new PrestoThriftBlock(null, null, null, null, booleanData, null, null, null, null, null); }
public static PrestoThriftBlock hyperLogLogData(PrestoThriftHyperLogLog hyperLogLogData) { return new PrestoThriftBlock(null, null, null, null, null, null, null, null, hyperLogLogData, null); }
public static PrestoThriftBlock dateData(PrestoThriftDate dateData) { return new PrestoThriftBlock(null, null, null, null, null, dateData, null, null, null, null); }
public static PrestoThriftBlock timestampData(PrestoThriftTimestamp timestampData) { return new PrestoThriftBlock(null, null, null, null, null, null, timestampData, null, null, null); }
public static PrestoThriftBlock bigintData(PrestoThriftBigint bigintData) { return new PrestoThriftBlock(null, bigintData, null, null, null, null, null, null, null, null); }