11d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert/* 21d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * Copyright (C) 2011 The Guava Authors 31d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * 41d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * Licensed under the Apache License, Version 2.0 (the "License"); 51d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * you may not use this file except in compliance with the License. 61d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * You may obtain a copy of the License at 71d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * 81d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * http://www.apache.org/licenses/LICENSE-2.0 91d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * 101d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * Unless required by applicable law or agreed to in writing, software 111d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * distributed under the License is distributed on an "AS IS" BASIS, 121d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 131d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * See the License for the specific language governing permissions and 141d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * limitations under the License. 151d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert */ 161d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 171d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertpackage com.google.common.collect; 181d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 191d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertimport com.google.common.annotations.GwtCompatible; 201d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 211d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertimport java.util.Map; 221d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertimport java.util.Set; 231d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 241d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert/** 251d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * Workaround for 261d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * <a href="http://bugs.sun.com/bugdatabase/view_bug.do?bug_id=6312706"> 271d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * EnumMap bug</a>. If you want to pass an {@code EnumMap}, with the 281d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * intention of using its {@code entrySet()} method, you should 291d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * wrap the {@code EnumMap} in this class instead. 301d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * 311d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * @author Dimitris Andreou 321d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert */ 331d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert@GwtCompatible 341d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertfinal class WellBehavedMap<K, V> extends ForwardingMap<K, V> { 351d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert private final Map<K, V> delegate; 361d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert private Set<Entry<K, V>> entrySet; 371d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 381d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert private WellBehavedMap(Map<K, V> delegate) { 391d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert this.delegate = delegate; 401d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 411d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 421d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert /** 431d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * Wraps the given map into a {@code WellBehavedEntriesMap}, which 441d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * intercepts its {@code entrySet()} method by taking the 451d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * {@code Set<K> keySet()} and transforming it to 461d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * {@code Set<Entry<K, V>>}. All other invocations are delegated as-is. 471d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert */ 481d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert static <K, V> WellBehavedMap<K, V> wrap(Map<K, V> delegate) { 491d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert return new WellBehavedMap<K, V>(delegate); 501d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 511d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 521d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert @Override protected Map<K, V> delegate() { 531d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert return delegate; 541d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 551d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 561d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert @Override public Set<Entry<K, V>> entrySet() { 571d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert Set<Entry<K, V>> es = entrySet; 581d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert if (es != null) { 591d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert return es; 601d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 611d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert return entrySet = Sets.transform( 621d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert delegate.keySet(), new KeyToEntryConverter<K, V>(this)); 631d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 641d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 651d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert private static class KeyToEntryConverter<K, V> 661d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert extends Sets.InvertibleFunction<K, Map.Entry<K, V>> { 671d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert final Map<K, V> map; 681d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 691d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert KeyToEntryConverter(Map<K, V> map) { 701d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert this.map = map; 711d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 721d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 731d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert @Override public Map.Entry<K, V> apply(final K key) { 741d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert return new AbstractMapEntry<K, V>() { 751d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert @Override public K getKey() { 761d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert return key; 771d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 781d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert @Override public V getValue() { 791d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert return map.get(key); 801d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 811d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert @Override public V setValue(V value) { 821d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert return map.put(key, value); 831d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 841d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert }; 851d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 861d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 871d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert @Override public K invert(Map.Entry<K, V> entry) { 881d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert return entry.getKey(); 891d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 901d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 911d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert} 92