/* File: SyncMap.java Originally written by Doug Lea and released into the public domain. This may be used for any purposes whatsoever without acknowledgment. Thanks for the assistance and support of Sun Microsystems Labs, and everyone contributing, testing, and using this code. History: Date Who What 1Aug1998 dl Create public version */ package com.alipay.zdal.datasource.resource.util.concurrent; import java.util.Collection; import java.util.Map; import java.util.Set; /** * SyncMaps wrap Sync-based control around java.util.Maps. * They operate in the same way as SyncCollection. * <p> * Reader operations are * <ul> * <li> size * <li> isEmpty * <li> get * <li> containsKey * <li> containsValue * <li> keySet * <li> entrySet * <li> values * </ul> * Writer operations are: * <ul> * <li> put * <li> putAll * <li> remove * <li> clear * </ul> * * <p>[<a href="http://gee.cs.oswego.edu/dl/classes/EDU/oswego/cs/dl/util/concurrent/intro.html"> Introduction to this package. </a>] * @see SyncCollection **/ public class SyncMap implements Map { protected final Map c_; // Backing Map protected final Sync rd_; // sync for read-only methods protected final Sync wr_; // sync for mutative methods protected final SynchronizedLong syncFailures_ = new SynchronizedLong(0); /** * Create a new SyncMap protecting the given map, * and using the given sync to control both reader and writer methods. * Common, reasonable choices for the sync argument include * Mutex, ReentrantLock, and Semaphores initialized to 1. **/ public SyncMap(Map map, Sync sync) { this(map, sync, sync); } /** * Create a new SyncMap protecting the given map, * and using the given ReadWriteLock to control reader and writer methods. **/ public SyncMap(Map map, ReadWriteLock rwl) { this(map, rwl.readLock(), rwl.writeLock()); } /** * Create a new SyncMap protecting the given map, * and using the given pair of locks to control reader and writer methods. **/ public SyncMap(Map map, Sync readLock, Sync writeLock) { c_ = map; rd_ = readLock; wr_ = writeLock; } /** * Return the Sync object managing read-only operations **/ public Sync readerSync() { return rd_; } /** * Return the Sync object managing mutative operations **/ public Sync writerSync() { return wr_; } /** * Return the number of synchronization failures for read-only operations **/ public long syncFailures() { return syncFailures_.get(); } /** Try to acquire sync before a reader operation; record failure **/ protected boolean beforeRead() { try { rd_.acquire(); return false; } catch (InterruptedException ex) { syncFailures_.increment(); return true; } } /** Clean up after a reader operation **/ protected void afterRead(boolean wasInterrupted) { if (wasInterrupted) { Thread.currentThread().interrupt(); } else rd_.release(); } @Override public int hashCode() { boolean wasInterrupted = beforeRead(); try { return c_.hashCode(); } finally { afterRead(wasInterrupted); } } @Override public boolean equals(Object o) { boolean wasInterrupted = beforeRead(); try { return c_.equals(o); } finally { afterRead(wasInterrupted); } } public int size() { boolean wasInterrupted = beforeRead(); try { return c_.size(); } finally { afterRead(wasInterrupted); } } public boolean isEmpty() { boolean wasInterrupted = beforeRead(); try { return c_.isEmpty(); } finally { afterRead(wasInterrupted); } } public boolean containsKey(Object o) { boolean wasInterrupted = beforeRead(); try { return c_.containsKey(o); } finally { afterRead(wasInterrupted); } } public boolean containsValue(Object o) { boolean wasInterrupted = beforeRead(); try { return c_.containsValue(o); } finally { afterRead(wasInterrupted); } } public Object get(Object key) { boolean wasInterrupted = beforeRead(); try { return c_.get(key); } finally { afterRead(wasInterrupted); } } public Object put(Object key, Object value) { try { wr_.acquire(); try { return c_.put(key, value); } finally { wr_.release(); } } catch (InterruptedException ex) { Thread.currentThread().interrupt(); throw new UnsupportedOperationException(); } } public Object remove(Object key) { try { wr_.acquire(); try { return c_.remove(key); } finally { wr_.release(); } } catch (InterruptedException ex) { Thread.currentThread().interrupt(); throw new UnsupportedOperationException(); } } public void putAll(Map coll) { try { wr_.acquire(); try { c_.putAll(coll); } finally { wr_.release(); } } catch (InterruptedException ex) { Thread.currentThread().interrupt(); throw new UnsupportedOperationException(); } } public void clear() { try { wr_.acquire(); try { c_.clear(); } finally { wr_.release(); } } catch (InterruptedException ex) { Thread.currentThread().interrupt(); throw new UnsupportedOperationException(); } } private transient Set keySet_ = null; private transient Set entrySet_ = null; private transient Collection values_ = null; public Set keySet() { boolean wasInterrupted = beforeRead(); try { if (keySet_ == null) keySet_ = new SyncSet(c_.keySet(), rd_, wr_); return keySet_; } finally { afterRead(wasInterrupted); } } public Set entrySet() { boolean wasInterrupted = beforeRead(); try { if (entrySet_ == null) entrySet_ = new SyncSet(c_.entrySet(), rd_, wr_); return entrySet_; } finally { afterRead(wasInterrupted); } } public Collection values() { boolean wasInterrupted = beforeRead(); try { if (values_ == null) values_ = new SyncCollection(c_.values(), rd_, wr_); return values_; } finally { afterRead(wasInterrupted); } } }