Coverage Report - org.mule.transport.ftp.FtpMuleMessageFactory
 
Classes in this File Line Coverage Branch Coverage Complexity
FtpMuleMessageFactory
0%
0/26
0%
0/8
0
 
 1  
 /*
 2  
  * $Id: FtpMuleMessageFactory.java 19191 2010-08-25 21:05:23Z tcarlson $
 3  
  * --------------------------------------------------------------------------------------
 4  
  * Copyright (c) MuleSoft, Inc.  All rights reserved.  http://www.mulesoft.com
 5  
  *
 6  
  * The software in this package is published under the terms of the CPAL v1.0
 7  
  * license, a copy of which has been included with this distribution in the
 8  
  * LICENSE.txt file.
 9  
  */
 10  
 
 11  
 package org.mule.transport.ftp;
 12  
 
 13  
 import org.mule.DefaultMuleMessage;
 14  
 import org.mule.api.MuleContext;
 15  
 import org.mule.transport.AbstractMuleMessageFactory;
 16  
 import org.mule.transport.file.FileConnector;
 17  
 
 18  
 import java.io.ByteArrayOutputStream;
 19  
 import java.io.IOException;
 20  
 import java.io.InputStream;
 21  
 import java.text.MessageFormat;
 22  
 
 23  
 import org.apache.commons.net.ftp.FTPClient;
 24  
 import org.apache.commons.net.ftp.FTPFile;
 25  
 
 26  
 public class FtpMuleMessageFactory extends AbstractMuleMessageFactory
 27  
 {
 28  
 
 29  
     private FTPClient ftpClient;
 30  
     private boolean streaming;
 31  
 
 32  
     public FtpMuleMessageFactory(MuleContext context)
 33  
     {
 34  0
         super(context);
 35  0
     }
 36  
 
 37  
     @Override
 38  
     protected Object extractPayload(Object transportMessage, String encoding) throws Exception
 39  
     {
 40  0
         FTPFile file = (FTPFile) transportMessage;
 41  0
         if (streaming)
 42  
         {
 43  0
             InputStream stream = ftpClient.retrieveFileStream(file.getName());
 44  0
             if (stream == null)
 45  
             {
 46  0
                 throw new IOException(MessageFormat.format("Failed to retrieve file {0}. Ftp error: {1}",
 47  
                     file.getName(), ftpClient.getReplyCode()));
 48  
             }
 49  0
             return stream;
 50  
         }
 51  
         else
 52  
         {
 53  0
             ByteArrayOutputStream baos = new ByteArrayOutputStream();
 54  0
             if (!ftpClient.retrieveFile(file.getName(), baos))
 55  
             {
 56  0
                 throw new IOException(MessageFormat.format("Failed to retrieve file {0}. Ftp error: {1}",
 57  
                     file.getName(), ftpClient.getReplyCode()));
 58  
             }
 59  0
             byte[] bytes = baos.toByteArray();
 60  0
             if (bytes.length > 0)
 61  
             {
 62  0
                 return bytes;
 63  
             }
 64  
             else
 65  
             {
 66  0
                 throw new IOException("File " + file.getName() + " is empty (zero bytes)");
 67  
             }
 68  
         }
 69  
     }
 70  
 
 71  
     @Override
 72  
     protected Class<?>[] getSupportedTransportMessageTypes()
 73  
     {
 74  0
         return new Class[]{FTPFile.class};
 75  
     }
 76  
 
 77  
     @Override
 78  
     protected void addProperties(DefaultMuleMessage message, Object transportMessage) throws Exception
 79  
     {
 80  0
         super.addProperties(message, transportMessage);
 81  
         
 82  0
         FTPFile file = (FTPFile) transportMessage;
 83  0
         message.setOutboundProperty(FileConnector.PROPERTY_ORIGINAL_FILENAME, file.getName());
 84  0
         message.setOutboundProperty(FileConnector.PROPERTY_FILE_SIZE, file.getSize());
 85  0
         message.setOutboundProperty(FileConnector.PROPERTY_FILE_TIMESTAMP, file.getTimestamp());
 86  0
     }
 87  
 
 88  
     public void setFtpClient(FTPClient ftpClient)
 89  
     {
 90  0
         this.ftpClient = ftpClient;
 91  0
     }
 92  
 
 93  
     public void setStreaming(boolean streaming)
 94  
     {
 95  0
         this.streaming = streaming;
 96  0
     }
 97  
 
 98  
 }