com.sun.xml.ws.transport.http.servlet.ServletConnectionImpl Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of jaxws-rt Show documentation
Show all versions of jaxws-rt Show documentation
JAX-WS Runtime with module descriptor
/*
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
*
* Copyright (c) 1997-2012 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.xml.ws.transport.http.servlet;
import com.sun.istack.NotNull;
import com.sun.xml.ws.api.ha.HaInfo;
import com.sun.xml.ws.api.message.Packet;
import com.sun.xml.ws.api.server.PortAddressResolver;
import com.sun.xml.ws.api.server.WSEndpoint;
import com.sun.xml.ws.api.server.WebServiceContextDelegate;
import com.sun.xml.ws.resources.WsservletMessages;
import com.sun.xml.ws.transport.Headers;
import com.sun.xml.ws.transport.http.HttpAdapter;
import com.sun.xml.ws.transport.http.WSHTTPConnection;
import com.sun.xml.ws.developer.JAXWSProperties;
import javax.servlet.ServletContext;
import javax.servlet.http.Cookie;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.xml.ws.WebServiceException;
import javax.xml.ws.handler.MessageContext;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.security.Principal;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Enumeration;
import java.util.List;
import java.util.Map;
import java.util.Set;
/**
*{@link WSHTTPConnection} implemented for {@link HttpServlet}.
*
* @author WS Development Team
*/
public class ServletConnectionImpl extends WSHTTPConnection implements WebServiceContextDelegate {
private final HttpServletRequest request;
private final HttpServletResponse response;
private final ServletContext context;
private int status;
private Headers requestHeaders;
private final ServletAdapter adapter;
private Headers responseHeaders;
private HaInfo haInfo;
public ServletConnectionImpl(@NotNull ServletAdapter adapter, ServletContext context, HttpServletRequest request, HttpServletResponse response) {
this.adapter = adapter;
this.context = context;
this.request = request;
this.response = response;
}
@Override
@Property({MessageContext.HTTP_REQUEST_HEADERS, Packet.INBOUND_TRANSPORT_HEADERS})
public @NotNull Map> getRequestHeaders() {
if (requestHeaders == null) {
requestHeaders = new Headers();
Enumeration enums = request.getHeaderNames();
while (enums.hasMoreElements()) {
String headerName = (String) enums.nextElement();
Enumeration e = request.getHeaders(headerName);
if (e != null) {
List values = requestHeaders.get(headerName);
while(e.hasMoreElements()) {
String headerValue = (String)e.nextElement();
if (values == null) {
values = new ArrayList();
requestHeaders.put(headerName, values);
}
values.add(headerValue);
}
}
}
}
return requestHeaders;
}
@Override
public Set getRequestHeaderNames() {
return getRequestHeaders().keySet();
}
@Override
public List getRequestHeaderValues(@NotNull String headerName) {
if (requestHeaders != null) {
return requestHeaders.get(headerName);
}
return null;
}
/**
* sets response headers.
*/
@Override
public void setResponseHeaders(Map> headers) {
if (headers == null) {
responseHeaders = null;
} else {
if (responseHeaders == null) {
responseHeaders = new Headers();
} else {
responseHeaders.clear();
}
responseHeaders.putAll(headers);
}
}
@Override
public void setResponseHeader(String key, String value) {
setResponseHeader(key, Collections.singletonList(value));
}
@Override
public void setResponseHeader(String key, List value) {
if (responseHeaders == null) {
responseHeaders = new Headers();
}
responseHeaders.put(key, value);
}
@Override
@Property({MessageContext.HTTP_RESPONSE_HEADERS, Packet.OUTBOUND_TRANSPORT_HEADERS})
public Map> getResponseHeaders() {
return responseHeaders;
}
@Override
public void setStatus(int status) {
this.status = status;
// Servlet containers don't seem to like setting of the value multiple times
// Moving the following to getOutput()
//response.setStatus(status);
}
@Override
@Property(MessageContext.HTTP_RESPONSE_CODE)
public int getStatus() {
return status;
}
@Override
public void setContentTypeResponseHeader(@NotNull String value) {
response.setContentType(value);
}
@Override
public @NotNull InputStream getInput() throws IOException {
return request.getInputStream();
}
@Override
public @NotNull OutputStream getOutput() throws IOException {
response.setStatus(status);
if (responseHeaders != null) {
for (Map.Entry> entry : responseHeaders.entrySet()) {
String name = entry.getKey();
if (name == null) {
continue;
}
if(name.equalsIgnoreCase("Content-Type") || name.equalsIgnoreCase("Content-Length"))
continue; // ignore headers that interfere with the operation
for (String value : entry.getValue()) {
response.addHeader(name, value);
}
}
}
return response.getOutputStream();
}
public @NotNull WebServiceContextDelegate getWebServiceContextDelegate() {
return this;
}
public Principal getUserPrincipal(Packet p) {
return request.getUserPrincipal();
}
public boolean isUserInRole(Packet p,String role) {
return request.isUserInRole(role);
}
public @NotNull String getEPRAddress(Packet p, WSEndpoint endpoint) {
PortAddressResolver resolver = adapter.owner.createPortAddressResolver(getBaseAddress(), endpoint.getImplementationClass());
String address = resolver.getAddressFor(endpoint.getServiceName(), endpoint.getPortName().getLocalPart());
if(address==null)
throw new WebServiceException(WsservletMessages.SERVLET_NO_ADDRESS_AVAILABLE(endpoint.getPortName()));
return address;
}
public String getWSDLAddress(@NotNull Packet request, @NotNull WSEndpoint endpoint) {
String eprAddress = getEPRAddress(request,endpoint);
if(adapter.getEndpoint().getPort() != null)
return eprAddress+"?wsdl";
else
return null;
}
@Override
@Property(MessageContext.HTTP_REQUEST_METHOD)
public @NotNull String getRequestMethod() {
return request.getMethod();
}
@Override
public boolean isSecure() {
return request.isSecure();
}
@Override
public Principal getUserPrincipal() {
return request.getUserPrincipal();
}
@Override
public boolean isUserInRole(String role) {
return request.isUserInRole(role);
}
@Override
public String getRequestHeader(String headerName) {
return request.getHeader(headerName);
}
@Override
@Property(MessageContext.QUERY_STRING)
public String getQueryString() {
return request.getQueryString();
}
@Override
@Property(MessageContext.PATH_INFO)
public @NotNull String getPathInfo() {
return request.getPathInfo();
}
@Override
public @NotNull String getRequestURI() {
return request.getRequestURI();
}
@Override
public @NotNull String getRequestScheme() {
return request.getScheme();
}
@Override
public @NotNull String getServerName() {
return request.getServerName();
}
@Override
public @NotNull int getServerPort() {
return request.getServerPort();
}
@Override
public @NotNull String getContextPath() {
return request.getContextPath();
}
public @NotNull String getBaseAddress() {
return getBaseAddress(request);
}
static @NotNull String getBaseAddress(HttpServletRequest request) {
StringBuilder buf = new StringBuilder();
buf.append(request.getScheme());
buf.append("://");
buf.append(request.getServerName());
buf.append(':');
buf.append(request.getServerPort());
buf.append(request.getContextPath());
return buf.toString();
}
@Override
public Object getRequestAttribute(String key) {
return request.getAttribute(key);
}
@Property(MessageContext.SERVLET_CONTEXT)
public ServletContext getContext() {
return context;
}
@Property(MessageContext.SERVLET_RESPONSE)
public HttpServletResponse getResponse() {
return response;
}
@Property(MessageContext.SERVLET_REQUEST)
public HttpServletRequest getRequest() {
return request;
}
@Property(JAXWSProperties.HTTP_REQUEST_URL)
public String getRequestURL() {
return request.getRequestURL().toString();
}
@Override
public String getProtocol() {
return request.getProtocol();
}
@Override
public void setContentLengthResponseHeader(int value) {
response.setContentLength(value);
}
@Override
public String getCookie(String name) {
Cookie[] cookies = request.getCookies();
if (cookies != null) {
for(Cookie cookie : cookies) {
if (cookie.getName().equals(name)) {
return cookie.getValue();
}
}
}
return null;
}
@Override
public void setCookie(String name, String value) {
Cookie cookie = new Cookie(name, value);
response.addCookie(cookie);
}
@Property(Packet.HA_INFO)
public HaInfo getHaInfo() {
if (haInfo == null) {
String replicaInstance = null;
String key = null;
String jrouteId = null;
Cookie[] cookies = request.getCookies();
if (cookies != null) {
for(Cookie cookie : cookies) {
if (replicaInstance == null && cookie.getName().equals("JREPLICA")) {
replicaInstance = cookie.getValue();
} else if (key == null && cookie.getName().equals("METRO_KEY")) {
key = cookie.getValue();
} else if (jrouteId == null && cookie.getName().equals("JROUTE")) {
jrouteId = cookie.getValue();
}
}
if (replicaInstance != null && key != null) {
String proxyJroute = request.getHeader("proxy-jroute");
boolean failOver = jrouteId != null && proxyJroute != null && !jrouteId.equals(proxyJroute);
haInfo = new HaInfo(key, replicaInstance, failOver);
}
}
}
return haInfo;
}
public void setHaInfo(HaInfo replicaInfo) {
this.haInfo = replicaInfo;
}
protected PropertyMap getPropertyMap() {
return model;
}
private static final PropertyMap model;
static {
model = parse(ServletConnectionImpl.class);
}
}