All Downloads are FREE. Search and download functionalities are using the official Maven repository.

com.hazelcast.simulator.tests.map.MapAsyncOpsTest Maven / Gradle / Ivy

The newest version!
/*
 * Copyright (c) 2008-2016, 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.simulator.tests.map;

import com.hazelcast.collection.IList;
import com.hazelcast.map.IMap;
import com.hazelcast.simulator.hz.HazelcastTest;
import com.hazelcast.simulator.test.BaseThreadState;
import com.hazelcast.simulator.test.annotations.AfterRun;
import com.hazelcast.simulator.test.annotations.Setup;
import com.hazelcast.simulator.test.annotations.TimeStep;
import com.hazelcast.simulator.test.annotations.Verify;
import com.hazelcast.simulator.tests.map.helpers.MapOperationCounter;

import java.util.concurrent.CompletableFuture;
import java.util.concurrent.TimeUnit;

import static com.hazelcast.simulator.utils.CommonUtils.sleepSeconds;

public class MapAsyncOpsTest extends HazelcastTest {

    // properties
    public int keyCount = 10;
    public int maxTTLExpirySeconds = 3;

    private final MapOperationCounter count = new MapOperationCounter();

    private IMap map;
    private IList results;

    @Setup
    public void setUp() {
        map = targetInstance.getMap(name);
        results = targetInstance.getList(name + "report");
    }

    @TimeStep(prob = 0.2)
    public CompletableFuture putAsync(ThreadState state) {
        int key = state.randomInt(keyCount);
        Object value = state.randomInt();
        count.putAsyncCount.incrementAndGet();
        return map.putAsync(key, value).toCompletableFuture();
    }

    @TimeStep(prob = 0.2)
    public CompletableFuture putAsyncTTL(ThreadState state) {
        int key = state.randomInt(keyCount);
        Object value = state.randomInt();
        int delay = 1 + state.randomInt(maxTTLExpirySeconds);
        count.putAsyncTTLCount.incrementAndGet();
        return map.putAsync(key, value, delay, TimeUnit.SECONDS).toCompletableFuture();
    }

    @TimeStep(prob = 0.2)
    public CompletableFuture getAsync(ThreadState state) {
        int key = state.randomInt(keyCount);
        count.getAsyncCount.incrementAndGet();
        return map.getAsync(key).toCompletableFuture();
    }

    @TimeStep(prob = 0.2)
    public CompletableFuture removeAsync(ThreadState state) {
        int key = state.randomInt(keyCount);
        count.removeAsyncCount.incrementAndGet();
        return map.removeAsync(key).toCompletableFuture();
    }

    @TimeStep(prob = 0.2)
    public void destroy(ThreadState state) {
        map.destroy();
        count.destroyCount.incrementAndGet();
    }

    @AfterRun
    public void afterRun() {
        results.add(count);
    }

    public class ThreadState extends BaseThreadState {
    }

    @Verify(global = true)
    public void globalVerify() {
        MapOperationCounter totalMapOperationsCount = new MapOperationCounter();
        for (MapOperationCounter mapOperationsCount : results) {
            totalMapOperationsCount.add(mapOperationsCount);
        }
        logger.info(name + ": " + totalMapOperationsCount + " total of " + results.size());
    }

    @Verify(global = false)
    public void verify() {
        sleepSeconds(maxTTLExpirySeconds * 2);

        logger.info(name + ": map size  =" + map.size());
    }
}