aoqi@0: /* aoqi@0: * Copyright (c) 1997, 2012, Oracle and/or its affiliates. All rights reserved. aoqi@0: * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. aoqi@0: * aoqi@0: * This code is free software; you can redistribute it and/or modify it aoqi@0: * under the terms of the GNU General Public License version 2 only, as aoqi@0: * published by the Free Software Foundation. Oracle designates this aoqi@0: * particular file as subject to the "Classpath" exception as provided aoqi@0: * by Oracle in the LICENSE file that accompanied this code. aoqi@0: * aoqi@0: * This code is distributed in the hope that it will be useful, but WITHOUT aoqi@0: * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or aoqi@0: * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License aoqi@0: * version 2 for more details (a copy is included in the LICENSE file that aoqi@0: * accompanied this code). aoqi@0: * aoqi@0: * You should have received a copy of the GNU General Public License version aoqi@0: * 2 along with this work; if not, write to the Free Software Foundation, aoqi@0: * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. aoqi@0: * aoqi@0: * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA aoqi@0: * or visit www.oracle.com if you need additional information or have any aoqi@0: * questions. aoqi@0: */ aoqi@0: aoqi@0: package com.sun.xml.internal.ws.encoding; aoqi@0: aoqi@0: import com.sun.xml.internal.org.jvnet.mimepull.MIMEPart; aoqi@0: aoqi@0: import javax.activation.DataSource; aoqi@0: import java.io.InputStream; aoqi@0: import java.io.IOException; aoqi@0: import java.io.OutputStream; aoqi@0: import java.io.File; aoqi@0: aoqi@0: import com.sun.xml.internal.ws.developer.StreamingDataHandler; aoqi@0: aoqi@0: /** aoqi@0: * Implementation of {@link StreamingDataHandler} to access MIME aoqi@0: * attachments efficiently. Applications can use the additional methods and decide aoqi@0: * on how to access the attachment data in JAX-WS applications. aoqi@0: * aoqi@0: *
aoqi@0: * for e.g.: aoqi@0: * aoqi@0: * DataHandler dh = proxy.getData(); aoqi@0: * StreamingDataHandler sdh = (StreamingDataHandler)dh; aoqi@0: * // readOnce() doesn't store attachment on the disk in some cases aoqi@0: * // for e.g when only one huge attachment after soap envelope part in MIME message aoqi@0: * InputStream in = sdh.readOnce(); aoqi@0: * ... aoqi@0: * in.close(); aoqi@0: * sdh.close(); aoqi@0: * aoqi@0: * @author Jitendra Kotamraju aoqi@0: */ aoqi@0: public class MIMEPartStreamingDataHandler extends StreamingDataHandler { aoqi@0: private final StreamingDataSource ds; aoqi@0: aoqi@0: public MIMEPartStreamingDataHandler(MIMEPart part) { aoqi@0: super(new StreamingDataSource(part)); aoqi@0: ds = (StreamingDataSource)getDataSource(); aoqi@0: } aoqi@0: aoqi@0: @Override aoqi@0: public InputStream readOnce() throws IOException { aoqi@0: return ds.readOnce(); aoqi@0: } aoqi@0: aoqi@0: @Override aoqi@0: public void moveTo(File file) throws IOException { aoqi@0: ds.moveTo(file); aoqi@0: } aoqi@0: aoqi@0: @Override aoqi@0: public void close() throws IOException { aoqi@0: ds.close(); aoqi@0: } aoqi@0: aoqi@0: private static final class StreamingDataSource implements DataSource { aoqi@0: private final MIMEPart part; aoqi@0: aoqi@0: StreamingDataSource(MIMEPart part) { aoqi@0: this.part = part; aoqi@0: } aoqi@0: aoqi@0: @Override aoqi@0: public InputStream getInputStream() throws IOException { aoqi@0: return part.read(); //readOnce() ?? aoqi@0: } aoqi@0: aoqi@0: InputStream readOnce() throws IOException { aoqi@0: try { aoqi@0: return part.readOnce(); aoqi@0: } catch(Exception e) { aoqi@0: throw new MyIOException(e); aoqi@0: } aoqi@0: } aoqi@0: aoqi@0: void moveTo(File file) throws IOException { aoqi@0: part.moveTo(file); aoqi@0: } aoqi@0: aoqi@0: @Override aoqi@0: public OutputStream getOutputStream() throws IOException { aoqi@0: return null; aoqi@0: } aoqi@0: aoqi@0: @Override aoqi@0: public String getContentType() { aoqi@0: return part.getContentType(); aoqi@0: } aoqi@0: aoqi@0: @Override aoqi@0: public String getName() { aoqi@0: return ""; aoqi@0: } aoqi@0: aoqi@0: public void close() throws IOException { aoqi@0: part.close(); aoqi@0: } aoqi@0: } aoqi@0: aoqi@0: private static final class MyIOException extends IOException { aoqi@0: private final Exception linkedException; aoqi@0: aoqi@0: MyIOException(Exception linkedException) { aoqi@0: this.linkedException = linkedException; aoqi@0: } aoqi@0: aoqi@0: @Override aoqi@0: public Throwable getCause() { aoqi@0: return linkedException; aoqi@0: } aoqi@0: } aoqi@0: aoqi@0: }