• file
• revisions
• annotate
• diff
• comparison
• raw

src/share/jaxws_classes/com/sun/xml/internal/ws/handler/MessageContextImpl.java@9c07ef4934dd (annotated)

src/share/jaxws_classes/com/sun/xml/internal/ws/handler/MessageContextImpl.java

Thu, 31 Aug 2017 15:18:52 +0800

author

aoqi

date

Thu, 31 Aug 2017 15:18:52 +0800

changeset 637

9c07ef4934dd

parent 368

0989ad8c0860

parent 0

373ffda63c9a

permissions

-rw-r--r--

merge

aoqi@0	1	/*
aoqi@0	2	* Copyright (c) 1997, 2013, Oracle and/or its affiliates. All rights reserved.
aoqi@0	3	* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
aoqi@0	4	*
aoqi@0	5	* This code is free software; you can redistribute it and/or modify it
aoqi@0	6	* under the terms of the GNU General Public License version 2 only, as
aoqi@0	7	* published by the Free Software Foundation. Oracle designates this
aoqi@0	8	* particular file as subject to the "Classpath" exception as provided
aoqi@0	9	* by Oracle in the LICENSE file that accompanied this code.
aoqi@0	10	*
aoqi@0	11	* This code is distributed in the hope that it will be useful, but WITHOUT
aoqi@0	12	* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
aoqi@0	13	* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
aoqi@0	14	* version 2 for more details (a copy is included in the LICENSE file that
aoqi@0	15	* accompanied this code).
aoqi@0	16	*
aoqi@0	17	* You should have received a copy of the GNU General Public License version
aoqi@0	18	* 2 along with this work; if not, write to the Free Software Foundation,
aoqi@0	19	* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
aoqi@0	20	*
aoqi@0	21	* Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
aoqi@0	22	* or visit www.oracle.com if you need additional information or have any
aoqi@0	23	* questions.
aoqi@0	24	*/
aoqi@0	25
aoqi@0	26	package com.sun.xml.internal.ws.handler;
aoqi@0	27
aoqi@0	28	import com.sun.xml.internal.ws.api.message.Attachment;
aoqi@0	29	import com.sun.xml.internal.ws.api.message.AttachmentSet;
aoqi@0	30	import com.sun.xml.internal.ws.api.message.Packet;
aoqi@0	31
aoqi@0	32	import javax.activation.DataHandler;
aoqi@0	33	import javax.xml.ws.handler.MessageContext;
aoqi@0	34	import java.util.Collection;
aoqi@0	35	import java.util.HashMap;
aoqi@0	36	import java.util.Map;
aoqi@0	37	import java.util.Set;
aoqi@0	38
aoqi@0	39
aoqi@0	40	/**
aoqi@0	41	*
aoqi@0	42	* @author WS Development Team
aoqi@0	43	*/
aoqi@0	44
aoqi@0	45	class MessageContextImpl implements MessageContext {
aoqi@0	46	private final Set<String> handlerScopeProps;
aoqi@0	47	private final Packet packet;
aoqi@0	48	private final Map<String, Object> asMapIncludingInvocationProperties;
aoqi@0	49
aoqi@0	50	/** Creates a new instance of MessageContextImpl */
aoqi@0	51	public MessageContextImpl(Packet packet) {
aoqi@0	52	this.packet = packet;
aoqi@0	53	this.asMapIncludingInvocationProperties = packet.asMapIncludingInvocationProperties();
aoqi@0	54	this.handlerScopeProps = packet.getHandlerScopePropertyNames(false);
aoqi@0	55	}
aoqi@0	56
aoqi@0	57	protected void updatePacket() {
aoqi@0	58	throw new UnsupportedOperationException("wrong call");
aoqi@0	59	}
aoqi@0	60
aoqi@0	61	public void setScope(String name, Scope scope) {
aoqi@0	62	if(!containsKey(name))
aoqi@0	63	throw new IllegalArgumentException("Property " + name + " does not exist.");
aoqi@0	64	if(scope == Scope.APPLICATION) {
aoqi@0	65	handlerScopeProps.remove(name);
aoqi@0	66	} else {
aoqi@0	67	handlerScopeProps.add(name);
aoqi@0	68
aoqi@0	69	}
aoqi@0	70	}
aoqi@0	71
aoqi@0	72	public Scope getScope(String name) {
aoqi@0	73	if(!containsKey(name))
aoqi@0	74	throw new IllegalArgumentException("Property " + name + " does not exist.");
aoqi@0	75	if(handlerScopeProps.contains(name)) {
aoqi@0	76	return Scope.HANDLER;
aoqi@0	77	} else {
aoqi@0	78	return Scope.APPLICATION;
aoqi@0	79	}
aoqi@0	80	}
aoqi@0	81
aoqi@0	82	public int size() {
aoqi@0	83	return asMapIncludingInvocationProperties.size();
aoqi@0	84	}
aoqi@0	85
aoqi@0	86	public boolean isEmpty() {
aoqi@0	87	return asMapIncludingInvocationProperties.isEmpty();
aoqi@0	88	}
aoqi@0	89
aoqi@0	90	public boolean containsKey(Object key) {
aoqi@0	91	return asMapIncludingInvocationProperties.containsKey(key);
aoqi@0	92	}
aoqi@0	93
aoqi@0	94	public boolean containsValue(Object value) {
aoqi@0	95	return asMapIncludingInvocationProperties.containsValue(value);
aoqi@0	96	}
aoqi@0	97
aoqi@0	98	public Object put(String key, Object value) {
aoqi@0	99	if (!asMapIncludingInvocationProperties.containsKey(key)) {
aoqi@0	100	//new property, default to Scope.HANDLER
aoqi@0	101	handlerScopeProps.add(key);
aoqi@0	102	}
aoqi@0	103	return asMapIncludingInvocationProperties.put(key, value);
aoqi@0	104	}
aoqi@0	105	public Object get(Object key) {
aoqi@0	106	if(key == null)
aoqi@0	107	return null;
aoqi@0	108	Object value = asMapIncludingInvocationProperties.get(key);
aoqi@0	109	//add the attachments from the Message to the corresponding attachment property
aoqi@0	110	if(key.equals(MessageContext.OUTBOUND_MESSAGE_ATTACHMENTS) ||
aoqi@0	111	key.equals(MessageContext.INBOUND_MESSAGE_ATTACHMENTS)){
aoqi@0	112	Map<String, DataHandler> atts = (Map<String, DataHandler>) value;
aoqi@0	113	if(atts == null)
aoqi@0	114	atts = new HashMap<String, DataHandler>();
aoqi@0	115	AttachmentSet attSet = packet.getMessage().getAttachments();
aoqi@0	116	for(Attachment att : attSet){
aoqi@0	117	String cid = att.getContentId();
aoqi@0	118	if (cid.indexOf("@jaxws.sun.com") == -1) {
aoqi@0	119	Object a = atts.get(cid);
aoqi@0	120	if (a == null) {
aoqi@0	121	a = atts.get("<" + cid + ">");
aoqi@0	122	if (a == null) atts.put(att.getContentId(), att.asDataHandler());
aoqi@0	123	}
aoqi@0	124	} else {
aoqi@0	125	atts.put(att.getContentId(), att.asDataHandler());
aoqi@0	126	}
aoqi@0	127	}
aoqi@0	128	return atts;
aoqi@0	129	}
aoqi@0	130	return value;
aoqi@0	131	}
aoqi@0	132
aoqi@0	133	public void putAll(Map<? extends String, ? extends Object> t) {
aoqi@0	134	for(String key: t.keySet()) {
aoqi@0	135	if(!asMapIncludingInvocationProperties.containsKey(key)) {
aoqi@0	136	//new property, default to Scope.HANDLER
aoqi@0	137	handlerScopeProps.add(key);
aoqi@0	138	}
aoqi@0	139	}
aoqi@0	140	asMapIncludingInvocationProperties.putAll(t);
aoqi@0	141	}
aoqi@0	142
aoqi@0	143	public void clear() {
aoqi@0	144	asMapIncludingInvocationProperties.clear();
aoqi@0	145	}
aoqi@0	146	public Object remove(Object key){
aoqi@0	147	handlerScopeProps.remove(key);
aoqi@0	148	return asMapIncludingInvocationProperties.remove(key);
aoqi@0	149	}
aoqi@0	150	public Set<String> keySet() {
aoqi@0	151	return asMapIncludingInvocationProperties.keySet();
aoqi@0	152	}
aoqi@0	153	public Set<Map.Entry<String, Object>> entrySet(){
aoqi@0	154	return asMapIncludingInvocationProperties.entrySet();
aoqi@0	155	}
aoqi@0	156	public Collection<Object> values() {
aoqi@0	157	return asMapIncludingInvocationProperties.values();
aoqi@0	158	}
aoqi@0	159	}