package com.sun.xml.internal.ws.client;
import com.sun.istack.internal.NotNull;
import com.sun.istack.internal.Nullable;
import com.sun.xml.internal.ws.Closeable;
import com.sun.xml.internal.ws.api.BindingID;
import com.sun.xml.internal.ws.api.EndpointAddress;
import com.sun.xml.internal.ws.api.WSBinding;
import com.sun.xml.internal.ws.api.WSService;
import com.sun.xml.internal.ws.api.addressing.WSEndpointReference;
import com.sun.xml.internal.ws.api.client.ServiceInterceptor;
import com.sun.xml.internal.ws.api.client.ServiceInterceptorFactory;
import com.sun.xml.internal.ws.api.model.SEIModel;
import com.sun.xml.internal.ws.api.pipe.*;
import com.sun.xml.internal.ws.api.server.Container;
import com.sun.xml.internal.ws.api.server.ContainerResolver;
import com.sun.xml.internal.ws.api.wsdl.parser.WSDLParserExtension;
import com.sun.xml.internal.ws.binding.BindingImpl;
import com.sun.xml.internal.ws.binding.WebServiceFeatureList;
import com.sun.xml.internal.ws.client.HandlerConfigurator.AnnotationConfigurator;
import com.sun.xml.internal.ws.client.HandlerConfigurator.HandlerResolverImpl;
import com.sun.xml.internal.ws.client.sei.SEIStub;
import com.sun.xml.internal.ws.developer.WSBindingProvider;
import com.sun.xml.internal.ws.developer.UsesJAXBContextFeature;
import com.sun.xml.internal.ws.model.AbstractSEIModelImpl;
import com.sun.xml.internal.ws.model.RuntimeModeler;
import com.sun.xml.internal.ws.model.SOAPSEIModel;
import com.sun.xml.internal.ws.model.wsdl.WSDLModelImpl;
import com.sun.xml.internal.ws.model.wsdl.WSDLPortImpl;
import com.sun.xml.internal.ws.model.wsdl.WSDLServiceImpl;
import com.sun.xml.internal.ws.resources.ClientMessages;
import com.sun.xml.internal.ws.resources.DispatchMessages;
import com.sun.xml.internal.ws.resources.ProviderApiMessages;
import com.sun.xml.internal.ws.util.JAXWSUtils;
import com.sun.xml.internal.ws.util.ServiceConfigurationError;
import com.sun.xml.internal.ws.util.ServiceFinder;
import static com.sun.xml.internal.ws.util.xml.XmlUtil.createDefaultCatalogResolver;
import com.sun.xml.internal.ws.wsdl.parser.RuntimeWSDLParser;
import org.xml.sax.SAXException;
import javax.jws.HandlerChain;
import javax.jws.WebService;
import javax.xml.bind.JAXBContext;
import javax.xml.namespace.QName;
import javax.xml.stream.XMLStreamException;
import javax.xml.transform.Source;
import javax.xml.transform.stream.StreamSource;
import javax.xml.ws.*;
import javax.xml.ws.handler.HandlerResolver;
import javax.xml.ws.soap.AddressingFeature;
import java.io.IOException;
import java.lang.reflect.Proxy;
import java.net.MalformedURLException;
import java.net.URL;
import java.security.AccessController;
import java.security.PrivilegedAction;
import java.util.*;
import java.util.concurrent.Executor;
import java.util.concurrent.Executors;
import java.util.concurrent.ThreadFactory;
public class WSServiceDelegate extends WSService {
private final Map<QName, PortInfo> ports = new HashMap<QName, PortInfo>();
private @NotNull HandlerConfigurator handlerConfigurator = new HandlerResolverImpl(null);
private final Class<? extends Service> serviceClass;
private final @NotNull QName serviceName;
private final Map<QName,SEIPortInfo> seiContext = new HashMap<QName,SEIPortInfo>();
private Executor executor;
private @Nullable WSDLServiceImpl wsdlService;
private final Container container;
final @NotNull ServiceInterceptor serviceInterceptor;
public WSServiceDelegate(URL wsdlDocumentLocation, QName serviceName, Class<? extends Service> serviceClass) {
this(
wsdlDocumentLocation==null ? null : new StreamSource(wsdlDocumentLocation.toExternalForm()),
serviceName,serviceClass);
}
public WSServiceDelegate(@Nullable Source wsdl, @NotNull QName serviceName, @NotNull final Class<? extends Service> serviceClass) {
if (serviceName == null)
throw new WebServiceException(ClientMessages.INVALID_SERVICE_NAME_NULL(serviceName));
InitParams initParams = INIT_PARAMS.get();
INIT_PARAMS.set(null); if(initParams==null) initParams = EMPTY_PARAMS;
this.serviceName = serviceName;
this.serviceClass = serviceClass;
Container tContainer = initParams.getContainer()!=null ? initParams.getContainer() : ContainerResolver.getInstance().getContainer();
if (tContainer == Container.NONE) {
tContainer = new ClientContainer();
}
this.container = tContainer;
ServiceInterceptor interceptor = ServiceInterceptorFactory.load(this, Thread.currentThread().getContextClassLoader());
ServiceInterceptor si = container.getSPI(ServiceInterceptor.class);
if (si != null) {
interceptor = ServiceInterceptor.aggregate(interceptor, si);
}
this.serviceInterceptor = interceptor;
if(wsdl == null){
if(serviceClass != Service.class){
WebServiceClient wsClient = AccessController.doPrivileged(new PrivilegedAction<WebServiceClient>() {
public WebServiceClient run() {
return serviceClass.getAnnotation(WebServiceClient.class);
}
});
String wsdlLocation = wsClient.wsdlLocation();
wsdlLocation = JAXWSUtils.absolutize(JAXWSUtils.getFileOrURLName(wsdlLocation));
wsdl = new StreamSource(wsdlLocation);
}
}
WSDLServiceImpl service=null;
if (wsdl != null) {
try {
URL url = wsdl.getSystemId()==null ? null : new URL(wsdl.getSystemId());
WSDLModelImpl model = parseWSDL(url, wsdl);
service = model.getService(this.serviceName);
if (service == null)
throw new WebServiceException(
ClientMessages.INVALID_SERVICE_NAME(this.serviceName,
buildNameList(model.getServices().keySet())));
for (WSDLPortImpl port : service.getPorts())
ports.put(port.getName(), new PortInfo(this, port));
} catch (MalformedURLException e) {
throw new WebServiceException(ClientMessages.INVALID_WSDL_URL(wsdl.getSystemId()));
}
}
this.wsdlService = service;
if (serviceClass != Service.class) {
HandlerChain handlerChain =
AccessController.doPrivileged(new PrivilegedAction<HandlerChain>() {
public HandlerChain run() {
return serviceClass.getAnnotation(HandlerChain.class);
}
});
if (handlerChain != null)
handlerConfigurator = new AnnotationConfigurator(this);
}
}
private WSDLModelImpl parseWSDL(URL wsdlDocumentLocation, Source wsdlSource) {
try {
return RuntimeWSDLParser.parse(wsdlDocumentLocation, wsdlSource, createDefaultCatalogResolver(),
true, getContainer(), ServiceFinder.find(WSDLParserExtension.class).toArray());
} catch (IOException e) {
throw new WebServiceException(e);
} catch (XMLStreamException e) {
throw new WebServiceException(e);
} catch (SAXException e) {
throw new WebServiceException(e);
} catch (ServiceConfigurationError e) {
throw new WebServiceException(e);
}
}
public Executor getExecutor() {
if (executor != null) {
return executor;
} else
executor = Executors.newCachedThreadPool(new DaemonThreadFactory());
return executor;
}
public void setExecutor(Executor executor) {
this.executor = executor;
}
public HandlerResolver getHandlerResolver() {
return handlerConfigurator.getResolver();
}
final HandlerConfigurator getHandlerConfigurator() {
return handlerConfigurator;
}
public void setHandlerResolver(HandlerResolver resolver) {
handlerConfigurator = new HandlerResolverImpl(resolver);
}
public <T> T getPort(QName portName, Class<T> portInterface) throws WebServiceException {
return getPort(portName, portInterface, EMPTY_FEATURES);
}
public <T> T getPort(QName portName, Class<T> portInterface, WebServiceFeature... features) {
if (portName == null || portInterface == null)
throw new IllegalArgumentException();
WSDLServiceImpl tWsdlService = this.wsdlService;
if (tWsdlService == null) {
tWsdlService = getWSDLModelfromSEI(portInterface);
if (tWsdlService == null) {
throw new WebServiceException(ProviderApiMessages.NO_WSDL_NO_PORT(portInterface.getName()));
}
}
WSDLPortImpl portModel = getPortModel(tWsdlService, portName);
return getPort(portModel.getEPR(), portName, portInterface, features);
}
public <T> T getPort(EndpointReference epr, Class<T> portInterface, WebServiceFeature... features) {
return getPort(WSEndpointReference.create(epr),portInterface,features);
}
public <T> T getPort(WSEndpointReference wsepr, Class<T> portInterface, WebServiceFeature... features) {
QName portTypeName = RuntimeModeler.getPortTypeName(portInterface);
QName portName = getPortNameFromEPR(wsepr, portTypeName);
return getPort(wsepr,portName,portInterface,features);
}
private <T> T getPort(WSEndpointReference wsepr, QName portName, Class<T> portInterface,
WebServiceFeature... features) {
SEIPortInfo spi = addSEI(portName, portInterface, features);
return createEndpointIFBaseProxy(wsepr,portName,portInterface,features, spi);
}
public <T> T getPort(Class<T> portInterface, WebServiceFeature... features) {
QName portTypeName = RuntimeModeler.getPortTypeName(portInterface);
WSDLServiceImpl wsdlService = this.wsdlService;
if(wsdlService == null) {
wsdlService = getWSDLModelfromSEI(portInterface);
if(wsdlService == null) {
throw new WebServiceException(ProviderApiMessages.NO_WSDL_NO_PORT(portInterface.getName()));
}
}
WSDLPortImpl port = wsdlService.getMatchingPort(portTypeName);
if (port == null)
throw new WebServiceException(ClientMessages.UNDEFINED_PORT_TYPE(portTypeName));
QName portName = port.getName();
return getPort(portName, portInterface,features);
}
public <T> T getPort(Class<T> portInterface) throws WebServiceException {
return getPort(portInterface, EMPTY_FEATURES);
}
public void addPort(QName portName, String bindingId, String endpointAddress) throws WebServiceException {
if (!ports.containsKey(portName)) {
BindingID bid = (bindingId == null) ? BindingID.SOAP11_HTTP : BindingID.parse(bindingId);
ports.put(portName,
new PortInfo(this, (endpointAddress == null) ? null :
EndpointAddress.create(endpointAddress), portName, bid));
} else
throw new WebServiceException(DispatchMessages.DUPLICATE_PORT(portName.toString()));
}
public <T> Dispatch<T> createDispatch(QName portName, Class<T> aClass, Service.Mode mode) throws WebServiceException {
return createDispatch(portName, aClass, mode, EMPTY_FEATURES);
}
@Override
public <T> Dispatch<T> createDispatch(QName portName, WSEndpointReference wsepr, Class<T> aClass, Service.Mode mode, WebServiceFeature... features) {
PortInfo port = safeGetPort(portName);
BindingImpl binding = port.createBinding(features,null);
Dispatch<T> dispatch = Stubs.createDispatch(portName, this, binding, aClass, mode, createPipeline(port, binding), wsepr);
serviceInterceptor.postCreateDispatch((WSBindingProvider) dispatch);
return dispatch;
}
public <T> Dispatch<T> createDispatch(QName portName, Class<T> aClass, Service.Mode mode, WebServiceFeature... features) {
WebServiceFeatureList featureList = new WebServiceFeatureList(features);
WSEndpointReference wsepr = null;
if(featureList.isEnabled(AddressingFeature.class) && wsdlService != null && wsdlService.get(portName) != null) {
wsepr = wsdlService.get(portName).getEPR();
}
return createDispatch(portName, wsepr, aClass, mode, features);
}
public <T> Dispatch<T> createDispatch(EndpointReference endpointReference, Class<T> type, Service.Mode mode, WebServiceFeature... features) {
WSEndpointReference wsepr = new WSEndpointReference(endpointReference);
QName portName = addPortEpr(wsepr);
return createDispatch(portName, wsepr, type, mode, features);
}
public
@NotNull
PortInfo safeGetPort(QName portName) {
PortInfo port = ports.get(portName);
if (port == null) {
throw new WebServiceException(ClientMessages.INVALID_PORT_NAME(portName, buildNameList(ports.keySet())));
}
return port;
}
private StringBuilder buildNameList(Collection<QName> names) {
StringBuilder sb = new StringBuilder();
for (QName qn : names) {
if (sb.length() > 0) sb.append(',');
sb.append(qn);
}
return sb;
}
private Tube createPipeline(PortInfo portInfo, WSBinding binding) {
checkAllWSDLExtensionsUnderstood(portInfo,binding);
SEIModel seiModel = null;
if(portInfo instanceof SEIPortInfo) {
seiModel = ((SEIPortInfo)portInfo).model;
}
BindingID bindingId = portInfo.bindingId;
TubelineAssembler assembler = TubelineAssemblerFactory.create(
Thread.currentThread().getContextClassLoader(), bindingId);
if (assembler == null)
throw new WebServiceException("Unable to process bindingID=" + bindingId); return assembler.createClient(
new ClientTubeAssemblerContext(
portInfo.targetEndpoint,
portInfo.portModel,
this, binding, container,((BindingImpl)binding).createCodec(),seiModel));
}
private void checkAllWSDLExtensionsUnderstood(PortInfo port, WSBinding binding) {
if (port.portModel != null && binding.isFeatureEnabled(RespectBindingFeature.class)) {
((WSDLPortImpl) port.portModel).areRequiredExtensionsUnderstood();
}
}
public EndpointAddress getEndpointAddress(QName qName) {
return ports.get(qName).targetEndpoint;
}
public Dispatch<Object> createDispatch(QName portName, JAXBContext jaxbContext, Service.Mode mode) throws WebServiceException {
return createDispatch(portName, jaxbContext, mode, EMPTY_FEATURES);
}
@Override
public Dispatch<Object> createDispatch(QName portName, WSEndpointReference wsepr, JAXBContext jaxbContext, Service.Mode mode, WebServiceFeature... features) {
PortInfo port = safeGetPort(portName);
BindingImpl binding = port.createBinding(features,null);
Dispatch<Object> dispatch = Stubs.createJAXBDispatch(
portName, this, binding, jaxbContext, mode,
createPipeline(port, binding), wsepr);
serviceInterceptor.postCreateDispatch((WSBindingProvider)dispatch);
return dispatch;
}
@Override
public @NotNull Container getContainer() {
return container;
}
public Dispatch<Object> createDispatch(QName portName, JAXBContext jaxbContext, Service.Mode mode, WebServiceFeature... webServiceFeatures) {
WebServiceFeatureList featureList = new WebServiceFeatureList(webServiceFeatures);
WSEndpointReference wsepr = null;
if(featureList.isEnabled(AddressingFeature.class) && wsdlService != null && wsdlService.get(portName) != null) {
wsepr = wsdlService.get(portName).getEPR();
}
return createDispatch(portName, wsepr, jaxbContext, mode, webServiceFeatures);
}
public Dispatch<Object> createDispatch(EndpointReference endpointReference, JAXBContext context, Service.Mode mode, WebServiceFeature... features) {
WSEndpointReference wsepr = new WSEndpointReference(endpointReference);
QName portName = addPortEpr(wsepr);
return createDispatch(portName, wsepr, context, mode, features);
}
private QName addPortEpr(WSEndpointReference wsepr) {
if (wsepr == null)
throw new WebServiceException(ProviderApiMessages.NULL_EPR());
QName eprPortName = getPortNameFromEPR(wsepr, null);
{
PortInfo portInfo = new PortInfo(this, (wsepr.getAddress() == null) ? null : EndpointAddress.create(wsepr.getAddress()), eprPortName,
getPortModel(wsdlService, eprPortName).getBinding().getBindingId());
if (!ports.containsKey(eprPortName)) {
ports.put(eprPortName, portInfo);
}
}
return eprPortName;
}
private QName getPortNameFromEPR(@NotNull WSEndpointReference wsepr, @Nullable QName portTypeName) {
QName portName;
WSEndpointReference.Metadata metadata = wsepr.getMetaData();
QName eprServiceName = metadata.getServiceName();
QName eprPortName = metadata.getPortName();
if ((eprServiceName != null ) && !eprServiceName.equals(serviceName)) {
throw new WebServiceException("EndpointReference WSDL ServiceName differs from Service Instance WSDL Service QName.\n"
+ " The two Service QNames must match");
}
if (wsdlService == null) {
Source eprWsdlSource = metadata.getWsdlSource();
if (eprWsdlSource == null) {
throw new WebServiceException(ProviderApiMessages.NULL_WSDL());
}
try {
WSDLModelImpl eprWsdlMdl = parseWSDL(new URL(wsepr.getAddress()), eprWsdlSource);
wsdlService = eprWsdlMdl.getService(serviceName);
if (wsdlService == null)
throw new WebServiceException(ClientMessages.INVALID_SERVICE_NAME(serviceName,
buildNameList(eprWsdlMdl.getServices().keySet())));
} catch (MalformedURLException e) {
throw new WebServiceException(ClientMessages.INVALID_ADDRESS(wsepr.getAddress()));
}
}
portName = eprPortName;
if (portName == null && portTypeName != null) {
WSDLPortImpl port = wsdlService.getMatchingPort(portTypeName);
if (port == null)
throw new WebServiceException(ClientMessages.UNDEFINED_PORT_TYPE(portTypeName));
portName = port.getName();
}
if (portName == null)
throw new WebServiceException(ProviderApiMessages.NULL_PORTNAME());
if (wsdlService.get(portName) == null)
throw new WebServiceException(ClientMessages.INVALID_EPR_PORT_NAME(portName, buildWsdlPortNames()));
return portName;
}
private WSDLServiceImpl getWSDLModelfromSEI(final Class sei) {
WebService ws = AccessController.doPrivileged(new PrivilegedAction<WebService>() {
public WebService run() {
return (WebService) sei.getAnnotation(WebService.class);
}
});
if (ws == null || ws.wsdlLocation().equals(""))
return null;
String wsdlLocation = ws.wsdlLocation();
wsdlLocation = JAXWSUtils.absolutize(JAXWSUtils.getFileOrURLName(wsdlLocation));
Source wsdl = new StreamSource(wsdlLocation);
WSDLServiceImpl service = null;
try {
URL url = wsdl.getSystemId() == null ? null : new URL(wsdl.getSystemId());
WSDLModelImpl model = parseWSDL(url, wsdl);
service = model.getService(this.serviceName);
if (service == null)
throw new WebServiceException(
ClientMessages.INVALID_SERVICE_NAME(this.serviceName,
buildNameList(model.getServices().keySet())));
} catch (MalformedURLException e) {
throw new WebServiceException(ClientMessages.INVALID_WSDL_URL(wsdl.getSystemId()));
}
return service;
}
public QName getServiceName() {
return serviceName;
}
protected Class getServiceClass() {
return serviceClass;
}
public Iterator<QName> getPorts() throws WebServiceException {
if (ports.isEmpty())
throw new WebServiceException("dii.service.no.wsdl.available");
return ports.keySet().iterator();
}
public URL getWSDLDocumentLocation() {
if(wsdlService==null) return null;
try {
return new URL(wsdlService.getParent().getLocation().getSystemId());
} catch (MalformedURLException e) {
throw new AssertionError(e); }
}
private <T> T createEndpointIFBaseProxy(@Nullable WSEndpointReference epr,QName portName, Class<T> portInterface,
WebServiceFeature[] webServiceFeatures, SEIPortInfo eif) {
if (wsdlService == null)
throw new WebServiceException(ClientMessages.INVALID_SERVICE_NO_WSDL(serviceName));
if (wsdlService.get(portName)==null) {
throw new WebServiceException(
ClientMessages.INVALID_PORT_NAME(portName,buildWsdlPortNames()));
}
BindingImpl binding = eif.createBinding(webServiceFeatures,portInterface);
SEIStub pis = new SEIStub(this, binding, eif.model, createPipeline(eif, binding), epr);
T proxy = portInterface.cast(Proxy.newProxyInstance(portInterface.getClassLoader(),
new Class[]{portInterface, WSBindingProvider.class, Closeable.class}, pis));
if (serviceInterceptor != null) {
serviceInterceptor.postCreateProxy((WSBindingProvider)proxy, portInterface);
}
return proxy;
}
private StringBuilder buildWsdlPortNames() {
Set<QName> wsdlPortNames = new HashSet<QName>();
for (WSDLPortImpl port : wsdlService.getPorts())
wsdlPortNames.add(port.getName());
return buildNameList(wsdlPortNames);
}
private @NotNull WSDLPortImpl getPortModel(WSDLServiceImpl wsdlService, QName portName) {
WSDLPortImpl port = wsdlService.get(portName);
if (port == null)
throw new WebServiceException(
ClientMessages.INVALID_PORT_NAME(portName, buildWsdlPortNames()));
return port;
}
private SEIPortInfo addSEI(QName portName, Class portInterface, WebServiceFeature... features) throws WebServiceException {
boolean ownModel = useOwnSEIModel(features);
if (ownModel) {
return createSEIPortInfo(portName, portInterface, features);
}
SEIPortInfo spi = seiContext.get(portName);
if (spi == null) {
spi = createSEIPortInfo(portName, portInterface, features);
seiContext.put(spi.portName, spi);
ports.put(spi.portName, spi);
}
return spi;
}
private SEIPortInfo createSEIPortInfo(QName portName, Class portInterface, WebServiceFeature... features) {
WSDLPortImpl wsdlPort = getPortModel(wsdlService, portName);
RuntimeModeler modeler = new RuntimeModeler(portInterface, serviceName, wsdlPort, features);
modeler.setClassLoader(portInterface.getClassLoader());
modeler.setPortName(portName);
AbstractSEIModelImpl model = modeler.buildRuntimeModel();
return new SEIPortInfo(this, portInterface, (SOAPSEIModel) model, wsdlPort);
}
private boolean useOwnSEIModel(WebServiceFeature... features) {
return WebServiceFeatureList.getFeature(features, UsesJAXBContextFeature.class) != null;
}
public WSDLServiceImpl getWsdlService() {
return wsdlService;
}
class DaemonThreadFactory implements ThreadFactory {
public Thread newThread(Runnable r) {
Thread daemonThread = new Thread(r);
daemonThread.setDaemon(Boolean.TRUE);
return daemonThread;
}
}
private static final WebServiceFeature[] EMPTY_FEATURES = new WebServiceFeature[0];
}