aoqi@0: /* aoqi@0: * Copyright (c) 1997, 2013, Oracle and/or its affiliates. All rights reserved. aoqi@0: * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. aoqi@0: * aoqi@0: * This code is free software; you can redistribute it and/or modify it aoqi@0: * under the terms of the GNU General Public License version 2 only, as aoqi@0: * published by the Free Software Foundation. Oracle designates this aoqi@0: * particular file as subject to the "Classpath" exception as provided aoqi@0: * by Oracle in the LICENSE file that accompanied this code. aoqi@0: * aoqi@0: * This code is distributed in the hope that it will be useful, but WITHOUT aoqi@0: * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or aoqi@0: * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License aoqi@0: * version 2 for more details (a copy is included in the LICENSE file that aoqi@0: * accompanied this code). aoqi@0: * aoqi@0: * You should have received a copy of the GNU General Public License version aoqi@0: * 2 along with this work; if not, write to the Free Software Foundation, aoqi@0: * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. aoqi@0: * aoqi@0: * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA aoqi@0: * or visit www.oracle.com if you need additional information or have any aoqi@0: * questions. aoqi@0: */ aoqi@0: aoqi@0: package com.sun.xml.internal.ws.model.wsdl; aoqi@0: aoqi@0: import com.sun.istack.internal.NotNull; aoqi@0: import com.sun.xml.internal.ws.api.model.ParameterBinding; aoqi@0: import com.sun.xml.internal.ws.api.model.wsdl.WSDLMessage; aoqi@0: import com.sun.xml.internal.ws.api.model.wsdl.WSDLModel; aoqi@0: import com.sun.xml.internal.ws.api.model.wsdl.WSDLOperation; aoqi@0: import com.sun.xml.internal.ws.api.model.wsdl.WSDLPortType; aoqi@0: import com.sun.xml.internal.ws.api.model.wsdl.editable.EditableWSDLBoundOperation; aoqi@0: import com.sun.xml.internal.ws.api.model.wsdl.editable.EditableWSDLBoundPortType; aoqi@0: import com.sun.xml.internal.ws.api.model.wsdl.editable.EditableWSDLMessage; aoqi@0: import com.sun.xml.internal.ws.api.model.wsdl.editable.EditableWSDLModel; aoqi@0: import com.sun.xml.internal.ws.api.model.wsdl.editable.EditableWSDLPart; aoqi@0: import com.sun.xml.internal.ws.api.model.wsdl.editable.EditableWSDLPort; aoqi@0: import com.sun.xml.internal.ws.api.model.wsdl.editable.EditableWSDLPortType; aoqi@0: import com.sun.xml.internal.ws.api.model.wsdl.editable.EditableWSDLService; aoqi@0: import com.sun.xml.internal.ws.policy.PolicyMap; aoqi@0: aoqi@0: import javax.jws.WebParam.Mode; aoqi@0: import javax.xml.namespace.QName; aoqi@0: aoqi@0: import java.util.Collections; aoqi@0: import java.util.HashMap; aoqi@0: import java.util.LinkedHashMap; aoqi@0: import java.util.Map; aoqi@0: aoqi@0: /** aoqi@0: * Implementation of {@link WSDLModel} aoqi@0: * aoqi@0: * @author Vivek Pandey aoqi@0: */ aoqi@0: public final class WSDLModelImpl extends AbstractExtensibleImpl implements EditableWSDLModel { aoqi@0: private final Map messages = new HashMap(); aoqi@0: private final Map portTypes = new HashMap(); aoqi@0: private final Map bindings = new HashMap(); aoqi@0: private final Map services = new LinkedHashMap(); aoqi@0: aoqi@0: private PolicyMap policyMap; aoqi@0: private final Map unmBindings aoqi@0: = Collections.unmodifiableMap(bindings); aoqi@0: aoqi@0: aoqi@0: public WSDLModelImpl(@NotNull String systemId) { aoqi@0: super(systemId,-1); aoqi@0: } aoqi@0: aoqi@0: /** aoqi@0: * To create {@link WSDLModelImpl} from WSDL that doesn't have a system ID. aoqi@0: */ aoqi@0: public WSDLModelImpl() { aoqi@0: super(null,-1); aoqi@0: } aoqi@0: aoqi@0: public void addMessage(EditableWSDLMessage msg){ aoqi@0: messages.put(msg.getName(), msg); aoqi@0: } aoqi@0: aoqi@0: public EditableWSDLMessage getMessage(QName name){ aoqi@0: return messages.get(name); aoqi@0: } aoqi@0: aoqi@0: public void addPortType(EditableWSDLPortType pt){ aoqi@0: portTypes.put(pt.getName(), pt); aoqi@0: } aoqi@0: aoqi@0: public EditableWSDLPortType getPortType(QName name){ aoqi@0: return portTypes.get(name); aoqi@0: } aoqi@0: aoqi@0: public void addBinding(EditableWSDLBoundPortType boundPortType){ aoqi@0: assert !bindings.containsValue(boundPortType); aoqi@0: bindings.put(boundPortType.getName(), boundPortType); aoqi@0: } aoqi@0: aoqi@0: public EditableWSDLBoundPortType getBinding(QName name){ aoqi@0: return bindings.get(name); aoqi@0: } aoqi@0: aoqi@0: public void addService(EditableWSDLService svc){ aoqi@0: services.put(svc.getName(), svc); aoqi@0: } aoqi@0: aoqi@0: public EditableWSDLService getService(QName name){ aoqi@0: return services.get(name); aoqi@0: } aoqi@0: aoqi@0: public Map getMessages() { aoqi@0: return messages; aoqi@0: } aoqi@0: aoqi@0: public @NotNull Map getPortTypes() { aoqi@0: return portTypes; aoqi@0: } aoqi@0: aoqi@0: public @NotNull Map getBindings() { aoqi@0: return unmBindings; aoqi@0: } aoqi@0: aoqi@0: public @NotNull Map getServices(){ aoqi@0: return services; aoqi@0: } aoqi@0: aoqi@0: /** aoqi@0: * Returns the first service QName from insertion order aoqi@0: */ aoqi@0: public QName getFirstServiceName(){ aoqi@0: if(services.isEmpty()) aoqi@0: return null; aoqi@0: return services.values().iterator().next().getName(); aoqi@0: } aoqi@0: aoqi@0: /** aoqi@0: * aoqi@0: * @param serviceName non-null service QName aoqi@0: * @param portName non-null port QName aoqi@0: * @return aoqi@0: * WSDLBoundOperation on success otherwise null. throws NPE if any of the parameters null aoqi@0: */ aoqi@0: public EditableWSDLBoundPortType getBinding(QName serviceName, QName portName){ aoqi@0: EditableWSDLService service = services.get(serviceName); aoqi@0: if(service != null){ aoqi@0: EditableWSDLPort port = service.get(portName); aoqi@0: if(port != null) aoqi@0: return port.getBinding(); aoqi@0: } aoqi@0: return null; aoqi@0: } aoqi@0: aoqi@0: public void finalizeRpcLitBinding(EditableWSDLBoundPortType boundPortType){ aoqi@0: assert(boundPortType != null); aoqi@0: QName portTypeName = boundPortType.getPortTypeName(); aoqi@0: if(portTypeName == null) aoqi@0: return; aoqi@0: WSDLPortType pt = portTypes.get(portTypeName); aoqi@0: if(pt == null) aoqi@0: return; aoqi@0: for (EditableWSDLBoundOperation bop : boundPortType.getBindingOperations()) { aoqi@0: WSDLOperation pto = pt.get(bop.getName().getLocalPart()); aoqi@0: WSDLMessage inMsgName = pto.getInput().getMessage(); aoqi@0: if(inMsgName == null) aoqi@0: continue; aoqi@0: EditableWSDLMessage inMsg = messages.get(inMsgName.getName()); aoqi@0: int bodyindex = 0; aoqi@0: if(inMsg != null){ aoqi@0: for(EditableWSDLPart part:inMsg.parts()){ aoqi@0: String name = part.getName(); aoqi@0: ParameterBinding pb = bop.getInputBinding(name); aoqi@0: if(pb.isBody()){ aoqi@0: part.setIndex(bodyindex++); aoqi@0: part.setBinding(pb); aoqi@0: bop.addPart(part, Mode.IN); aoqi@0: } aoqi@0: } aoqi@0: } aoqi@0: bodyindex=0; aoqi@0: if(pto.isOneWay()) aoqi@0: continue; aoqi@0: WSDLMessage outMsgName = pto.getOutput().getMessage(); aoqi@0: if(outMsgName == null) aoqi@0: continue; aoqi@0: EditableWSDLMessage outMsg = messages.get(outMsgName.getName()); aoqi@0: if(outMsg!= null){ aoqi@0: for(EditableWSDLPart part:outMsg.parts()){ aoqi@0: String name = part.getName(); aoqi@0: ParameterBinding pb = bop.getOutputBinding(name); aoqi@0: if(pb.isBody()){ aoqi@0: part.setIndex(bodyindex++); aoqi@0: part.setBinding(pb); aoqi@0: bop.addPart(part, Mode.OUT); aoqi@0: } aoqi@0: } aoqi@0: } aoqi@0: } aoqi@0: } aoqi@0: aoqi@0: /** aoqi@0: * Gives the PolicyMap associated with the WSDLModel aoqi@0: * aoqi@0: * @return PolicyMap aoqi@0: */ aoqi@0: public PolicyMap getPolicyMap() { aoqi@0: return policyMap; aoqi@0: } aoqi@0: aoqi@0: /** aoqi@0: * Set PolicyMap for the WSDLModel. aoqi@0: * @param policyMap aoqi@0: */ aoqi@0: public void setPolicyMap(PolicyMap policyMap) { aoqi@0: this.policyMap = policyMap; aoqi@0: } aoqi@0: aoqi@0: /** aoqi@0: * Invoked at the end of the model construction to fix up references, etc. aoqi@0: */ aoqi@0: public void freeze() { aoqi@0: for (EditableWSDLService service : services.values()) { aoqi@0: service.freeze(this); aoqi@0: } aoqi@0: for (EditableWSDLBoundPortType bp : bindings.values()) { aoqi@0: bp.freeze(); aoqi@0: } aoqi@0: // Enforce freeze all the portTypes referenced by this endpoints, see Bug8966673 for detail aoqi@0: for (EditableWSDLPortType pt : portTypes.values()) { aoqi@0: pt.freeze(); aoqi@0: } aoqi@0: } aoqi@0: }