com.sun.jersey.core.spi.factory.MessageBodyFactory Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of jersey-bundle Show documentation
Show all versions of jersey-bundle Show documentation
A bundle containing code of all jar-based modules that provide
JAX-RS and Jersey-related features. Such a bundle is *only intended* for
developers that do not use Maven's dependency system.
The bundle does not include code for contributes, tests and samples.
/*
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
*
* Copyright (c) 2010-2011 Oracle and/or its affiliates. All rights reserved.
*
* The contents of this file are subject to the terms of either the GNU
* General Public License Version 2 only ("GPL") or the Common Development
* and Distribution License("CDDL") (collectively, the "License"). You
* may not use this file except in compliance with the License. You can
* obtain a copy of the License at
* http://glassfish.java.net/public/CDDL+GPL_1_1.html
* or packager/legal/LICENSE.txt. See the License for the specific
* language governing permissions and limitations under the License.
*
* When distributing the software, include this License Header Notice in each
* file and include the License file at packager/legal/LICENSE.txt.
*
* GPL Classpath Exception:
* Oracle designates this particular file as subject to the "Classpath"
* exception as provided by Oracle in the GPL Version 2 section of the License
* file that accompanied this code.
*
* Modifications:
* If applicable, add the following below the License Header, with the fields
* enclosed by brackets [] replaced by your own identifying information:
* "Portions Copyright [year] [name of copyright owner]"
*
* Contributor(s):
* If you wish your version of this file to be governed by only the CDDL or
* only the GPL Version 2, indicate your decision by adding "[Contributor]
* elects to include this software in this distribution under the [CDDL or GPL
* Version 2] license." If you don't indicate a single choice of license, a
* recipient has the option to distribute your version of this file under
* either the CDDL, the GPL Version 2 or to extend the choice of license to
* its licensees as provided above. However, if you add GPL Version 2 code
* and therefore, elected the GPL Version 2 license, then the option applies
* only if the new code is made subject to such option by the copyright
* holder.
*/
package com.sun.jersey.core.spi.factory;
import com.sun.jersey.core.header.MediaTypes;
import com.sun.jersey.core.reflection.ReflectionHelper;
import com.sun.jersey.core.reflection.ReflectionHelper.DeclaringClassInterfacePair;
import com.sun.jersey.core.spi.component.ProviderServices;
import com.sun.jersey.core.util.KeyComparator;
import com.sun.jersey.core.util.KeyComparatorHashMap;
import com.sun.jersey.core.util.KeyComparatorLinkedHashMap;
import com.sun.jersey.spi.MessageBodyWorkers;
import javax.ws.rs.Consumes;
import javax.ws.rs.Produces;
import javax.ws.rs.core.MediaType;
import javax.ws.rs.ext.MessageBodyReader;
import javax.ws.rs.ext.MessageBodyWriter;
import java.io.PrintWriter;
import java.io.StringWriter;
import java.lang.annotation.Annotation;
import java.lang.reflect.Type;
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.Set;
/**
* A factory for managing {@link MessageBodyReader} and {@link MessageBodyWriter}
* instances.
*
* @author [email protected]
*/
public class MessageBodyFactory implements MessageBodyWorkers {
/* package */ static final KeyComparator MEDIA_TYPE_COMPARATOR =
new KeyComparator() {
public boolean equals(MediaType x, MediaType y) {
return x.getType().equalsIgnoreCase(y.getType())
&& x.getSubtype().equalsIgnoreCase(y.getSubtype());
}
public int hash(MediaType k) {
return k.getType().toLowerCase().hashCode() +
k.getSubtype().toLowerCase().hashCode();
}
public int compare(MediaType o1, MediaType o2) {
throw new UnsupportedOperationException("Not supported yet.");
}
};
private final ProviderServices providerServices;
private final boolean deprecatedProviderPrecedence;
private Map> readerProviders;
private Map> writerProviders;
private List writerListProviders;
private Map> customReaderProviders;
private Map> customWriterProviders;
private List customWriterListProviders;
private static class MessageBodyWriterPair {
final MessageBodyWriter mbw;
final List types;
MessageBodyWriterPair(MessageBodyWriter mbw, List types) {
this.mbw = mbw;
this.types = types;
}
}
public MessageBodyFactory(ProviderServices providerServices, boolean deprecatedProviderPrecedence) {
this.providerServices = providerServices;
this.deprecatedProviderPrecedence = deprecatedProviderPrecedence;
}
private static class DistanceComparator implements Comparator {
private final Class c;
private final Map distanceMap = new HashMap();
DistanceComparator(Class c) {
this.c = c;
}
public int compare(T o1, T o2) {
int d1 = getDistance(o1);
int d2 = getDistance(o2);
return d2 - d1;
}
int getDistance(T t) {
Integer d = distanceMap.get(t.getClass());
if (d != null)
return d;
DeclaringClassInterfacePair p = ReflectionHelper.getClass(
t.getClass(), c);
Class[] as = ReflectionHelper.getParameterizedClassArguments(p);
Class a = (as != null) ? as[0] : null;
d = 0;
while (a != null && a != Object.class) {
d++;
a = a.getSuperclass();
}
distanceMap.put(t.getClass(), d);
return d;
}
}
public void init() {
initReaders();
initWriters();
}
private void initReaders() {
this.customReaderProviders = new KeyComparatorHashMap>(
MEDIA_TYPE_COMPARATOR);
this.readerProviders = new KeyComparatorHashMap>(
MEDIA_TYPE_COMPARATOR);
if(deprecatedProviderPrecedence) {
initReaders(this.readerProviders, providerServices.getProvidersAndServices(MessageBodyReader.class));
} else {
initReaders(this.customReaderProviders, providerServices.getProviders(MessageBodyReader.class));
initReaders(this.readerProviders, providerServices.getServices(MessageBodyReader.class));
}
}
private void initReaders(Map> providersMap, Set providersSet) {
for (MessageBodyReader provider : providersSet) {
List values = MediaTypes.createMediaTypes(
provider.getClass().getAnnotation(Consumes.class));
for (MediaType type : values)
getClassCapability(providersMap, provider, type);
}
DistanceComparator dc = new DistanceComparator(MessageBodyReader.class);
for (Map.Entry> e : providersMap.entrySet()) {
Collections.sort(e.getValue(), dc);
}
}
private void initWriters() {
this.customWriterProviders = new KeyComparatorHashMap>(
MEDIA_TYPE_COMPARATOR);
this.customWriterListProviders = new ArrayList();
this.writerProviders = new KeyComparatorHashMap>(
MEDIA_TYPE_COMPARATOR);
this.writerListProviders = new ArrayList();
if(deprecatedProviderPrecedence) {
initWriters(writerProviders,writerListProviders, providerServices.getProvidersAndServices(MessageBodyWriter.class));
} else {
initWriters(customWriterProviders, customWriterListProviders, providerServices.getProviders(MessageBodyWriter.class));
initWriters(writerProviders, writerListProviders, providerServices.getServices(MessageBodyWriter.class));
}
}
private void initWriters(Map> providersMap, List listProviders, Set providersSet) {
for (MessageBodyWriter provider : providersSet) {
List values = MediaTypes.createMediaTypes(
provider.getClass().getAnnotation(Produces.class));
for (MediaType type : values)
getClassCapability(providersMap, provider, type);
listProviders.add(new MessageBodyWriterPair(provider, values));
}
final DistanceComparator dc = new DistanceComparator(MessageBodyWriter.class);
for (Map.Entry> e : providersMap.entrySet()) {
Collections.sort(e.getValue(), dc);
}
Collections.sort(listProviders, new Comparator() {
public int compare(MessageBodyWriterPair p1, MessageBodyWriterPair p2) {
return dc.compare(p1.mbw, p2.mbw);
}
});
}
private void getClassCapability(Map> capabilities,
T provider, MediaType mediaType) {
if (!capabilities.containsKey(mediaType))
capabilities.put(mediaType, new ArrayList());
List providers = capabilities.get(mediaType);
providers.add(provider);
}
// MessageBodyWorkers
public Map> getReaders(MediaType mediaType) {
Map> subSet =
new KeyComparatorLinkedHashMap>(
MEDIA_TYPE_COMPARATOR);
if(!customReaderProviders.isEmpty())
getCompatibleReadersWritersMap(mediaType, customReaderProviders, subSet);
getCompatibleReadersWritersMap(mediaType, readerProviders, subSet);
return subSet;
}
public Map> getWriters(MediaType mediaType) {
Map> subSet =
new KeyComparatorLinkedHashMap>(
MEDIA_TYPE_COMPARATOR);
if(!customWriterProviders.isEmpty())
getCompatibleReadersWritersMap(mediaType, customWriterProviders, subSet);
getCompatibleReadersWritersMap(mediaType, writerProviders, subSet);
return subSet;
}
public String readersToString(Map> readers) {
return toString(readers);
}
public String writersToString(Map> writers) {
return toString(writers);
}
private String toString(Map> set) {
StringWriter sw = new StringWriter();
PrintWriter pw = new PrintWriter(sw);
for (Map.Entry> e : set.entrySet()) {
pw.append(e.getKey().toString()).println(" ->");
for (T t : e.getValue()) {
pw.append(" ").println(t.getClass().getName());
}
}
pw.flush();
return sw.toString();
}
public MessageBodyReader getMessageBodyReader(Class c, Type t,
Annotation[] as,
MediaType mediaType) {
MessageBodyReader reader;
if(!customReaderProviders.isEmpty()) {
reader = _getMessageBodyReader(c, t, as, mediaType, customReaderProviders);
if(reader != null)
return reader;
}
reader = _getMessageBodyReader(c, t, as, mediaType, readerProviders);
return reader;
}
private MessageBodyReader _getMessageBodyReader(Class c, Type t,
Annotation[] as,
MediaType mediaType,
Map> providers) {
MessageBodyReader p = null;
if (mediaType != null) {
p = _getMessageBodyReader(c, t, as, mediaType, mediaType, providers);
if (p == null)
p = _getMessageBodyReader(c, t, as, mediaType,
MediaTypes.getTypeWildCart(mediaType), providers);
}
if (p == null)
p = _getMessageBodyReader(c, t, as, mediaType, MediaTypes.GENERAL_MEDIA_TYPE, providers);
return p;
}
private MessageBodyReader _getMessageBodyReader(Class c, Type t,
Annotation[] as,
MediaType mediaType, MediaType lookup) {
MessageBodyReader reader;
if(!customReaderProviders.isEmpty()) {
reader = _getMessageBodyReader(c, t, as, mediaType, lookup, customReaderProviders);
if(reader != null)
return reader;
}
reader = _getMessageBodyReader(c, t, as, mediaType, lookup, readerProviders);
return reader;
}
private MessageBodyReader _getMessageBodyReader(Class c, Type t,
Annotation[] as,
MediaType mediaType, MediaType lookup,
Map> providers) {
List readers = providers.get(lookup);
if (readers == null)
return null;
for (MessageBodyReader p : readers) {
if (p.isReadable(c, t, as, mediaType)) {
return p;
}
}
return null;
}
public MessageBodyWriter getMessageBodyWriter(Class c, Type t,
Annotation[] as,
MediaType mediaType) {
MessageBodyWriter p;
if(!customWriterProviders.isEmpty()) {
p = _getMessageBodyWriter(c, t, as, mediaType, customWriterProviders);
if(p != null)
return p;
}
p = _getMessageBodyWriter(c, t, as, mediaType, writerProviders);
return p;
}
private MessageBodyWriter _getMessageBodyWriter(Class c, Type t,
Annotation[] as,
MediaType mediaType,
Map> providers) {
MessageBodyWriter p = null;
if (mediaType != null) {
p = _getMessageBodyWriter(c, t, as, mediaType, mediaType, providers);
if (p == null)
p = _getMessageBodyWriter(c, t, as, mediaType,
MediaTypes.getTypeWildCart(mediaType), providers);
}
if (p == null)
p = _getMessageBodyWriter(c, t, as, mediaType, MediaTypes.GENERAL_MEDIA_TYPE, providers);
return p;
}
private MessageBodyWriter _getMessageBodyWriter(Class c, Type t,
Annotation[] as,
MediaType mediaType, MediaType lookup,
Map> providers) {
List writers = providers.get(lookup);
if (writers == null)
return null;
for (MessageBodyWriter p : writers) {
if (p.isWriteable(c, t, as, mediaType)) {
return p;
}
}
return null;
}
private void getCompatibleReadersWritersMap(MediaType mediaType,
Map> set,
Map> subSet) {
if (mediaType.isWildcardType()) {
getCompatibleReadersWritersList(mediaType, set, subSet);
} else if (mediaType.isWildcardSubtype()) {
getCompatibleReadersWritersList(mediaType, set, subSet);
getCompatibleReadersWritersList(MediaTypes.GENERAL_MEDIA_TYPE, set, subSet);
} else {
getCompatibleReadersWritersList(mediaType, set, subSet);
getCompatibleReadersWritersList(
MediaTypes.getTypeWildCart(mediaType),
set, subSet);
getCompatibleReadersWritersList(MediaTypes.GENERAL_MEDIA_TYPE, set, subSet);
}
}
private void getCompatibleReadersWritersList(MediaType mediaType,
Map> set,
Map> subSet) {
List readers = set.get(mediaType);
if (readers != null) {
subSet.put(mediaType, Collections.unmodifiableList(readers));
}
}
public List getMessageBodyWriterMediaTypes(Class c, Type t,
Annotation[] as) {
List mtl = new ArrayList();
for (MessageBodyWriterPair mbwp : customWriterListProviders) {
if (mbwp.mbw.isWriteable(c, t, as, MediaType.APPLICATION_OCTET_STREAM_TYPE)) {
mtl.addAll(mbwp.types);
}
}
for (MessageBodyWriterPair mbwp : writerListProviders) {
if (mbwp.mbw.isWriteable(c, t, as, MediaType.APPLICATION_OCTET_STREAM_TYPE)) {
mtl.addAll(mbwp.types);
}
}
Collections.sort(mtl, MediaTypes.MEDIA_TYPE_COMPARATOR);
return mtl;
}
public MediaType getMessageBodyWriterMediaType(Class c, Type t,
Annotation[] as, List acceptableMediaTypes) {
for (MediaType acceptable : acceptableMediaTypes) {
for (MessageBodyWriterPair mbwp : customWriterListProviders) {
for (MediaType mt : mbwp.types) {
if (mt.isCompatible(acceptable) &&
mbwp.mbw.isWriteable(c, t, as, acceptable)) {
return MediaTypes.mostSpecific(mt, acceptable);
}
}
}
for (MessageBodyWriterPair mbwp : writerListProviders) {
for (MediaType mt : mbwp.types) {
if (mt.isCompatible(acceptable) &&
mbwp.mbw.isWriteable(c, t, as, acceptable)) {
return MediaTypes.mostSpecific(mt, acceptable);
}
}
}
}
return null;
}
}