1.1 --- /dev/null Thu Jan 01 00:00:00 1970 +0000 1.2 +++ b/src/share/jaxws_classes/com/sun/xml/internal/bind/v2/util/DataSourceSource.java Tue Mar 06 16:09:35 2012 -0800 1.3 @@ -0,0 +1,123 @@ 1.4 +/* 1.5 + * Copyright (c) 1997, 2011, Oracle and/or its affiliates. All rights reserved. 1.6 + * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. 1.7 + * 1.8 + * This code is free software; you can redistribute it and/or modify it 1.9 + * under the terms of the GNU General Public License version 2 only, as 1.10 + * published by the Free Software Foundation. Oracle designates this 1.11 + * particular file as subject to the "Classpath" exception as provided 1.12 + * by Oracle in the LICENSE file that accompanied this code. 1.13 + * 1.14 + * This code is distributed in the hope that it will be useful, but WITHOUT 1.15 + * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 1.16 + * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License 1.17 + * version 2 for more details (a copy is included in the LICENSE file that 1.18 + * accompanied this code). 1.19 + * 1.20 + * You should have received a copy of the GNU General Public License version 1.21 + * 2 along with this work; if not, write to the Free Software Foundation, 1.22 + * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. 1.23 + * 1.24 + * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA 1.25 + * or visit www.oracle.com if you need additional information or have any 1.26 + * questions. 1.27 + */ 1.28 + 1.29 +package com.sun.xml.internal.bind.v2.util; 1.30 + 1.31 +import java.io.IOException; 1.32 +import java.io.InputStream; 1.33 +import java.io.InputStreamReader; 1.34 +import java.io.Reader; 1.35 + 1.36 +import javax.activation.DataHandler; 1.37 +import javax.activation.DataSource; 1.38 +import javax.activation.MimeType; 1.39 +import javax.activation.MimeTypeParseException; 1.40 +import javax.xml.transform.Source; 1.41 +import javax.xml.transform.stream.StreamSource; 1.42 + 1.43 +/** 1.44 + * {@link Source} implementation backed by {@link DataHandler}. 1.45 + * 1.46 + * <p> 1.47 + * This implementation allows the same {@link Source} to be used 1.48 + * mutliple times. 1.49 + * 1.50 + * <p> 1.51 + * {@link Source} isn't really pluggable. As a consequence, 1.52 + * this implementation is clunky --- weak against unexpected 1.53 + * usage of the class. 1.54 + * 1.55 + * @author Kohsuke Kawaguchi 1.56 + */ 1.57 +public final class DataSourceSource extends StreamSource { 1.58 + private final DataSource source; 1.59 + 1.60 + /** 1.61 + * If null, default to the encoding declaration 1.62 + */ 1.63 + private final String charset; 1.64 + 1.65 + // remember the value we returned so that the 2nd invocation 1.66 + // will return the same object, which is what's expeted out of 1.67 + // StreamSource 1.68 + private Reader r; 1.69 + private InputStream is; 1.70 + 1.71 + public DataSourceSource(DataHandler dh) throws MimeTypeParseException { 1.72 + this(dh.getDataSource()); 1.73 + } 1.74 + 1.75 + public DataSourceSource(DataSource source) throws MimeTypeParseException { 1.76 + this.source = source; 1.77 + 1.78 + String ct = source.getContentType(); 1.79 + if(ct==null) { 1.80 + charset = null; 1.81 + } else { 1.82 + MimeType mimeType = new MimeType(ct); 1.83 + this.charset = mimeType.getParameter("charset"); 1.84 + } 1.85 + } 1.86 + 1.87 + @Override 1.88 + public void setReader(Reader reader) { 1.89 + throw new UnsupportedOperationException(); 1.90 + } 1.91 + 1.92 + @Override 1.93 + public void setInputStream(InputStream inputStream) { 1.94 + throw new UnsupportedOperationException(); 1.95 + } 1.96 + 1.97 + @Override 1.98 + public Reader getReader() { 1.99 + try { 1.100 + if(charset==null) return null; 1.101 + if(r==null) 1.102 + r = new InputStreamReader(source.getInputStream(),charset); 1.103 + return r; 1.104 + } catch (IOException e) { 1.105 + // argh 1.106 + throw new RuntimeException(e); 1.107 + } 1.108 + } 1.109 + 1.110 + @Override 1.111 + public InputStream getInputStream() { 1.112 + try { 1.113 + if(charset!=null) return null; 1.114 + if(is==null) 1.115 + is = source.getInputStream(); 1.116 + return is; 1.117 + } catch (IOException e) { 1.118 + // argh 1.119 + throw new RuntimeException(e); 1.120 + } 1.121 + } 1.122 + 1.123 + public DataSource getDataSource() { 1.124 + return source; 1.125 + } 1.126 +}