/*
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
*
* Copyright (c) 1997-2013 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.db;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.lang.reflect.Method;
import java.util.HashMap;
import java.util.Map;
import javax.xml.ws.WebServiceFeature;
import com.oracle.webservices.api.databinding.JavaCallInfo;
import com.oracle.webservices.api.message.MessageContext;
import com.sun.xml.ws.api.databinding.EndpointCallBridge;
import com.sun.xml.ws.api.databinding.WSDLGenInfo;
import com.sun.xml.ws.api.databinding.Databinding;
import com.sun.xml.ws.api.databinding.DatabindingConfig;
import com.sun.xml.ws.api.databinding.ClientCallBridge;
import com.sun.xml.ws.api.message.Message;
import com.sun.xml.ws.api.message.MessageContextFactory;
import com.sun.xml.ws.api.message.Packet;
import com.sun.xml.ws.api.model.MEP;
import com.sun.xml.ws.api.model.SEIModel;
import com.sun.xml.ws.api.model.WSDLOperationMapping;
import com.sun.xml.ws.api.model.wsdl.WSDLPort;
import com.sun.xml.ws.api.pipe.Codec;
import com.sun.xml.ws.api.pipe.ContentType;
import com.sun.xml.ws.binding.BindingImpl;
import com.sun.xml.ws.client.sei.StubAsyncHandler;
import com.sun.xml.ws.client.sei.StubHandler;
import com.sun.xml.ws.model.AbstractSEIModelImpl;
import com.sun.xml.ws.model.JavaMethodImpl;
import com.sun.xml.ws.model.RuntimeModeler;
import com.sun.xml.ws.server.sei.TieHandler;
import com.sun.xml.ws.wsdl.ActionBasedOperationSignature;
import com.sun.xml.ws.wsdl.DispatchException;
import com.sun.xml.ws.wsdl.OperationDispatcher;
/**
* WsRuntimeImpl is the databinding processor built on SEIModel
*
* @author shih-chang.chen@oracle.com
*/
public final class DatabindingImpl implements Databinding {
AbstractSEIModelImpl seiModel;
Map<Method, StubHandler> stubHandlers;
// QNameMap<TieHandler> wsdlOpMap = new QNameMap<TieHandler>();
Map<JavaMethodImpl, TieHandler> wsdlOpMap = new HashMap<JavaMethodImpl, TieHandler>();
Map<Method, TieHandler> tieHandlers = new HashMap<Method, TieHandler>();
OperationDispatcher operationDispatcher;
OperationDispatcher operationDispatcherNoWsdl;
boolean clientConfig = false;
Codec codec;
MessageContextFactory packetFactory = null;
public DatabindingImpl(DatabindingProviderImpl p, DatabindingConfig config) {
RuntimeModeler modeler = new RuntimeModeler(config);
modeler.setClassLoader(config.getClassLoader());
seiModel = modeler.buildRuntimeModel();
WSDLPort wsdlport = config.getWsdlPort();
packetFactory = new MessageContextFactory(seiModel.getWSBinding().getFeatures());
clientConfig = isClientConfig(config);
if (clientConfig) {
initStubHandlers();
}
seiModel.setDatabinding(this);
if (wsdlport != null) {
freeze(wsdlport);
}
if (operationDispatcher == null) {
operationDispatcherNoWsdl = new OperationDispatcher(null, seiModel.getWSBinding(), seiModel);
}
// if(!clientConfig) {
for (JavaMethodImpl jm : seiModel.getJavaMethods()) {
if (!jm.isAsync()) {
TieHandler th = new TieHandler(jm, seiModel.getWSBinding(), packetFactory);
wsdlOpMap.put(jm, th);
tieHandlers.put(th.getMethod(), th);
}
}
// }
}
//TODO isClientConfig
private boolean isClientConfig(DatabindingConfig config) {
if (config.getContractClass() == null) {
return false;
}
if (!config.getContractClass().isInterface()) {
return false;
}
return (config.getEndpointClass() == null || config.getEndpointClass().isInterface());
}
//TODO fix freeze
public void freeze(WSDLPort port) {
if (clientConfig) {
return;
}
synchronized(this) {
if (operationDispatcher == null) {
operationDispatcher = (port == null) ? null : new OperationDispatcher(port, seiModel.getWSBinding(), seiModel);
}
}
}
public SEIModel getModel() {
return seiModel;
}
//Refactored from SEIStub
private void initStubHandlers() {
stubHandlers = new HashMap<Method, StubHandler>();
Map<ActionBasedOperationSignature, JavaMethodImpl> syncs = new HashMap<ActionBasedOperationSignature, JavaMethodImpl>();
// fill in methodHandlers.
// first fill in sychronized versions
for (JavaMethodImpl m : seiModel.getJavaMethods()) {
if (!m.getMEP().isAsync) {
StubHandler handler = new StubHandler(m, packetFactory);
syncs.put(m.getOperationSignature(), m);
stubHandlers.put(m.getMethod(), handler);
}
}
for (JavaMethodImpl jm : seiModel.getJavaMethods()) {
JavaMethodImpl sync = syncs.get(jm.getOperationSignature());
if (jm.getMEP() == MEP.ASYNC_CALLBACK || jm.getMEP() == MEP.ASYNC_POLL) {
Method m = jm.getMethod();
StubAsyncHandler handler = new StubAsyncHandler(jm, sync, packetFactory);
stubHandlers.put(m, handler);
}
}
}
JavaMethodImpl resolveJavaMethod(Packet req) throws DispatchException {
WSDLOperationMapping m = req.getWSDLOperationMapping();
if (m == null) {
synchronized (this) {
m = (operationDispatcher != null)
? operationDispatcher.getWSDLOperationMapping(req)
: operationDispatcherNoWsdl.getWSDLOperationMapping(req);
}
}
return (JavaMethodImpl) m.getJavaMethod();
}
public JavaCallInfo deserializeRequest(Packet req) {
com.sun.xml.ws.api.databinding.JavaCallInfo call = new com.sun.xml.ws.api.databinding.JavaCallInfo();
try {
JavaMethodImpl wsdlOp = resolveJavaMethod(req);
TieHandler tie = wsdlOpMap.get(wsdlOp);
call.setMethod(tie.getMethod());
Object[] args = tie.readRequest(req.getMessage());
call.setParameters(args);
} catch (DispatchException e) {
call.setException(e);
}
return call;
}
public JavaCallInfo deserializeResponse(Packet res, JavaCallInfo call) {
StubHandler stubHandler = stubHandlers.get(call.getMethod());
try {
return stubHandler.readResponse(res, call);
} catch (Throwable e) {
call.setException(e);
return call;
}
}
public WebServiceFeature[] getFeatures() {
// TODO Auto-generated method stub
return null;
}
@Override
public Packet serializeRequest(JavaCallInfo call) {
StubHandler stubHandler = stubHandlers.get(call.getMethod());
Packet p = stubHandler.createRequestPacket(call);
p.setState(Packet.State.ClientRequest);
return p;
}
@Override
public Packet serializeResponse(JavaCallInfo call) {
Method method = call.getMethod();
Message message = null;
if (method != null) {
TieHandler th = tieHandlers.get(method);
if (th != null) {
return th.serializeResponse(call);
}
}
if (call.getException() instanceof DispatchException) {
message = ((DispatchException) call.getException()).fault;
}
Packet p = (Packet) packetFactory.createContext(message);
p.setState(Packet.State.ServerResponse);
return p;
}
@Override
public ClientCallBridge getClientBridge(Method method) {
return stubHandlers.get(method);
}
@Override
public void generateWSDL(WSDLGenInfo info) {
com.sun.xml.ws.wsdl.writer.WSDLGenerator wsdlGen = new com.sun.xml.ws.wsdl.writer.WSDLGenerator(
seiModel,
info.getWsdlResolver(),
seiModel.getWSBinding(),
info.getContainer(), seiModel.getEndpointClass(),
info.isInlineSchemas(),
info.isSecureXmlProcessingDisabled(),
info.getExtensions());
wsdlGen.doGeneration();
}
@Override
public EndpointCallBridge getEndpointBridge(Packet req) throws DispatchException {
JavaMethodImpl wsdlOp = resolveJavaMethod(req);
return wsdlOpMap.get(wsdlOp);
}
Codec getCodec() {
if (codec == null) {
codec = ((BindingImpl) seiModel.getWSBinding()).createCodec();
}
return codec;
}
@Override
public ContentType encode(Packet packet, OutputStream out) throws IOException {
return getCodec().encode(packet, out);
}
@Override
public void decode(InputStream in, String ct, Packet p) throws IOException {
getCodec().decode(in, ct, p);
}
@Override
public com.oracle.webservices.api.databinding.JavaCallInfo createJavaCallInfo(Method method, Object[] args) {
return new com.sun.xml.ws.api.databinding.JavaCallInfo(method, args);
}
@Override
public com.oracle.webservices.api.databinding.JavaCallInfo deserializeResponse(
MessageContext message, com.oracle.webservices.api.databinding.JavaCallInfo call) {
return deserializeResponse((Packet) message, (JavaCallInfo) call);
}
@Override
public com.oracle.webservices.api.databinding.JavaCallInfo deserializeRequest(MessageContext message) {
return deserializeRequest((Packet) message);
}
@Override
public MessageContextFactory getMessageContextFactory() {
return packetFactory;
}
}