1 /*
2 * Copyright (c) 1997, 2013, Oracle and/or its affiliates. All rights reserved.
3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
4 *
5 * This code is free software; you can redistribute it and/or modify it
6 * under the terms of the GNU General Public License version 2 only, as
7 * published by the Free Software Foundation. Oracle designates this
8 * particular file as subject to the "Classpath" exception as provided
9 * by Oracle in the LICENSE file that accompanied this code.
10 *
11 * This code is distributed in the hope that it will be useful, but WITHOUT
12 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
13 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
14 * version 2 for more details (a copy is included in the LICENSE file that
15 * accompanied this code).
16 *
17 * You should have received a copy of the GNU General Public License version
18 * 2 along with this work; if not, write to the Free Software Foundation,
19 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
20 *
21 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
22 * or visit www.oracle.com if you need additional information or have any
23 * questions.
24 */
25
26 package com.sun.xml.internal.ws.db;
27
28 import java.io.IOException;
29 import java.io.InputStream;
30 import java.io.OutputStream;
31 import java.lang.reflect.Method;
32 import java.util.HashMap;
33 import java.util.Map;
34
35 import javax.xml.ws.WebServiceFeature;
36
37 import com.oracle.webservices.internal.api.databinding.JavaCallInfo;
38 import com.oracle.webservices.internal.api.message.MessageContext;
39 import com.sun.xml.internal.ws.api.databinding.EndpointCallBridge;
40 import com.sun.xml.internal.ws.api.databinding.WSDLGenInfo;
41 import com.sun.xml.internal.ws.api.databinding.Databinding;
42 import com.sun.xml.internal.ws.api.databinding.DatabindingConfig;
43 import com.sun.xml.internal.ws.api.databinding.ClientCallBridge;
44 import com.sun.xml.internal.ws.api.message.Message;
45 import com.sun.xml.internal.ws.api.message.MessageContextFactory;
46 import com.sun.xml.internal.ws.api.message.Packet;
47 import com.sun.xml.internal.ws.api.model.MEP;
48 import com.sun.xml.internal.ws.api.model.SEIModel;
49 import com.sun.xml.internal.ws.api.model.WSDLOperationMapping;
50 import com.sun.xml.internal.ws.api.model.wsdl.WSDLPort;
51 import com.sun.xml.internal.ws.api.pipe.Codec;
52 import com.sun.xml.internal.ws.api.pipe.ContentType;
53 import com.sun.xml.internal.ws.binding.BindingImpl;
54 import com.sun.xml.internal.ws.client.sei.StubAsyncHandler;
55 import com.sun.xml.internal.ws.client.sei.StubHandler;
56 import com.sun.xml.internal.ws.model.AbstractSEIModelImpl;
57 import com.sun.xml.internal.ws.model.JavaMethodImpl;
58 import com.sun.xml.internal.ws.model.RuntimeModeler;
59 import com.sun.xml.internal.ws.server.sei.TieHandler;
60 import com.sun.xml.internal.ws.wsdl.ActionBasedOperationSignature;
61 import com.sun.xml.internal.ws.wsdl.DispatchException;
62 import com.sun.xml.internal.ws.wsdl.OperationDispatcher;
63
64 /**
65 * WsRuntimeImpl is the databinding processor built on SEIModel
66 *
67 * @author shih-chang.chen@oracle.com
68 */
69 public final class DatabindingImpl implements Databinding {
70
71 AbstractSEIModelImpl seiModel;
72 Map<Method, StubHandler> stubHandlers;
73 // QNameMap<TieHandler> wsdlOpMap = new QNameMap<TieHandler>();
74 Map<JavaMethodImpl, TieHandler> wsdlOpMap = new HashMap<JavaMethodImpl, TieHandler>();
75 Map<Method, TieHandler> tieHandlers = new HashMap<Method, TieHandler>();
76 OperationDispatcher operationDispatcher;
77 OperationDispatcher operationDispatcherNoWsdl;
78 boolean clientConfig = false;
79 Codec codec;
80 MessageContextFactory packetFactory = null;
81
82 public DatabindingImpl(DatabindingProviderImpl p, DatabindingConfig config) {
83 RuntimeModeler modeler = new RuntimeModeler(config);
84 modeler.setClassLoader(config.getClassLoader());
85 seiModel = modeler.buildRuntimeModel();
86 WSDLPort wsdlport = config.getWsdlPort();
87 packetFactory = new MessageContextFactory(seiModel.getWSBinding().getFeatures());
88 clientConfig = isClientConfig(config);
89 if (clientConfig) {
90 initStubHandlers();
91 }
92 seiModel.setDatabinding(this);
93 if (wsdlport != null) {
94 freeze(wsdlport);
95 }
96 if (operationDispatcher == null) {
97 operationDispatcherNoWsdl = new OperationDispatcher(null, seiModel.getWSBinding(), seiModel);
98 }
99 // if(!clientConfig) {
100 for (JavaMethodImpl jm : seiModel.getJavaMethods()) {
101 if (!jm.isAsync()) {
102 TieHandler th = new TieHandler(jm, seiModel.getWSBinding(), packetFactory);
103 wsdlOpMap.put(jm, th);
104 tieHandlers.put(th.getMethod(), th);
105 }
106 }
107 // }
108 }
109
110 //TODO isClientConfig
111 private boolean isClientConfig(DatabindingConfig config) {
112 if (config.getContractClass() == null) {
113 return false;
114 }
115 if (!config.getContractClass().isInterface()) {
116 return false;
117 }
118 return (config.getEndpointClass() == null || config.getEndpointClass().isInterface());
119 }
120 //TODO fix freeze
121
122 public void freeze(WSDLPort port) {
123 if (clientConfig) {
124 return;
125 }
126 synchronized(this) {
127 if (operationDispatcher == null) {
128 operationDispatcher = (port == null) ? null : new OperationDispatcher(port, seiModel.getWSBinding(), seiModel);
129 }
130 }
131 }
132
133 public SEIModel getModel() {
134 return seiModel;
135 }
136 //Refactored from SEIStub
137
138 private void initStubHandlers() {
139 stubHandlers = new HashMap<Method, StubHandler>();
140 Map<ActionBasedOperationSignature, JavaMethodImpl> syncs = new HashMap<ActionBasedOperationSignature, JavaMethodImpl>();
141 // fill in methodHandlers.
142 // first fill in sychronized versions
143 for (JavaMethodImpl m : seiModel.getJavaMethods()) {
144 if (!m.getMEP().isAsync) {
145 StubHandler handler = new StubHandler(m, packetFactory);
146 syncs.put(m.getOperationSignature(), m);
147 stubHandlers.put(m.getMethod(), handler);
148 }
149 }
150 for (JavaMethodImpl jm : seiModel.getJavaMethods()) {
151 JavaMethodImpl sync = syncs.get(jm.getOperationSignature());
152 if (jm.getMEP() == MEP.ASYNC_CALLBACK || jm.getMEP() == MEP.ASYNC_POLL) {
153 Method m = jm.getMethod();
154 StubAsyncHandler handler = new StubAsyncHandler(jm, sync, packetFactory);
155 stubHandlers.put(m, handler);
156 }
157 }
158 }
159
160 JavaMethodImpl resolveJavaMethod(Packet req) throws DispatchException {
161 WSDLOperationMapping m = req.getWSDLOperationMapping();
162 if (m == null) {
163 synchronized (this) {
164 m = (operationDispatcher != null)
165 ? operationDispatcher.getWSDLOperationMapping(req)
166 : operationDispatcherNoWsdl.getWSDLOperationMapping(req);
167 }
168 }
169 return (JavaMethodImpl) m.getJavaMethod();
170 }
171
172 public JavaCallInfo deserializeRequest(Packet req) {
173 com.sun.xml.internal.ws.api.databinding.JavaCallInfo call = new com.sun.xml.internal.ws.api.databinding.JavaCallInfo();
174 try {
175 JavaMethodImpl wsdlOp = resolveJavaMethod(req);
176 TieHandler tie = wsdlOpMap.get(wsdlOp);
177 call.setMethod(tie.getMethod());
178 Object[] args = tie.readRequest(req.getMessage());
179 call.setParameters(args);
180 } catch (DispatchException e) {
181 call.setException(e);
182 }
183 return call;
184 }
185
186 public JavaCallInfo deserializeResponse(Packet res, JavaCallInfo call) {
187 StubHandler stubHandler = stubHandlers.get(call.getMethod());
188 try {
189 return stubHandler.readResponse(res, call);
190 } catch (Throwable e) {
191 call.setException(e);
192 return call;
193 }
194 }
195
196 public WebServiceFeature[] getFeatures() {
197 // TODO Auto-generated method stub
198 return null;
199 }
200
201 @Override
202 public Packet serializeRequest(JavaCallInfo call) {
203 StubHandler stubHandler = stubHandlers.get(call.getMethod());
204 Packet p = stubHandler.createRequestPacket(call);
205 p.setState(Packet.State.ClientRequest);
206 return p;
207 }
208
209 @Override
210 public Packet serializeResponse(JavaCallInfo call) {
211 Method method = call.getMethod();
212 Message message = null;
213 if (method != null) {
214 TieHandler th = tieHandlers.get(method);
215 if (th != null) {
216 return th.serializeResponse(call);
217 }
218 }
219 if (call.getException() instanceof DispatchException) {
220 message = ((DispatchException) call.getException()).fault;
221 }
222 Packet p = (Packet) packetFactory.createContext(message);
223 p.setState(Packet.State.ServerResponse);
224 return p;
225 }
226
227 @Override
228 public ClientCallBridge getClientBridge(Method method) {
229 return stubHandlers.get(method);
230 }
231
232 @Override
233 public void generateWSDL(WSDLGenInfo info) {
234 com.sun.xml.internal.ws.wsdl.writer.WSDLGenerator wsdlGen = new com.sun.xml.internal.ws.wsdl.writer.WSDLGenerator(
235 seiModel,
236 info.getWsdlResolver(),
237 seiModel.getWSBinding(),
238 info.getContainer(), seiModel.getEndpointClass(),
239 info.isInlineSchemas(),
240 info.isSecureXmlProcessingDisabled(),
241 info.getExtensions());
242 wsdlGen.doGeneration();
243 }
244
245 @Override
246 public EndpointCallBridge getEndpointBridge(Packet req) throws DispatchException {
247 JavaMethodImpl wsdlOp = resolveJavaMethod(req);
248 return wsdlOpMap.get(wsdlOp);
249 }
250
251 Codec getCodec() {
252 if (codec == null) {
253 codec = ((BindingImpl) seiModel.getWSBinding()).createCodec();
254 }
255 return codec;
256 }
257
258 @Override
259 public ContentType encode(Packet packet, OutputStream out) throws IOException {
260 return getCodec().encode(packet, out);
261 }
262
263 @Override
264 public void decode(InputStream in, String ct, Packet p) throws IOException {
265 getCodec().decode(in, ct, p);
266 }
267
268 @Override
269 public com.oracle.webservices.internal.api.databinding.JavaCallInfo createJavaCallInfo(Method method, Object[] args) {
270 return new com.sun.xml.internal.ws.api.databinding.JavaCallInfo(method, args);
271 }
272
273 @Override
274 public com.oracle.webservices.internal.api.databinding.JavaCallInfo deserializeResponse(
275 MessageContext message, com.oracle.webservices.internal.api.databinding.JavaCallInfo call) {
276 return deserializeResponse((Packet) message, (JavaCallInfo) call);
277 }
278
279 @Override
280 public com.oracle.webservices.internal.api.databinding.JavaCallInfo deserializeRequest(MessageContext message) {
281 return deserializeRequest((Packet) message);
282 }
283
284 @Override
285 public MessageContextFactory getMessageContextFactory() {
286 return packetFactory;
287 }
288 }