ohair@286: /*
alanb@368:  * Copyright (c) 1997, 2012, Oracle and/or its affiliates. All rights reserved.
ohair@286:  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
ohair@286:  *
ohair@286:  * This code is free software; you can redistribute it and/or modify it
ohair@286:  * under the terms of the GNU General Public License version 2 only, as
ohair@286:  * published by the Free Software Foundation.  Oracle designates this
ohair@286:  * particular file as subject to the "Classpath" exception as provided
ohair@286:  * by Oracle in the LICENSE file that accompanied this code.
ohair@286:  *
ohair@286:  * This code is distributed in the hope that it will be useful, but WITHOUT
ohair@286:  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
ohair@286:  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
ohair@286:  * version 2 for more details (a copy is included in the LICENSE file that
ohair@286:  * accompanied this code).
ohair@286:  *
ohair@286:  * You should have received a copy of the GNU General Public License version
ohair@286:  * 2 along with this work; if not, write to the Free Software Foundation,
ohair@286:  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
ohair@286:  *
ohair@286:  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
ohair@286:  * or visit www.oracle.com if you need additional information or have any
ohair@286:  * questions.
ohair@286:  */
ohair@286: 
ohair@286: package com.sun.xml.internal.ws.model;
ohair@286: 
ohair@286: import com.sun.xml.internal.ws.api.model.ParameterBinding;
ohair@286: import com.sun.xml.internal.ws.binding.WebServiceFeatureList;
ohair@286: 
ohair@286: import javax.jws.WebParam.Mode;
ohair@286: import javax.xml.namespace.QName;
ohair@286: import java.util.HashSet;
ohair@286: import java.util.Iterator;
ohair@286: import java.util.Set;
ohair@286: 
ohair@286: /**
ohair@286:  * Creates SOAP specific RuntimeModel
ohair@286:  *
ohair@286:  * @author Vivek Pandey
ohair@286:  */
ohair@286: public class SOAPSEIModel extends AbstractSEIModelImpl {
ohair@286: 
ohair@286:     public SOAPSEIModel(WebServiceFeatureList features) {
ohair@286:         super(features);
ohair@286:     }
ohair@286: 
ohair@286:     @Override
ohair@286:     protected void populateMaps() {
ohair@286:         int emptyBodyCount = 0;
ohair@286:         for(JavaMethodImpl jm : getJavaMethods()){
ohair@286:             put(jm.getMethod(), jm);
ohair@286:             boolean bodyFound = false;
ohair@286:             for(ParameterImpl p:jm.getRequestParameters()){
ohair@286:                 ParameterBinding binding = p.getBinding();
ohair@286:                 if(binding.isBody()){
ohair@286:                     put(p.getName(), jm);
ohair@286:                     bodyFound = true;
ohair@286:                 }
ohair@286:             }
ohair@286:             if(!bodyFound){
ohair@286:                 put(emptyBodyName, jm);
ohair@286: //                System.out.println("added empty body for: "+jm.getMethod().getName());
ohair@286:                 emptyBodyCount++;
ohair@286:             }
ohair@286:         }
ohair@286:         if(emptyBodyCount > 1){
ohair@286:             //TODO throw exception
ohair@286: //            System.out.println("Error: Unqiue signature violation - more than 1 empty body!");
ohair@286:         }
ohair@286:     }
ohair@286: 
ohair@286:     public Set<QName> getKnownHeaders() {
ohair@286:         Set<QName> headers = new HashSet<QName>();
ohair@286:         for (JavaMethodImpl method : getJavaMethods()) {
ohair@286:             // fill in request headers
ohair@286:             Iterator<ParameterImpl> params = method.getRequestParameters().iterator();
ohair@286:             fillHeaders(params, headers, Mode.IN);
ohair@286: 
ohair@286:             // fill in response headers
ohair@286:             params = method.getResponseParameters().iterator();
ohair@286:             fillHeaders(params, headers, Mode.OUT);
ohair@286:         }
ohair@286:         return headers;
ohair@286:     }
ohair@286: 
ohair@286:     /**
ohair@286:      * @param params
ohair@286:      * @param headers
ohair@286:      */
ohair@286:     private void fillHeaders(Iterator<ParameterImpl> params, Set<QName> headers, Mode mode) {
ohair@286:         while (params.hasNext()) {
ohair@286:             ParameterImpl param = params.next();
ohair@286:             ParameterBinding binding = (mode == Mode.IN)?param.getInBinding():param.getOutBinding();
ohair@286:             QName name = param.getName();
ohair@286:             if (binding.isHeader() && !headers.contains(name)) {
ohair@286:                 headers.add(name);
ohair@286:             }
ohair@286:         }
ohair@286:     }
ohair@286: }