/*
* Copyright (c) 2005, 2006, Oracle and/or its affiliates. All rights reserved.
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
*
* This code is free software; you can redistribute it and/or modify it
* under the terms of the GNU General Public License version 2 only, as
* published by the Free Software Foundation. Oracle designates this
* particular file as subject to the "Classpath" exception as provided
* by Oracle in the LICENSE file that accompanied this code.
*
* This code is distributed in the hope that it will be useful, but WITHOUT
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
* version 2 for more details (a copy is included in the LICENSE file that
* accompanied this code).
*
* You should have received a copy of the GNU General Public License version
* 2 along with this work; if not, write to the Free Software Foundation,
* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
*
* Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
* or visit www.oracle.com if you need additional information or have any
* questions.
*/
package com.sun.xml.internal.ws.client.sei;
//import com.sun.tools.internal.ws.wsdl.document.soap.SOAPBinding;
import com.sun.istack.internal.NotNull;
import com.sun.istack.internal.Nullable;
import com.sun.xml.internal.ws.api.message.Message;
import com.sun.xml.internal.ws.api.message.Packet;
import com.sun.xml.internal.ws.api.pipe.Fiber;
import com.sun.xml.internal.ws.client.AsyncInvoker;
import com.sun.xml.internal.ws.client.AsyncResponseImpl;
import com.sun.xml.internal.ws.client.RequestContext;
import com.sun.xml.internal.ws.client.ResponseContext;
import com.sun.xml.internal.ws.fault.SOAPFaultBuilder;
import com.sun.xml.internal.ws.model.JavaMethodImpl;
import com.sun.xml.internal.ws.model.ParameterImpl;
import com.sun.xml.internal.ws.model.WrapperParameter;
import javax.jws.soap.SOAPBinding.Style;
import javax.xml.ws.AsyncHandler;
import javax.xml.ws.Response;
import javax.xml.ws.WebServiceException;
import java.util.List;
/**
* Common part between {@link CallbackMethodHandler} and {@link PollingMethodHandler}.
*
* @author Kohsuke Kawaguchi
* @author Jitendra Kotamraju
*/
abstract class AsyncMethodHandler extends SEIMethodHandler {
private final ResponseBuilder responseBuilder;
/**
* Async bean class that has setters for all out parameters
*/
private final @Nullable Class asyncBeanClass;
AsyncMethodHandler(SEIStub owner, JavaMethodImpl jm, JavaMethodImpl sync) {
super(owner, sync);
List<ParameterImpl> rp = sync.getResponseParameters();
int size = 0;
for( ParameterImpl param : rp ) {
if (param.isWrapperStyle()) {
WrapperParameter wrapParam = (WrapperParameter)param;
size += wrapParam.getWrapperChildren().size();
if (sync.getBinding().getStyle() == Style.DOCUMENT) {
// doc/asyncBeanClass - asyncBeanClass bean is in async signature
// Add 2 or more so that it is considered as async bean case
size += 2;
}
} else {
++size;
}
}
Class tempWrap = null;
if (size > 1) {
rp = jm.getResponseParameters();
for(ParameterImpl param : rp) {
if (param.isWrapperStyle()) {
WrapperParameter wrapParam = (WrapperParameter)param;
if (sync.getBinding().getStyle() == Style.DOCUMENT) {
// doc/asyncBeanClass style
tempWrap = (Class)wrapParam.getTypeReference().type;
break;
}
for(ParameterImpl p : wrapParam.getWrapperChildren()) {
if (p.getIndex() == -1) {
tempWrap = (Class)p.getTypeReference().type;
break;
}
}
if (tempWrap != null) {
break;
}
} else {
if (param.getIndex() == -1) {
tempWrap = (Class)param.getTypeReference().type;
break;
}
}
}
}
asyncBeanClass = tempWrap;
switch(size) {
case 0 :
responseBuilder = buildResponseBuilder(sync, ValueSetterFactory.NONE);
break;
case 1 :
responseBuilder = buildResponseBuilder(sync, ValueSetterFactory.SINGLE);
break;
default :
responseBuilder = buildResponseBuilder(sync, new ValueSetterFactory.AsyncBeanValueSetterFactory(asyncBeanClass));
}
}
protected final Response<Object> doInvoke(Object proxy, Object[] args, AsyncHandler handler) {
AsyncInvoker invoker = new SEIAsyncInvoker(proxy, args);
AsyncResponseImpl<Object> ft = new AsyncResponseImpl<Object>(invoker,handler);
invoker.setReceiver(ft);
// TODO: Do we set this executor on Engine and run the AsyncInvoker in this thread ?
owner.getExecutor().execute(ft);
return ft;
}
private class SEIAsyncInvoker extends AsyncInvoker {
// snapshot the context now. this is necessary to avoid concurrency issue,
// and is required by the spec
private final RequestContext rc = owner.requestContext.copy();
private final Object[] args;
SEIAsyncInvoker(Object proxy, Object[] args) {
this.args = args;
}
public void do_run () {
Packet req = new Packet(createRequestMessage(args));
req.soapAction = soapAction;
req.expectReply = !isOneWay;
req.getMessage().assertOneWay(isOneWay);
Fiber.CompletionCallback callback = new Fiber.CompletionCallback() {
public void onCompletion(@NotNull Packet response) {
responseImpl.setResponseContext(new ResponseContext(response));
Message msg = response.getMessage();
if (msg == null) {
return;
}
try {
if(msg.isFault()) {
SOAPFaultBuilder faultBuilder = SOAPFaultBuilder.create(msg);
throw faultBuilder.createException(checkedExceptions);
} else {
Object[] rargs = new Object[1];
if (asyncBeanClass != null) {
rargs[0] = asyncBeanClass.newInstance();
}
responseBuilder.readResponse(msg, rargs);
responseImpl.set(rargs[0], null);
}
} catch (Throwable t) {
if (t instanceof RuntimeException) {
if (t instanceof WebServiceException) {
responseImpl.set(null, t);
return;
}
} else if (t instanceof Exception) {
responseImpl.set(null, t);
return;
}
//its RuntimeException or some other exception resulting from user error, wrap it in
// WebServiceException
responseImpl.set(null, new WebServiceException(t));
}
}
public void onCompletion(@NotNull Throwable error) {
if (error instanceof WebServiceException) {
responseImpl.set(null, error);
} else {
//its RuntimeException or some other exception resulting from user error, wrap it in
// WebServiceException
responseImpl.set(null, new WebServiceException(error));
}
}
};
owner.doProcessAsync(req, rc, callback);
}
}
ValueGetterFactory getValueGetterFactory() {
return ValueGetterFactory.ASYNC;
}
}