/*
* Copyright (c) 2011, 2019 Oracle and/or its affiliates. All rights reserved.
*
* This program and the accompanying materials are made available under the
* terms of the Eclipse Public License v. 2.0, which is available at
* http://www.eclipse.org/legal/epl-2.0.
*
* This Source Code may also be made available under the following Secondary
* Licenses when the conditions for such availability set forth in the
* Eclipse Public License v. 2.0 are satisfied: GNU General Public License,
* version 2 with the GNU Classpath Exception, which is available at
* https://www.gnu.org/software/classpath/license.html.
*
* SPDX-License-Identifier: EPL-2.0 OR GPL-2.0 WITH Classpath-exception-2.0
*/
package org.glassfish.jersey.message.internal;
import java.util.Collections;
import java.util.HashSet;
import java.util.Iterator;
import java.util.LinkedList;
import java.util.List;
import java.util.Locale;
import java.util.Set;
import java.util.stream.Collectors;
import javax.ws.rs.core.HttpHeaders;
import javax.ws.rs.core.MediaType;
import javax.ws.rs.core.Variant;
import org.glassfish.jersey.internal.util.collection.Ref;
Utility for selecting variant that best matches request from a list of variants.
Author: Paul Sandoz, Marek Potociar
/**
* Utility for selecting variant that best matches request from a list of variants.
*
* @author Paul Sandoz
* @author Marek Potociar
*/
public final class VariantSelector {
private VariantSelector() {
}
Interface to get a dimension value from a variant and check if an
acceptable dimension value is compatible with a dimension value.
/**
* Interface to get a dimension value from a variant and check if an
* acceptable dimension value is compatible with a dimension value.
*/
private interface DimensionChecker<T, U> {
Get the dimension value from the variant.
Params: - v – the variant.
Returns: the dimension value.
/**
* Get the dimension value from the variant.
*
* @param v the variant.
* @return the dimension value.
*/
U getDimension(VariantHolder v);
Get the quality source of the dimension.
Returns: quality source.
/**
* Get the quality source of the dimension.
*
* @return quality source.
*/
int getQualitySource(VariantHolder v, U u);
Ascertain if the acceptable dimension value is compatible with
the dimension value.
Params: - t – the acceptable dimension value.
- u – the dimension value.
Returns: true
if the acceptable dimension value is compatible with the dimension value.
/**
* Ascertain if the acceptable dimension value is compatible with
* the dimension value.
*
* @param t the acceptable dimension value.
* @param u the dimension value.
* @return {@code true} if the acceptable dimension value is compatible with
* the dimension value.
*/
boolean isCompatible(T t, U u);
Get the value of the Vary header.
Returns: the value of the Vary header.
/**
* Get the value of the Vary header.
*
* @return the value of the Vary header.
*/
String getVaryHeaderValue();
}
private static final DimensionChecker<AcceptableMediaType, MediaType> MEDIA_TYPE_DC =
new DimensionChecker<AcceptableMediaType, MediaType>() {
@Override
public MediaType getDimension(final VariantHolder v) {
return v.v.getMediaType();
}
@Override
public boolean isCompatible(final AcceptableMediaType t, final MediaType u) {
return t.isCompatible(u);
}
@Override
public int getQualitySource(final VariantHolder v, final MediaType u) {
return v.mediaTypeQs;
}
@Override
public String getVaryHeaderValue() {
return HttpHeaders.ACCEPT;
}
};
private static final DimensionChecker<AcceptableLanguageTag, Locale> LANGUAGE_TAG_DC =
new DimensionChecker<AcceptableLanguageTag, Locale>() {
@Override
public Locale getDimension(final VariantHolder v) {
return v.v.getLanguage();
}
@Override
public boolean isCompatible(final AcceptableLanguageTag t, final Locale u) {
return t.isCompatible(u);
}
@Override
public int getQualitySource(final VariantHolder qsv, final Locale u) {
return Quality.MINIMUM;
}
@Override
public String getVaryHeaderValue() {
return HttpHeaders.ACCEPT_LANGUAGE;
}
};
private static final DimensionChecker<AcceptableToken, String> CHARSET_DC =
new DimensionChecker<AcceptableToken, String>() {
@Override
public String getDimension(final VariantHolder v) {
final MediaType m = v.v.getMediaType();
return (m != null) ? m.getParameters().get("charset") : null;
}
@Override
public boolean isCompatible(final AcceptableToken t, final String u) {
return t.isCompatible(u);
}
@Override
public int getQualitySource(final VariantHolder qsv, final String u) {
return Quality.MINIMUM;
}
@Override
public String getVaryHeaderValue() {
return HttpHeaders.ACCEPT_CHARSET;
}
};
private static final DimensionChecker<AcceptableToken, String> ENCODING_DC =
new DimensionChecker<AcceptableToken, String>() {
@Override
public String getDimension(final VariantHolder v) {
return v.v.getEncoding();
}
@Override
public boolean isCompatible(final AcceptableToken t, final String u) {
return t.isCompatible(u);
}
@Override
public int getQualitySource(final VariantHolder qsv, final String u) {
return Quality.MINIMUM;
}
@Override
public String getVaryHeaderValue() {
return HttpHeaders.ACCEPT_ENCODING;
}
};
Select variants for a given dimension.
Params: - variantHolders – collection of variants.
- acceptableValues – the list of acceptable dimension values, ordered by the quality
parameter, with the highest quality dimension value occurring
first.
- dimensionChecker – the dimension checker
- vary – output list of generated vary headers.
/**
* Select variants for a given dimension.
*
* @param variantHolders collection of variants.
* @param acceptableValues the list of acceptable dimension values, ordered by the quality
* parameter, with the highest quality dimension value occurring
* first.
* @param dimensionChecker the dimension checker
* @param vary output list of generated vary headers.
*/
private static <T extends Qualified, U> LinkedList<VariantHolder> selectVariants(
final List<VariantHolder> variantHolders,
final List<T> acceptableValues,
final DimensionChecker<T, U> dimensionChecker,
final Set<String> vary) {
int cq = Quality.MINIMUM;
int cqs = Quality.MINIMUM;
final LinkedList<VariantHolder> selected = new LinkedList<>();
// Iterate over the acceptable entries
// This assumes the entries are ordered by the quality
for (final T a : acceptableValues) {
final int q = a.getQuality();
final Iterator<VariantHolder> iv = variantHolders.iterator();
while (iv.hasNext()) {
final VariantHolder v = iv.next();
// Get the dimension value of the variant to check
final U d = dimensionChecker.getDimension(v);
if (d != null) {
vary.add(dimensionChecker.getVaryHeaderValue());
// Check if the acceptable entry is compatable with
// the dimension value
final int qs = dimensionChecker.getQualitySource(v, d);
if (qs >= cqs && dimensionChecker.isCompatible(a, d)) {
if (qs > cqs) {
cqs = qs;
cq = q;
// Remove all entries that were added for qs < cqs
selected.clear();
selected.add(v);
} else if (q > cq) {
cq = q;
// Add variant with higher accept quality at the front
selected.addFirst(v);
} else if (q == cq) {
// Ensure selection is stable with order of variants
// with same quality of source and accept quality
selected.add(v);
}
iv.remove();
}
}
}
}
// Add all variants that are not compatible with this dimension
// to the end
for (final VariantHolder v : variantHolders) {
if (dimensionChecker.getDimension(v) == null) {
selected.add(v);
}
}
return selected;
}
private static class VariantHolder {
private final Variant v;
private final int mediaTypeQs;
VariantHolder(final Variant v) {
this(v, Quality.DEFAULT);
}
VariantHolder(final Variant v, final int mediaTypeQs) {
this.v = v;
this.mediaTypeQs = mediaTypeQs;
}
}
private static LinkedList<VariantHolder> getVariantHolderList(final List<Variant> variants) {
final LinkedList<VariantHolder> l = new LinkedList<>();
for (final Variant v : variants) {
final MediaType mt = v.getMediaType();
if (mt != null) {
if (mt instanceof QualitySourceMediaType || mt.getParameters()
.containsKey(Quality.QUALITY_SOURCE_PARAMETER_NAME)) {
final int qs = QualitySourceMediaType.getQualitySource(mt);
l.add(new VariantHolder(v, qs));
} else {
l.add(new VariantHolder(v));
}
} else {
l.add(new VariantHolder(v));
}
}
return l;
}
Select the representation variant that best matches the request. More explicit
variants are chosen ahead of less explicit ones.
Params: - context – inbound message context.
- variants – list of possible variants.
- varyHeaderValue – an output reference of vary header value that should be put
into the response Vary header.
Returns: selected variant.
/**
* Select the representation variant that best matches the request. More explicit
* variants are chosen ahead of less explicit ones.
*
* @param context inbound message context.
* @param variants list of possible variants.
* @param varyHeaderValue an output reference of vary header value that should be put
* into the response Vary header.
* @return selected variant.
*/
public static Variant selectVariant(final InboundMessageContext context,
final List<Variant> variants,
final Ref<String> varyHeaderValue) {
final List<Variant> selectedVariants = selectVariants(context, variants, varyHeaderValue);
return selectedVariants.isEmpty() ? null : selectedVariants.get(0);
}
Select possible representation variants in order in which they best matches the request.
Params: - context – inbound message context.
- variants – list of possible variants.
- varyHeaderValue – an output reference of vary header value that should be put into the response Vary header.
Returns: possible variants.
/**
* Select possible representation variants in order in which they best matches the request.
*
* @param context inbound message context.
* @param variants list of possible variants.
* @param varyHeaderValue an output reference of vary header value that should be put into the response Vary header.
* @return possible variants.
*/
public static List<Variant> selectVariants(final InboundMessageContext context,
final List<Variant> variants,
final Ref<String> varyHeaderValue) {
LinkedList<VariantHolder> vhs = getVariantHolderList(variants);
final Set<String> vary = new HashSet<>();
vhs = selectVariants(vhs, context.getQualifiedAcceptableMediaTypes(), MEDIA_TYPE_DC, vary);
vhs = selectVariants(vhs, context.getQualifiedAcceptableLanguages(), LANGUAGE_TAG_DC, vary);
vhs = selectVariants(vhs, context.getQualifiedAcceptCharset(), CHARSET_DC, vary);
vhs = selectVariants(vhs, context.getQualifiedAcceptEncoding(), ENCODING_DC, vary);
if (vhs.isEmpty()) {
return Collections.emptyList();
} else {
final StringBuilder varyHeader = new StringBuilder();
for (final String v : vary) {
if (varyHeader.length() > 0) {
varyHeader.append(',');
}
varyHeader.append(v);
}
final String varyValue = varyHeader.toString();
if (!varyValue.isEmpty()) {
varyHeaderValue.set(varyValue);
}
return vhs.stream()
.map(variantHolder -> variantHolder.v)
.collect(Collectors.toList());
}
}
}