org.apache.airavata.wsmg.client.protocol.WSNTProtocolClient Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of airavata-messenger-client Show documentation
The newest version!
/*
 *
 * 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.airavata.wsmg.client.protocol;

import java.util.Map;

import javax.xml.namespace.QName;

import org.apache.airavata.wsmg.commons.WsmgCommonConstants;
import org.apache.airavata.wsmg.commons.NameSpaceConstants;
import org.apache.axiom.om.OMAbstractFactory;
import org.apache.axiom.om.OMElement;
import org.apache.axiom.om.OMFactory;
import org.apache.axis2.AxisFault;
import org.apache.axis2.addressing.EndpointReference;
import org.apache.axis2.addressing.EndpointReferenceHelper;

public class WSNTProtocolClient {

    public static OMElement createSubscriptionMsg(EndpointReference eventSinkLocation, String topicExpression,
            String xpathExpression) throws AxisFault {
        OMFactory factory = OMAbstractFactory.getOMFactory();

        OMElement message = factory.createOMElement("SubscribeRequest", NameSpaceConstants.WSNT_NS);

        if (topicExpression != null) {
            OMElement topicExpEl = factory.createOMElement("TopicExpression", NameSpaceConstants.WSNT_NS, message);

            topicExpEl.addAttribute("Dialect", WsmgCommonConstants.TOPIC_EXPRESSION_SIMPLE_DIALECT,
                    NameSpaceConstants.WSNT_NS);
            topicExpEl.declareNamespace(NameSpaceConstants.WIDGET_NS);
            topicExpEl.setText(NameSpaceConstants.WIDGET_NS.getPrefix() + ":" + topicExpression);
        }

        if (xpathExpression != null) {
            OMElement xpathExpEl = factory.createOMElement("Selector", NameSpaceConstants.WSNT_NS, message);
            xpathExpEl.addAttribute("Dialect", WsmgCommonConstants.XPATH_DIALECT, null);
            xpathExpEl.setText(xpathExpression);
        }

        OMElement useNotifyEl = factory.createOMElement("UseNotify", message.getNamespace(), message);
        useNotifyEl.setText("true");// check wether we still need this

        OMElement eprCrEl = EndpointReferenceHelper.toOM(factory, eventSinkLocation, new QName("ConsumerReference"),
                NameSpaceConstants.WSA_NS.getNamespaceURI());

        message.addChild(eprCrEl);
        eprCrEl.setNamespace(message.getNamespace());

        return message;
    }

    public static String decodeSubscriptionResponse(OMElement subscriptionReference) throws AxisFault {

        String subscriptionId = null;

        EndpointReference subscriptionReferenceEPR = EndpointReferenceHelper.fromOM(subscriptionReference);

        Map referenceParams = subscriptionReferenceEPR.getAllReferenceParameters();

        if (referenceParams != null) {
            QName identifierQName = new QName(NameSpaceConstants.WSNT_NS.getNamespaceURI(),
                    WsmgCommonConstants.SUBSCRIPTION_ID);

            OMElement identifierEl = referenceParams.get(identifierQName);
            subscriptionId = (identifierEl != null) ? identifierEl.getText() : null;

        }

        return subscriptionId;
    }

    public static OMElement createUnsubscribeMsg() {
        OMFactory factory = OMAbstractFactory.getOMFactory();
        OMElement message = factory.createOMElement("UnsubsribeRequest", NameSpaceConstants.WSNT_NS);

        return message;
    }

    public static OMElement encodeNotification(String topic, OMElement message, EndpointReference producerReference)
            throws AxisFault {
        OMFactory factory = OMAbstractFactory.getOMFactory();

        OMElement topicExpEl = factory.createOMElement("Topic", NameSpaceConstants.WSNT_NS);
        topicExpEl.addAttribute("Dialect", WsmgCommonConstants.TOPIC_EXPRESSION_SIMPLE_DIALECT, null);
        topicExpEl.declareNamespace(NameSpaceConstants.WIDGET_NS);
        topicExpEl.setText(NameSpaceConstants.WIDGET_NS.getPrefix() + ":" + topic);

        OMElement messageToNotify = factory.createOMElement("Notify", NameSpaceConstants.WSNT_NS);
        messageToNotify.declareNamespace(NameSpaceConstants.WSNT_NS);
        messageToNotify.declareNamespace(NameSpaceConstants.WSA_NS);
        OMElement notificationMesssageEl = factory.createOMElement("NotificationMessage",
                messageToNotify.getNamespace(), messageToNotify);

        notificationMesssageEl.addChild(topicExpEl);

        notificationMesssageEl.addChild(EndpointReferenceHelper.toOM(factory, producerReference, new QName(
                notificationMesssageEl.getNamespace().getNamespaceURI(), "ProducerReference", notificationMesssageEl
                        .getNamespace().getPrefix()), NameSpaceConstants.WSA_NS.getNamespaceURI()));

        OMElement messageEl = factory.createOMElement("Message", notificationMesssageEl.getNamespace(),
                notificationMesssageEl);

        messageEl.addChild(message);
        return messageToNotify;
    }

}