diff -r 88b85470e72c -r f50545b5e2f1 src/share/jaxws_classes/com/sun/xml/internal/ws/client/sei/AsyncMethodHandler.java --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/src/share/jaxws_classes/com/sun/xml/internal/ws/client/sei/AsyncMethodHandler.java Tue Mar 06 16:09:35 2012 -0800 @@ -0,0 +1,224 @@ +/* + * Copyright (c) 1997, 2011, 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.databinding.ClientCallBridge; +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.api.pipe.FiberContextSwitchInterceptor; +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 com.sun.xml.internal.org.jvnet.ws.databinding.JavaCallInfo; + +import javax.jws.soap.SOAPBinding.Style; +import javax.xml.ws.AsyncHandler; +import javax.xml.ws.Response; +import javax.xml.ws.WebServiceException; + +import java.lang.reflect.Method; +import java.util.List; + +/** + * Common part between {@link CallbackMethodHandler} and {@link PollingMethodHandler}. + * + * @author Kohsuke Kawaguchi + * @author Jitendra Kotamraju + */ +abstract class AsyncMethodHandler extends MethodHandler { + + AsyncMethodHandler(SEIStub owner, Method m) { + super(owner, m); + } + +// 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 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 doInvoke(Object proxy, Object[] args, AsyncHandler handler) { + + AsyncInvoker invoker = new SEIAsyncInvoker(proxy, args); + invoker.setNonNullAsyncHandlerGiven(handler != null); + AsyncResponseImpl ft = new AsyncResponseImpl(invoker,handler); + invoker.setReceiver(ft); + ft.run(); + 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 () { + JavaCallInfo call = owner.databinding.createJavaCallInfo(method, args); + Packet req = (Packet)owner.databinding.serializeRequest(call); + + 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 { + Object[] rargs = new Object[1]; + JavaCallInfo call = owner.databinding.createJavaCallInfo(method, rargs); + call = owner.databinding.deserializeResponse(response, call); + if (call.getException() != null) { + throw call.getException(); + } else { + responseImpl.set(rargs[0], null); + } +// dbHandler.readResponse(response, call); +// responseImpl.set(rargs[0], null); +// 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(responseImpl, req, rc, callback); + } + } + + ValueGetterFactory getValueGetterFactory() { + return ValueGetterFactory.ASYNC; + } + +}