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

com.github.dm.jrt.util.WeakIdentityHashMap Maven / Gradle / Ivy

Go to download

Parallel programming on the go

There is a newer version: 5.9.0
Show newest version
/*
 * 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.github.dm.jrt.util;

import org.jetbrains.annotations.NotNull;

import java.lang.ref.ReferenceQueue;
import java.lang.ref.WeakReference;
import java.util.AbstractSet;
import java.util.Collection;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;
import java.util.Set;

/**
 * Map implementation combining the features of {@link java.util.IdentityHashMap} and
 * {@link java.util.WeakHashMap}.
 * 

 * Note that the map entries might change each time the object is explicitly accessed.
 * 

 * Created by davide-maestroni on 11/17/2014.
 *
 * @param  the key type.
 * @param  the value type.
 */
public class WeakIdentityHashMap implements Map {

    private final HashMap mMap;

    private final ReferenceQueue mQueue = new ReferenceQueue();

    private AbstractSet> mEntrySet;

    private AbstractSet mKeySet;

    /**
     * Constructor.
     */
    public WeakIdentityHashMap() {

        mMap = new HashMap();
    }

    /**
     * Constructor.
     *
     * @param map the initial content.
     * @see java.util.HashMap#HashMap(java.util.Map)
     */
    public WeakIdentityHashMap(@NotNull final Map map) {

        mMap = new HashMap(map.size());
        putAll(map);
    }

    /**
     * Constructor.
     *
     * @param initialCapacity the initial capacity.
     * @see java.util.HashMap#HashMap(int)
     */
    public WeakIdentityHashMap(final int initialCapacity) {

        mMap = new HashMap(initialCapacity);
    }

    /**
     * Constructor.
     *
     * @param initialCapacity the initial capacity.
     * @param loadFactor      the load factor.
     * @see java.util.HashMap#HashMap(int, float)
     */
    public WeakIdentityHashMap(final int initialCapacity, final float loadFactor) {

        mMap = new HashMap(initialCapacity, loadFactor);
    }

    @Override
    public int hashCode() {

        return mMap.hashCode();
    }

    public int size() {

        cleanUp();
        return mMap.size();
    }

    public boolean isEmpty() {

        cleanUp();
        return mMap.isEmpty();
    }

    public boolean containsKey(final Object o) {

        cleanUp();
        return mMap.containsKey(new IdentityWeakReference(o));
    }

    public boolean containsValue(final Object o) {

        cleanUp();
        return mMap.containsValue(o);
    }

    public V get(final Object o) {

        cleanUp();
        return mMap.get(new IdentityWeakReference(o));
    }

    public V put(final K k, final V v) {

        cleanUp();
        return mMap.put(new IdentityWeakReference(k, mQueue), v);
    }

    public V remove(final Object o) {

        cleanUp();
        return mMap.remove(new IdentityWeakReference(o));
    }

    public void putAll(@NotNull final Map map) {

        cleanUp();
        final ReferenceQueue queue = mQueue;
        final HashMap referenceMap = mMap;

        for (final Entry entry : map.entrySet()) {

            referenceMap.put(new IdentityWeakReference(entry.getKey(), queue), entry.getValue());
        }
    }

    public void clear() {

        mMap.clear();
    }

    @NotNull
    public Set keySet() {

        if (mKeySet == null) {

            mKeySet = new AbstractSet() {

                @NotNull
                @Override
                public Iterator iterator() {

                    return new KeyIterator();
                }

                @Override
                public int size() {

                    return mMap.size();
                }
            };
        }

        return mKeySet;
    }

    @NotNull
    public Collection values() {

        return mMap.values();
    }

    @NotNull
    public Set> entrySet() {

        if (mEntrySet == null) {

            mEntrySet = new AbstractSet>() {

                @NotNull
                @Override
                public Iterator> iterator() {

                    return new EntryIterator();
                }

                @Override
                public int size() {

                    return mMap.size();
                }
            };
        }

        return mEntrySet;
    }

    @SuppressWarnings("unchecked")
    private void cleanUp() {

        final HashMap map = mMap;
        final ReferenceQueue queue = mQueue;
        IdentityWeakReference reference = (IdentityWeakReference) queue.poll();

        while (reference != null) {

            map.remove(reference);
            reference = (IdentityWeakReference) queue.poll();
        }
    }

    /**
     * Weak reference using object identity for comparison.
     */
    private static class IdentityWeakReference extends WeakReference {

        private final int mHashCode;

        /**
         * Constructor.
         *
         * @param referent the referent instance.
         * @param queue    the reference queue.
         * @see java.lang.ref.WeakReference#WeakReference(Object, ReferenceQueue)
         */
        private IdentityWeakReference(final Object referent,
                final ReferenceQueue queue) {

            super(referent, queue);
            mHashCode = System.identityHashCode(referent);
        }

        /**
         * Constructor.
         *
         * @param referent the referent instance.
         * @see java.lang.ref.WeakReference#WeakReference(Object)
         */
        private IdentityWeakReference(final Object referent) {

            super(referent);
            mHashCode = System.identityHashCode(referent);
        }

        @Override
        public int hashCode() {

            return mHashCode;
        }

        @Override
        public boolean equals(final Object obj) {

            if (this == obj) {

                return true;
            }

            if (!(obj instanceof IdentityWeakReference)) {

                return false;
            }

            final IdentityWeakReference that = (IdentityWeakReference) obj;

            if (mHashCode != that.mHashCode) {

                return false;
            }

            final Object referent = get();
            return (referent != null) && (referent == that.get());
        }
    }

    /**
     * Map entry iterator.
     */
    private class EntryIterator implements Iterator> {

        private final Iterator mIterator = mMap.keySet().iterator();

        public boolean hasNext() {

            return mIterator.hasNext();
        }

        public Entry next() {

            return new WeakEntry(mIterator.next());
        }

        public void remove() {

            mIterator.remove();
        }
    }

    /**
     * Map key iterator.
     */
    private class KeyIterator implements Iterator {

        private final Iterator mIterator = mMap.keySet().iterator();

        public boolean hasNext() {

            return mIterator.hasNext();
        }

        @SuppressWarnings("unchecked")
        public K next() {

            return (K) mIterator.next().get();
        }

        public void remove() {

            mIterator.remove();
        }
    }

    /**
     * Map entry implementation.
     */
    private class WeakEntry implements Entry {

        private final IdentityWeakReference mReference;

        /**
         * Constructor.
         *
         * @param key the key reference.
         */
        private WeakEntry(@NotNull final IdentityWeakReference key) {

            mReference = key;
        }

        @SuppressWarnings("unchecked")
        public K getKey() {

            return (K) mReference.get();
        }

        public V getValue() {

            return mMap.get(mReference);
        }

        public V setValue(final V v) {

            return mMap.put(mReference, v);
        }
    }

    @Override
    @SuppressWarnings("EqualsBetweenInconvertibleTypes")
    public boolean equals(final Object o) {

        if (this == o) {

            return true;
        }

        if (!(o instanceof WeakIdentityHashMap)) {

            return (o instanceof Map) && o.equals(this);
        }

        final WeakIdentityHashMap that = (WeakIdentityHashMap) o;
        return mMap.equals(that.mMap);
    }
}
    

    

    

            

    

            



    
    






    
© 2015 - 2024 Weber Informatics LLC | Privacy Policy