You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@cxf.apache.org by dk...@apache.org on 2012/08/09 17:31:59 UTC

svn commit: r1371257 - /cxf/branches/2.6.x-fixes/api/src/main/java/org/apache/cxf/attachment/ImageDataContentHandler.java

Author: dkulp
Date: Thu Aug  9 15:31:59 2012
New Revision: 1371257

URL: http://svn.apache.org/viewvc?rev=1371257&view=rev
Log:
Merged revisions 1368661 via  git cherry-pick from
https://svn.apache.org/repos/asf/cxf/trunk

........
  r1368661 | dkulp | 2012-08-02 15:43:29 -0400 (Thu, 02 Aug 2012) | 2 lines

  Add some extra formats for the image handler

........

Modified:
    cxf/branches/2.6.x-fixes/api/src/main/java/org/apache/cxf/attachment/ImageDataContentHandler.java

Modified: cxf/branches/2.6.x-fixes/api/src/main/java/org/apache/cxf/attachment/ImageDataContentHandler.java
URL: http://svn.apache.org/viewvc/cxf/branches/2.6.x-fixes/api/src/main/java/org/apache/cxf/attachment/ImageDataContentHandler.java?rev=1371257&r1=1371256&r2=1371257&view=diff
==============================================================================
--- cxf/branches/2.6.x-fixes/api/src/main/java/org/apache/cxf/attachment/ImageDataContentHandler.java (original)
+++ cxf/branches/2.6.x-fixes/api/src/main/java/org/apache/cxf/attachment/ImageDataContentHandler.java Thu Aug  9 15:31:59 2012
@@ -26,7 +26,10 @@ import java.awt.MediaTracker;
 import java.awt.datatransfer.DataFlavor;
 import java.awt.datatransfer.UnsupportedFlavorException;
 import java.awt.image.BufferedImage;
+import java.io.File;
+import java.io.FileInputStream;
 import java.io.IOException;
+import java.io.InputStream;
 import java.io.OutputStream;
 import java.util.Iterator;
 
@@ -37,6 +40,8 @@ import javax.imageio.ImageIO;
 import javax.imageio.ImageWriter;
 import javax.imageio.stream.ImageOutputStream;
 
+import org.apache.cxf.helpers.IOUtils;
+
 /**
  * 
  */
@@ -51,6 +56,10 @@ public class ImageDataContentHandler imp
         }
     }
 
+    public ImageDataContentHandler() {
+        
+    }
+    
     public Object getContent(DataSource ds) throws IOException {
         return ImageIO.read(ds.getInputStream());
     }
@@ -70,17 +79,27 @@ public class ImageDataContentHandler imp
     }
 
     public void writeTo(Object obj, String mimeType, OutputStream os) throws IOException {
-        Iterator<ImageWriter> writers = ImageIO.getImageWritersByMIMEType(mimeType);
-        if (writers.hasNext()) {
-            ImageWriter writer = writers.next();
-            
-            BufferedImage bimg = convertToBufferedImage((Image)obj);
-            ImageOutputStream out = ImageIO.createImageOutputStream(os); 
-            writer.setOutput(out);
-            writer.write(bimg);
-            writer.dispose();
-            out.flush();
-            out.close();
+        if (obj instanceof Image) {
+            Iterator<ImageWriter> writers = ImageIO.getImageWritersByMIMEType(mimeType);
+            if (writers.hasNext()) {
+                ImageWriter writer = writers.next();
+                
+                BufferedImage bimg = convertToBufferedImage((Image)obj);
+                ImageOutputStream out = ImageIO.createImageOutputStream(os); 
+                writer.setOutput(out);
+                writer.write(bimg);
+                writer.dispose();
+                out.flush();
+                out.close();
+                return;
+            }
+        } else if (obj instanceof byte[]) {
+            os.write((byte[])obj);
+        } else if (obj instanceof InputStream) {
+            IOUtils.copyAndCloseInput((InputStream)obj, os);
+        } else if (obj instanceof File) {
+            FileInputStream file = new FileInputStream((File)obj);
+            IOUtils.copyAndCloseInput(file, os);
         } else {
             throw new IOException("Attachment type not spported " + obj.getClass());                    
         }