/*
* Copyright (c) 2005, 2006, Oracle and/or its affiliates. All rights reserved.
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
*
* This code is free software; you can redistribute it and/or modify it
* under the terms of the GNU General Public License version 2 only, as
* published by the Free Software Foundation. Oracle designates this
* particular file as subject to the "Classpath" exception as provided
* by Oracle in the LICENSE file that accompanied this code.
*
* This code is distributed in the hope that it will be useful, but WITHOUT
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
* version 2 for more details (a copy is included in the LICENSE file that
* accompanied this code).
*
* You should have received a copy of the GNU General Public License version
* 2 along with this work; if not, write to the Free Software Foundation,
* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
*
* Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
* or visit www.oracle.com if you need additional information or have any
* questions.
*/
package com.sun.xml.internal.ws.server;
import com.sun.xml.internal.ws.api.message.Packet;
import com.sun.xml.internal.ws.api.message.AttachmentSet;
import com.sun.xml.internal.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}.
*
* <p>
* 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().
*
* <p>
* 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
*/
@SuppressWarnings({"SuspiciousMethodCalls"})
public final class EndpointMessageContextImpl extends AbstractMap<String,Object> implements MessageContext {
/**
* Lazily computed.
*/
private Set<Map.Entry<String,Object>> entrySet;
private final Packet packet;
/**
* @param packet
* The {@link Packet} to wrap.
*/
public EndpointMessageContextImpl(Packet packet) {
this.packet = packet;
}
@Override
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<String, DataHandler> atts = (Map<String, DataHandler>) value;
if(atts == null)
atts = new HashMap<String, DataHandler>();
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
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<Map.Entry<String, Object>> 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<Map.Entry<String, Object>> {
public Iterator<Map.Entry<String, Object>> iterator() {
final Iterator<Map.Entry<String, Object>> it = createBackupMap().entrySet().iterator();
return new Iterator<Map.Entry<String, Object>>() {
Map.Entry<String, Object> cur;
public boolean hasNext() {
return it.hasNext();
}
public Map.Entry<String, Object> next() {
cur = it.next();
return cur;
}
public void remove() {
it.remove();
EndpointMessageContextImpl.this.remove(cur.getKey());
}
};
}
public int size() {
return createBackupMap().size();
}
}
private Map<String, Object> createBackupMap() {
Map<String, Object> backupMap = new HashMap<String, Object>();
backupMap.putAll(packet.createMapView());
Set<String> handlerProps = packet.getHandlerScopePropertyNames(true);
for(Map.Entry<String, Object> e : packet.invocationProperties.entrySet()) {
if (!handlerProps.contains(e.getKey())) {
backupMap.put(e.getKey(), e.getValue());
}
}
return backupMap;
}
}