XmlRpcWriter.java

Go to the documentation of this file.

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.    
 */
package org.apache.xmlrpc.serializer;

import java.io.ByteArrayOutputStream;
import java.io.ObjectOutputStream;
import java.util.HashMap;
import java.util.Map;

import org.apache.xmlrpc.XmlRpcRequest;
import org.apache.xmlrpc.XmlRpcRequestConfig;
import org.apache.xmlrpc.common.TypeFactory;
import org.apache.xmlrpc.common.XmlRpcStreamConfig;
import org.apache.xmlrpc.common.XmlRpcStreamRequestConfig;
import org.xml.sax.Attributes;
import org.xml.sax.ContentHandler;
import org.xml.sax.SAXException;
import org.xml.sax.helpers.AttributesImpl;


public class XmlRpcWriter {
        public static final String EXTENSIONS_URI = "http://ws.apache.org/xmlrpc/namespaces/extensions";
        private static final Attributes ZERO_ATTRIBUTES = new AttributesImpl();
        private final XmlRpcStreamConfig config;
        private final TypeFactory typeFactory;
        private final ContentHandler handler;

        public XmlRpcWriter(XmlRpcStreamConfig pConfig, ContentHandler pHandler,
                                            TypeFactory pTypeFactory) {
                config = pConfig;
                handler = pHandler;
                typeFactory = pTypeFactory;
        }

        public void write(XmlRpcRequest pRequest) throws SAXException {
                handler.startDocument();
                boolean extensions = pRequest.getConfig().isEnabledForExtensions();
                if (extensions) {
                        handler.startPrefixMapping("ex", XmlRpcWriter.EXTENSIONS_URI);
                }
                handler.startElement("", "methodCall", "methodCall", ZERO_ATTRIBUTES);
                handler.startElement("", "methodName", "methodName", ZERO_ATTRIBUTES);
                String s = pRequest.getMethodName();
                handler.characters(s.toCharArray(), 0, s.length());
                handler.endElement("", "methodName", "methodName");
                handler.startElement("", "params", "params", ZERO_ATTRIBUTES);
                int num = pRequest.getParameterCount();
                for (int i = 0;  i < num;  i++) {
                        handler.startElement("", "param", "param", ZERO_ATTRIBUTES);
                        writeValue(pRequest.getParameter(i));
                        handler.endElement("", "param", "param");
                }
                handler.endElement("", "params", "params");
        handler.endElement("", "methodCall", "methodCall");
                if (extensions) {
                        handler.endPrefixMapping("ex");
                }
                handler.endDocument();
        }

        public void write(XmlRpcRequestConfig pConfig, Object pResult) throws SAXException {
                handler.startDocument();
                boolean extensions = pConfig.isEnabledForExtensions();
                if (extensions) {
                        handler.startPrefixMapping("ex", XmlRpcWriter.EXTENSIONS_URI);
                }
                handler.startElement("", "methodResponse", "methodResponse", ZERO_ATTRIBUTES);
                handler.startElement("", "params", "params", ZERO_ATTRIBUTES);
                handler.startElement("", "param", "param", ZERO_ATTRIBUTES);
                writeValue(pResult);
                handler.endElement("", "param", "param");
                handler.endElement("", "params", "params");
                handler.endElement("", "methodResponse", "methodResponse");
                if (extensions) {
                        handler.endPrefixMapping("ex");
                }
                handler.endDocument();
        }

    public void write(XmlRpcRequestConfig pConfig, int pCode, String pMessage) throws SAXException {
        write(pConfig, pCode, pMessage, null);
    }

        public void write(XmlRpcRequestConfig pConfig, int pCode, String pMessage,
            Throwable pThrowable) throws SAXException {
                handler.startDocument();
                boolean extensions = pConfig.isEnabledForExtensions();
                if (extensions) {
                        handler.startPrefixMapping("ex", XmlRpcWriter.EXTENSIONS_URI);
                }
                handler.startElement("", "methodResponse", "methodResponse", ZERO_ATTRIBUTES);
                handler.startElement("", "fault", "fault", ZERO_ATTRIBUTES);
                Map map = new HashMap();
        map.put("faultCode", new Integer(pCode));
        map.put("faultString", pMessage == null ? "" : pMessage);
        if (pThrowable != null  &&  extensions  &&  (pConfig instanceof XmlRpcStreamRequestConfig)  &&
                ((XmlRpcStreamRequestConfig) pConfig).isEnabledForExceptions()) {
            try {
                ByteArrayOutputStream baos = new ByteArrayOutputStream();
                ObjectOutputStream oos = new ObjectOutputStream(baos);
                oos.writeObject(pThrowable);
                oos.close();
                baos.close();
                map.put("faultCause", baos.toByteArray());
            } catch (Throwable t) {
                // Ignore me
            }
        }
        writeValue(map);
                handler.endElement("", "fault", "fault");
                handler.endElement("", "methodResponse", "methodResponse");
                if (extensions) {
                        handler.endPrefixMapping("ex");
                }
                handler.endDocument();
        }

        protected void writeValue(Object pObject) throws SAXException {
                TypeSerializer serializer = typeFactory.getSerializer(config, pObject);
                if (serializer == null) {
                        throw new SAXException("Unsupported Java type: " + pObject.getClass().getName());
                }
                serializer.write(handler, pObject);
        }
}