ignite-commits mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From se...@apache.org
Subject [42/45] incubator-ignite git commit: ignite-sql-old - tests
Date Sun, 15 Feb 2015 03:12:33 GMT
ignite-sql-old - tests


Project: http://git-wip-us.apache.org/repos/asf/incubator-ignite/repo
Commit: http://git-wip-us.apache.org/repos/asf/incubator-ignite/commit/2e834e6b
Tree: http://git-wip-us.apache.org/repos/asf/incubator-ignite/tree/2e834e6b
Diff: http://git-wip-us.apache.org/repos/asf/incubator-ignite/diff/2e834e6b

Branch: refs/heads/ignite-sql-old
Commit: 2e834e6bcfa256ecaf988fed343b433976d75def
Parents: a50e052
Author: S.Vladykin <svladykin@gridgain.com>
Authored: Sun Feb 15 05:46:16 2015 +0300
Committer: S.Vladykin <svladykin@gridgain.com>
Committed: Sun Feb 15 05:46:16 2015 +0300

----------------------------------------------------------------------
 .../processors/cache/IgniteCacheProxy.java      |   6 +-
 .../GridCacheCrossCacheQuerySelfTestNewApi.java | 410 +++++++++++++++++++
 2 files changed, 414 insertions(+), 2 deletions(-)
----------------------------------------------------------------------


http://git-wip-us.apache.org/repos/asf/incubator-ignite/blob/2e834e6b/modules/core/src/main/java/org/apache/ignite/internal/processors/cache/IgniteCacheProxy.java
----------------------------------------------------------------------
diff --git a/modules/core/src/main/java/org/apache/ignite/internal/processors/cache/IgniteCacheProxy.java
b/modules/core/src/main/java/org/apache/ignite/internal/processors/cache/IgniteCacheProxy.java
index 1361b8b..30a9708 100644
--- a/modules/core/src/main/java/org/apache/ignite/internal/processors/cache/IgniteCacheProxy.java
+++ b/modules/core/src/main/java/org/apache/ignite/internal/processors/cache/IgniteCacheProxy.java
@@ -304,7 +304,8 @@ public class IgniteCacheProxy<K, V> extends AsyncSupportAdapter<IgniteCache<K,
V
             if (grp != null)
                 qry.projection(grp);
 
-            qry.pageSize(q.getPageSize());
+            if (q.getPageSize() > 0)
+                qry.pageSize(q.getPageSize());
 
             qry.enableDedup(false);
             qry.includeBackups(false);
@@ -424,7 +425,8 @@ public class IgniteCacheProxy<K, V> extends AsyncSupportAdapter<IgniteCache<K,
V
 
             CacheQuery<List<?>> q = ((GridCacheQueriesEx<K,V>)delegate.queries()).createSqlFieldsQuery(qry.getSql(),
false);
 
-            q.pageSize(qry.getPageSize());
+            if (qry.getPageSize() > 0)
+                q.pageSize(qry.getPageSize());
 
             CacheQueryFuture<List<?>> fut = q.execute(qry.getArgs());
 

http://git-wip-us.apache.org/repos/asf/incubator-ignite/blob/2e834e6b/modules/indexing/src/test/java/org/apache/ignite/internal/processors/cache/GridCacheCrossCacheQuerySelfTestNewApi.java
----------------------------------------------------------------------
diff --git a/modules/indexing/src/test/java/org/apache/ignite/internal/processors/cache/GridCacheCrossCacheQuerySelfTestNewApi.java
b/modules/indexing/src/test/java/org/apache/ignite/internal/processors/cache/GridCacheCrossCacheQuerySelfTestNewApi.java
new file mode 100644
index 0000000..6dcf5dd
--- /dev/null
+++ b/modules/indexing/src/test/java/org/apache/ignite/internal/processors/cache/GridCacheCrossCacheQuerySelfTestNewApi.java
@@ -0,0 +1,410 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License.  You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.apache.ignite.internal.processors.cache;
+
+import org.apache.ignite.*;
+import org.apache.ignite.cache.*;
+import org.apache.ignite.cache.query.annotations.*;
+import org.apache.ignite.configuration.*;
+import org.apache.ignite.internal.processors.cache.query.*;
+import org.apache.ignite.internal.util.typedef.*;
+import org.apache.ignite.lang.*;
+import org.apache.ignite.marshaller.optimized.*;
+import org.apache.ignite.spi.discovery.tcp.*;
+import org.apache.ignite.spi.discovery.tcp.ipfinder.*;
+import org.apache.ignite.spi.discovery.tcp.ipfinder.vm.*;
+import org.apache.ignite.testframework.junits.common.*;
+import org.jetbrains.annotations.*;
+
+import javax.cache.*;
+import java.util.*;
+
+import static org.apache.ignite.cache.CacheAtomicityMode.*;
+import static org.apache.ignite.cache.CacheDistributionMode.*;
+import static org.apache.ignite.cache.CachePreloadMode.*;
+import static org.apache.ignite.cache.query.Query.*;
+
+/**
+ * Tests cross cache queries.
+ */
+public class GridCacheCrossCacheQuerySelfTestNewApi extends GridCommonAbstractTest {
+    /** */
+    private static final TcpDiscoveryIpFinder ipFinder = new TcpDiscoveryVmIpFinder(true);
+
+    /** */
+    private Ignite ignite;
+
+    /** {@inheritDoc} */
+    @Override protected IgniteConfiguration getConfiguration(String gridName) throws Exception
{
+        IgniteConfiguration c = super.getConfiguration(gridName);
+
+        TcpDiscoverySpi disco = new TcpDiscoverySpi();
+
+        disco.setIpFinder(ipFinder);
+
+        c.setDiscoverySpi(disco);
+
+        c.setMarshaller(new OptimizedMarshaller(false));
+
+        c.setCacheConfiguration(createCache("replicated", CacheMode.REPLICATED),
+            createCache("partitioned", CacheMode.PARTITIONED));
+
+        return c;
+    }
+
+    /** {@inheritDoc} */
+    @Override protected void beforeTest() throws Exception {
+        ignite = startGridsMultiThreaded(3);
+    }
+
+    /** {@inheritDoc} */
+    @Override protected void afterTest() throws Exception {
+        stopAllGrids();
+
+        ignite = null;
+    }
+
+    /**
+     * Creates new cache configuration.
+     *
+     * @param name Cache name.
+     * @param mode Cache mode.
+     * @return Cache configuration.
+     */
+    private static CacheConfiguration createCache(String name, CacheMode mode) {
+        CacheConfiguration cc = defaultCacheConfiguration();
+
+        cc.setName(name);
+        cc.setCacheMode(mode);
+        cc.setWriteSynchronizationMode(CacheWriteSynchronizationMode.FULL_SYNC);
+        cc.setPreloadMode(SYNC);
+        cc.setSwapEnabled(true);
+        cc.setEvictNearSynchronized(false);
+        cc.setAtomicityMode(TRANSACTIONAL);
+        cc.setDistributionMode(NEAR_PARTITIONED);
+
+        return cc;
+    }
+
+    public void testNewApi() throws IgniteCheckedException, InterruptedException {
+        fillCaches();
+
+        IgniteCache<Object,Object> p = ignite.jcache("partitioned");
+
+        List<Cache.Entry<Object,Object>> res = p.query(sql(FactPurchase.class,
+            "price = 5")).getAll();
+
+        assertEquals(1, res.size());
+
+        IgniteReducer<List<?>,Long> rdc = new IgniteReducer<List<?>,Long>()
{
+            /** */
+            long l;
+
+            @Override public boolean collect(@Nullable List<?> row) {
+                l += ((Number)row.get(0)).longValue();
+
+                return true;
+            }
+
+            @Override public Long reduce() {
+                return l;
+            }
+        };
+
+        final long cnt = F.reduce(p.queryFields(sql("select count(*) from FactPurchase where
price > 5")), rdc);
+
+        X.println("all facts: " + cnt);
+
+        long cnt0 = 0;
+
+        for (int i = 0; i < 3; i++) {
+            List<List<?>> rows = grid(i).jcache("partitioned").localQueryFields(
+                sql("select count(*) from FactPurchase where price > 5")).getAll();
+
+            assertEquals(1, rows.size());
+
+            X.println("node facts: " + rows);
+
+            cnt0 += ((Number)rows.get(0).get(0)).longValue();
+        }
+
+        assertEquals(cnt, cnt0);
+
+        cnt0 = 0;
+
+        for (int i = 0; i < 3; i++) {
+            List<Cache.Entry<Object,Object>> rows = grid(i).jcache("partitioned").localQuery(
+                sql(FactPurchase.class, "price > 5")).getAll();
+
+            X.println("node facts: " + rows.size());
+
+            cnt0 += rows.size();
+        }
+
+        assertEquals(cnt, cnt0);
+    }
+    /**
+     * @throws IgniteCheckedException If failed.
+     */
+    private void fillCaches() throws IgniteCheckedException, InterruptedException {
+        int idGen = 0;
+
+        GridCache<Integer, Object> dimCache = ignite.cache("replicated");
+
+        for (int i = 0; i < 2; i++) {
+            int id = idGen++;
+
+            dimCache.put(id, new DimStore(id, "Store" + id));
+        }
+
+        for (int i = 0; i < 5; i++) {
+            int id = idGen++;
+
+            dimCache.put(id, new DimProduct(id, "Product" + id));
+        }
+
+        CacheProjection<Integer, DimStore> stores = dimCache.projection(Integer.class,
DimStore.class);
+        CacheProjection<Integer, DimProduct> prods = dimCache.projection(Integer.class,
DimProduct.class);
+
+        GridCache<Integer, FactPurchase> factCache = ignite.cache("partitioned");
+
+        List<DimStore> dimStores = new ArrayList<>(stores.values());
+        Collections.sort(dimStores, new Comparator<DimStore>() {
+            @Override public int compare(DimStore o1, DimStore o2) {
+                return o1.getId() > o2.getId() ? 1 : o1.getId() < o2.getId() ? -1 :
0;
+            }
+        });
+
+        List<DimProduct> dimProds = new ArrayList<>(prods.values());
+        Collections.sort(dimProds, new Comparator<DimProduct>() {
+            @Override
+            public int compare(DimProduct o1, DimProduct o2) {
+                return o1.getId() > o2.getId() ? 1 : o1.getId() < o2.getId() ? -1 :
0;
+            }
+        });
+
+        for (int i = 0; i < 10; i++) {
+            int id = idGen++;
+
+            DimStore store = dimStores.get(i % dimStores.size());
+            DimProduct prod = dimProds.get(i % dimProds.size());
+
+            factCache.put(id, new FactPurchase(id, prod.getId(), store.getId(), i + 5));
+        }
+    }
+
+    /**
+     * Fills the caches with data and executes the query.
+     *
+     * @param prj Cache projection.
+     * @throws Exception If failed.
+     * @return Result.
+     */
+    private List<Map.Entry<Integer, FactPurchase>> body(CacheProjection<Integer,
FactPurchase> prj)
+        throws Exception {
+        CacheQuery<Map.Entry<Integer, FactPurchase>> qry = (prj == null ?
+            ignite.<Integer, FactPurchase>cache("partitioned") : prj).queries().createSqlQuery(FactPurchase.class,
+            "from \"replicated\".DimStore, \"partitioned\".FactPurchase where DimStore.id
= FactPurchase.storeId");
+
+        List<Map.Entry<Integer, FactPurchase>> res = new ArrayList<>(qry.execute().get());
+        Collections.sort(res, new Comparator<Map.Entry<Integer, FactPurchase>>()
{
+            @Override public int compare(Map.Entry<Integer, FactPurchase> o1, Map.Entry<Integer,
FactPurchase> o2) {
+                return o1.getKey() > o2.getKey() ? 1 : o1.getKey() < o2.getKey() ?
-1 : 0;
+            }
+        });
+
+        return res;
+    }
+
+    /**
+     * Checks result.
+     * @param res Result to check.
+     */
+    private static void check(List<Map.Entry<Integer, FactPurchase>> res) {
+        assertEquals("Result size", 4, res.size());
+
+        checkPurchase(res.get(0), 13, 3, 0);
+        checkPurchase(res.get(1), 14, 4, 1);
+        checkPurchase(res.get(2), 15, 5, 0);
+        checkPurchase(res.get(3), 16, 6, 1);
+    }
+
+    /**
+     * Checks purchase.
+     * @param entry Entry to check.
+     * @param id ID.
+     * @param productId Product ID.
+     * @param storeId Store ID.
+     */
+    private static void checkPurchase(Map.Entry<Integer, FactPurchase> entry, int id,
int productId, int storeId) {
+        FactPurchase purchase = entry.getValue();
+
+        assertEquals("Id", id, entry.getKey().intValue());
+        assertEquals("Id", id, purchase.getId());
+        assertEquals("ProductId", productId, purchase.getProductId());
+        assertEquals("StoreId", storeId, purchase.getStoreId());
+    }
+
+    /**
+     * Represents a product available for purchase. In our {@code snowflake} schema a {@code
product} is a {@code
+     * 'dimension'} and will be cached in {@link CacheMode#REPLICATED} cache.
+     */
+    private static class DimProduct {
+        /** Primary key. */
+        @QuerySqlField
+        private int id;
+
+        /** Product name. */
+        @QuerySqlField
+        private String name;
+
+        /**
+         * Constructs a product instance.
+         *
+         * @param id Product ID.
+         * @param name Product name.
+         */
+        DimProduct(int id, String name) {
+            this.id = id;
+            this.name = name;
+        }
+
+        /**
+         * Gets product ID.
+         *
+         * @return Product ID.
+         */
+        public int getId() {
+            return id;
+        }
+
+        /**
+         * Gets product name.
+         *
+         * @return Product name.
+         */
+        public String getName() {
+            return name;
+        }
+    }
+
+    /**
+     * Represents a physical store location. In our {@code snowflake} schema a {@code store}
is a {@code 'dimension'}
+     * and will be cached in {@link CacheMode#REPLICATED} cache.
+     */
+    private static class DimStore {
+        /** Primary key. */
+        @QuerySqlField
+        private int id;
+
+        /** Store name. */
+        @QuerySqlField
+        private String name;
+
+        /**
+         * Constructs a store instance.
+         *
+         * @param id Store ID.
+         * @param name Store name.
+         */
+        DimStore(int id, String name) {
+            this.id = id;
+            this.name = name;
+        }
+
+        /**
+         * Gets store ID.
+         *
+         * @return Store ID.
+         */
+        public int getId() {
+            return id;
+        }
+
+        /**
+         * Gets store name.
+         *
+         * @return Store name.
+         */
+        public String getName() {
+            return name;
+        }
+    }
+
+    /**
+     * Represents a purchase record. In our {@code snowflake} schema purchase is a {@code
'fact'} and will be cached in
+     * larger {@link CacheMode#PARTITIONED} cache.
+     */
+    private static class FactPurchase {
+        /** Primary key. */
+        @QuerySqlField
+        private int id;
+
+        /** Foreign key to store at which purchase occurred. */
+        @QuerySqlField
+        private int storeId;
+
+        /** Foreign key to purchased product. */
+        @QuerySqlField
+        private int productId;
+
+        @QuerySqlField
+        private int price;
+
+        /**
+         * Constructs a purchase record.
+         *
+         * @param id Purchase ID.
+         * @param productId Purchased product ID.
+         * @param storeId Store ID.
+         */
+        FactPurchase(int id, int productId, int storeId, int price) {
+            this.id = id;
+            this.productId = productId;
+            this.storeId = storeId;
+            this.price = price;
+        }
+
+        /**
+         * Gets purchase ID.
+         *
+         * @return Purchase ID.
+         */
+        public int getId() {
+            return id;
+        }
+
+        /**
+         * Gets purchased product ID.
+         *
+         * @return Product ID.
+         */
+        public int getProductId() {
+            return productId;
+        }
+
+        /**
+         * Gets ID of store at which purchase was made.
+         *
+         * @return Store ID.
+         */
+        public int getStoreId() {
+            return storeId;
+        }
+    }
+}


Mime
View raw message