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

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

src/share/jaxws_classes/com/sun/xml/internal/ws/encoding/XmlDataContentHandler.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, 2012, 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.encoding;
aoqi@0	27
aoqi@0	28	import com.sun.xml.internal.ws.util.xml.XmlUtil;
aoqi@0	29
aoqi@0	30	import javax.activation.ActivationDataFlavor;
aoqi@0	31	import javax.activation.DataContentHandler;
aoqi@0	32	import javax.activation.DataSource;
aoqi@0	33	import javax.xml.transform.OutputKeys;
aoqi@0	34	import javax.xml.transform.Source;
aoqi@0	35	import javax.xml.transform.Transformer;
aoqi@0	36	import javax.xml.transform.stream.StreamResult;
aoqi@0	37	import javax.xml.transform.stream.StreamSource;
aoqi@0	38	import java.awt.datatransfer.DataFlavor;
aoqi@0	39	import java.io.IOException;
aoqi@0	40	import java.io.InputStreamReader;
aoqi@0	41	import java.io.OutputStream;
aoqi@0	42	import java.io.OutputStreamWriter;
aoqi@0	43	import java.util.Arrays;
aoqi@0	44
aoqi@0	45	/**
aoqi@0	46	* JAF data handler for XML content
aoqi@0	47	*
aoqi@0	48	* @author Jitendra Kotamraju
aoqi@0	49	*/
aoqi@0	50	public class XmlDataContentHandler implements DataContentHandler {
aoqi@0	51
aoqi@0	52	private final DataFlavor[] flavors;
aoqi@0	53
aoqi@0	54	public XmlDataContentHandler() throws ClassNotFoundException {
aoqi@0	55	flavors = new DataFlavor[3];
aoqi@0	56	flavors[0] = new ActivationDataFlavor(StreamSource.class, "text/xml", "XML");
aoqi@0	57	flavors[1] = new ActivationDataFlavor(StreamSource.class, "application/xml", "XML");
aoqi@0	58	flavors[2] = new ActivationDataFlavor(String.class, "text/xml", "XML String");
aoqi@0	59	}
aoqi@0	60
aoqi@0	61	public DataFlavor[] getTransferDataFlavors() {
aoqi@0	62	return Arrays.copyOf(flavors, flavors.length);
aoqi@0	63	}
aoqi@0	64
aoqi@0	65	public Object getTransferData(DataFlavor df, DataSource ds)
aoqi@0	66	throws IOException {
aoqi@0	67
aoqi@0	68	for (DataFlavor aFlavor : flavors) {
aoqi@0	69	if (aFlavor.equals(df)) {
aoqi@0	70	return getContent(ds);
aoqi@0	71	}
aoqi@0	72	}
aoqi@0	73	return null;
aoqi@0	74	}
aoqi@0	75
aoqi@0	76	/**
aoqi@0	77	* Create an object from the input stream
aoqi@0	78	*/
aoqi@0	79	public Object getContent(DataSource ds) throws IOException {
aoqi@0	80	String ctStr = ds.getContentType();
aoqi@0	81	String charset = null;
aoqi@0	82	if (ctStr != null) {
aoqi@0	83	ContentType ct = new ContentType(ctStr);
aoqi@0	84	if (!isXml(ct)) {
aoqi@0	85	throw new IOException(
aoqi@0	86	"Cannot convert DataSource with content type \""
aoqi@0	87	+ ctStr + "\" to object in XmlDataContentHandler");
aoqi@0	88	}
aoqi@0	89	charset = ct.getParameter("charset");
aoqi@0	90	}
aoqi@0	91	return (charset != null)
aoqi@0	92	? new StreamSource(new InputStreamReader(ds.getInputStream()), charset)
aoqi@0	93	: new StreamSource(ds.getInputStream());
aoqi@0	94	}
aoqi@0	95
aoqi@0	96	/**
aoqi@0	97	* Convert the object to a byte stream
aoqi@0	98	*/
aoqi@0	99	public void writeTo(Object obj, String mimeType, OutputStream os)
aoqi@0	100	throws IOException {
aoqi@0	101
aoqi@0	102	if (!(obj instanceof DataSource || obj instanceof Source || obj instanceof String)) {
aoqi@0	103	throw new IOException("Invalid Object type = "+obj.getClass()+
aoqi@0	104	". XmlDataContentHandler can only convert DataSource|Source|String to XML.");
aoqi@0	105	}
aoqi@0	106
aoqi@0	107	ContentType ct = new ContentType(mimeType);
aoqi@0	108	if (!isXml(ct)) {
aoqi@0	109	throw new IOException(
aoqi@0	110	"Invalid content type \"" + mimeType + "\" for XmlDataContentHandler");
aoqi@0	111	}
aoqi@0	112
aoqi@0	113	String charset = ct.getParameter("charset");
aoqi@0	114	if (obj instanceof String) {
aoqi@0	115	String s = (String) obj;
aoqi@0	116	if (charset == null) {
aoqi@0	117	charset = "utf-8";
aoqi@0	118	}
aoqi@0	119	OutputStreamWriter osw = new OutputStreamWriter(os, charset);
aoqi@0	120	osw.write(s, 0, s.length());
aoqi@0	121	osw.flush();
aoqi@0	122	return;
aoqi@0	123	}
aoqi@0	124
aoqi@0	125	Source source = (obj instanceof DataSource)
aoqi@0	126	? (Source)getContent((DataSource)obj) : (Source)obj;
aoqi@0	127	try {
aoqi@0	128	Transformer transformer = XmlUtil.newTransformer();
aoqi@0	129	if (charset != null) {
aoqi@0	130	transformer.setOutputProperty(OutputKeys.ENCODING, charset);
aoqi@0	131	}
aoqi@0	132	StreamResult result = new StreamResult(os);
aoqi@0	133	transformer.transform(source, result);
aoqi@0	134	} catch (Exception ex) {
aoqi@0	135	throw new IOException(
aoqi@0	136	"Unable to run the JAXP transformer in XmlDataContentHandler "
aoqi@0	137	+ ex.getMessage());
aoqi@0	138	}
aoqi@0	139	}
aoqi@0	140
aoqi@0	141	private boolean isXml(ContentType ct) {
aoqi@0	142	return ct.getSubType().equals("xml") &&
aoqi@0	143	(ct.getPrimaryType().equals("text") || ct.getPrimaryType().equals("application"));
aoqi@0	144	}
aoqi@0	145
aoqi@0	146	}