Examples with StructSetOrResultsSet org.apache.geode.cache.query.functional.StructSetOrResultsSet used on opensource projects

Example 16 with StructSetOrResultsSet

use of org.apache.geode.cache.query.functional.StructSetOrResultsSet in project geode by apache.

the class QueryIndexUsingXMLDUnitTest method executeQueryAndCompareResult.

private CacheSerializableRunnable executeQueryAndCompareResult(final boolean compareHash) {
    return new CacheSerializableRunnable("execute query and compare results.") {

        @Override
        public void run2() {
            QueryService qs = getCache().getQueryService();
            StructSetOrResultsSet resultsSet = new StructSetOrResultsSet();
            SelectResults[][] selectResults = new SelectResults[1][2];
            String[] queryStrings = new String[2];
            int numQueries = QUERY_STR.length;
            for (int j = 0; j < numQueries; j++) {
                String[] queryArray = QUERY_STR[j];
                int numQueriesToCheck = compareHash ? queryArray.length : 3;
                for (int i = 0; i < numQueriesToCheck; i++) {
                    QueryObserverImpl observer = new QueryObserverImpl();
                    QueryObserverHolder.setInstance(observer);
                    // Query using index.
                    queryStrings[0] = QUERY_STR[j][i];
                    // Execute query with index.
                    Query query = qs.newQuery(queryStrings[0]);
                    try {
                        selectResults[0][0] = (SelectResults) query.execute();
                    } catch (Exception ex) {
                        throw new AssertionError("Failed to execute the query.", ex);
                    }
                    assertThat(observer.isIndexesUsed).isTrue().as("Index not used for query. " + queryStrings[0]);
                    // Query using no index.
                    queryStrings[1] = QUERY_STR_NO_INDEX[j];
                    try {
                        query = qs.newQuery(queryStrings[1]);
                        selectResults[0][1] = (SelectResults) query.execute();
                    } catch (Exception ex) {
                        throw new AssertionError("Failed to execute the query on no index region.", ex);
                    }
                    // compare.
                    getLogWriter().info("Execute query : " + System.getProperty("line.separator") + " QUERY_STR with index: " + queryStrings[0] + " " + System.getProperty("line.separator") + " QUERY_STR without index: " + queryStrings[1]);
                    resultsSet.CompareQueryResultsWithoutAndWithIndexes(selectResults, 1, queryStrings);
                }
            }
        }
    };
}

Example 17 with StructSetOrResultsSet

use of org.apache.geode.cache.query.functional.StructSetOrResultsSet in project geode by apache.

the class IndexUseJUnitTest method testRangeGroupingBehaviourOfCompactMapIndex.

@Test
public void testRangeGroupingBehaviourOfCompactMapIndex() throws Exception {
    QueryService qs;
    qs = CacheUtils.getQueryService();
    LocalRegion testRgn = (LocalRegion) CacheUtils.createRegion("testRgn", null);
    int ID = 1;
    // and so on
    for (; ID <= 30; ++ID) {
        MapKeyIndexData mkid = new MapKeyIndexData(ID);
        for (int j = 1; j <= ID; ++j) {
            mkid.addKeyValue("key1", j * 1);
            mkid.addKeyValue("key2", j * 2);
            mkid.addKeyValue("key3", j * 3);
        }
        testRgn.put(ID, mkid);
    }
    qs = CacheUtils.getQueryService();
    String[] queries = { "SELECT DISTINCT * FROM /testRgn itr1  WHERE itr1.maap['key2'] >= 3 and itr1.maap['key2'] <=18", "SELECT DISTINCT * FROM /testRgn itr1  WHERE itr1.maap['key3'] >= 3  and  itr1.maap['key3'] >= 13 ", "SELECT DISTINCT * FROM /testRgn itr1  WHERE itr1.maap['key2'] >= 3  and  itr1.maap['key3'] >= 13 ", "SELECT DISTINCT * FROM /testRgn itr1  WHERE itr1.maap['key2'] >= 3  and  itr1.maap['key3'] < 18 " };
    Object[][] r = new Object[queries.length][2];
    // Execute Queries without Indexes
    for (int i = 0; i < queries.length; i++) {
        Query q = null;
        try {
            q = CacheUtils.getQueryService().newQuery(queries[i]);
            CacheUtils.getLogger().info("Executing query: " + queries[i]);
            r[i][0] = q.execute();
            CacheUtils.log("Executed query: " + queries[i]);
        } catch (Exception e) {
            e.printStackTrace();
            fail(q.getQueryString());
        }
    }
    Index i1 = qs.createIndex("Index1", IndexType.FUNCTIONAL, "objs.maap['key2','key3']", "/testRgn objs");
    assertTrue(i1 instanceof CompactMapRangeIndex);
    // Execute Queries with Indexes
    for (int i = 0; i < queries.length; i++) {
        Query q = null;
        try {
            q = CacheUtils.getQueryService().newQuery(queries[i]);
            CacheUtils.getLogger().info("Executing query: " + queries[i]);
            QueryObserverImpl observer = new QueryObserverImpl();
            QueryObserverHolder.setInstance(observer);
            r[i][1] = q.execute();
            CacheUtils.log("Executing query: " + queries[i] + " with index created");
            if (!observer.isIndexesUsed) {
                fail("Index is NOT uesd");
            }
            Iterator itr = observer.indexesUsed.iterator();
            assertTrue(itr.hasNext());
            String temp = itr.next().toString();
            assertEquals(temp, "Index1");
        } catch (Exception e) {
            e.printStackTrace();
            fail(q.getQueryString());
        }
    }
    StructSetOrResultsSet ssOrrs = new StructSetOrResultsSet();
    ssOrrs.CompareQueryResultsWithoutAndWithIndexes(r, queries.length, queries);
}

Example 18 with StructSetOrResultsSet

use of org.apache.geode.cache.query.functional.StructSetOrResultsSet in project geode by apache.

the class IndexUseJUnitTest method testIndexUsageWithOrderBy.

@Test
public void testIndexUsageWithOrderBy() throws Exception {
    QueryService qs;
    qs = CacheUtils.getQueryService();
    LocalRegion testRgn = (LocalRegion) CacheUtils.createRegion("testRgn", null);
    int numObjects = 30;
    // and so on
    for (int i = 0; i < numObjects; i++) {
        Portfolio p = new Portfolio(i);
        p.pkid = ("" + (numObjects - i));
        testRgn.put("" + i, p);
    }
    qs = CacheUtils.getQueryService();
    String[] queries = { "SELECT DISTINCT * FROM /testRgn p  WHERE p.ID <= 10 order by p.pkid asc limit 1", "SELECT DISTINCT * FROM /testRgn p  WHERE p.ID <= 10 order by p.pkid desc limit 1" };
    Object[][] r = new Object[queries.length][2];
    // Execute Queries without Indexes
    for (int i = 0; i < queries.length; i++) {
        Query q = null;
        try {
            q = CacheUtils.getQueryService().newQuery(queries[i]);
            CacheUtils.getLogger().info("Executing query: " + queries[i]);
            // verify individual result
            SelectResults sr = (SelectResults) q.execute();
            List results = sr.asList();
            for (int rows = 0; rows < results.size(); rows++) {
                Portfolio p = (Portfolio) results.get(0);
                CacheUtils.getLogger().info("p: " + p);
                if (i == 0) {
                    assertEquals(p.getID(), 10);
                    assertEquals(p.pkid, "" + (numObjects - 10));
                } else if (i == 1) {
                    assertEquals(p.getID(), 0);
                    assertEquals(p.pkid, "" + numObjects);
                }
            }
            r[i][0] = sr;
            CacheUtils.log("Executed query: " + queries[i]);
        } catch (Exception e) {
            e.printStackTrace();
            fail(q.getQueryString());
        }
    }
    Index i1 = qs.createIndex("Index1", IndexType.FUNCTIONAL, "p.ID", "/testRgn p");
    // Execute Queries with Indexes
    for (int i = 0; i < queries.length; i++) {
        Query q = null;
        try {
            q = CacheUtils.getQueryService().newQuery(queries[i]);
            CacheUtils.getLogger().info("Executing query: " + queries[i]);
            QueryObserverImpl observer = new QueryObserverImpl();
            QueryObserverHolder.setInstance(observer);
            SelectResults sr = (SelectResults) q.execute();
            List results = sr.asList();
            for (int rows = 0; rows < results.size(); rows++) {
                Portfolio p = (Portfolio) results.get(0);
                CacheUtils.getLogger().info("index p: " + p);
                if (i == 0) {
                    assertEquals(p.getID(), 10);
                    assertEquals(p.pkid, "" + (numObjects - 10));
                } else if (i == 1) {
                    assertEquals(p.getID(), 0);
                    assertEquals(p.pkid, "" + numObjects);
                }
            }
            r[i][1] = sr;
            // r[i][1] = q.execute();
            CacheUtils.log("Executing query: " + queries[i] + " with index created");
            if (!observer.isIndexesUsed) {
                fail("Index is NOT uesd");
            }
            Iterator itr = observer.indexesUsed.iterator();
            assertTrue(itr.hasNext());
            String temp = itr.next().toString();
            assertEquals(temp, "Index1");
        } catch (Exception e) {
            e.printStackTrace();
            fail(q.getQueryString());
        }
    }
    StructSetOrResultsSet ssOrrs = new StructSetOrResultsSet();
    ssOrrs.CompareQueryResultsWithoutAndWithIndexes(r, queries.length, queries);
}

Example 19 with StructSetOrResultsSet

use of org.apache.geode.cache.query.functional.StructSetOrResultsSet in project geode by apache.

the class IndexUseJUnitTest method testIndexUsageWithOrderBy2.

@Test
public void testIndexUsageWithOrderBy2() throws Exception {
    QueryService qs;
    qs = CacheUtils.getQueryService();
    LocalRegion testRgn = (LocalRegion) CacheUtils.createRegion("testRgn", null);
    int numObjects = 30;
    // and so on
    for (int i = 0; i < numObjects; i++) {
        Portfolio p = new Portfolio(i % 2);
        p.createTime = (numObjects - i);
        testRgn.put("" + i, p);
    }
    qs = CacheUtils.getQueryService();
    String[] queries = { "SELECT DISTINCT p.key, p.value FROM /testRgn.entrySet p  WHERE p.value.ID <= 10 order by p.value.createTime asc limit 1", "SELECT DISTINCT p.key, p.value FROM /testRgn.entrySet p  WHERE p.value.ID <= 10 order by p.value.createTime desc limit 1" };
    Object[][] r = new Object[queries.length][2];
    // Execute Queries without Indexes
    for (int i = 0; i < queries.length; i++) {
        Query q = null;
        try {
            q = CacheUtils.getQueryService().newQuery(queries[i]);
            CacheUtils.getLogger().info("Executing query: " + queries[i]);
            // verify individual result
            SelectResults sr = (SelectResults) q.execute();
            List results = sr.asList();
            for (int rows = 0; rows < results.size(); rows++) {
                Struct s = (Struct) results.get(0);
                Portfolio p = (Portfolio) s.get("value");
                CacheUtils.getLogger().info("p: " + p);
                if (i == 0) {
                    assertEquals(p.createTime, 1);
                } else if (i == 1) {
                    assertEquals(p.createTime, numObjects);
                }
            }
            r[i][0] = sr;
            CacheUtils.log("Executed query: " + queries[i]);
        } catch (Exception e) {
            e.printStackTrace();
            fail(q.getQueryString());
        }
    }
    Index i1 = qs.createIndex("Index1", IndexType.FUNCTIONAL, "p.value.ID", "/testRgn.entrySet p");
    // Execute Queries with Indexes
    for (int i = 0; i < queries.length; i++) {
        Query q = null;
        try {
            q = CacheUtils.getQueryService().newQuery(queries[i]);
            CacheUtils.getLogger().info("Executing query: " + queries[i]);
            QueryObserverImpl observer = new QueryObserverImpl();
            QueryObserverHolder.setInstance(observer);
            SelectResults sr = (SelectResults) q.execute();
            List results = sr.asList();
            for (int rows = 0; rows < results.size(); rows++) {
                Struct s = (Struct) results.get(0);
                Portfolio p = (Portfolio) s.get("value");
                CacheUtils.getLogger().info("index p: " + p);
                if (i == 0) {
                    assertEquals(p.createTime, 1);
                } else if (i == 1) {
                    assertEquals(p.createTime, numObjects);
                }
            }
            r[i][1] = sr;
            // r[i][1] = q.execute();
            CacheUtils.log("Executing query: " + queries[i] + " with index created");
            if (!observer.isIndexesUsed) {
                fail("Index is NOT uesd");
            }
            Iterator itr = observer.indexesUsed.iterator();
            assertTrue(itr.hasNext());
            String temp = itr.next().toString();
            assertEquals(temp, "Index1");
        } catch (Exception e) {
            e.printStackTrace();
            fail(q.getQueryString());
        }
    }
    StructSetOrResultsSet ssOrrs = new StructSetOrResultsSet();
    ssOrrs.CompareQueryResultsWithoutAndWithIndexes(r, queries.length, queries);
}

Example 20 with StructSetOrResultsSet

use of org.apache.geode.cache.query.functional.StructSetOrResultsSet in project geode by apache.

the class IndexUseJUnitTest method evaluateMapTypeIndexUsage.

private void evaluateMapTypeIndexUsage(String indexExpression, String fromClause, String[] queries, String[] queriesIndexNotUsed, Class expectedIndexClass) throws Exception {
    QueryService qs = CacheUtils.getQueryService();
    Object[][] r = new Object[queries.length][2];
    // Execute Queries without Indexes
    for (int i = 0; i < queries.length; i++) {
        Query q = null;
        try {
            q = CacheUtils.getQueryService().newQuery(queries[i]);
            CacheUtils.getLogger().info("Executing query: " + queries[i]);
            r[i][0] = q.execute();
            CacheUtils.log("Executed query: " + queries[i]);
        } catch (Exception e) {
            e.printStackTrace();
            fail(q.getQueryString());
        }
    }
    Index i1 = qs.createIndex("Index1", IndexType.FUNCTIONAL, indexExpression, fromClause);
    assertTrue(i1.getClass().equals(expectedIndexClass));
    // Execute Queries with Indexes
    for (int i = 0; i < queries.length; i++) {
        Query q = null;
        try {
            q = CacheUtils.getQueryService().newQuery(queries[i]);
            CacheUtils.getLogger().info("Executing query: " + queries[i]);
            QueryObserverImpl observer = new QueryObserverImpl();
            QueryObserverHolder.setInstance(observer);
            r[i][1] = q.execute();
            CacheUtils.log("Executing query: " + queries[i] + " with index created");
            if (!observer.isIndexesUsed) {
                fail("Index is NOT used");
            }
            Iterator itr = observer.indexesUsed.iterator();
            assertTrue(itr.hasNext());
            String temp = itr.next().toString();
            assertEquals(temp, "Index1");
        } catch (Exception e) {
            e.printStackTrace();
            fail(q.getQueryString());
        }
    }
    StructSetOrResultsSet ssOrrs = new StructSetOrResultsSet();
    ssOrrs.CompareQueryResultsWithoutAndWithIndexes(r, queries.length, queries);
    // Test queries index not used
    for (int i = 0; i < queriesIndexNotUsed.length; i++) {
        Query q = null;
        try {
            q = CacheUtils.getQueryService().newQuery(queriesIndexNotUsed[i]);
            CacheUtils.getLogger().info("Executing query: " + queriesIndexNotUsed[i]);
            QueryObserverImpl observer = new QueryObserverImpl();
            QueryObserverHolder.setInstance(observer);
            CacheUtils.log("Executing query: " + queriesIndexNotUsed[i] + " with index created");
            q.execute();
            assertFalse(observer.isIndexesUsed);
            Iterator itr = observer.indexesUsed.iterator();
            assertFalse(itr.hasNext());
        } catch (Exception e) {
            e.printStackTrace();
            fail(q.getQueryString());
        }
    }
}