View Javadoc
1   /*
2    * Licensed to the Apache Software Foundation (ASF) under one or more
3    * contributor license agreements.  See the NOTICE file distributed with
4    * this work for additional information regarding copyright ownership.
5    * The ASF licenses this file to You under the Apache License, Version 2.0
6    * (the "License"); you may not use this file except in compliance with
7    * the License.  You may obtain a copy of the License at
8    *
9    *      http://www.apache.org/licenses/LICENSE-2.0
10   *
11   * Unless required by applicable law or agreed to in writing, software
12   * distributed under the License is distributed on an "AS IS" BASIS,
13   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14   * See the License for the specific language governing permissions and
15   * limitations under the License.
16   */
17  package org.apache.commons.collections4.map;
18  
19  import java.io.IOException;
20  import java.io.ObjectInputStream;
21  import java.io.ObjectOutputStream;
22  import java.io.Serializable;
23  import java.util.Collection;
24  import java.util.Map;
25  import java.util.Set;
26  import java.util.SortedMap;
27  
28  import org.apache.commons.collections4.BoundedMap;
29  import org.apache.commons.collections4.CollectionUtils;
30  import org.apache.commons.collections4.collection.UnmodifiableCollection;
31  import org.apache.commons.collections4.set.UnmodifiableSet;
32  
33  /**
34   * Decorates another {@code SortedMap} to fix the size blocking add/remove.
35   * <p>
36   * Any action that would change the size of the map is disallowed.
37   * The put method is allowed to change the value associated with an existing
38   * key however.
39   * </p>
40   * <p>
41   * If trying to remove or clear the map, an UnsupportedOperationException is
42   * thrown. If trying to put a new mapping into the map, an
43   * IllegalArgumentException is thrown. This is because the put method can
44   * succeed if the mapping's key already exists in the map, so the put method
45   * is not always unsupported.
46   * </p>
47   * <p>
48   * <strong>Note that FixedSizeSortedMap is not synchronized and is not thread-safe.</strong>
49   * If you wish to use this map from multiple threads concurrently, you must use
50   * appropriate synchronization. The simplest approach is to wrap this map
51   * using {@link java.util.Collections#synchronizedSortedMap}. This class may throw
52   * exceptions when accessed by concurrent threads without synchronization.
53   * </p>
54   * <p>
55   * This class is Serializable from Commons Collections 3.1.
56   * </p>
57   *
58   * @param <K> the type of the keys in this map
59   * @param <V> the type of the values in this map
60   * @since 3.0
61   */
62  public class FixedSizeSortedMap<K, V>
63          extends AbstractSortedMapDecorator<K, V>
64          implements BoundedMap<K, V>, Serializable {
65  
66      /** Serialization version */
67      private static final long serialVersionUID = 3126019624511683653L;
68  
69      /**
70       * Factory method to create a fixed size sorted map.
71       *
72       * @param <K>  the key type
73       * @param <V>  the value type
74       * @param map  the map to decorate, must not be null
75       * @return a new fixed size sorted map
76       * @throws NullPointerException if map is null
77       * @since 4.0
78       */
79      public static <K, V> FixedSizeSortedMap<K, V> fixedSizeSortedMap(final SortedMap<K, V> map) {
80          return new FixedSizeSortedMap<>(map);
81      }
82  
83      /**
84       * Constructor that wraps (not copies).
85       *
86       * @param map  the map to decorate, must not be null
87       * @throws NullPointerException if map is null
88       */
89      protected FixedSizeSortedMap(final SortedMap<K, V> map) {
90          super(map);
91      }
92  
93      @Override
94      public void clear() {
95          throw new UnsupportedOperationException("Map is fixed size");
96      }
97  
98      @Override
99      public Set<Map.Entry<K, V>> entrySet() {
100         return UnmodifiableSet.unmodifiableSet(map.entrySet());
101     }
102 
103     /**
104      * Gets the map being decorated.
105      *
106      * @return the decorated map
107      */
108     protected SortedMap<K, V> getSortedMap() {
109         return (SortedMap<K, V>) map;
110     }
111 
112     @Override
113     public SortedMap<K, V> headMap(final K toKey) {
114         return new FixedSizeSortedMap<>(getSortedMap().headMap(toKey));
115     }
116 
117     @Override
118     public boolean isFull() {
119         return true;
120     }
121 
122     @Override
123     public Set<K> keySet() {
124         return UnmodifiableSet.unmodifiableSet(map.keySet());
125     }
126 
127     @Override
128     public int maxSize() {
129         return size();
130     }
131 
132     @Override
133     public V put(final K key, final V value) {
134         if (!map.containsKey(key)) {
135             throw new IllegalArgumentException("Cannot put new key/value pair - Map is fixed size");
136         }
137         return map.put(key, value);
138     }
139 
140     @Override
141     public void putAll(final Map<? extends K, ? extends V> mapToCopy) {
142         if (CollectionUtils.isSubCollection(mapToCopy.keySet(), keySet())) {
143             throw new IllegalArgumentException("Cannot put new key/value pair - Map is fixed size");
144         }
145         map.putAll(mapToCopy);
146     }
147 
148     /**
149      * Read the map in using a custom routine.
150      *
151      * @param in the input stream
152      * @throws IOException if an error occurs while reading from the stream
153      * @throws ClassNotFoundException if an object read from the stream can not be loaded
154      */
155     @SuppressWarnings("unchecked") // (1) should only fail if input stream is incorrect
156     private void readObject(final ObjectInputStream in) throws IOException, ClassNotFoundException {
157         in.defaultReadObject();
158         map = (Map<K, V>) in.readObject(); // (1)
159     }
160 
161     @Override
162     public V remove(final Object key) {
163         throw new UnsupportedOperationException("Map is fixed size");
164     }
165 
166     @Override
167     public SortedMap<K, V> subMap(final K fromKey, final K toKey) {
168         return new FixedSizeSortedMap<>(getSortedMap().subMap(fromKey, toKey));
169     }
170 
171     @Override
172     public SortedMap<K, V> tailMap(final K fromKey) {
173         return new FixedSizeSortedMap<>(getSortedMap().tailMap(fromKey));
174     }
175 
176     @Override
177     public Collection<V> values() {
178         return UnmodifiableCollection.unmodifiableCollection(map.values());
179     }
180 
181     /**
182      * Write the map out using a custom routine.
183      *
184      * @param out  the output stream
185      * @throws IOException if an error occurs while writing to the stream
186      */
187     private void writeObject(final ObjectOutputStream out) throws IOException {
188         out.defaultWriteObject();
189         out.writeObject(map);
190     }
191 
192 }