com.sun.xml.ws.server.EndpointMessageContextImpl Maven / Gradle / Ivy
Show all versions of webservices-rt Show documentation
/*
* Copyright (c) 1997, 2018 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 javax.xml.ws.handler.MessageContext;
import javax.xml.ws.WebServiceContext;
import javax.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;
}
}