org.mule.module.cxf.support.StaxFeature Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of mule-module-cxf Show documentation
A Mule module for web service connectivity using CXF.
The newest version!
/*
 * Copyright (c) MuleSoft, Inc.  All rights reserved.  http://www.mulesoft.com
 * The software in this package is published under the terms of the CPAL v1.0
 * license, a copy of which has been included with this distribution in the
 * LICENSE.txt file.
 */
package org.mule.module.cxf.support;

import javax.xml.stream.XMLInputFactory;
import javax.xml.stream.XMLOutputFactory;

import org.apache.cxf.Bus;
import org.apache.cxf.endpoint.Client;
import org.apache.cxf.endpoint.Server;
import org.apache.cxf.feature.AbstractFeature;
import org.apache.cxf.interceptor.Fault;
import org.apache.cxf.message.Message;
import org.apache.cxf.phase.AbstractPhaseInterceptor;
import org.apache.cxf.phase.Phase;
import org.apache.cxf.service.Service;

/**
 * Configures the StAX XMLInputFactory and XMLOutputFactory which CXF uses.
 * 
 */
public class StaxFeature extends AbstractFeature {
    private String xmlInputFactory;
    private String xmlOutputFactory;
    
    @Override
    public void initialize(Client client, Bus bus) {
        Service service = client.getEndpoint().getService();
        
        setProperties(service);
    }

    private void setProperties(Service service) {
        if (xmlInputFactory != null) {
            service.put(XMLInputFactory.class.getName(), xmlInputFactory);
        }
        
        if (xmlOutputFactory != null) {
            service.put(XMLOutputFactory.class.getName(), xmlOutputFactory);
        }
    }

    @Override
    public void initialize(Server server, Bus bus) {
        Service service = server.getEndpoint().getService();
        
        setProperties(service);
    }

    @Override
    public void initialize(Bus bus) {
        AbstractPhaseInterceptor in = new AbstractPhaseInterceptor(Phase.RECEIVE) {
            public void handleMessage(Message message) throws Fault {
                if (xmlInputFactory != null) {
                    message.put(XMLInputFactory.class.getName(), xmlInputFactory);
                }
            }
        };
        
        bus.getInInterceptors().add(in);
        bus.getInFaultInterceptors().add(in);
        
        AbstractPhaseInterceptor out = new AbstractPhaseInterceptor(Phase.SETUP) {
            public void handleMessage(Message message) throws Fault {
                if (xmlOutputFactory != null) {
                    message.put(XMLOutputFactory.class.getName(), xmlOutputFactory);
                }
            }
        };
        
        bus.getOutInterceptors().add(out);
        bus.getOutFaultInterceptors().add(out);
    }

    public String getXmlInputFactory() {
        return xmlInputFactory;
    }

    public void setXmlInputFactory(String xmlInputFactory) {
        this.xmlInputFactory = xmlInputFactory;
    }

    public String getXmlOutputFactory() {
        return xmlOutputFactory;
    }

    public void setXmlOutputFactory(String xmlOutputFactory) {
        this.xmlOutputFactory = xmlOutputFactory;
    }

}