All Downloads are FREE. Search and download functionalities are using the official Maven repository.

com.sun.xml.ws.server.EndpointMessageContextImpl Maven / Gradle / Ivy

There is a newer version: 4.0.4
Show newest version
/*
 * Copyright (c) 1997, 2020 Oracle and/or its affiliates. All rights reserved.
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Distribution License v. 1.0, which is available at
 * http://www.eclipse.org/org/documents/edl-v10.php.
 *
 * SPDX-License-Identifier: BSD-3-Clause
 */

package com.sun.xml.ws.server;

import com.sun.xml.ws.api.message.Packet;
import com.sun.xml.ws.api.message.AttachmentSet;
import com.sun.xml.ws.api.message.Attachment;

import java.util.*;
import jakarta.xml.ws.handler.MessageContext;
import jakarta.xml.ws.WebServiceContext;
import jakarta.activation.DataHandler;

/**
 * Implements {@link WebServiceContext}'s {@link MessageContext} on top of {@link Packet}.
 *
 * 

* This class creates a {@link Map} view for APPLICATION scoped properties that * gets exposed to endpoint implementations during the invocation * of web methods. The implementations access this map using * WebServiceContext.getMessageContext(). * *

* Some of the {@link Map} methods requre this class to * build the complete {@link Set} of properties, but we * try to avoid that as much as possible. * * * @author Jitendra Kotamraju */ public final class EndpointMessageContextImpl extends AbstractMap implements MessageContext { /** * Lazily computed. */ private Set> entrySet; private final Packet packet; /** * @param packet * The {@link Packet} to wrap. */ public EndpointMessageContextImpl(Packet packet) { this.packet = packet; } @Override @SuppressWarnings("element-type-mismatch") public Object get(Object key) { if (packet.supports(key)) { return packet.get(key); // strongly typed } if (packet.getHandlerScopePropertyNames(true).contains(key)) { return null; // no such application-scope property } Object value = packet.invocationProperties.get(key); //add the attachments from the Message to the corresponding attachment property if(key.equals(MessageContext.OUTBOUND_MESSAGE_ATTACHMENTS) || key.equals(MessageContext.INBOUND_MESSAGE_ATTACHMENTS)){ Map atts = (Map) value; if(atts == null) atts = new HashMap(); AttachmentSet attSet = packet.getMessage().getAttachments(); for(Attachment att : attSet){ atts.put(att.getContentId(), att.asDataHandler()); } return atts; } return value; } @Override public Object put(String key, Object value) { if (packet.supports(key)) { return packet.put(key, value); // strongly typed } Object old = packet.invocationProperties.get(key); if (old != null) { if (packet.getHandlerScopePropertyNames(true).contains(key)) { throw new IllegalArgumentException("Cannot overwrite property in HANDLER scope"); } // Overwrite existing APPLICATION scoped property packet.invocationProperties.put(key, value); return old; } // No existing property. So Add a new property packet.invocationProperties.put(key, value); return null; } @Override @SuppressWarnings("element-type-mismatch") public Object remove(Object key) { if (packet.supports(key)) { return packet.remove(key); } Object old = packet.invocationProperties.get(key); if (old != null) { if (packet.getHandlerScopePropertyNames(true).contains(key)) { throw new IllegalArgumentException("Cannot remove property in HANDLER scope"); } // Remove existing APPLICATION scoped property packet.invocationProperties.remove(key); return old; } // No existing property. return null; } public Set> entrySet() { if (entrySet == null) { entrySet = new EntrySet(); } return entrySet; } public void setScope(String name, MessageContext.Scope scope) { throw new UnsupportedOperationException( "All the properties in this context are in APPLICATION scope. Cannot do setScope()."); } public MessageContext.Scope getScope(String name) { throw new UnsupportedOperationException( "All the properties in this context are in APPLICATION scope. Cannot do getScope()."); } private class EntrySet extends AbstractSet> { public Iterator> iterator() { final Iterator> it = createBackupMap().entrySet().iterator(); return new Iterator>() { Map.Entry cur; public boolean hasNext() { return it.hasNext(); } public Map.Entry next() { cur = it.next(); return cur; } public void remove() { it.remove(); EndpointMessageContextImpl.this.remove(cur.getKey()); } }; } public int size() { return createBackupMap().size(); } } private Map createBackupMap() { Map backupMap = new HashMap(); backupMap.putAll(packet.createMapView()); Set handlerProps = packet.getHandlerScopePropertyNames(true); for(Map.Entry e : packet.invocationProperties.entrySet()) { if (!handlerProps.contains(e.getKey())) { backupMap.put(e.getKey(), e.getValue()); } } return backupMap; } }





© 2015 - 2024 Weber Informatics LLC | Privacy Policy