/*
* 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.rest;
import java.lang.reflect.Field;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.UUID;
import java.util.concurrent.Callable;
import org.apache.ignite.IgniteCache;
import org.apache.ignite.IgniteCheckedException;
import org.apache.ignite.compute.ComputeJob;
import org.apache.ignite.compute.ComputeJobAdapter;
import org.apache.ignite.compute.ComputeJobResult;
import org.apache.ignite.compute.ComputeTaskSplitAdapter;
import org.apache.ignite.configuration.CacheConfiguration;
import org.apache.ignite.configuration.ConnectorConfiguration;
import org.apache.ignite.configuration.IgniteConfiguration;
import org.apache.ignite.internal.IgniteKernal;
import org.apache.ignite.internal.processors.rest.client.message.GridClientNodeBean;
import org.apache.ignite.internal.processors.rest.client.message.GridClientTaskResultBean;
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.GridTestUtils;
import org.apache.ignite.testframework.junits.common.GridCommonAbstractTest;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;
import static org.apache.ignite.cache.CacheMode.LOCAL;
import static org.apache.ignite.cache.CacheWriteSynchronizationMode.FULL_SYNC;
/**
* TCP protocol test.
*/
@SuppressWarnings("unchecked")
public class RestBinaryProtocolSelfTest extends GridCommonAbstractTest {
/** */
private static final TcpDiscoveryIpFinder IP_FINDER = new TcpDiscoveryVmIpFinder(true);
/** */
private static final String CACHE_NAME = "cache";
/** */
private static final String HOST = "127.0.0.1";
/** */
private static final int PORT = 11212;
/** */
private TestBinaryClient client;
/** {@inheritDoc} */
@Override protected void beforeTestsStarted() throws Exception {
startGrid();
}
/** {@inheritDoc} */
@Override protected void afterTestsStopped() throws Exception {
stopAllGrids();
}
/** {@inheritDoc} */
@Override protected void beforeTest() throws Exception {
client = client();
}
/** {@inheritDoc} */
@Override protected void afterTest() throws Exception {
client.shutdown();
grid().cache(DEFAULT_CACHE_NAME).clear();
grid().cache(CACHE_NAME).clear();
}
/** {@inheritDoc} */
@Override protected IgniteConfiguration getConfiguration(String igniteInstanceName) throws Exception {
IgniteConfiguration cfg = super.getConfiguration(igniteInstanceName);
cfg.setLocalHost(HOST);
assert cfg.getConnectorConfiguration() == null;
ConnectorConfiguration clientCfg = new ConnectorConfiguration();
clientCfg.setPort(PORT);
cfg.setConnectorConfiguration(clientCfg);
TcpDiscoverySpi disco = new TcpDiscoverySpi();
disco.setIpFinder(IP_FINDER);
cfg.setDiscoverySpi(disco);
cfg.setCacheConfiguration(cacheConfiguration(DEFAULT_CACHE_NAME), cacheConfiguration(CACHE_NAME));
return cfg;
}
/**
* @param cacheName Cache name.
* @return Cache configuration.
* @throws Exception In case of error.
*/
private CacheConfiguration cacheConfiguration(@NotNull String cacheName) throws Exception {
CacheConfiguration cfg = defaultCacheConfiguration();
cfg.setCacheMode(LOCAL);
cfg.setName(cacheName);
cfg.setWriteSynchronizationMode(FULL_SYNC);
cfg.setStatisticsEnabled(true);
return cfg;
}
/**
* @return Client.
* @throws IgniteCheckedException In case of error.
*/
private TestBinaryClient client() throws IgniteCheckedException {
return new TestBinaryClient(HOST, PORT);
}
/**
* @throws Exception If failed.
*/
public void testPut() throws Exception {
assertTrue(client.cachePut(DEFAULT_CACHE_NAME, "key1", "val1"));
assertEquals("val1", grid().cache(DEFAULT_CACHE_NAME).get("key1"));
assertTrue(client.cachePut(CACHE_NAME, "key1", "val1"));
assertEquals("val1", grid().cache(CACHE_NAME).get("key1"));
}
/**
* @throws Exception If failed.
*/
public void testPutAll() throws Exception {
client.cachePutAll(DEFAULT_CACHE_NAME, F.asMap("key1", "val1", "key2", "val2"));
Map<String, String> map = grid().<String, String>cache(DEFAULT_CACHE_NAME).getAll(F.asSet("key1", "key2"));
assertEquals(2, map.size());
assertEquals("val1", map.get("key1"));
assertEquals("val2", map.get("key2"));
client.cachePutAll(CACHE_NAME, F.asMap("key1", "val1", "key2", "val2"));
map = grid().<String, String>cache(CACHE_NAME).getAll(F.asSet("key1", "key2"));
assertEquals(2, map.size());
assertEquals("val1", map.get("key1"));
assertEquals("val2", map.get("key2"));
}
/**
* @throws Exception If failed.
*/
public void testGet() throws Exception {
grid().cache(DEFAULT_CACHE_NAME).put("key", "val");
assertEquals("val", client.cacheGet(DEFAULT_CACHE_NAME, "key"));
grid().cache(CACHE_NAME).put("key", "val");
assertEquals("val", client.cacheGet(CACHE_NAME, "key"));
}
/**
* @throws Exception If failed.
*/
public void testFailure() throws Exception {
IgniteKernal kernal = ((IgniteKernal)grid());
GridRestProcessor proc = kernal.context().rest();
// Clearing handlers collection to force failure.
Field hndField = proc.getClass().getDeclaredField("handlers");
hndField.setAccessible(true);
Map handlers = (Map)hndField.get(proc);
Map cp = new HashMap(handlers);
handlers.clear();
try {
GridTestUtils.assertThrows(
log,
new Callable<Object>() {
@Override public Object call() throws Exception {
return client.cacheGet(DEFAULT_CACHE_NAME, "key");
}
},
IgniteCheckedException.class,
"Failed to process client request: Failed to find registered handler for command: CACHE_GET");
}
finally {
handlers.putAll(cp);
}
}
/**
* @throws Exception If failed.
*/
public void testGetAll() throws Exception {
IgniteCache<Object, Object> jcacheDflt = grid().cache(DEFAULT_CACHE_NAME);
IgniteCache<Object, Object> jcacheName = grid().cache(CACHE_NAME);
jcacheDflt.put("key1", "val1");
jcacheDflt.put("key2", "val2");
Map<String, String> map = client.cacheGetAll(DEFAULT_CACHE_NAME, "key1", "key2");
assertEquals(2, map.size());
assertEquals("val1", map.get("key1"));
assertEquals("val2", map.get("key2"));
jcacheDflt.put("key3", "val3");
jcacheDflt.put("key4", "val4");
map = client.cacheGetAll(DEFAULT_CACHE_NAME, "key3", "key4");
assertEquals(2, map.size());
assertEquals("val3", map.get("key3"));
assertEquals("val4", map.get("key4"));
jcacheName.put("key1", "val1");
jcacheName.put("key2", "val2");
map = client.cacheGetAll(CACHE_NAME, "key1", "key2");
assertEquals(2, map.size());
assertEquals("val1", map.get("key1"));
assertEquals("val2", map.get("key2"));
jcacheName.put("key3", "val3");
jcacheName.put("key4", "val4");
map = client.cacheGetAll(CACHE_NAME, "key3", "key4");
assertEquals(2, map.size());
assertEquals("val3", map.get("key3"));
assertEquals("val4", map.get("key4"));
}
/**
* @throws Exception If failed.
*/
public void testRemove() throws Exception {
IgniteCache<Object, Object> jcacheDflt = grid().cache(DEFAULT_CACHE_NAME);
IgniteCache<Object, Object> jcacheName = grid().cache(CACHE_NAME);
jcacheDflt.put("key", "val");
assertTrue(client.cacheRemove(DEFAULT_CACHE_NAME, "key"));
assertFalse(client.cacheRemove(DEFAULT_CACHE_NAME, "wrongKey"));
assertNull(jcacheDflt.get("key"));
jcacheName.put("key", "val");
assertTrue(client.cacheRemove(CACHE_NAME, "key"));
assertFalse(client.cacheRemove(CACHE_NAME, "wrongKey"));
assertNull(jcacheName.get("key"));
}
/**
* @throws Exception If failed.
*/
public void testRemoveAll() throws Exception {
IgniteCache<Object, Object> jcacheDflt = grid().cache(DEFAULT_CACHE_NAME);
jcacheDflt.put("key1", "val1");
jcacheDflt.put("key2", "val2");
jcacheDflt.put("key3", "val3");
jcacheDflt.put("key4", "val4");
client.cacheRemoveAll(DEFAULT_CACHE_NAME, "key1", "key2");
assertNull(jcacheDflt.get("key1"));
assertNull(jcacheDflt.get("key2"));
assertNotNull(jcacheDflt.get("key3"));
assertNotNull(jcacheDflt.get("key4"));
IgniteCache<Object, Object> jcacheName = grid().cache(CACHE_NAME);
jcacheName.put("key1", "val1");
jcacheName.put("key2", "val2");
jcacheName.put("key3", "val3");
jcacheName.put("key4", "val4");
client.cacheRemoveAll(CACHE_NAME, "key1", "key2");
assertNull(jcacheName.get("key1"));
assertNull(jcacheName.get("key2"));
assertNotNull(jcacheName.get("key3"));
assertNotNull(jcacheName.get("key4"));
}
/**
* @throws Exception If failed.
*/
public void testReplace() throws Exception {
assertFalse(client.cacheReplace(DEFAULT_CACHE_NAME, "key1", "val1"));
IgniteCache<Object, Object> jcacheDflt = grid().cache(DEFAULT_CACHE_NAME);
jcacheDflt.put("key1", "val1");
assertTrue(client.cacheReplace(DEFAULT_CACHE_NAME, "key1", "val2"));
assertFalse(client.cacheReplace(DEFAULT_CACHE_NAME, "key2", "val1"));
jcacheDflt.put("key2", "val1");
assertTrue(client.cacheReplace(DEFAULT_CACHE_NAME, "key2", "val2"));
jcacheDflt.clear();
assertFalse(client.cacheReplace(CACHE_NAME, "key1", "val1"));
grid().cache(CACHE_NAME).put("key1", "val1");
assertTrue(client.cacheReplace(CACHE_NAME, "key1", "val2"));
}
/**
* @throws Exception If failed.
*/
public void testCompareAndSet() throws Exception {
assertFalse(client.cacheCompareAndSet(DEFAULT_CACHE_NAME, "key", null, null));
IgniteCache<Object, Object> jcache = grid().cache(DEFAULT_CACHE_NAME);
jcache.put("key", "val");
assertTrue(client.cacheCompareAndSet(DEFAULT_CACHE_NAME, "key", null, null));
assertNull(jcache.get("key"));
assertFalse(client.cacheCompareAndSet(DEFAULT_CACHE_NAME, "key", null, "val"));
jcache.put("key", "val");
assertFalse(client.cacheCompareAndSet(DEFAULT_CACHE_NAME, "key", null, "wrongVal"));
assertEquals("val", jcache.get("key"));
assertTrue(client.cacheCompareAndSet(DEFAULT_CACHE_NAME, "key", null, "val"));
assertNull(jcache.get("key"));
assertTrue(client.cacheCompareAndSet(DEFAULT_CACHE_NAME, "key", "val", null));
assertEquals("val", jcache.get("key"));
assertFalse(client.cacheCompareAndSet(DEFAULT_CACHE_NAME, "key", "newVal", null));
assertEquals("val", jcache.get("key"));
jcache.remove("key");
assertFalse(client.cacheCompareAndSet(DEFAULT_CACHE_NAME, "key", "val1", "val2"));
jcache.put("key", "val2");
assertFalse(client.cacheCompareAndSet(DEFAULT_CACHE_NAME, "key", "val1", "wrongVal"));
assertEquals("val2", jcache.get("key"));
assertTrue(client.cacheCompareAndSet(DEFAULT_CACHE_NAME, "key", "val1", "val2"));
assertEquals("val1", jcache.get("key"));
jcache.remove("key");
assertFalse(client.cacheCompareAndSet(CACHE_NAME, "key", null, null));
IgniteCache<Object, Object> jcacheName = grid().cache(CACHE_NAME);
jcacheName.put("key", "val");
assertTrue(client.cacheCompareAndSet(CACHE_NAME, "key", null, null));
assertNull(jcacheName.get("key"));
assertFalse(client.cacheCompareAndSet(CACHE_NAME, "key", null, "val"));
jcacheName.put("key", "val");
assertFalse(client.cacheCompareAndSet(CACHE_NAME, "key", null, "wrongVal"));
assertEquals("val", jcacheName.get("key"));
assertTrue(client.cacheCompareAndSet(CACHE_NAME, "key", null, "val"));
assertNull(jcacheName.get("key"));
assertTrue(client.cacheCompareAndSet(CACHE_NAME, "key", "val", null));
assertEquals("val", jcacheName.get("key"));
assertFalse(client.cacheCompareAndSet(CACHE_NAME, "key", "newVal", null));
assertEquals("val", jcacheName.get("key"));
jcacheName.remove("key");
assertFalse(client.cacheCompareAndSet(CACHE_NAME, "key", "val1", "val2"));
jcacheName.put("key", "val2");
assertFalse(client.cacheCompareAndSet(CACHE_NAME, "key", "val1", "wrongVal"));
assertEquals("val2", jcacheName.get("key"));
assertTrue(client.cacheCompareAndSet(CACHE_NAME, "key", "val1", "val2"));
assertEquals("val1", jcacheName.get("key"));
jcacheName.remove("key");
}
/**
* @throws Exception If failed.
*/
public void testMetrics() throws Exception {
IgniteCache<Object, Object> jcacheDft = grid().cache(DEFAULT_CACHE_NAME);
IgniteCache<Object, Object> jcacheName = grid().cache(CACHE_NAME);
jcacheDft.localMxBean().clear();
jcacheName.localMxBean().clear();
jcacheDft.put("key1", "val");
jcacheDft.put("key2", "val");
jcacheDft.put("key2", "val");
jcacheDft.get("key1");
jcacheDft.get("key2");
jcacheDft.get("key2");
jcacheName.put("key1", "val");
jcacheName.put("key2", "val");
jcacheName.put("key2", "val");
jcacheName.get("key1");
jcacheName.get("key2");
jcacheName.get("key2");
Map<String, Long> m = client.cacheMetrics(DEFAULT_CACHE_NAME);
assertNotNull(m);
assertEquals(4, m.size());
assertEquals(3, m.get("reads").longValue());
assertEquals(3, m.get("writes").longValue());
m = client.cacheMetrics(CACHE_NAME);
assertNotNull(m);
assertEquals(4, m.size());
assertEquals(3, m.get("reads").longValue());
assertEquals(3, m.get("writes").longValue());
}
/**
* @throws Exception If failed.
*/
public void testAppend() throws Exception {
grid().cache(DEFAULT_CACHE_NAME).remove("key");
grid().cache(CACHE_NAME).remove("key");
assertFalse(client.cacheAppend(DEFAULT_CACHE_NAME, "key", ".val"));
assertFalse(client.cacheAppend(CACHE_NAME, "key", ".val"));
grid().cache(DEFAULT_CACHE_NAME).put("key", "orig");
grid().cache(CACHE_NAME).put("key", "orig");
assertTrue(client.cacheAppend(DEFAULT_CACHE_NAME, "key", ".val"));
assertEquals("orig.val", grid().cache(DEFAULT_CACHE_NAME).get("key"));
assertTrue(client.cacheAppend(DEFAULT_CACHE_NAME, "key", ".newVal"));
assertEquals("orig.val.newVal", grid().cache(DEFAULT_CACHE_NAME).get("key"));
assertTrue(client.cacheAppend(CACHE_NAME, "key", ".val"));
assertEquals("orig.val", grid().cache(CACHE_NAME).get("key"));
assertTrue(client.cacheAppend(CACHE_NAME, "key", ".newVal"));
assertEquals("orig.val.newVal", grid().cache(CACHE_NAME).get("key"));
}
/**
* @throws Exception If failed.
*/
public void testPrepend() throws Exception {
grid().cache(DEFAULT_CACHE_NAME).remove("key");
grid().cache(CACHE_NAME).remove("key");
assertFalse(client.cachePrepend(DEFAULT_CACHE_NAME, "key", ".val"));
assertFalse(client.cachePrepend(CACHE_NAME, "key", ".val"));
grid().cache(DEFAULT_CACHE_NAME).put("key", "orig");
grid().cache(CACHE_NAME).put("key", "orig");
assertTrue(client.cachePrepend(DEFAULT_CACHE_NAME, "key", "val."));
assertEquals("val.orig", grid().cache(DEFAULT_CACHE_NAME).get("key"));
assertTrue(client.cachePrepend(DEFAULT_CACHE_NAME, "key", "newVal."));
assertEquals("newVal.val.orig", grid().cache(DEFAULT_CACHE_NAME).get("key"));
assertTrue(client.cachePrepend(CACHE_NAME, "key", "val."));
assertEquals("val.orig", grid().cache(CACHE_NAME).get("key"));
assertTrue(client.cachePrepend(CACHE_NAME, "key", "newVal."));
assertEquals("newVal.val.orig", grid().cache(CACHE_NAME).get("key"));
}
/**
* @throws Exception If failed.
*/
public void testExecute() throws Exception {
GridClientTaskResultBean res = client.execute(TestTask.class.getName(),
Arrays.asList("executing", 3, "test", 5, "task"));
assertTrue(res.isFinished());
assertEquals(new Integer(25), res.getResult());
}
/**
* @throws Exception If failed.
*/
public void testNode() throws Exception {
assertNull(client.node(UUID.randomUUID(), false, false));
assertNull(client.node("wrongHost", false, false));
GridClientNodeBean node = client.node(grid().localNode().id(), true, true);
assertNotNull(node);
assertFalse(node.getAttributes().isEmpty());
assertNotNull(node.getMetrics());
assertNotNull(node.getTcpAddresses());
assertEquals(PORT, node.getTcpPort());
assertEquals(grid().localNode().id(), node.getNodeId());
node = client.node(grid().localNode().id(), false, false);
assertNotNull(node);
assertNull(node.getAttributes());
assertNull(node.getMetrics());
assertNotNull(node.getTcpAddresses());
assertEquals(PORT, node.getTcpPort());
assertEquals(grid().localNode().id(), node.getNodeId());
node = client.node(HOST, true, true);
assertNotNull(node);
assertFalse(node.getAttributes().isEmpty());
assertNotNull(node.getMetrics());
assertNotNull(node.getTcpAddresses());
assertEquals(PORT, node.getTcpPort());
assertEquals(grid().localNode().id(), node.getNodeId());
node = client.node(HOST, false, false);
assertNotNull(node);
assertNull(node.getAttributes());
assertNull(node.getMetrics());
assertNotNull(node.getTcpAddresses());
assertEquals(PORT, node.getTcpPort());
assertEquals(grid().localNode().id(), node.getNodeId());
}
/**
* @throws Exception If failed.
*/
public void testTopology() throws Exception {
List<GridClientNodeBean> top = client.topology(true, true);
assertNotNull(top);
assertEquals(1, top.size());
GridClientNodeBean node = F.first(top);
assertNotNull(node);
assertFalse(node.getAttributes().isEmpty());
assertNotNull(node.getMetrics());
assertNotNull(node.getTcpAddresses());
assertEquals(grid().localNode().id(), node.getNodeId());
top = client.topology(false, false);
assertNotNull(top);
assertEquals(1, top.size());
node = F.first(top);
assertNotNull(node);
assertNull(node.getAttributes());
assertNull(node.getMetrics());
assertNotNull(node.getTcpAddresses());
assertEquals(grid().localNode().id(), node.getNodeId());
}
/**
* Test task.
*/
private static class TestTask extends ComputeTaskSplitAdapter<List<Object>, Integer> {
/** {@inheritDoc} */
@Override protected Collection<? extends ComputeJob> split(int gridSize, List<Object> args) {
Collection<ComputeJobAdapter> jobs = new ArrayList<>(args.size());
for (final Object arg : args) {
jobs.add(new ComputeJobAdapter() {
@SuppressWarnings("OverlyStrongTypeCast")
@Override public Object execute() {
try {
return ((String)arg).length();
}
catch (ClassCastException ignored) {
assert arg instanceof Integer;
return arg;
}
}
});
}
return jobs;
}
/** {@inheritDoc} */
@Override public Integer reduce(List<ComputeJobResult> results) {
int sum = 0;
for (ComputeJobResult res : results)
sum += res.<Integer>getData();
return sum;
}
}
}