001    /** 
002     * 
003     * Copyright 2004 Protique Ltd
004     * 
005     * Licensed under the Apache License, Version 2.0 (the "License"); 
006     * you may not use this file except in compliance with the License. 
007     * You may obtain a copy of the License at 
008     * 
009     * http://www.apache.org/licenses/LICENSE-2.0
010     * 
011     * Unless required by applicable law or agreed to in writing, software
012     * distributed under the License is distributed on an "AS IS" BASIS, 
013     * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 
014     * See the License for the specific language governing permissions and 
015     * limitations under the License. 
016     * 
017     **/
018    package org.codehaus.activesoap.handler.xmlbeans;
019    
020    import org.apache.xmlbeans.XmlObject;
021    import org.codehaus.activesoap.Handler;
022    import org.codehaus.activesoap.MessageExchange;
023    
024    import javax.xml.stream.XMLStreamReader;
025    import javax.xml.stream.XMLStreamWriter;
026    
027    /**
028     * Generates a REST/SOAP invocation body from a method call with arguments.
029     *
030     * @version $Revision: 1.3 $
031     */
032    public class XMLBeansOutputHandler implements Handler {
033        private Object[] args;
034    
035        public XMLBeansOutputHandler(Object[] args) {
036            this.args = args;
037        }
038    
039        public XMLBeansOutputHandler(Object argument) {
040            this(new Object[]{argument});
041        }
042    
043        public void invoke(MessageExchange exchange) throws Exception {
044            XMLStreamReader in = exchange.getIn();
045            XMLStreamWriter out = exchange.getOut();
046    
047            // TODO should we set the method name as the SOAP action?
048    
049            for (int i = 0, size = args.length; i < size; i++) {
050                Object arg = args[i];
051                if (arg instanceof XmlObject) {
052                    XMLBeansHelper.save((XmlObject) arg, out, exchange.isRepairingNamespace());
053                }
054                else {
055                    throw new IllegalArgumentException("Argument: " + (i + 1) + " is not an XMLBean: " + arg);
056                }
057            }
058        }
059    }