/*
 * Written by Doug Lea with assistance from members of JCP JSR-166
 * Expert Group and released to the public domain, as explained at
 * http://creativecommons.org/publicdomain/zero/1.0/
 */
import java.util.*;
import java.util.concurrent.*;
import java.util.concurrent.locks.*;

/**
 * This is an incomplete implementation of a wrapper class
 * that places read-write locks around unsynchronized Maps.
 * Exists as a sample input for MapLoops test.
 */
public class SRWTreeMap implements ConcurrentMap {
    private final Map m;
    private final StampedLock lock = new StampedLock();

    public SRWTreeMap(Map m) {
        if (m == null)
            throw new NullPointerException();
        this.m = m;
    }

    public SRWTreeMap() {
        this(new TreeMap()); // use TreeMap by default
        //        this(new IdentityHashMap());
    }

    public int size() {
        StampedLock l = this.lock;
        long stamp = l.readLock();
        try { return m.size(); }
        finally { l.unlockRead(stamp); }
    }

    public boolean isEmpty() {
        StampedLock l = this.lock;
        long stamp = l.readLock();
        try { return m.isEmpty(); }
        finally { l.unlockRead(stamp); }
    }

    public Object get(Object key) {
        StampedLock l = this.lock;
        long stamp = l.tryOptimisticRead();
        Object x = m.get(key);
        if (l.validate(stamp))
            return x;
        stamp = l.readLock();
        try { return m.get(key); }
        finally { l.unlockRead(stamp); }
    }

    public boolean containsKey(Object key) {
        StampedLock l = this.lock;
        long stamp = l.readLock();
        try { return m.containsKey(key); }
        finally { l.unlockRead(stamp); }
    }

    public boolean containsValue(Object value) {
        StampedLock l = this.lock;
        long stamp = l.readLock();
        try { return m.containsValue(value); }
        finally { l.unlockRead(stamp); }
    }

    public Set keySet() { // Not implemented
        return m.keySet();
    }

    public Set entrySet() { // Not implemented
        return m.entrySet();
    }

    public Collection values() { // Not implemented
        return m.values();
    }

    public boolean equals(Object o) {
        StampedLock l = this.lock;
        long stamp = l.readLock();
        try { return m.equals(o); }
        finally { l.unlockRead(stamp); }
    }

    public int hashCode() {
        StampedLock l = this.lock;
        long stamp = l.readLock();
        try { return m.hashCode(); }
        finally { l.unlockRead(stamp); }
    }

    public String toString() {
        StampedLock l = this.lock;
        long stamp = l.readLock();
        try { return m.toString(); }
        finally { l.unlockRead(stamp); }
    }

    public Object put(Object key, Object value) {
        StampedLock l = this.lock;
        long stamp = l.writeLock();
        try { return m.put(key, value); }
        finally { l.unlockWrite(stamp); }
    }

    public Object putIfAbsent(Object key, Object value) {
        StampedLock l = this.lock;
        long stamp = l.writeLock();
        try {
            Object v = m.get(key);
            return (v == null) ? m.put(key, value) : v;
        }
        finally { l.unlockWrite(stamp); }
    }

    public boolean replace(Object key, Object oldValue, Object newValue) {
        StampedLock l = this.lock;
        long stamp = l.writeLock();
        try {
            if (m.get(key).equals(oldValue)) {
                m.put(key, newValue);
                return true;
            }
            return false;
        }
        finally { l.unlockWrite(stamp); }
    }

    public Object replace(Object key, Object newValue) {
        StampedLock l = this.lock;
        long stamp = l.writeLock();
        try {
            if (m.containsKey(key))
                return m.put(key, newValue);
            return null;
        }
        finally { l.unlockWrite(stamp); }
    }

    public Object remove(Object key) {
        StampedLock l = this.lock;
        long stamp = l.writeLock();
        try { return m.remove(key); }
        finally { l.unlockWrite(stamp); }
    }

    public boolean remove(Object key, Object value) {
        StampedLock l = this.lock;
        long stamp = l.writeLock();
        try {
            if (m.get(key).equals(value)) {
                m.remove(key);
                return true;
            }
            return false;
        }
        finally { l.unlockWrite(stamp); }
    }

    public void putAll(Map map) {
        StampedLock l = this.lock;
        long stamp = l.writeLock();
        try { m.putAll(map); }
        finally { l.unlockWrite(stamp); }
    }

    public void clear() {
        StampedLock l = this.lock;
        long stamp = l.writeLock();
        try { m.clear(); }
        finally { l.unlockWrite(stamp); }
    }

}