org.mule.providers.soap.axis.wsdl.wsrf.aspect.WsAddressingAdvice.java Source code

Introduction

Here is the source code for org.mule.providers.soap.axis.wsdl.wsrf.aspect.WsAddressingAdvice.java

Source

/*
 * $Id$
 * --------------------------------------------------------------------------------------
 * Copyright (c) MuleSource, Inc.  All rights reserved.  http://www.mulesource.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.providers.soap.axis.wsdl.wsrf.aspect;

import org.mule.providers.soap.axis.wsdl.wsrf.BasePriorityAdvice;
import org.mule.providers.soap.axis.wsdl.wsrf.util.WSRFParameter;
import org.mule.providers.soap.wsdl.wsrf.instance.GenericPortTypeSoapBindingsStub;

import org.mule.umo.UMOEvent;

import java.lang.reflect.Method;

import javax.xml.namespace.QName;
import javax.xml.rpc.ServiceException;

import org.apache.axis.AxisFault;
import org.apache.axis.client.Call;
import org.apache.axis.description.OperationDesc;
import org.apache.axis.message.addressing.AddressingHeaders;
import org.apache.axis.message.addressing.Constants;
import org.apache.axis.message.addressing.ReferencePropertiesType;
import org.apache.log4j.Level;
import org.apache.log4j.Logger;
import org.globus.wsrf.encoding.SerializationException;
import org.globus.wsrf.impl.SimpleResourceKey;

import org.springframework.aop.MethodBeforeAdvice;

/**
 * WsAddressing Aspect to inject Ws-Addressing information in SOAP - Header and/or
 * manage WSDL Stub.
 */
public class WsAddressingAdvice extends BasePriorityAdvice implements MethodBeforeAdvice {

    /**
     * Default Constructor.
     */
    public WsAddressingAdvice() {
        Logger.getLogger(this.getClass()).log(Level.INFO, this.getClass().getName() + " : started.");
    }

    /**
     * Inject call informations.
     * 
     * @param arg0 .
     * @param arg1 .
     * @param arg2 .
     * @throws Throwable .
     */
    public void before(Method arg0, Object[] arg1, Object arg2) throws Throwable {

        Call call = (Call) arg1[0];

        if (call == null) {
            return;
        }

        setSerializableClasses(call);
        UMOEvent event = (UMOEvent) arg1[1];
        setOperationsDesc(call, event);
        call.setUseSOAPAction(true);
        call.setSOAPActionURI((String) event.getMessage().getProperty(WSRFParameter.SOAP_ACTION_URI));

        call.setEncodingStyle(null);
        call.setProperty(org.apache.axis.client.Call.SEND_TYPE_ATTR, Boolean.FALSE);
        call.setProperty(org.apache.axis.AxisEngine.PROP_DOMULTIREFS, Boolean.FALSE);
        call.setSOAPVersion(org.apache.axis.soap.SOAPConstants.SOAP11_CONSTANTS);
        setHeader(call, event);

    }

    /**
     * set Serializable Classes.
     * 
     * @param call call
     * @throws AxisFault exception on axis fault
     * @throws ServiceException service exception
     */
    private void setSerializableClasses(Call call) throws AxisFault, ServiceException {
        GenericPortTypeSoapBindingsStub stub = new GenericPortTypeSoapBindingsStub(call.getService());
        stub._createCall();
    }

    /**
     * set Operations Desc.
     * 
     * @param call call
     * @param event event
     */
    private void setOperationsDesc(Call call, UMOEvent event) {
        OperationDesc oper;
        if (call.getOperation() != null) {
            oper = call.getOperation();
        } else {
            oper = new org.apache.axis.description.OperationDesc();
        }

        try {
            oper.setName((String) event.getMessage().getProperty(WSRFParameter.METHOD));
            oper.setReturnType((QName) event.getMessage().getProperty(WSRFParameter.RETURN_QTYPE));
            oper.setReturnClass((Class) event.getMessage().getProperty(WSRFParameter.RETURN_CLASS));
            String serviceNamespace = (String) event.getMessage().getProperty(WSRFParameter.SERVICE_NAMESPACE);
            String returnQName = (String) event.getMessage().getProperty(WSRFParameter.RETURN_QNAME);
            oper.setReturnQName(new javax.xml.namespace.QName(serviceNamespace, returnQName));
            oper.setStyle(org.apache.axis.constants.Style.DOCUMENT);
            oper.setUse(org.apache.axis.constants.Use.LITERAL);
            oper.getParameter(0).setName((String) event.getMessage().getProperty(WSRFParameter.METHOD));

            call.setOperation(oper);
        } catch (Exception e) {
            Logger.getLogger(this.getClass()).log(Level.WARN,
                    this.getClass().getName() + " OperationDesc :  not injected.." + e.getMessage());
            e.printStackTrace();
            return;
        }

        Logger.getLogger(this.getClass()).log(Level.DEBUG,
                this.getClass().getName() + " OperationDesc :  injected..");
    }

    /**
     * set header Informations.
     * 
     * @param call call
     * @param event event
     */
    private void setHeader(Call call, UMOEvent event) {
        AddressingHeaders headers;
        if (call.getProperty(Constants.ENV_ADDRESSING_REQUEST_HEADERS) == null) {
            headers = new AddressingHeaders();
        } else {
            headers = (AddressingHeaders) call.getProperty(Constants.ENV_ADDRESSING_REQUEST_HEADERS);
        }
        setReferenceProperties(headers, event);
        call.setProperty(Constants.ENV_ADDRESSING_REQUEST_HEADERS, headers);
        Logger.getLogger(this.getClass()).log(Level.DEBUG,
                this.getClass().getName() + " ENV_ADDRESSING_REQUEST_HEADERS :  injected..");
    }

    /**
     * set Reference Properties. injecting Resource Key Ws-Addressing information
     * 
     * @param headers headers
     * @param event event
     */
    private void setReferenceProperties(AddressingHeaders headers, UMOEvent event) {

        String serviceNamespace = (String) event.getMessage().getProperty(WSRFParameter.SERVICE_NAMESPACE);
        String resourceKeyName = (String) event.getMessage().getProperty(WSRFParameter.RESOURCE_KEY_NAME);
        QName keyName = new QName(serviceNamespace, resourceKeyName);
        String keyValue = (String) event.getMessage().getProperty(WSRFParameter.RESOURCE_KEY);
        //avoid to add ReferencesProperties if resource key is not define
        if (keyValue == null) {
            Logger.getLogger(this.getClass()).log(Level.DEBUG,
                    this.getClass().getName() + " : ReferencesProperties IGNORED . Resource key not found..");
            return;
        }
        SimpleResourceKey key = new SimpleResourceKey(keyName, keyValue);

        ReferencePropertiesType props = headers.getReferenceProperties();
        if (props == null) {
            props = new ReferencePropertiesType();
        }
        try {
            props.add(key.toSOAPElement());

        } catch (SerializationException e) {
            e.printStackTrace();
        }
        Logger.getLogger(this.getClass()).log(Level.DEBUG,
                this.getClass().getName() + " : ReferencesProperties injected..");
        headers.setReferenceProperties(props);
    }

}