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:  * <p>
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: }