/*
* 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.jdbc2;
import java.io.Serializable;
import java.sql.DriverManager;
import java.sql.ResultSet;
import java.sql.Statement;
import org.apache.ignite.IgniteCache;
import org.apache.ignite.cache.affinity.AffinityKey;
import org.apache.ignite.cache.query.annotations.QuerySqlField;
import org.apache.ignite.configuration.CacheConfiguration;
import org.apache.ignite.configuration.ConnectorConfiguration;
import org.apache.ignite.configuration.IgniteConfiguration;
import org.apache.ignite.internal.util.typedef.F;
import org.apache.ignite.spi.discovery.tcp.TcpDiscoverySpi;
import org.apache.ignite.spi.discovery.tcp.ipfinder.TcpDiscoveryIpFinder;
import org.apache.ignite.spi.discovery.tcp.ipfinder.vm.TcpDiscoveryVmIpFinder;
import org.apache.ignite.testframework.junits.common.GridCommonAbstractTest;
import org.jetbrains.annotations.NotNull;
import static org.apache.ignite.IgniteJdbcDriver.CFG_URL_PREFIX;
import static org.apache.ignite.cache.CacheAtomicityMode.TRANSACTIONAL;
import static org.apache.ignite.cache.CacheMode.PARTITIONED;
import static org.apache.ignite.cache.CacheWriteSynchronizationMode.FULL_SYNC;
/**
* Tests for complex queries (joins, etc.).
*/
public class JdbcComplexQuerySelfTest extends GridCommonAbstractTest {
/** IP finder. */
private static final TcpDiscoveryIpFinder IP_FINDER = new TcpDiscoveryVmIpFinder(true);
/** JDBC URL. */
private static final String BASE_URL = CFG_URL_PREFIX + "cache=pers@modules/clients/src/test/config/jdbc-config.xml";
/** Statement. */
private Statement stmt;
/** {@inheritDoc} */
@Override protected IgniteConfiguration getConfiguration(String igniteInstanceName) throws Exception {
IgniteConfiguration cfg = super.getConfiguration(igniteInstanceName);
cfg.setCacheConfiguration(
cacheConfiguration("pers", AffinityKey.class, Person.class),
cacheConfiguration("org", String.class, Organization.class));
TcpDiscoverySpi disco = new TcpDiscoverySpi();
disco.setIpFinder(IP_FINDER);
cfg.setDiscoverySpi(disco);
cfg.setConnectorConfiguration(new ConnectorConfiguration());
return cfg;
}
/**
* @param name Name.
* @param clsK Class k.
* @param clsV Class v.
* @return Cache configuration.
*/
protected CacheConfiguration cacheConfiguration(@NotNull String name, Class<?> clsK, Class<?> clsV) {
CacheConfiguration<?,?> cache = defaultCacheConfiguration();
cache.setName(name);
cache.setCacheMode(PARTITIONED);
cache.setBackups(1);
cache.setWriteSynchronizationMode(FULL_SYNC);
cache.setAtomicityMode(TRANSACTIONAL);
cache.setIndexedTypes(clsK, clsV);
return cache;
}
/** {@inheritDoc} */
@Override protected void beforeTestsStarted() throws Exception {
startGrids(3);
IgniteCache<String, Organization> orgCache = grid(0).cache("org");
assert orgCache != null;
orgCache.put("o1", new Organization(1, "A"));
orgCache.put("o2", new Organization(2, "B"));
IgniteCache<AffinityKey, Person> personCache = grid(0).cache("pers");
assert personCache != null;
personCache.put(new AffinityKey<>("p1", "o1"), new Person(1, "John White", 25, 1));
personCache.put(new AffinityKey<>("p2", "o1"), new Person(2, "Joe Black", 35, 1));
personCache.put(new AffinityKey<>("p3", "o2"), new Person(3, "Mike Green", 40, 2));
Class.forName("org.apache.ignite.IgniteJdbcDriver");
}
/** {@inheritDoc} */
@Override protected void afterTestsStopped() throws Exception {
stopAllGrids();
}
/** {@inheritDoc} */
@Override protected void beforeTest() throws Exception {
stmt = DriverManager.getConnection(BASE_URL).createStatement();
assert stmt != null;
assert !stmt.isClosed();
}
/** {@inheritDoc} */
@Override protected void afterTest() throws Exception {
if (stmt != null) {
stmt.getConnection().close();
stmt.close();
assert stmt.isClosed();
}
}
/**
* @throws Exception If failed.
*/
public void testJoin() throws Exception {
ResultSet rs = stmt.executeQuery(
"select p.id, p.name, o.name as orgName from \"pers\".Person p, \"org\".Organization o where p.orgId = o.id");
assert rs != null;
int cnt = 0;
while (rs.next()) {
int id = rs.getInt("id");
if (id == 1) {
assert "John White".equals(rs.getString("name"));
assert "A".equals(rs.getString("orgName"));
}
else if (id == 2) {
assert "Joe Black".equals(rs.getString("name"));
assert "A".equals(rs.getString("orgName"));
}
else if (id == 3) {
assert "Mike Green".equals(rs.getString("name"));
assert "B".equals(rs.getString("orgName"));
}
else
assert false : "Wrong ID: " + id;
cnt++;
}
assert cnt == 3;
}
/**
* @throws Exception If failed.
*/
public void testJoinWithoutAlias() throws Exception {
ResultSet rs = stmt.executeQuery(
"select p.id, p.name, o.name from \"pers\".Person p, \"org\".Organization o where p.orgId = o.id");
assert rs != null;
int cnt = 0;
while (rs.next()) {
int id = rs.getInt(1);
if (id == 1) {
assert "John White".equals(rs.getString("name"));
assert "John White".equals(rs.getString(2));
assert "A".equals(rs.getString(3));
}
else if (id == 2) {
assert "Joe Black".equals(rs.getString("name"));
assert "Joe Black".equals(rs.getString(2));
assert "A".equals(rs.getString(3));
}
else if (id == 3) {
assert "Mike Green".equals(rs.getString("name"));
assert "Mike Green".equals(rs.getString(2));
assert "B".equals(rs.getString(3));
}
else
assert false : "Wrong ID: " + id;
cnt++;
}
assert cnt == 3;
}
/**
* @throws Exception If failed.
*/
public void testIn() throws Exception {
ResultSet rs = stmt.executeQuery("select name from \"pers\".Person where age in (25, 35)");
assert rs != null;
int cnt = 0;
while (rs.next()) {
assert "John White".equals(rs.getString("name")) ||
"Joe Black".equals(rs.getString("name"));
cnt++;
}
assert cnt == 2;
}
/**
* @throws Exception If failed.
*/
public void testBetween() throws Exception {
ResultSet rs = stmt.executeQuery("select name from \"pers\".Person where age between 24 and 36");
assert rs != null;
int cnt = 0;
while (rs.next()) {
assert "John White".equals(rs.getString("name")) ||
"Joe Black".equals(rs.getString("name"));
cnt++;
}
assert cnt == 2;
}
/**
* @throws Exception If failed.
*/
public void testCalculatedValue() throws Exception {
ResultSet rs = stmt.executeQuery("select age * 2 from \"pers\".Person");
assert rs != null;
int cnt = 0;
while (rs.next()) {
assert rs.getInt(1) == 50 ||
rs.getInt(1) == 70 ||
rs.getInt(1) == 80;
cnt++;
}
assert cnt == 3;
}
/**
* Person.
*/
@SuppressWarnings("UnusedDeclaration")
private static class Person implements Serializable {
/** ID. */
@QuerySqlField
private final int id;
/** Name. */
@QuerySqlField(index = false)
private final String name;
/** Age. */
@QuerySqlField
private final int age;
/** Organization ID. */
@QuerySqlField
private final int orgId;
/**
* @param id ID.
* @param name Name.
* @param age Age.
* @param orgId Organization ID.
*/
private Person(int id, String name, int age, int orgId) {
assert !F.isEmpty(name);
assert age > 0;
assert orgId > 0;
this.id = id;
this.name = name;
this.age = age;
this.orgId = orgId;
}
}
/**
* Organization.
*/
@SuppressWarnings("UnusedDeclaration")
private static class Organization implements Serializable {
/** ID. */
@QuerySqlField
private final int id;
/** Name. */
@QuerySqlField(index = false)
private final String name;
/**
* @param id ID.
* @param name Name.
*/
private Organization(int id, String name) {
this.id = id;
this.name = name;
}
}
}