Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance. Project price only 1 $
You can buy this project and download/modify it how often you want.
/*
* Copyright 2010-2013 Ning, Inc.
*
* Ning licenses this file to you 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:
*
* http://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.killbill.billing.jaxrs.util;
import java.net.URI;
import java.util.HashMap;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Objects;
import javax.annotation.Nullable;
import javax.inject.Inject;
import javax.servlet.ServletRequest;
import javax.ws.rs.Path;
import javax.ws.rs.core.Response;
import javax.ws.rs.core.Response.ResponseBuilder;
import javax.ws.rs.core.UriBuilder;
import javax.ws.rs.core.UriInfo;
import org.killbill.billing.jaxrs.resources.JaxRsResourceBase;
import org.killbill.billing.jaxrs.resources.JaxrsResource;
import org.killbill.billing.util.config.definition.JaxrsConfig;
public class JaxrsUriBuilder {
private final JaxrsConfig jaxrsConfig;
private final Map, UriBuilder> classToUriBuilder = new HashMap<>();
private final Map classAndMethodToUriBuilder = new HashMap<>();
private final Map pathAndClassToUriBuilder = new HashMap<>();
private final Map pathClassAndMethodToUriBuilder = new HashMap<>();
@Inject
public JaxrsUriBuilder(final JaxrsConfig jaxrsConfig) {
this.jaxrsConfig = jaxrsConfig;
}
public Response buildResponse(final UriInfo uriInfo,
final Class theClass,
final String getMethodName,
final Object objectId,
final ServletRequest request) {
return buildResponse(uriInfo, theClass, getMethodName, objectId, null, request);
}
public Response buildResponse(final UriInfo uriInfo,
final Class theClass,
final String getMethodName,
final Object objectId,
final Map params,
final ServletRequest request) {
return buildResponse(Response.status(Response.Status.CREATED), uriInfo, theClass, getMethodName, objectId, params, request);
}
public Response buildResponse(final ResponseBuilder responseBuilder,
final UriInfo uriInfo,
final Class theClass,
final String getMethodName,
final Object objectId,
final ServletRequest request) {
return buildResponse(responseBuilder, uriInfo, theClass, getMethodName, objectId, null, request);
}
public Response buildResponse(final ResponseBuilder responseBuilder,
final UriInfo uriInfo,
final Class theClass,
final String getMethodName,
final Object objectId,
final Map params,
final ServletRequest request) {
final URI location = buildLocation(uriInfo, theClass, getMethodName, objectId, params, request);
return !jaxrsConfig.isJaxrsLocationFullUrl() ?
responseBuilder.header("Location", location.getPath()).build() :
responseBuilder.location(location).build();
}
private URI buildLocation(final UriInfo uriInfo,
final Class theClass,
final String getMethodName,
final Object objectId,
final Map params,
final ServletRequest request) {
final UriBuilder uriBuilder = getUriBuilder(uriInfo.getBaseUri().getPath(), theClass, getMethodName);
if (null != params && !params.isEmpty()) {
for (final Entry entry : params.entrySet()) {
uriBuilder.queryParam(entry.getKey(), entry.getValue());
}
}
if (jaxrsConfig.isJaxrsLocationFullUrl()) {
if (jaxrsConfig.isJaxrsLocationUseForwardHeaders()) {
// Use "remote" value to support X-Forwarded headers (assumes RemoteIpValve or similar is configured)
// See https://github.com/killbill/killbill/issues/566
uriBuilder.scheme(request.getScheme())
.host(Objects.requireNonNullElse(jaxrsConfig.getJaxrsLocationHost(), request.getServerName()))
.port(request.getServerPort());
} else {
uriBuilder.scheme(uriInfo.getAbsolutePath().getScheme())
.host(uriInfo.getAbsolutePath().getHost())
.port(uriInfo.getAbsolutePath().getPort());
}
}
return objectId != null ? uriBuilder.build(objectId) : uriBuilder.build();
}
public URI nextPage(final Class theClass,
final String getMethodName,
final Long nextOffset,
final Long limit,
final Map queryParams,
final Map pathParams) {
if (nextOffset == null || limit == null) {
// End of pagination?
return null;
}
final UriBuilder uriBuilder = getUriBuilder(theClass, getMethodName).queryParam(JaxRsResourceBase.QUERY_SEARCH_OFFSET, nextOffset)
.queryParam(JaxRsResourceBase.QUERY_SEARCH_LIMIT, limit);
for (final Entry entry : queryParams.entrySet()) {
uriBuilder.queryParam(entry.getKey(), entry.getValue());
}
for (final Entry entry : pathParams.entrySet()) {
uriBuilder.resolveTemplate(entry.getKey(), entry.getValue());
}
return uriBuilder.build();
}
private UriBuilder getUriBuilder(final String path, final Class theClassMaybeEnhanced, @Nullable final String getMethodName) {
final Class theClass = getNonEnhancedClass(theClassMaybeEnhanced);
return getMethodName != null ? fromPath("/".equals(path) ? path.substring(1) : path, theClass, getMethodName) : fromPath(path, theClass);
}
private UriBuilder fromPath(final String path, final Class theClass, final String getMethodName) {
final String key = path + theClass.getName() + getMethodName;
UriBuilder uriBuilder = pathClassAndMethodToUriBuilder.get(key);
if (uriBuilder == null) {
synchronized (pathClassAndMethodToUriBuilder) {
uriBuilder = pathClassAndMethodToUriBuilder.get(key);
if (uriBuilder == null) {
uriBuilder = fromPath(path, theClass).path(theClass, getMethodName);
pathClassAndMethodToUriBuilder.put(key, uriBuilder);
}
}
}
return uriBuilder.clone();
}
private UriBuilder fromPath(final String path, final Class theClass) {
final String key = path + theClass.getName();
UriBuilder uriBuilder = pathAndClassToUriBuilder.get(key);
if (uriBuilder == null) {
synchronized (pathAndClassToUriBuilder) {
uriBuilder = pathAndClassToUriBuilder.get(key);
if (uriBuilder == null) {
uriBuilder = UriBuilder.fromPath(path).path(theClass);
pathAndClassToUriBuilder.put(key, uriBuilder);
}
}
}
return uriBuilder.clone();
}
private UriBuilder getUriBuilder(final Class theClassMaybeEnhanced, @Nullable final String getMethodName) {
final Class theClass = getNonEnhancedClass(theClassMaybeEnhanced);
return getMethodName != null ? fromResource(theClass, getMethodName) : fromResource(theClass);
}
private UriBuilder fromResource(final Class theClass, final String getMethodName) {
final String key = theClass.getName() + getMethodName;
UriBuilder uriBuilder = classAndMethodToUriBuilder.get(key);
if (uriBuilder == null) {
synchronized (classAndMethodToUriBuilder) {
uriBuilder = classAndMethodToUriBuilder.get(key);
if (uriBuilder == null) {
uriBuilder = fromResource(theClass).path(theClass, getMethodName);
classAndMethodToUriBuilder.put(key, uriBuilder);
}
}
}
return uriBuilder.clone();
}
private UriBuilder fromResource(final Class theClass) {
UriBuilder uriBuilder = classToUriBuilder.get(theClass);
if (uriBuilder == null) {
synchronized (classToUriBuilder) {
uriBuilder = classToUriBuilder.get(theClass);
if (uriBuilder == null) {
uriBuilder = UriBuilder.fromResource(theClass);
classToUriBuilder.put(theClass, uriBuilder);
}
}
}
return uriBuilder.clone();
}
private Class getNonEnhancedClass(final Class theClassMaybeEnhanced) {
// If Guice is proxying the class for example ($EnhancerByGuice$), we want the real class.
// See https://java.net/projects/jersey/lists/users/archive/2008-10/message/291
Class theClass = theClassMaybeEnhanced;
while (theClass.getAnnotation(Path.class) == null && JaxRsResourceBase.class.isAssignableFrom(theClass)) {
theClass = theClass.getSuperclass();
}
return theClass;
}
}
