Browse Source
This commit extracts MultiValueMapAdapter from CollectionUtils and reuses its implementation as base class of LinkedMultiValueMap. Closes gh-25140pull/25187/head
Juergen Hoeller
6 years ago
4 changed files with 202 additions and 299 deletions
@ -0,0 +1,178 @@
@@ -0,0 +1,178 @@
|
||||
/* |
||||
* Copyright 2002-2020 the original author or authors. |
||||
* |
||||
* 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 |
||||
* |
||||
* https://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 org.springframework.util; |
||||
|
||||
import java.io.Serializable; |
||||
import java.util.Collection; |
||||
import java.util.LinkedHashMap; |
||||
import java.util.LinkedList; |
||||
import java.util.List; |
||||
import java.util.Map; |
||||
import java.util.Set; |
||||
|
||||
import org.springframework.lang.Nullable; |
||||
|
||||
/** |
||||
* Adapts a given {@link Map} to the {@link MultiValueMap} contract. |
||||
* |
||||
* @author Arjen Poutsma |
||||
* @author Juergen Hoeller |
||||
* @since 3.1 |
||||
* @param <K> the key type |
||||
* @param <V> the value element type |
||||
* @see CollectionUtils#toMultiValueMap |
||||
* @see LinkedMultiValueMap |
||||
*/ |
||||
@SuppressWarnings("serial") |
||||
class MultiValueMapAdapter<K, V> implements MultiValueMap<K, V>, Serializable { |
||||
|
||||
private final Map<K, List<V>> targetMap; |
||||
|
||||
|
||||
MultiValueMapAdapter(Map<K, List<V>> targetMap) { |
||||
this.targetMap = targetMap; |
||||
} |
||||
|
||||
|
||||
@Override |
||||
@Nullable |
||||
public V getFirst(K key) { |
||||
List<V> values = this.targetMap.get(key); |
||||
return (values != null && !values.isEmpty() ? values.get(0) : null); |
||||
} |
||||
|
||||
@Override |
||||
public void add(K key, @Nullable V value) { |
||||
List<V> values = this.targetMap.computeIfAbsent(key, k -> new LinkedList<>()); |
||||
values.add(value); |
||||
} |
||||
|
||||
@Override |
||||
public void addAll(K key, List<? extends V> values) { |
||||
List<V> currentValues = this.targetMap.computeIfAbsent(key, k -> new LinkedList<>()); |
||||
currentValues.addAll(values); |
||||
} |
||||
|
||||
@Override |
||||
public void addAll(MultiValueMap<K, V> values) { |
||||
for (Entry<K, List<V>> entry : values.entrySet()) { |
||||
addAll(entry.getKey(), entry.getValue()); |
||||
} |
||||
} |
||||
|
||||
@Override |
||||
public void set(K key, @Nullable V value) { |
||||
List<V> values = new LinkedList<>(); |
||||
values.add(value); |
||||
this.targetMap.put(key, values); |
||||
} |
||||
|
||||
@Override |
||||
public void setAll(Map<K, V> values) { |
||||
values.forEach(this::set); |
||||
} |
||||
|
||||
@Override |
||||
public Map<K, V> toSingleValueMap() { |
||||
Map<K, V> singleValueMap = new LinkedHashMap<>(this.targetMap.size()); |
||||
this.targetMap.forEach((key, values) -> { |
||||
if (values != null && !values.isEmpty()) { |
||||
singleValueMap.put(key, values.get(0)); |
||||
} |
||||
}); |
||||
return singleValueMap; |
||||
} |
||||
|
||||
@Override |
||||
public int size() { |
||||
return this.targetMap.size(); |
||||
} |
||||
|
||||
@Override |
||||
public boolean isEmpty() { |
||||
return this.targetMap.isEmpty(); |
||||
} |
||||
|
||||
@Override |
||||
public boolean containsKey(Object key) { |
||||
return this.targetMap.containsKey(key); |
||||
} |
||||
|
||||
@Override |
||||
public boolean containsValue(Object value) { |
||||
return this.targetMap.containsValue(value); |
||||
} |
||||
|
||||
@Override |
||||
@Nullable |
||||
public List<V> get(Object key) { |
||||
return this.targetMap.get(key); |
||||
} |
||||
|
||||
@Override |
||||
@Nullable |
||||
public List<V> put(K key, List<V> value) { |
||||
return this.targetMap.put(key, value); |
||||
} |
||||
|
||||
@Override |
||||
@Nullable |
||||
public List<V> remove(Object key) { |
||||
return this.targetMap.remove(key); |
||||
} |
||||
|
||||
@Override |
||||
public void putAll(Map<? extends K, ? extends List<V>> map) { |
||||
this.targetMap.putAll(map); |
||||
} |
||||
|
||||
@Override |
||||
public void clear() { |
||||
this.targetMap.clear(); |
||||
} |
||||
|
||||
@Override |
||||
public Set<K> keySet() { |
||||
return this.targetMap.keySet(); |
||||
} |
||||
|
||||
@Override |
||||
public Collection<List<V>> values() { |
||||
return this.targetMap.values(); |
||||
} |
||||
|
||||
@Override |
||||
public Set<Entry<K, List<V>>> entrySet() { |
||||
return this.targetMap.entrySet(); |
||||
} |
||||
|
||||
@Override |
||||
public boolean equals(@Nullable Object other) { |
||||
return (this == other || this.targetMap.equals(other)); |
||||
} |
||||
|
||||
@Override |
||||
public int hashCode() { |
||||
return this.targetMap.hashCode(); |
||||
} |
||||
|
||||
@Override |
||||
public String toString() { |
||||
return this.targetMap.toString(); |
||||
} |
||||
|
||||
} |
||||
Loading…
Reference in new issue