mirror of
https://github.com/OpenHFT/Chronicle-Map.git
synced 2021-09-13 01:26:12 +00:00
554 lines
20 KiB
Java
554 lines
20 KiB
Java
/*
|
|
* Copyright 2014 Higher Frequency Trading
|
|
*
|
|
* http://www.higherfrequencytrading.com
|
|
*
|
|
* Licensed 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 net.openhft.chronicle.map;
|
|
|
|
import net.openhft.chronicle.hash.replication.TcpTransportAndNetworkConfig;
|
|
import org.junit.After;
|
|
import org.junit.Assert;
|
|
import org.junit.Before;
|
|
import org.junit.Test;
|
|
|
|
import java.io.IOException;
|
|
import java.net.InetSocketAddress;
|
|
import java.util.*;
|
|
import java.util.concurrent.ExecutionException;
|
|
|
|
import static org.junit.Assert.assertEquals;
|
|
|
|
/**
|
|
* @author Rob Austin.
|
|
*/
|
|
public class StatelessClientTest {
|
|
|
|
public static final int SIZE = 2500;
|
|
static int s_port = 9050;
|
|
|
|
enum ToString implements Function<Object, String> {
|
|
INSTANCE;
|
|
|
|
@Override
|
|
public String apply(Object obj) {
|
|
return obj.toString();
|
|
}
|
|
}
|
|
|
|
Set<Thread> threads;
|
|
|
|
@Before
|
|
public void sampleThreads() {
|
|
threads = Thread.getAllStackTraces().keySet();
|
|
}
|
|
|
|
@After
|
|
public void checkThreadsShutdown() {
|
|
checkThreadsShutdown(threads);
|
|
}
|
|
|
|
public static void checkThreadsShutdown(Set<Thread> threads) {
|
|
// give them a change to stop if there were killed.
|
|
try {
|
|
Thread.sleep(100);
|
|
} catch (InterruptedException e) {
|
|
e.printStackTrace();
|
|
}
|
|
Map<Thread, StackTraceElement[]> threadMap = Thread.getAllStackTraces();
|
|
threadMap.keySet().removeAll(threads);
|
|
if (!threadMap.isEmpty()) {
|
|
System.out.println("### threads still running after the test ###");
|
|
for (Map.Entry<Thread, StackTraceElement[]> entry : threadMap.entrySet()) {
|
|
System.out.println(entry.getKey());
|
|
for (StackTraceElement ste : entry.getValue()) {
|
|
System.out.println("\t" + ste);
|
|
}
|
|
}
|
|
try {
|
|
for (Thread thread : threadMap.keySet()) {
|
|
if (thread.isAlive()) {
|
|
System.out.println("Waiting for " + thread);
|
|
thread.join(1000);
|
|
if (thread.isAlive()) {
|
|
System.out.println("Forcing " + thread + " to die");
|
|
thread.stop();
|
|
}
|
|
}
|
|
}
|
|
} catch (InterruptedException e) {
|
|
e.printStackTrace();
|
|
}
|
|
}
|
|
}
|
|
|
|
@Test(timeout = 10000)
|
|
public void testMapForKeyWithEntry() throws IOException, InterruptedException {
|
|
int port = s_port++;
|
|
try (ChronicleMap<Integer, StringBuilder> serverMap = ChronicleMapBuilder
|
|
.of(Integer.class, StringBuilder.class)
|
|
.putReturnsNull(true)
|
|
.entries(SIZE)
|
|
.defaultValue(new StringBuilder())
|
|
.replication((byte) 2, TcpTransportAndNetworkConfig.of(port))
|
|
.create()) {
|
|
|
|
serverMap.put(10, new StringBuilder("Hello World"));
|
|
|
|
try (ChronicleMap<Integer, StringBuilder> statelessMap = ChronicleMapBuilder.of(Integer
|
|
.class, StringBuilder.class)
|
|
.putReturnsNull(true)
|
|
.statelessClient(new InetSocketAddress("localhost", port)).create()) {
|
|
|
|
String actual = statelessMap.mapForKey(10, ToString.INSTANCE);
|
|
|
|
assertEquals("Hello World", actual);
|
|
}
|
|
}
|
|
}
|
|
|
|
@Test(timeout = 10000)
|
|
public void testMapForKeyWhenNoEntry() throws IOException, InterruptedException {
|
|
int port = s_port++;
|
|
try (ChronicleMap<Integer, StringBuilder> serverMap = ChronicleMapBuilder
|
|
.of(Integer.class, StringBuilder.class)
|
|
.defaultValue(new StringBuilder())
|
|
.entries(SIZE)
|
|
.replication((byte) 2, TcpTransportAndNetworkConfig.of(port)).create()) {
|
|
|
|
serverMap.put(10, new StringBuilder("Hello World"));
|
|
|
|
try (ChronicleMap<Integer, StringBuilder> statelessMap = ChronicleMapBuilder.of(Integer
|
|
.class, StringBuilder.class)
|
|
.statelessClient(new InetSocketAddress("localhost", port)).create()) {
|
|
|
|
String actual = statelessMap.mapForKey(11, ToString.INSTANCE);
|
|
|
|
assertEquals(null, actual);
|
|
}
|
|
}
|
|
}
|
|
|
|
@Test(timeout = 10000)
|
|
public void testBufferOverFlowPutAllAndEntrySet() throws IOException, InterruptedException {
|
|
int port = s_port++;
|
|
try (ChronicleMap<Integer, CharSequence> serverMap = ChronicleMapBuilder
|
|
.of(Integer.class, CharSequence.class)
|
|
.entries(SIZE)
|
|
.replication((byte) 2, TcpTransportAndNetworkConfig.of(port))
|
|
.create()) {
|
|
|
|
try (ChronicleMap<Integer, CharSequence> statelessMap = ChronicleMapBuilder
|
|
.of(Integer.class, CharSequence.class)
|
|
.statelessClient(new InetSocketAddress("localhost", port))
|
|
.create()) {
|
|
|
|
Map<Integer, CharSequence> payload = new HashMap<Integer, CharSequence>();
|
|
|
|
for (int i = 0; i < SIZE; i++) {
|
|
payload.put(i, "some value=" + i);
|
|
}
|
|
|
|
statelessMap.putAll(payload);
|
|
|
|
Set<Map.Entry<Integer, CharSequence>> entries = statelessMap.entrySet();
|
|
|
|
Map.Entry<Integer, CharSequence> next = entries.iterator().next();
|
|
assertEquals("some value=" + next.getKey(), next.getValue());
|
|
|
|
assertEquals(SIZE, entries.size());
|
|
|
|
}
|
|
}
|
|
}
|
|
|
|
@Test(timeout = 10000)
|
|
public void testBufferOverFlowPutAllAndValues() throws IOException, InterruptedException {
|
|
int port = s_port++;
|
|
|
|
try (ChronicleMap<Integer, CharSequence> serverMap = ChronicleMapBuilder
|
|
.of(Integer.class, CharSequence.class)
|
|
.entries(SIZE)
|
|
.replication((byte) 2, TcpTransportAndNetworkConfig.of(port))
|
|
.create()) {
|
|
|
|
try (ChronicleMap<Integer, CharSequence> statelessMap = ChronicleMapBuilder
|
|
.of(Integer.class, CharSequence.class)
|
|
.statelessClient(new InetSocketAddress("localhost", port))
|
|
.create()) {
|
|
|
|
Map<Integer, CharSequence> payload = new HashMap<Integer, CharSequence>();
|
|
|
|
for (int i = 0; i < SIZE; i++) {
|
|
payload.put(i, "some value=" + i);
|
|
}
|
|
|
|
statelessMap.putAll(payload);
|
|
|
|
Collection<CharSequence> values = statelessMap.values();
|
|
|
|
assertEquals(values.size(), SIZE);
|
|
|
|
}
|
|
}
|
|
}
|
|
|
|
@Test(timeout = 10000)
|
|
public void testBufferOverFlowPutAllAndKeySet() throws IOException, InterruptedException {
|
|
int port = s_port++;
|
|
|
|
try (ChronicleMap<Integer, CharSequence> serverMap = ChronicleMapBuilder
|
|
.of(Integer.class, CharSequence.class)
|
|
.entries(SIZE)
|
|
.replication((byte) 2, TcpTransportAndNetworkConfig.of(port))
|
|
.create()) {
|
|
|
|
try (ChronicleMap<Integer, CharSequence> statelessMap = ChronicleMapBuilder
|
|
.of(Integer.class, CharSequence.class)
|
|
.statelessClient(new InetSocketAddress("localhost", port))
|
|
.create()) {
|
|
Map<Integer, CharSequence> payload = new HashMap<Integer, CharSequence>();
|
|
|
|
for (int i = 0; i < SIZE; i++) {
|
|
payload.put(i, "some value=" + i);
|
|
}
|
|
|
|
statelessMap.putAll(payload);
|
|
|
|
final Set<Integer> keys = statelessMap.keySet();
|
|
|
|
assertEquals(keys.size(), payload.size());
|
|
|
|
}
|
|
}
|
|
}
|
|
|
|
@Test(timeout = 10000)
|
|
public void test() throws IOException, InterruptedException {
|
|
int port = s_port++;
|
|
|
|
try (ChronicleMap<Integer, CharSequence> serverMap = ChronicleMapBuilder
|
|
.of(Integer.class, CharSequence.class)
|
|
.entries(SIZE)
|
|
.replication((byte) 2, TcpTransportAndNetworkConfig.of(port))
|
|
.create()) {
|
|
|
|
serverMap.put(10, "EXAMPLE-10");
|
|
try (ChronicleMap<Integer, CharSequence> statelessMap = ChronicleMapBuilder.of(Integer
|
|
.class, CharSequence.class)
|
|
.statelessClient(new InetSocketAddress("localhost", port)).create()) {
|
|
|
|
assertEquals("EXAMPLE-10", statelessMap.get(10));
|
|
assertEquals(1, statelessMap.size());
|
|
}
|
|
}
|
|
}
|
|
|
|
@Test(timeout = 10000)
|
|
public void testClientCreatedBeforeServer() throws IOException, InterruptedException {
|
|
int port = s_port++;
|
|
|
|
try (ChronicleMap<Integer, CharSequence> serverMap = ChronicleMapBuilder.of(Integer.class, CharSequence.class)
|
|
.replication((byte) 2, TcpTransportAndNetworkConfig.of(port)).create()) {
|
|
|
|
try (ChronicleMap<Integer, CharSequence> statelessMap = ChronicleMapBuilder.of(Integer
|
|
.class, CharSequence.class)
|
|
.statelessClient(new InetSocketAddress("localhost", port)).create()) {
|
|
|
|
serverMap.put(10, "EXAMPLE-10");
|
|
|
|
assertEquals("EXAMPLE-10", statelessMap.get(10));
|
|
assertEquals(1, statelessMap.size());
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
@Test(timeout = 10000)
|
|
public void testServerPutStringKeyMap() throws IOException, InterruptedException {
|
|
|
|
try (ChronicleMap<String, Map> serverMap = ChronicleMapBuilder
|
|
.of(String.class, Map.class)
|
|
.entries(SIZE)
|
|
.replication((byte) 2, TcpTransportAndNetworkConfig.of(8056))
|
|
.create()) {
|
|
|
|
try (ChronicleMap<String, Map> statelessMap = ChronicleMapBuilder
|
|
.of(String.class, Map.class)
|
|
.statelessClient(new InetSocketAddress("localhost", 8056))
|
|
.create()) {
|
|
|
|
serverMap.put("hello", Collections.singletonMap("hello", "world"));
|
|
|
|
assertEquals(Collections.singletonMap("hello", "world"), statelessMap.get("hello"));
|
|
assertEquals(1, statelessMap.size());
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
@Test(timeout = 10000)
|
|
public void testStringKeyMapPutIntoStatelessMap() throws IOException, InterruptedException {
|
|
|
|
final Map<String, String> data = new HashMap<String, String>();
|
|
|
|
String value = new String(new char[10]);
|
|
for (int i = 0; i < 1000; i++) {
|
|
data.put("" + i, value);
|
|
}
|
|
|
|
try (ChronicleMap<String, Map> serverMap = ChronicleMapBuilder
|
|
.of(String.class, Map.class)
|
|
.entries(SIZE)
|
|
.replication((byte) 2, TcpTransportAndNetworkConfig.of(8056))
|
|
.create()) {
|
|
|
|
try (ChronicleMap<String, Map> statelessMap = ChronicleMapBuilder
|
|
.of(String.class, Map.class)
|
|
.statelessClient(new InetSocketAddress("localhost", 8056))
|
|
.create()) {
|
|
|
|
statelessMap.put("hello", data);
|
|
|
|
assertEquals(data, serverMap.get("hello"));
|
|
assertEquals(1, statelessMap.size());
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
@Test(timeout = 10000)
|
|
public void testBufferOverFlowPutAll() throws IOException, InterruptedException {
|
|
try (ChronicleMap<Integer, CharSequence> serverMap = ChronicleMapBuilder
|
|
.of(Integer.class, CharSequence.class)
|
|
.entries(SIZE)
|
|
.replication((byte) 2, TcpTransportAndNetworkConfig.of(8056))
|
|
.create()) {
|
|
|
|
try (ChronicleMap<Integer, CharSequence> statelessMap = ChronicleMapBuilder
|
|
.of(Integer.class, CharSequence.class)
|
|
.statelessClient(new InetSocketAddress("localhost", 8056))
|
|
.create()) {
|
|
|
|
Map<Integer, CharSequence> payload = new HashMap<Integer, CharSequence>();
|
|
|
|
for (int i = 0; i < SIZE; i++) {
|
|
payload.put(i, "some value=" + i);
|
|
}
|
|
|
|
statelessMap.putAll(payload);
|
|
|
|
int value = SIZE - 10;
|
|
|
|
assertEquals("some value=" + value, statelessMap.get(value));
|
|
assertEquals(SIZE, statelessMap.size());
|
|
|
|
}
|
|
}
|
|
}
|
|
|
|
@Test(timeout = 10000)
|
|
public void testBufferOverFlowPutAllWherePutReturnsNull() throws IOException, InterruptedException {
|
|
|
|
try (ChronicleMap<Integer, CharSequence> serverMap = ChronicleMapBuilder
|
|
.of(Integer.class, CharSequence.class)
|
|
.entries(SIZE)
|
|
.replication((byte) 2, TcpTransportAndNetworkConfig.of(8056))
|
|
.create()) {
|
|
|
|
try (ChronicleMap<Integer, CharSequence> statelessMap = ChronicleMapBuilder
|
|
.of(Integer.class, CharSequence.class)
|
|
.statelessClient(new InetSocketAddress("localhost", 8056))
|
|
.create()) {
|
|
|
|
Map<Integer, CharSequence> payload = new HashMap<Integer, CharSequence>();
|
|
|
|
for (int i = 0; i < SIZE; i++) {
|
|
payload.put(i, "some value=" + i);
|
|
}
|
|
|
|
statelessMap.putAll(payload);
|
|
|
|
int value = SIZE - 10;
|
|
|
|
assertEquals("some value=" + value, statelessMap.get(value));
|
|
assertEquals(SIZE, statelessMap.size());
|
|
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
@Test(timeout = 10000)
|
|
public void testPutWherePutReturnsNull() throws IOException,
|
|
InterruptedException {
|
|
|
|
try (ChronicleMap<Integer, CharSequence> serverMap = ChronicleMapBuilder.of(Integer.class, CharSequence.class)
|
|
.replication((byte) 2, TcpTransportAndNetworkConfig.of(8056)).create()) {
|
|
|
|
try (ChronicleMap<Integer, CharSequence> statelessMap = ChronicleMapBuilder.of(Integer
|
|
.class, CharSequence.class)
|
|
.statelessClient(new InetSocketAddress("localhost", 8056)).create()) {
|
|
|
|
statelessMap.put(1, "some value");
|
|
|
|
assertEquals("some value", statelessMap.get(1));
|
|
assertEquals(1, statelessMap.size());
|
|
|
|
}
|
|
}
|
|
}
|
|
|
|
@Test(timeout = 10000)
|
|
public void testRemoveWhereRemoveReturnsNull() throws IOException,
|
|
InterruptedException {
|
|
|
|
try (ChronicleMap<Integer, CharSequence> serverMap = ChronicleMapBuilder.of(Integer.class, CharSequence.class)
|
|
.replication((byte) 2, TcpTransportAndNetworkConfig.of(8056)).create()) {
|
|
|
|
try (ChronicleMap<Integer, CharSequence> statelessMap = ChronicleMapBuilder.of(Integer
|
|
.class, CharSequence.class)
|
|
.statelessClient(new InetSocketAddress("localhost", 8056)).create()) {
|
|
|
|
statelessMap.put(1, "some value");
|
|
|
|
assertEquals("some value", statelessMap.get(1));
|
|
assertEquals(1, statelessMap.size());
|
|
|
|
statelessMap.remove(1);
|
|
|
|
assertEquals(null, statelessMap.get(1));
|
|
assertEquals(0, statelessMap.size());
|
|
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
@Test(timeout = 10000)
|
|
public void testGetLater() throws IOException,
|
|
InterruptedException, ExecutionException {
|
|
|
|
try (ChronicleMap<Integer, CharSequence> serverMap = ChronicleMapBuilder.of(Integer.class, CharSequence.class)
|
|
.replication((byte) 2, TcpTransportAndNetworkConfig.of(8056)).create()) {
|
|
|
|
try (ChronicleMap<Integer, CharSequence> statelessMap = ChronicleMapBuilder.of(Integer
|
|
.class, CharSequence.class)
|
|
.statelessClient(new InetSocketAddress("localhost", 8056)).create()) {
|
|
|
|
statelessMap.put(1, "some value");
|
|
|
|
assertEquals("some value", statelessMap.getLater(1).get());
|
|
assertEquals(1, statelessMap.size());
|
|
|
|
statelessMap.remove(1);
|
|
|
|
assertEquals(null, statelessMap.getLater(1).get());
|
|
assertEquals(0, statelessMap.size());
|
|
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
@Test
|
|
public void testPutLater() throws IOException,
|
|
InterruptedException, ExecutionException {
|
|
|
|
try (ChronicleMap<Integer, CharSequence> serverMap = ChronicleMapBuilder.of(Integer.class, CharSequence.class)
|
|
.replication((byte) 2, TcpTransportAndNetworkConfig.of(8056)).create()) {
|
|
|
|
try (ChronicleMap<Integer, CharSequence> statelessMap = ChronicleMapBuilder.of(Integer
|
|
.class, CharSequence.class)
|
|
.statelessClient(new InetSocketAddress("localhost", 8056)).create()) {
|
|
|
|
CharSequence oldValue = statelessMap.putLater(1, "some value").get();
|
|
assertEquals("some value", statelessMap.get(1));
|
|
assertEquals(1, statelessMap.size());
|
|
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
@Test(timeout = 10000)
|
|
public void testRemoveLater() throws IOException,
|
|
InterruptedException, ExecutionException {
|
|
|
|
try (ChronicleMap<Integer, CharSequence> serverMap = ChronicleMapBuilder.of(Integer.class, CharSequence.class)
|
|
.replication((byte) 2, TcpTransportAndNetworkConfig.of(8056)).create()) {
|
|
|
|
try (ChronicleMap<Integer, CharSequence> statelessMap = ChronicleMapBuilder.of(Integer
|
|
.class, CharSequence.class)
|
|
.statelessClient(new InetSocketAddress("localhost", 8056)).create()) {
|
|
statelessMap.put(1, "some value");
|
|
statelessMap.removeLater(1).get();
|
|
assertEquals(null, statelessMap.get(1));
|
|
assertEquals(0, statelessMap.size());
|
|
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
@Test(timeout = 10000)
|
|
public void testEquals() throws IOException, InterruptedException {
|
|
|
|
final ChronicleMap<Integer, CharSequence> serverMap1;
|
|
final ChronicleMap<Integer, CharSequence> serverMap2;
|
|
final ChronicleMap<Integer, CharSequence> statelessMap1;
|
|
final ChronicleMap<Integer, CharSequence> statelessMap2;
|
|
|
|
// server
|
|
{
|
|
serverMap1 = ChronicleMapBuilder.of(Integer.class, CharSequence.class)
|
|
.replication((byte) 2, TcpTransportAndNetworkConfig.of(8056)).create();
|
|
serverMap2 = ChronicleMapBuilder.of(Integer.class, CharSequence.class)
|
|
.replication((byte) 2, TcpTransportAndNetworkConfig.of(8077)).create();
|
|
|
|
}
|
|
|
|
// stateless client
|
|
{
|
|
statelessMap1 = ChronicleMapBuilder.of(Integer
|
|
.class, CharSequence.class)
|
|
.statelessClient(new InetSocketAddress("localhost", 8056)).create();
|
|
statelessMap2 = ChronicleMapBuilder.of(Integer
|
|
.class, CharSequence.class)
|
|
.statelessClient(new InetSocketAddress("localhost", 8077)).create();
|
|
}
|
|
|
|
Map<Integer, CharSequence> payload = new HashMap<Integer, CharSequence>();
|
|
for (int i = 0; i < 1000; i++) {
|
|
payload.put(i, "some value=" + i);
|
|
}
|
|
|
|
statelessMap1.putAll(payload);
|
|
Assert.assertTrue(statelessMap1.equals(payload));
|
|
|
|
statelessMap1.close();
|
|
statelessMap2.close();
|
|
|
|
serverMap1.close();
|
|
serverMap2.close();
|
|
|
|
}
|
|
} |