private long getLong(TpchColumn<E> tpchColumn) { if (tpchColumn.getType().getBase() == TpchColumnType.Base.DATE) { return tpchColumn.getDate(row); } if (tpchColumn.getType().getBase() == TpchColumnType.Base.INTEGER) { return tpchColumn.getInteger(row); } return tpchColumn.getIdentifier(row); }
@Override public long getIdentifier(E entity) { return entity.getRowNumber(); }
private double getDouble(TpchColumn<E> tpchColumn) { return tpchColumn.getDouble(row); }
public TestRedisDistributed(EmbeddedRedis embeddedRedis) { super(() -> RedisQueryRunner.createRedisQueryRunner(embeddedRedis, "string", TpchTable.getTables())); this.embeddedRedis = embeddedRedis; }
private <E extends TpchEntity> Comparable<?> getTpchValue(E row, TpchColumn<E> column) { TpchColumnType.Base baseType = column.getType().getBase(); switch (baseType) { case IDENTIFIER: return column.getIdentifier(row); case INTEGER: return column.getInteger(row); case DATE: return column.getDate(row); case DOUBLE: return column.getDouble(row); case VARCHAR: return column.getString(row); } throw new UnsupportedOperationException(format("Unsupported TPCH base type [%s]", baseType)); } }
private static String redisTableName(TpchTable<?> table) { return TPCH_SCHEMA + ":" + table.getTableName().toLowerCase(ENGLISH); }
<E extends TpchEntity> TableStatisticsData recordStatistics(TpchTable<E> tpchTable, Predicate<E> constraint, double scaleFactor) { Iterable<E> rows = tpchTable.createGenerator(scaleFactor, 1, 1); return recordStatistics(rows, tpchTable.getColumns(), constraint); }
private static Type getColumnType(TpchColumn<?> input) { switch (input.getType().getBase()) { case IDENTIFIER: return BIGINT; case INTEGER: return INTEGER; case DATE: return DATE; case DOUBLE: return DOUBLE; case VARCHAR: return createUnboundedVarcharType(); } throw new IllegalArgumentException("Unsupported type " + input.getType()); }
private Slice getSlice(TpchColumn<E> tpchColumn) { return Slices.utf8Slice(tpchColumn.getString(row)); }
public TestRedisDistributedHash(EmbeddedRedis embeddedRedis) { super(() -> RedisQueryRunner.createRedisQueryRunner(embeddedRedis, "hash", TpchTable.getTables())); this.embeddedRedis = embeddedRedis; }
private static String kafkaTopicName(TpchTable<?> table) { return TPCH_SCHEMA + "." + table.getTableName().toLowerCase(ENGLISH); }
public TestMySqlDistributedQueries(TestingMySqlServer mysqlServer) { super(() -> createMySqlQueryRunner(mysqlServer, TpchTable.getTables())); this.mysqlServer = mysqlServer; }
public TestElasticsearchIntegrationSmokeTest(EmbeddedElasticsearchNode embeddedElasticsearchNode) { super(() -> createElasticsearchQueryRunner(embeddedElasticsearchNode, TpchTable.getTables())); this.embeddedElasticsearchNode = embeddedElasticsearchNode; }
public TestHiveDistributedAggregations() { super(() -> createQueryRunner(getTables())); } }
public TestPostgreSqlDistributedQueries(TestingPostgreSqlServer postgreSqlServer) { super(() -> createPostgreSqlQueryRunner(postgreSqlServer, TpchTable.getTables())); this.postgreSqlServer = postgreSqlServer; }
public TestKafkaDistributed(EmbeddedKafka embeddedKafka) { super(() -> createKafkaQueryRunner(embeddedKafka, TpchTable.getTables())); this.embeddedKafka = embeddedKafka; }
public TestJdbcDistributedQueries() { super(() -> createJdbcQueryRunner(TpchTable.getTables())); }
public TestHiveDistributedQueries() { super(() -> createQueryRunner(getTables())); }
public TestMongoDistributedQueries() { super(() -> createMongoQueryRunner(TpchTable.getTables())); }