Search in sources :

Example 1 with PlayerStatsView

use of example.PlayerStatsView in project elide by yahoo.

the class QueryEngineTest method testAllArgumentQuery.

/**
 * Test group by, having, dimension, metric at the same time.
 *
 * @throws Exception exception
 */
@Test
public void testAllArgumentQuery() throws Exception {
    Map<String, Sorting.SortOrder> sortMap = new TreeMap<>();
    sortMap.put("countryName", Sorting.SortOrder.asc);
    Query query = Query.builder().source(playerStatsViewTable).metricProjection(playerStatsViewTable.getMetricProjection("highScore")).dimensionProjection(playerStatsViewTable.getDimensionProjection("countryName")).whereFilter(filterParser.parseFilterExpression("countryName=='United States'", playerStatsViewType, false)).havingFilter(filterParser.parseFilterExpression("highScore > 300", playerStatsViewType, false)).sorting(new SortingImpl(sortMap, PlayerStatsView.class, dictionary)).arguments(playerStatsViewTableArgs).build();
    List<Object> results = toList(engine.executeQuery(query, transaction).getData());
    PlayerStatsView stats2 = new PlayerStatsView();
    stats2.setId("0");
    stats2.setHighScore(2412);
    stats2.setCountryName("United States");
    assertEquals(ImmutableList.of(stats2), results);
}
Also used : PlayerStatsView(example.PlayerStatsView) Query(com.yahoo.elide.datastores.aggregation.query.Query) SortingImpl(com.yahoo.elide.core.sort.SortingImpl) TreeMap(java.util.TreeMap) SQLUnitTest(com.yahoo.elide.datastores.aggregation.framework.SQLUnitTest) Test(org.junit.jupiter.api.Test)

Example 2 with PlayerStatsView

use of example.PlayerStatsView in project elide by yahoo.

the class QueryEngineTest method testFromSubQuery.

/**
 * Test loading records using {@link FromSubquery}.
 *
 * @throws Exception exception
 */
@Test
public void testFromSubQuery() throws Exception {
    Query query = Query.builder().source(playerStatsViewTable.toQueryable()).metricProjection(playerStatsViewTable.getMetricProjection("highScore")).arguments(playerStatsViewTableArgs).build();
    List<Object> results = toList(engine.executeQuery(query, transaction).getData());
    PlayerStatsView stats2 = new PlayerStatsView();
    stats2.setId("0");
    stats2.setHighScore(2412);
    assertEquals(ImmutableList.of(stats2), results);
}
Also used : PlayerStatsView(example.PlayerStatsView) Query(com.yahoo.elide.datastores.aggregation.query.Query) SQLUnitTest(com.yahoo.elide.datastores.aggregation.framework.SQLUnitTest) Test(org.junit.jupiter.api.Test)

Example 3 with PlayerStatsView

use of example.PlayerStatsView in project elide by yahoo.

the class SQLUnitTest method init.

public static void init(SQLDialect sqlDialect, Set<Optimizer> optimizers, MetaDataStore metaDataStore) {
    Properties properties = new Properties();
    properties.put("driverClassName", "org.h2.Driver");
    String jdbcUrl = "jdbc:h2:mem:db1;DB_CLOSE_DELAY=-1" + ";NON_KEYWORDS=VALUE,USER" + ";DATABASE_TO_UPPER=FALSE" + getCompatabilityMode(sqlDialect.getDialectType());
    properties.put("jdbcUrl", jdbcUrl);
    HikariConfig config = new HikariConfig(properties);
    DataSource dataSource = new HikariDataSource(config);
    try (Connection h2Conn = dataSource.getConnection()) {
        h2Conn.createStatement().execute("RUNSCRIPT FROM 'classpath:prepare_tables.sql'");
    } catch (SQLException e) {
        ((HikariDataSource) dataSource).close();
        throw new IllegalStateException(e);
    }
    SQLUnitTest.metaDataStore = metaDataStore;
    dictionary = EntityDictionary.builder().build();
    dictionary.bindEntity(PlayerStatsWithView.class);
    dictionary.bindEntity(PlayerStatsView.class);
    dictionary.bindEntity(PlayerStats.class);
    dictionary.bindEntity(Country.class);
    dictionary.bindEntity(SubCountry.class);
    dictionary.bindEntity(Player.class);
    dictionary.bindEntity(CountryView.class);
    dictionary.bindEntity(CountryViewNested.class);
    dictionary.bindEntity(Continent.class);
    dictionary.bindEntity(GameRevenue.class);
    filterParser = RSQLFilterDialect.builder().dictionary(dictionary).build();
    // Manually register the serdes because we are not running a complete Elide service.
    CoerceUtil.register(Day.class, new Day.DaySerde());
    CoerceUtil.register(Hour.class, new Hour.HourSerde());
    CoerceUtil.register(ISOWeek.class, new ISOWeek.ISOWeekSerde());
    CoerceUtil.register(Minute.class, new Minute.MinuteSerde());
    CoerceUtil.register(Month.class, new Month.MonthSerde());
    CoerceUtil.register(Quarter.class, new Quarter.QuarterSerde());
    CoerceUtil.register(Second.class, new Second.SecondSerde());
    CoerceUtil.register(Week.class, new Week.WeekSerde());
    CoerceUtil.register(Year.class, new Year.YearSerde());
    metaDataStore.populateEntityDictionary(dictionary);
    // Need to provide details for connections used by all available models.
    Map<String, ConnectionDetails> connectionDetailsMap = new HashMap<>();
    connectionDetailsMap.put("mycon", new ConnectionDetails(dataSource, sqlDialect));
    connectionDetailsMap.put("SalesDBConnection", new ConnectionDetails(DUMMY_DATASOURCE, sqlDialect));
    Function<String, ConnectionDetails> connectionLookup = (name) -> connectionDetailsMap.getOrDefault(name, new ConnectionDetails(dataSource, sqlDialect));
    engine = new SQLQueryEngine(metaDataStore, connectionLookup, optimizers, new DefaultQueryPlanMerger(metaDataStore), new DefaultQueryValidator(metaDataStore.getMetadataDictionary()));
    playerStatsTable = (SQLTable) metaDataStore.getTable("playerStats", NO_VERSION);
    videoGameTable = (SQLTable) metaDataStore.getTable("videoGame", NO_VERSION);
    playerStatsViewTable = (SQLTable) metaDataStore.getTable("playerStatsView", NO_VERSION);
    playerStatsViewTableArgs = new HashMap<>();
    playerStatsViewTableArgs.put("rating", Argument.builder().name("overallRating").value("Great").build());
    playerStatsViewTableArgs.put("minScore", Argument.builder().name("minScore").value("0").build());
}
Also used : Quarter(com.yahoo.elide.datastores.aggregation.timegrains.Quarter) Year(com.yahoo.elide.datastores.aggregation.timegrains.Year) BeforeEach(org.junit.jupiter.api.BeforeEach) FilterPredicate(com.yahoo.elide.core.filter.predicates.FilterPredicate) Arrays(java.util.Arrays) Path(com.yahoo.elide.core.Path) Connection(java.sql.Connection) Provider(javax.inject.Provider) Include(com.yahoo.elide.annotation.Include) ParseException(com.yahoo.elide.core.filter.dialect.ParseException) SortingImpl(com.yahoo.elide.core.sort.SortingImpl) ClassType(com.yahoo.elide.core.type.ClassType) Argument(com.yahoo.elide.core.request.Argument) RSQLFilterDialect(com.yahoo.elide.core.filter.dialect.RSQLFilterDialect) Continent(example.dimensions.Continent) ISOWeek(com.yahoo.elide.datastores.aggregation.timegrains.ISOWeek) OrFilterExpression(com.yahoo.elide.core.filter.expression.OrFilterExpression) Map(java.util.Map) MetricProjection(com.yahoo.elide.datastores.aggregation.query.MetricProjection) DefaultQueryPlanMerger(com.yahoo.elide.datastores.aggregation.query.DefaultQueryPlanMerger) SQLDialectFactory(com.yahoo.elide.datastores.aggregation.queryengines.sql.dialects.SQLDialectFactory) STRING_TYPE(com.yahoo.elide.core.type.ClassType.STRING_TYPE) DefaultClassScanner(com.yahoo.elide.core.utils.DefaultClassScanner) Set(java.util.Set) CoerceUtil(com.yahoo.elide.core.utils.coerce.CoerceUtil) Collectors(java.util.stream.Collectors) EntityDictionary(com.yahoo.elide.core.dictionary.EntityDictionary) PlayerStats(example.PlayerStats) ConnectionDetails(com.yahoo.elide.datastores.aggregation.queryengines.sql.ConnectionDetails) Query(com.yahoo.elide.datastores.aggregation.query.Query) List(java.util.List) Hour(com.yahoo.elide.datastores.aggregation.timegrains.Hour) ImmutablePagination(com.yahoo.elide.datastores.aggregation.query.ImmutablePagination) TypeHelper.getClassType(com.yahoo.elide.core.utils.TypeHelper.getClassType) HikariDataSource(com.zaxxer.hikari.HikariDataSource) AndFilterExpression(com.yahoo.elide.core.filter.expression.AndFilterExpression) Optimizer(com.yahoo.elide.datastores.aggregation.query.Optimizer) Pattern(java.util.regex.Pattern) MetaDataStore(com.yahoo.elide.datastores.aggregation.metadata.MetaDataStore) Assertions.assertDoesNotThrow(org.junit.jupiter.api.Assertions.assertDoesNotThrow) Attribute(com.yahoo.elide.core.request.Attribute) Assertions.assertNotNull(org.junit.jupiter.api.Assertions.assertNotNull) SQLQueryEngine(com.yahoo.elide.datastores.aggregation.queryengines.sql.SQLQueryEngine) CountryViewNested(example.dimensions.CountryViewNested) Day(com.yahoo.elide.datastores.aggregation.timegrains.Day) Quarter(com.yahoo.elide.datastores.aggregation.timegrains.Quarter) SubCountry(example.dimensions.SubCountry) Second(com.yahoo.elide.datastores.aggregation.timegrains.Second) Week(com.yahoo.elide.datastores.aggregation.timegrains.Week) HashMap(java.util.HashMap) TimeGrain(com.yahoo.elide.datastores.aggregation.metadata.enums.TimeGrain) CountryView(example.dimensions.CountryView) Function(java.util.function.Function) ArrayList(java.util.ArrayList) HashSet(java.util.HashSet) SQLException(java.sql.SQLException) SQLDialect(com.yahoo.elide.datastores.aggregation.queryengines.sql.dialects.SQLDialect) Month(com.yahoo.elide.datastores.aggregation.timegrains.Month) DataSource(javax.sql.DataSource) NO_VERSION(com.yahoo.elide.core.dictionary.EntityDictionary.NO_VERSION) StreamSupport(java.util.stream.StreamSupport) Assertions.assertEquals(org.junit.jupiter.api.Assertions.assertEquals) FilterExpression(com.yahoo.elide.core.filter.expression.FilterExpression) LinkedHashSet(java.util.LinkedHashSet) PlayerStatsView(example.PlayerStatsView) QueryEngine(com.yahoo.elide.datastores.aggregation.QueryEngine) DimensionProjection(com.yahoo.elide.datastores.aggregation.query.DimensionProjection) Properties(java.util.Properties) Sorting(com.yahoo.elide.core.request.Sorting) GameRevenue(example.GameRevenue) Minute(com.yahoo.elide.datastores.aggregation.timegrains.Minute) DefaultQueryValidator(com.yahoo.elide.datastores.aggregation.DefaultQueryValidator) Player(example.Player) HikariConfig(com.zaxxer.hikari.HikariConfig) AfterEach(org.junit.jupiter.api.AfterEach) Country(example.dimensions.Country) TreeMap(java.util.TreeMap) SQLTable(com.yahoo.elide.datastores.aggregation.queryengines.sql.metadata.SQLTable) Type(com.yahoo.elide.core.type.Type) Operator(com.yahoo.elide.core.filter.Operator) PlayerStatsWithView(example.PlayerStatsWithView) SQLQueryEngine(com.yahoo.elide.datastores.aggregation.queryengines.sql.SQLQueryEngine) HikariDataSource(com.zaxxer.hikari.HikariDataSource) SQLException(java.sql.SQLException) HashMap(java.util.HashMap) ConnectionDetails(com.yahoo.elide.datastores.aggregation.queryengines.sql.ConnectionDetails) DefaultQueryPlanMerger(com.yahoo.elide.datastores.aggregation.query.DefaultQueryPlanMerger) Properties(java.util.Properties) DefaultQueryValidator(com.yahoo.elide.datastores.aggregation.DefaultQueryValidator) Month(com.yahoo.elide.datastores.aggregation.timegrains.Month) Minute(com.yahoo.elide.datastores.aggregation.timegrains.Minute) Hour(com.yahoo.elide.datastores.aggregation.timegrains.Hour) Connection(java.sql.Connection) HikariConfig(com.zaxxer.hikari.HikariConfig) HikariDataSource(com.zaxxer.hikari.HikariDataSource) DataSource(javax.sql.DataSource) ISOWeek(com.yahoo.elide.datastores.aggregation.timegrains.ISOWeek) Second(com.yahoo.elide.datastores.aggregation.timegrains.Second) Year(com.yahoo.elide.datastores.aggregation.timegrains.Year) Day(com.yahoo.elide.datastores.aggregation.timegrains.Day) ISOWeek(com.yahoo.elide.datastores.aggregation.timegrains.ISOWeek) Week(com.yahoo.elide.datastores.aggregation.timegrains.Week)

Example 4 with PlayerStatsView

use of example.PlayerStatsView in project elide by yahoo.

the class QueryEngineTest method testNotProjectedFilter.

/**
 * Test filtering on an attribute that's not present in the query.
 *
 * @throws Exception exception
 */
@Test
public void testNotProjectedFilter() throws Exception {
    Query query = Query.builder().source(playerStatsViewTable).metricProjection(playerStatsViewTable.getMetricProjection("highScore")).whereFilter(filterParser.parseFilterExpression("countryName=='United States'", playerStatsViewType, false)).arguments(playerStatsViewTableArgs).build();
    List<Object> results = toList(engine.executeQuery(query, transaction).getData());
    PlayerStatsView stats2 = new PlayerStatsView();
    stats2.setId("0");
    stats2.setHighScore(2412);
    assertEquals(ImmutableList.of(stats2), results);
}
Also used : PlayerStatsView(example.PlayerStatsView) Query(com.yahoo.elide.datastores.aggregation.query.Query) SQLUnitTest(com.yahoo.elide.datastores.aggregation.framework.SQLUnitTest) Test(org.junit.jupiter.api.Test)

Aggregations

Query (com.yahoo.elide.datastores.aggregation.query.Query)4 PlayerStatsView (example.PlayerStatsView)4 SQLUnitTest (com.yahoo.elide.datastores.aggregation.framework.SQLUnitTest)3 Test (org.junit.jupiter.api.Test)3 SortingImpl (com.yahoo.elide.core.sort.SortingImpl)2 TreeMap (java.util.TreeMap)2 Include (com.yahoo.elide.annotation.Include)1 Path (com.yahoo.elide.core.Path)1 EntityDictionary (com.yahoo.elide.core.dictionary.EntityDictionary)1 NO_VERSION (com.yahoo.elide.core.dictionary.EntityDictionary.NO_VERSION)1 Operator (com.yahoo.elide.core.filter.Operator)1 ParseException (com.yahoo.elide.core.filter.dialect.ParseException)1 RSQLFilterDialect (com.yahoo.elide.core.filter.dialect.RSQLFilterDialect)1 AndFilterExpression (com.yahoo.elide.core.filter.expression.AndFilterExpression)1 FilterExpression (com.yahoo.elide.core.filter.expression.FilterExpression)1 OrFilterExpression (com.yahoo.elide.core.filter.expression.OrFilterExpression)1 FilterPredicate (com.yahoo.elide.core.filter.predicates.FilterPredicate)1 Argument (com.yahoo.elide.core.request.Argument)1 Attribute (com.yahoo.elide.core.request.Attribute)1 Sorting (com.yahoo.elide.core.request.Sorting)1