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
26package com.sun.tools.internal.ws.processor.model;
27
28import com.sun.tools.internal.ws.processor.model.java.JavaMethod;
29import com.sun.tools.internal.ws.wsdl.document.soap.SOAPStyle;
30import com.sun.tools.internal.ws.wsdl.document.soap.SOAPUse;
31import com.sun.tools.internal.ws.wsdl.framework.Entity;
32import com.sun.xml.internal.ws.spi.db.BindingHelper;
33
34import javax.xml.namespace.QName;
35import java.util.HashSet;
36import java.util.Iterator;
37import java.util.Set;
38
39/**
40 *
41 * @author WS Development Team
42 */
43public class Operation extends ModelObject {
44
45    public Operation(Entity entity) {
46        super(entity);
47    }
48
49    public Operation(Operation operation, Entity entity){
50        this(operation._name, entity);
51        this._style = operation._style;
52        this._use = operation._use;
53        this.customizedName = operation.customizedName;
54    }
55    public Operation(QName name, Entity entity) {
56        super(entity);
57        _name = name;
58        _uniqueName = name.getLocalPart();
59        _faultNames = new HashSet<String>();
60        _faults = new HashSet<Fault>();
61    }
62
63    public QName getName() {
64        return _name;
65    }
66
67    public void setName(QName n) {
68        _name = n;
69    }
70
71    public String getUniqueName() {
72        return _uniqueName;
73    }
74
75    public void setUniqueName(String s) {
76        _uniqueName = s;
77    }
78
79    public Request getRequest() {
80        return _request;
81    }
82
83    public void setRequest(Request r) {
84        _request = r;
85    }
86
87    public Response getResponse() {
88        return _response;
89    }
90
91    public void setResponse(Response r) {
92        _response = r;
93    }
94
95    public boolean isOverloaded() {
96        return !_name.getLocalPart().equals(_uniqueName);
97    }
98
99    public void addFault(Fault f) {
100        if (_faultNames.contains(f.getName())) {
101            throw new ModelException("model.uniqueness");
102        }
103        _faultNames.add(f.getName());
104        _faults.add(f);
105    }
106
107    public Iterator<Fault> getFaults() {
108        return _faults.iterator();
109    }
110
111    public Set<Fault> getFaultsSet() {
112        return _faults;
113    }
114
115    /* serialization */
116    public void setFaultsSet(Set<Fault> s) {
117        _faults = s;
118        initializeFaultNames();
119    }
120
121    private void initializeFaultNames() {
122        _faultNames = new HashSet<String>();
123        if (_faults != null) {
124            for (Iterator iter = _faults.iterator(); iter.hasNext();) {
125                Fault f = (Fault) iter.next();
126                if (f.getName() != null && _faultNames.contains(f.getName())) {
127                    throw new ModelException("model.uniqueness");
128                }
129                _faultNames.add(f.getName());
130            }
131        }
132    }
133
134    public Iterator<Fault> getAllFaults() {
135        Set<Fault> allFaults = getAllFaultsSet();
136        return allFaults.iterator();
137    }
138
139    public Set<Fault> getAllFaultsSet() {
140        Set transSet = new HashSet();
141        transSet.addAll(_faults);
142        Iterator iter = _faults.iterator();
143        Fault fault;
144        Set tmpSet;
145        while (iter.hasNext()) {
146            tmpSet = ((Fault)iter.next()).getAllFaultsSet();
147            transSet.addAll(tmpSet);
148        }
149        return transSet;
150    }
151
152    public int getFaultCount() {
153        return _faults.size();
154    }
155
156    public Set<Block> getAllFaultBlocks(){
157        Set<Block> blocks = new HashSet<Block>();
158        Iterator faults = _faults.iterator();
159        while(faults.hasNext()){
160            Fault f = (Fault)faults.next();
161            blocks.add(f.getBlock());
162        }
163        return blocks;
164    }
165
166    public JavaMethod getJavaMethod() {
167        return _javaMethod;
168    }
169
170    public void setJavaMethod(JavaMethod i) {
171        _javaMethod = i;
172    }
173
174    public String getSOAPAction() {
175        return _soapAction;
176    }
177
178    public void setSOAPAction(String s) {
179        _soapAction = s;
180    }
181
182    public SOAPStyle getStyle() {
183        return _style;
184    }
185
186    public void setStyle(SOAPStyle s) {
187        _style = s;
188    }
189
190    public SOAPUse getUse() {
191        return _use;
192    }
193
194    public void setUse(SOAPUse u) {
195        _use = u;
196    }
197
198    public boolean isWrapped() {
199        return _isWrapped;
200    }
201
202    public void setWrapped(boolean isWrapped) {
203        _isWrapped = isWrapped;
204    }
205
206
207    public void accept(ModelVisitor visitor) throws Exception {
208        visitor.visit(this);
209    }
210
211    public void setCustomizedName(String name){
212        this.customizedName = name;
213    }
214
215    public String getCustomizedName(){
216        return customizedName;
217    }
218
219    public String getJavaMethodName(){
220        //if JavaMethod is created return the name
221        if(_javaMethod != null){
222            return _javaMethod.getName();
223        }
224
225        //return the customized operation name if any without mangling
226        if(customizedName != null){
227            return customizedName;
228        }
229
230        return BindingHelper.mangleNameToVariableName(_name.getLocalPart());
231    }
232
233    public com.sun.tools.internal.ws.wsdl.document.Operation getWSDLPortTypeOperation(){
234        return wsdlOperation;
235    }
236
237    public void setWSDLPortTypeOperation(com.sun.tools.internal.ws.wsdl.document.Operation wsdlOperation){
238        this.wsdlOperation = wsdlOperation;
239    }
240
241
242
243    private String customizedName;
244    private boolean _isWrapped = true;
245    private QName _name;
246    private String _uniqueName;
247    private Request _request;
248    private Response _response;
249    private JavaMethod _javaMethod;
250    private String _soapAction;
251    private SOAPStyle _style = SOAPStyle.DOCUMENT;
252    private SOAPUse _use = SOAPUse.LITERAL;
253    private Set<String> _faultNames;
254    private Set<Fault> _faults;
255    private com.sun.tools.internal.ws.wsdl.document.Operation wsdlOperation;
256
257}
258