/*
* 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.server.impl.container.grizzly;
import com.sun.grizzly.tcp.http11.GrizzlyAdapter;
import com.sun.grizzly.tcp.http11.GrizzlyRequest;
import com.sun.grizzly.tcp.http11.GrizzlyResponse;
import com.sun.jersey.api.container.ContainerException;
import com.sun.jersey.api.container.grizzly.GrizzlyServerFactory;
import com.sun.jersey.api.core.ResourceConfig;
import com.sun.jersey.core.header.InBoundHeaders;
import com.sun.jersey.server.impl.ThreadLocalInvoker;
import com.sun.jersey.spi.container.ContainerListener;
import com.sun.jersey.spi.container.ContainerRequest;
import com.sun.jersey.spi.container.ContainerResponse;
import com.sun.jersey.spi.container.ContainerResponseWriter;
import com.sun.jersey.spi.container.ReloadListener;
import com.sun.jersey.spi.container.WebApplication;
import com.sun.jersey.spi.inject.SingletonTypeInjectableProvider;
import java.io.IOException;
import java.io.OutputStream;
import java.lang.reflect.Type;
import java.net.URI;
import java.net.URISyntaxException;
import java.util.Enumeration;
import java.util.List;
import java.util.Map;
import javax.ws.rs.core.Context;
import javax.ws.rs.core.GenericEntity;
/**
*
* @author Marc.Hadley@Sun.Com
*/
public final class GrizzlyContainer extends GrizzlyAdapter implements ContainerListener {
private WebApplication application;
private String basePath = "/";
private final ThreadLocalInvoker<GrizzlyRequest> requestInvoker =
new ThreadLocalInvoker<GrizzlyRequest>();
private final ThreadLocalInvoker<GrizzlyResponse> responseInvoker =
new ThreadLocalInvoker<GrizzlyResponse>();
private static class ContextInjectableProvider<T> extends
SingletonTypeInjectableProvider<Context, T> {
protected ContextInjectableProvider(Type type, T instance) {
super(type, instance);
}
}
public GrizzlyContainer(ResourceConfig rc, WebApplication app) throws ContainerException {
this.application = app;
setAllowEncodedSlash(rc.getFeature(GrizzlyServerFactory.FEATURE_ALLOW_ENCODED_SLASH));
GenericEntity<ThreadLocal<GrizzlyRequest>> requestThreadLocal =
new GenericEntity<ThreadLocal<GrizzlyRequest>>(requestInvoker.getImmutableThreadLocal()) {};
rc.getSingletons().add(new ContextInjectableProvider<ThreadLocal<GrizzlyRequest>>(
requestThreadLocal.getType(), requestThreadLocal.getEntity()));
GenericEntity<ThreadLocal<GrizzlyResponse>> responseThreadLocal =
new GenericEntity<ThreadLocal<GrizzlyResponse>>(responseInvoker.getImmutableThreadLocal()) {};
rc.getSingletons().add(new ContextInjectableProvider<ThreadLocal<GrizzlyResponse>>(
responseThreadLocal.getType(), responseThreadLocal.getEntity()));
}
private final static class Writer implements ContainerResponseWriter {
final GrizzlyResponse response;
Writer(GrizzlyResponse response) {
this.response = response;
}
public OutputStream writeStatusAndHeaders(long contentLength,
ContainerResponse cResponse) throws IOException {
response.setStatus(cResponse.getStatus());
if (contentLength != -1 && contentLength < Integer.MAX_VALUE)
response.setContentLength((int)contentLength);
for (Map.Entry<String, List<Object>> e : cResponse.getHttpHeaders().entrySet()) {
for (Object value : e.getValue()) {
response.addHeader(e.getKey(), ContainerResponse.getHeaderValue(value));
}
}
String contentType = response.getHeader("Content-Type");
if (contentType != null) {
response.setContentType(contentType);
}
return response.getOutputStream();
}
public void finish() throws IOException {
}
}
public void service(GrizzlyRequest request, GrizzlyResponse response) {
try {
requestInvoker.set(request);
responseInvoker.set(response);
_service(request, response);
} finally {
requestInvoker.set(null);
responseInvoker.set(null);
}
}
private void _service(GrizzlyRequest request, GrizzlyResponse response) {
WebApplication _application = application;
final URI baseUri = getBaseUri(request);
/*
* request.unparsedURI() is a URI in encoded form that contains
* the URI path and URI query components.
*/
final URI requestUri = baseUri.resolve(
request.getRequest().unparsedURI().toString());
/**
* Check if the request URI path starts with the base URI path
*/
if (!requestUri.getRawPath().startsWith(basePath)) {
response.setStatus(404);
return;
}
try {
final ContainerRequest cRequest = new ContainerRequest(
_application,
request.getMethod(),
baseUri,
requestUri,
getHeaders(request),
request.getInputStream());
_application.handleRequest(cRequest, new Writer(response));
} catch (IOException ex) {
throw new RuntimeException(ex);
}
}
@Override
public void afterService(GrizzlyRequest request, GrizzlyResponse response)
throws Exception {
}
@Override
public void setResourcesContextPath(String resourcesContextPath) {
super.setResourcesContextPath(resourcesContextPath);
if (resourcesContextPath == null || resourcesContextPath.length() == 0) {
basePath = "/";
} else if (resourcesContextPath.charAt(resourcesContextPath.length() - 1) != '/') {
basePath = resourcesContextPath + "/";
} else {
basePath = resourcesContextPath;
}
if (basePath.charAt(0) != '/')
throw new IllegalArgumentException("The resource context path, " + resourcesContextPath +
", must start with a '/'");
}
private URI getBaseUri(GrizzlyRequest request) {
try {
return new URI(
request.getScheme(),
null,
request.getServerName(),
request.getServerPort(),
basePath,
null, null);
} catch (URISyntaxException ex) {
throw new IllegalArgumentException(ex);
}
}
private InBoundHeaders getHeaders(GrizzlyRequest request) {
InBoundHeaders rh = new InBoundHeaders();
Enumeration names = request.getHeaderNames();
while (names.hasMoreElements()) {
String name = (String)names.nextElement();
String value = request.getHeader(name);
rh.add(name, value);
}
return rh;
}
// ContainerListener
@Override
public void onReload() {
WebApplication oldApplication = application;
application = application.clone();
if (application.getFeaturesAndProperties() instanceof ReloadListener)
((ReloadListener) application.getFeaturesAndProperties()).onReload();
oldApplication.destroy();
}
}