Search in sources :

Example 1 with RelatesToHeader

use of com.sun.xml.ws.message.RelatesToHeader in project metro-jax-ws by eclipse-ee4j.

the class Packet method populateAddressingHeaders.

private void populateAddressingHeaders(Packet responsePacket, AddressingVersion av, SOAPVersion sv, String action, boolean mustUnderstand) {
    // populate WS-A headers only if WS-A is enabled
    if (av == null)
        return;
    // if one-way, then dont populate any WS-A headers
    if (responsePacket.getMessage() == null)
        return;
    MessageHeaders hl = responsePacket.getMessage().getHeaders();
    WsaPropertyBag wpb = getSatellite(WsaPropertyBag.class);
    Message msg = getMessage();
    // wsa:To
    WSEndpointReference replyTo = null;
    Header replyToFromRequestMsg = AddressingUtils.getFirstHeader(msg.getHeaders(), av.replyToTag, true, sv);
    Header replyToFromResponseMsg = hl.get(av.toTag, false);
    boolean replaceToTag = true;
    try {
        if (replyToFromRequestMsg != null) {
            replyTo = replyToFromRequestMsg.readAsEPR(av);
        }
        if (replyToFromResponseMsg != null && replyTo == null) {
            replaceToTag = false;
        }
    } catch (XMLStreamException e) {
        throw new WebServiceException(AddressingMessages.REPLY_TO_CANNOT_PARSE(), e);
    }
    if (replyTo == null) {
        replyTo = AddressingUtils.getReplyTo(msg.getHeaders(), av, sv);
    }
    // false for Provider with no wsdl, Expects User to set the coresponding header on the Message.
    if (AddressingUtils.getAction(responsePacket.getMessage().getHeaders(), av, sv) == null) {
        // wsa:Action header is not set in the message, so use the wsa:Action  passed as the parameter.
        hl.add(new StringHeader(av.actionTag, action, sv, mustUnderstand));
    }
    // wsa:MessageID
    if (responsePacket.getMessage().getHeaders().get(av.messageIDTag, false) == null) {
        // if header doesn't exist, method getID creates a new random id
        String newID = Message.generateMessageID();
        hl.add(new StringHeader(av.messageIDTag, newID));
    }
    // wsa:RelatesTo
    String mid = null;
    if (wpb != null) {
        mid = wpb.getMessageID();
    }
    if (mid == null) {
        mid = AddressingUtils.getMessageID(msg.getHeaders(), av, sv);
    }
    if (mid != null) {
        hl.addOrReplace(new RelatesToHeader(av.relatesToTag, mid));
    }
    // populate reference parameters
    WSEndpointReference refpEPR = null;
    if (responsePacket.getMessage().isFault()) {
        // choose FaultTo
        if (wpb != null) {
            refpEPR = wpb.getFaultToFromRequest();
        }
        if (refpEPR == null) {
            refpEPR = AddressingUtils.getFaultTo(msg.getHeaders(), av, sv);
        }
        // if FaultTo is null, then use ReplyTo
        if (refpEPR == null) {
            refpEPR = replyTo;
        }
    } else {
        // choose ReplyTo
        refpEPR = replyTo;
    }
    if (replaceToTag && refpEPR != null) {
        hl.addOrReplace(new StringHeader(av.toTag, refpEPR.getAddress()));
        refpEPR.addReferenceParametersToList(hl);
    }
}
Also used : WsaPropertyBag(com.sun.xml.ws.addressing.WsaPropertyBag) StringHeader(com.sun.xml.ws.message.StringHeader) SOAPMessage(jakarta.xml.soap.SOAPMessage) StringHeader(com.sun.xml.ws.message.StringHeader) RelatesToHeader(com.sun.xml.ws.message.RelatesToHeader) XMLStreamException(javax.xml.stream.XMLStreamException) WebServiceException(jakarta.xml.ws.WebServiceException) RelatesToHeader(com.sun.xml.ws.message.RelatesToHeader) WSEndpointReference(com.sun.xml.ws.api.addressing.WSEndpointReference)

Example 2 with RelatesToHeader

use of com.sun.xml.ws.message.RelatesToHeader in project metro-jax-ws by eclipse-ee4j.

the class AddressingUtils method fillRequestAddressingHeaders.

private static void fillRequestAddressingHeaders(@NotNull MessageHeaders headers, @NotNull Packet packet, @NotNull AddressingVersion av, @NotNull SOAPVersion sv, @NotNull OneWayFeature oneWayFeature, boolean oneway, @NotNull String action) {
    if (!oneway && !oneWayFeature.isUseAsyncWithSyncInvoke() && Boolean.TRUE.equals(packet.isSynchronousMEP)) {
        fillRequestAddressingHeaders(headers, packet, av, sv, oneway, action);
    } else {
        fillCommonAddressingHeaders(headers, packet, av, sv, action, false);
        boolean isMessageIdAdded = false;
        // add if it doesn't already exist and OneWayFeature requests a specific ReplyTo
        if (headers.get(av.replyToTag, false) == null) {
            WSEndpointReference replyToEpr = oneWayFeature.getReplyTo();
            if (replyToEpr != null) {
                headers.add(replyToEpr.createHeader(av.replyToTag));
                // add wsa:MessageID only for non-null ReplyTo
                if (packet.getMessage().getHeaders().get(av.messageIDTag, false) == null) {
                    // if header doesn't exist, method getID creates a new random id
                    String newID = oneWayFeature.getMessageId() == null ? Message.generateMessageID() : oneWayFeature.getMessageId();
                    headers.add(new StringHeader(av.messageIDTag, newID));
                    isMessageIdAdded = true;
                }
            }
        }
        // If the user sets a messageId, use it.
        final String messageId = oneWayFeature.getMessageId();
        if (!isMessageIdAdded && messageId != null) {
            headers.add(new StringHeader(av.messageIDTag, messageId));
        }
        // add if it doesn't already exist and OneWayFeature requests a specific FaultTo
        if (headers.get(av.faultToTag, false) == null) {
            WSEndpointReference faultToEpr = oneWayFeature.getFaultTo();
            if (faultToEpr != null) {
                headers.add(faultToEpr.createHeader(av.faultToTag));
                // add wsa:MessageID only for non-null FaultTo
                if (headers.get(av.messageIDTag, false) == null) {
                    headers.add(new StringHeader(av.messageIDTag, Message.generateMessageID()));
                }
            }
        }
        // wsa:From
        if (oneWayFeature.getFrom() != null) {
            headers.addOrReplace(oneWayFeature.getFrom().createHeader(av.fromTag));
        }
        // wsa:RelatesTo
        if (oneWayFeature.getRelatesToID() != null) {
            headers.addOrReplace(new RelatesToHeader(av.relatesToTag, oneWayFeature.getRelatesToID()));
        }
    }
}
Also used : StringHeader(com.sun.xml.ws.message.StringHeader) RelatesToHeader(com.sun.xml.ws.message.RelatesToHeader) WSEndpointReference(com.sun.xml.ws.api.addressing.WSEndpointReference)

Aggregations

WSEndpointReference (com.sun.xml.ws.api.addressing.WSEndpointReference)2 RelatesToHeader (com.sun.xml.ws.message.RelatesToHeader)2 StringHeader (com.sun.xml.ws.message.StringHeader)2 WsaPropertyBag (com.sun.xml.ws.addressing.WsaPropertyBag)1 SOAPMessage (jakarta.xml.soap.SOAPMessage)1 WebServiceException (jakarta.xml.ws.WebServiceException)1 XMLStreamException (javax.xml.stream.XMLStreamException)1