/*
* Copyright (c) 2008-2013, Hazelcast, Inc. All Rights Reserved.
*
* 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 com.hazelcast.demo;
import com.hazelcast.core.Hazelcast;
import com.hazelcast.core.HazelcastInstance;
import com.hazelcast.core.Member;
import com.hazelcast.core.MultiMap;
import com.hazelcast.core.Partition;
import java.util.Random;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.logging.Logger;
/**
* Tests for MultiMap
*/
public final class SimpleMultiMapTest {
private static final int STATS_SECONDS = 10;
private static final Logger LOGGER = Logger.getLogger(SimpleMapTest.class.getName());
private static final Random RANDOM = new Random();
private static final HazelcastInstance INSTANCE = Hazelcast.newHazelcastInstance(null);
private static int threadCount = 40;
private static int entryCount = 10 * 1000;
private static int valueSize = 1000;
private static int getPercentage = 40;
private static int putPercentage = 40;
private SimpleMultiMapTest() {
}
public static void main(String[] args) {
boolean load = init(args);
ExecutorService es = Executors.newFixedThreadPool(threadCount);
final MultiMap<String, byte[]> map = INSTANCE.getMultiMap("default");
final AtomicInteger gets = new AtomicInteger(0);
final AtomicInteger puts = new AtomicInteger(0);
final AtomicInteger removes = new AtomicInteger(0);
load(load, es, map);
for (int i = 0; i < threadCount; i++) {
es.execute(new Runnable() {
public void run() {
while (true) {
int key = (int) (RANDOM.nextFloat() * entryCount);
int operation = ((int) (RANDOM.nextFloat() * 100));
if (operation < getPercentage) {
map.get(String.valueOf(key));
gets.incrementAndGet();
} else if (operation < getPercentage + putPercentage) {
map.put(String.valueOf(key), new byte[valueSize]);
puts.incrementAndGet();
} else {
map.remove(String.valueOf(key));
removes.incrementAndGet();
}
}
}
});
}
Executors.newSingleThreadExecutor().execute(new Runnable() {
public void run() {
while (true) {
try {
Thread.sleep(STATS_SECONDS * 1000);
LOGGER.info("cluster size: " + INSTANCE.getCluster().getMembers().size());
int putCount = puts.getAndSet(0);
int getCount = gets.getAndSet(0);
int removeCount = removes.getAndSet(0);
LOGGER.info("TOTAL: " + (removeCount + putCount + getCount) / STATS_SECONDS);
LOGGER.info("PUTS: " + putCount / STATS_SECONDS);
LOGGER.info("GETS: " + getCount / STATS_SECONDS);
LOGGER.info("REMOVES:" + removeCount / STATS_SECONDS);
} catch (InterruptedException ignored) {
return;
}
}
}
});
}
private static void load(boolean load, ExecutorService es, final MultiMap<String, byte[]> map) {
if (load) {
Member thisMember = INSTANCE.getCluster().getLocalMember();
for (int i = 0; i < entryCount; i++) {
final String key = String.valueOf(i);
Partition partition = INSTANCE.getPartitionService().getPartition(key);
if (thisMember.equals(partition.getOwner())) {
es.execute(new Runnable() {
public void run() {
map.put(key, new byte[valueSize]);
}
});
}
}
}
}
private static boolean init(String[] args) {
boolean load = false;
if (args != null && args.length > 0) {
for (String arg : args) {
arg = arg.trim();
if (arg.startsWith("t")) {
threadCount = Integer.parseInt(arg.substring(1));
} else if (arg.startsWith("c")) {
entryCount = Integer.parseInt(arg.substring(1));
} else if (arg.startsWith("v")) {
valueSize = Integer.parseInt(arg.substring(1));
} else if (arg.startsWith("g")) {
getPercentage = Integer.parseInt(arg.substring(1));
} else if (arg.startsWith("p")) {
putPercentage = Integer.parseInt(arg.substring(1));
} else if (arg.startsWith("load")) {
load = true;
}
}
} else {
LOGGER.info("Help: sh test.sh t200 v130 p10 g85 ");
LOGGER.info(" // means 200 threads, value-size 130 bytes, 10% put, 85% get");
LOGGER.info("");
}
LOGGER.info("Starting Test with ");
LOGGER.info(" Thread Count: " + threadCount);
LOGGER.info(" Entry Count: " + entryCount);
LOGGER.info(" Value Size: " + valueSize);
LOGGER.info(" Get Percentage: " + getPercentage);
LOGGER.info(" Put Percentage: " + putPercentage);
LOGGER.info(" Remove Percentage: " + (100 - (putPercentage + getPercentage)));
return load;
}
}