ru.yandex.qatools.camelot.common.LocalMemoryStorage Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of camelot-commons Show documentation

There is a newer version: 2.5.4

package ru.yandex.qatools.camelot.common;

import org.slf4j.Logger;
import ru.yandex.qatools.camelot.api.Storage;

import java.util.HashMap;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.locks.ReentrantLock;

import static org.slf4j.LoggerFactory.getLogger;

/**
 * @author Ilya Sadykov (mailto: [email protected])
 */
public class LocalMemoryStorage implements Storage {
    private Map storage = new HashMap<>();
    private Map locks = new ConcurrentHashMap<>();
    private final Logger logger = getLogger(getClass());

    public LocalMemoryStorage() {
    }

    public LocalMemoryStorage(Map storage) {
        this.storage = storage;
    }

    @Override
    public void put(String key, T value) {
        storage.put(key, value);
    }

    @Override
    public T get(String key) {
        return storage.get(key);
    }

    @Override
    public Set keys() {
        return storage.keySet();
    }

    @Override
    public Set localKeys() {
        return keys();
    }

    @Override
    public boolean lock(String key, long timeout, TimeUnit ofUnit) {
        try {
            return getLock(key).tryLock(timeout, ofUnit);
        } catch (Exception e) {
            logger.warn(String.format("Failed to lock storage by key %s", key), e);
        }
        return false;
    }

    @Override
    public void unlock(String key) {
        try {
            getLock(key).unlock();
        } catch (Exception e) {
            logger.warn(String.format("Failed to unlock storage by key %s", key), e);
        }
    }

    private synchronized ReentrantLock getLock(String key) {
        if (!locks.containsKey(key)) {
            locks.put(key, new ReentrantLock());
        }
        return locks.get(key);
    }
}