package org.jboss.resteasy.core;
import org.jboss.resteasy.util.MediaTypeHelper;
import javax.ws.rs.core.MediaType;
import java.io.Serializable;
import java.lang.ref.WeakReference;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.CopyOnWriteArrayList;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
public class MediaTypeMap<T>
{
public interface Typed
{
Class<?> getType();
}
private static class TypedEntryComparator implements Comparator<Entry<?>>, Serializable
{
private static final long serialVersionUID = -8815419198743440920L;
private Class<?> type;
TypedEntryComparator(final Class<?> type)
{
this.type = type;
}
private boolean isAssignableFrom(Typed typed)
{
if (typed.getType() == null) return false;
return typed.getType().isAssignableFrom(type);
}
private int compareTypes(Entry<?> entry, Entry<?> entry1)
{
int val = 0;
if (entry.object instanceof Typed && entry1.object instanceof Typed && type != null)
{
Typed one = (Typed) entry.object;
Typed two = (Typed) entry1.object;
boolean oneTyped = isAssignableFrom(one);
boolean twoTyped = isAssignableFrom(two);
if (oneTyped == twoTyped && (!oneTyped && !twoTyped))
{
val = 0;
}
else if (oneTyped == twoTyped)
{
if (one.getType().equals(two.getType()))
{
val = 0;
}
else if (one.getType().isAssignableFrom(two.getType()))
{
val = 1;
}
else
{
val = -1;
}
}
else if (oneTyped) val = -1;
else val = 1;
}
return val;
}
public int compare(Entry<?> entry, Entry<?> entry1)
{
int val = compareTypes(entry, entry1);
if (val == 0) val = entry.compareTo(entry1);
return val;
}
}
private static class Entry<T> implements Comparable<Entry<?>>
{
public MediaType mediaType;
public T object;
private Entry(final MediaType mediaType, final T object)
{
this.mediaType = mediaType;
this.object = object;
}
@SuppressWarnings({"rawtypes", "unchecked"})
public int compareTo(Entry<?> entry)
{
int val = MediaTypeHelper.compareWeight(mediaType, entry.mediaType);
if (val == 0 && object instanceof Comparable && entry.object instanceof Comparable)
{
return ((Comparable) object).compareTo(entry.object);
}
return val;
}
}
private static Pattern COMPOSITE_PATTERN = Pattern.compile("([^\\+]+)\\+(.+)");
public static Pattern COMPOSITE_SUBTYPE_WILDCARD_PATTERN = Pattern.compile("\\*\\+(.+)");
public static Pattern WILD_SUBTYPE_COMPOSITE_PATTERN = Pattern.compile("([^\\+]+)\\+\\*");
private static class SubtypeMap<T>
{
private volatile Map<String, List<Entry<T>>> index;
private volatile Map<String, List<Entry<T>>> compositeIndex;
private volatile Map<String, List<Entry<T>>> wildCompositeIndex;
private final CopyOnWriteArrayList<Entry<T>> wildcards;
private final CopyOnWriteArrayList<Entry<T>> all;
private SubtypeMap() {
index = new HashMap<>();
compositeIndex = new HashMap<>();
wildCompositeIndex = new HashMap<>();
wildcards = new CopyOnWriteArrayList<>();
all = new CopyOnWriteArrayList<>();
}
private SubtypeMap(final SubtypeMap<T> subtypeMap) {
index = subtypeMap.index;
compositeIndex = subtypeMap.compositeIndex;
wildCompositeIndex = subtypeMap.wildCompositeIndex;
wildcards = new CopyOnWriteArrayList<>(subtypeMap.wildcards);
all = new CopyOnWriteArrayList<>(subtypeMap.all);
}
private void add(final MediaType type, final T obj)
{
final Entry<T> entry = new Entry<>(type, obj);
all.add(entry);
final Matcher matcher = COMPOSITE_SUBTYPE_WILDCARD_PATTERN.matcher(type.getSubtype());
final Matcher wildCompositeMatcher = WILD_SUBTYPE_COMPOSITE_PATTERN.matcher(type.getSubtype());
index = copy(index);
compositeIndex = copy(compositeIndex);
wildCompositeIndex = copy(wildCompositeIndex);
if (type.isWildcardSubtype()) wildcards.add(entry);
else if (matcher.matches())
{
add(compositeIndex, matcher.group(1), entry);
}
else if (wildCompositeMatcher.matches())
{
add(wildCompositeIndex, wildCompositeMatcher.group(1), entry);
}
else
{
add(index, type.getSubtype(), entry);
}
}
private Map<String, List<Entry<T>>> copy(final Map<String, List<Entry<T>>> original) {
final Map<String, List<Entry<T>>> copy = new HashMap<>(original.size());
original.forEach((key, value) -> copy.put(key, new CopyOnWriteArrayList<>(value)));
return copy;
}
private void add(final Map<String, List<Entry<T>>> map,
final String key,
final Entry<T> entry) {
map.putIfAbsent(key, new CopyOnWriteArrayList<>());
map.get(key).add(entry);
}
private List<Entry<T>> getPossible(MediaType accept)
{
if (accept.isWildcardSubtype())
{
return all;
}
else
{
List<Entry<T>> matches = new ArrayList<Entry<T>>();
List<Entry<T>> indexed = index.get(accept.getSubtype());
if (indexed != null) matches.addAll(indexed);
Matcher matcher = COMPOSITE_PATTERN.matcher(accept.getSubtype());
String compositeKey = accept.getSubtype();
if (matcher.matches())
{
String wildCompositeKey = matcher.group(1);
List<Entry<T>> windex = wildCompositeIndex.get(wildCompositeKey);
if (windex != null) matches.addAll(windex);
compositeKey = matcher.group(2);
}
List<Entry<T>> indexed2 = compositeIndex.get(compositeKey);
if (indexed2 != null) matches.addAll(indexed2);
matches.addAll(wildcards);
return matches;
}
}
}
public Map<CachedMediaTypeAndClass, List<T>> getClassCache()
{
return classCache;
}
public static class CachedMediaTypeAndClass
{
private WeakReference<Class<?>> clazz;
private MediaType mediaType;
private final int hash;
@SuppressWarnings({"rawtypes", "unchecked"})
private CachedMediaTypeAndClass(final Class clazz, final MediaType mediaType)
{
this.clazz = new WeakReference(clazz);
this.mediaType = mediaType;
int result = clazz.hashCode();
result = 31 * result + (mediaType.getType() != null ? mediaType.getType().hashCode() : 0) + (mediaType.getSubtype() != null ? mediaType.getSubtype().hashCode() : 0);
hash = result;
}
private Class<?> getClazz()
{
return clazz.get();
}
@Override
public boolean equals(Object o)
{
if (this == o) return true;
if (o == null || getClass() != o.getClass()) return false;
CachedMediaTypeAndClass that = (CachedMediaTypeAndClass) o;
Class<?> clazz = getClazz();
if (clazz == null || that.getClazz() == null) return false;
if (!clazz.equals(that.getClazz())) return false;
if (mediaType.getType() != null)
{
if (!mediaType.getType().equals(that.mediaType.getType())) return false;
}
else if ((mediaType.getType() != that.mediaType.getType())) return false;
if (mediaType.getSubtype() != null)
{
if (!mediaType.getSubtype().equals(that.mediaType.getSubtype())) return false;
}
else if ((mediaType.getSubtype() != that.mediaType.getSubtype())) return false;
return true;
}
@Override
public int hashCode()
{
return hash;
}
}
private volatile Map<String, SubtypeMap<T>> index = new HashMap<>();
private volatile Map<CachedMediaTypeAndClass, List<T>> classCache = new HashMap<>();
private final CopyOnWriteArrayList<Entry<T>> wildcards;
private final CopyOnWriteArrayList<Entry<T>> all;
public MediaTypeMap() {
wildcards = new CopyOnWriteArrayList<>();
all = new CopyOnWriteArrayList<>();
}
public MediaTypeMap(final MediaTypeMap<T> mediaTypeMap) {
index = mediaTypeMap.index;
wildcards = new CopyOnWriteArrayList<>(mediaTypeMap.wildcards);
all = new CopyOnWriteArrayList<>(mediaTypeMap.all);
classCache = mediaTypeMap.classCache;
}
public synchronized void add(final MediaType type, T obj)
{
final MediaType newType =
new MediaType(type.getType().toLowerCase(), type.getSubtype().toLowerCase(), type.getParameters());
final Entry<T> entry = new Entry<>(newType, obj);
classCache = new HashMap<>();
all.add(entry);
Collections.sort(all);
final Map<String, SubtypeMap<T>> oldIndex = index;
index = new HashMap<>();
oldIndex.forEach((key, value) -> index.put(key, new SubtypeMap<>(value)));
if (newType.isWildcardType())
{
wildcards.add(entry);
}
else
{
index.putIfAbsent(newType.getType(), new SubtypeMap<>());
index.get(newType.getType()).add(newType, obj);
}
}
private static <T> List<T> convert(List<Entry<T>> list)
{
List<T> newList = new ArrayList<T>(list.size());
for (Entry<T> entry : list)
{
newList.add(entry.object);
}
return newList;
}
public List<T> getPossible(MediaType accept)
{
accept = new MediaType(accept.getType().toLowerCase(), accept.getSubtype().toLowerCase(), accept.getParameters());
List<Entry<T>> matches = new ArrayList<Entry<T>>();
if (accept.isWildcardType())
{
return convert(all);
}
else
{
matches.addAll(wildcards);
SubtypeMap<T> indexed = index.get(accept.getType());
if (indexed != null)
{
matches.addAll(indexed.getPossible(accept));
}
}
Collections.sort(matches);
return convert(matches);
}
public static boolean useCache = true;
public List<T> getPossible(MediaType accept, Class<?> type)
{
List<T> cached = null;
CachedMediaTypeAndClass cacheEntry = null;
if (useCache)
{
cacheEntry = new CachedMediaTypeAndClass(type, accept);
cached = classCache.get(cacheEntry);
if (cached != null) return cached;
}
accept = new MediaType(accept.getType().toLowerCase(), accept.getSubtype().toLowerCase(), accept.getParameters());
List<Entry<T>> matches = new ArrayList<Entry<T>>();
if (accept.isWildcardType())
{
matches.addAll(all);
}
else
{
matches.addAll(wildcards);
SubtypeMap<T> indexed = index.get(accept.getType());
if (indexed != null)
{
matches.addAll(indexed.getPossible(accept));
}
}
Collections.sort(matches, new TypedEntryComparator(type));
cached = convert(matches);
if (useCache) classCache.put(cacheEntry, cached);
return cached;
}
}