com.hazelcast.simulator.tests.syntheticmap.SyntheticMapService Maven / Gradle / Ivy

Go to download
/*
 * Copyright (c) 2008-2017, 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.syntheticmap;

import com.hazelcast.core.DistributedObject;
import com.hazelcast.nio.serialization.Data;
import com.hazelcast.spi.ManagedService;
import com.hazelcast.spi.NodeEngine;
import com.hazelcast.spi.RemoteService;

import java.util.HashMap;
import java.util.Map;
import java.util.Properties;

public class SyntheticMapService implements ManagedService, RemoteService {

    public static final String SERVICE_NAME = "hz:impl:syntheticMapService";

    private NodeEngine nodeEngine;
    private Partition[] partitions;

    @Override
    public DistributedObject createDistributedObject(String objectName) {
        return new SyntheticMapProxy(objectName, nodeEngine, this);
    }

    @Override
    public void destroyDistributedObject(String objectName) {
    }

    @Override
    public void init(NodeEngine nodeEngine, Properties properties) {
        this.nodeEngine = nodeEngine;
        int partitionCount = nodeEngine.getPartitionService().getPartitionCount();
        partitions = new Partition[partitionCount];
        for (int i = 0; i < partitionCount; i++) {
            partitions[i] = new Partition();
        }
    }

    public Data get(int partitionId, String mapName, Data key) {
        Partition partition = partitions[partitionId];
        RecordStore recordStore = partition.get(mapName);
        return recordStore.get(key);
    }

    public void put(int partitionId, String mapName, Data key, Data value) {
        Partition partition = partitions[partitionId];
        RecordStore recordStore = partition.get(mapName);
        recordStore.put(key, value);
    }

    @Override
    public void reset() {
        // no-op
    }

    @Override
    public void shutdown(boolean terminate) {
        // no-op
    }

    private static class Partition {
        private Map stores = new HashMap();

        public RecordStore get(String name) {
            RecordStore store = stores.get(name);
            if (store == null) {
                store = new RecordStore();
                stores.put(name, store);
            }
            return store;
        }
    }

    private static class RecordStore {
        private final Map map = new HashMap();

        public Data get(Data key) {
            return map.get(key);
        }

        public void put(Data key, Data value) {
            map.put(key, value);
        }
    }
}