| /* |
| * 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 java.io.Serializable; |
| import java.util.List; |
| import org.apache.ignite.IgniteCache; |
| import org.apache.ignite.cache.CacheMode; |
| import org.apache.ignite.cache.query.FieldsQueryCursor; |
| import org.apache.ignite.cache.query.SqlFieldsQuery; |
| import org.apache.ignite.cache.query.annotations.QuerySqlField; |
| import org.apache.ignite.configuration.CacheConfiguration; |
| import org.apache.ignite.testframework.junits.common.GridCommonAbstractTest; |
| import org.junit.Test; |
| |
| /** |
| * |
| */ |
| public class SqlFieldsQuerySelfTest extends GridCommonAbstractTest { |
| /** INSERT statement. */ |
| private static final String INSERT = "insert into Person(_key, name) values (5, 'x')"; |
| |
| /** {@inheritDoc} */ |
| @Override protected void afterTest() throws Exception { |
| stopAllGrids(); |
| } |
| |
| /** |
| * @throws Exception If error. |
| */ |
| @Test |
| public void testSqlFieldsQuery() throws Exception { |
| startGrids(2); |
| |
| createAndFillCache(); |
| |
| executeQuery(); |
| } |
| |
| /** |
| * @throws Exception If error. |
| */ |
| @Test |
| public void testSqlFieldsQueryWithTopologyChanges() throws Exception { |
| startGrid(0); |
| |
| createAndFillCache(); |
| |
| startGrid(1); |
| |
| executeQuery(); |
| } |
| |
| /** |
| * |
| */ |
| private void executeQuery() { |
| IgniteCache<?, ?> cache = grid(1).cache("person"); |
| |
| SqlFieldsQuery qry = new SqlFieldsQuery("select name as \"Full Name\", age from person where age > 10"); |
| |
| FieldsQueryCursor<List<?>> qryCursor = cache.query(qry); |
| |
| assertEquals(2, qryCursor.getAll().size()); |
| |
| assertEquals(2, qryCursor.getColumnsCount()); // Row contains "name" and "age" fields. |
| |
| assertEquals("Full Name", qryCursor.getFieldName(0)); |
| |
| assertEquals("AGE", qryCursor.getFieldName(1)); |
| } |
| |
| /** |
| * |
| */ |
| private IgniteCache<Integer, Person> createAndFillCache() { |
| CacheConfiguration<Integer, Person> cacheConf = new CacheConfiguration<>(DEFAULT_CACHE_NAME); |
| |
| cacheConf.setCacheMode(CacheMode.PARTITIONED); |
| cacheConf.setBackups(0); |
| |
| cacheConf.setIndexedTypes(Integer.class, Person.class); |
| |
| cacheConf.setName("person"); |
| |
| IgniteCache<Integer, Person> cache = grid(0).createCache(cacheConf); |
| |
| cache.put(1, new Person("sun", 100)); |
| cache.put(2, new Person("moon", 50)); |
| |
| return cache; |
| } |
| |
| /** */ |
| private void destroyCache() { |
| grid(0).destroyCache("person"); |
| } |
| |
| /** |
| * |
| */ |
| public static class Person implements Serializable { |
| /** Name. */ |
| @QuerySqlField |
| private String name; |
| |
| /** Age. */ |
| @QuerySqlField |
| private int age; |
| |
| /** |
| * @param name Name. |
| * @param age Age. |
| */ |
| public Person(String name, int age) { |
| this.name = name; |
| this.age = age; |
| } |
| |
| /** |
| * |
| */ |
| public String getName() { |
| return name; |
| } |
| |
| /** |
| * @param name Name. |
| */ |
| public void setName(String name) { |
| this.name = name; |
| } |
| |
| /** |
| * |
| */ |
| public int getAge() { |
| return age; |
| } |
| |
| /** |
| * @param age Age. |
| */ |
| public void setAge(int age) { |
| this.age = age; |
| } |
| } |
| } |