src/eu/alefzero/webdav/FileRequestEntity.java

   1 package eu.alefzero.webdav;
   2
   3 import java.io.File;
   4 import java.io.IOException;
   5 import java.io.OutputStream;
   6 import java.io.RandomAccessFile;
   7 import java.nio.ByteBuffer;
   8 import java.nio.channels.FileChannel;
   9 import java.nio.channels.FileLock;
  10 import java.util.Collection;
  11 import java.util.HashSet;
  12 import java.util.Iterator;
  13 import java.util.Set;
  14
  15 import org.apache.commons.httpclient.methods.RequestEntity;
  16
  17 import eu.alefzero.webdav.OnDatatransferProgressListener;
  18
  19 import android.util.Log;
  20
  21
  22 /**
  23  * A RequestEntity that represents a File.
  24  *
  25  */
  26 public class FileRequestEntity implements RequestEntity {
  27
  28     final File mFile;
  29     final String mContentType;
  30     Set<OnDatatransferProgressListener> mDataTransferListeners = new HashSet<OnDatatransferProgressListener>();
  31
  32     public FileRequestEntity(final File file, final String contentType) {
  33         super();
  34         this.mFile = file;
  35         this.mContentType = contentType;
  36         if (file == null) {
  37             throw new IllegalArgumentException("File may not be null");
  38         }
  39     }
  40
  41     @Override
  42     public long getContentLength() {
  43         return mFile.length();
  44     }
  45
  46     @Override
  47     public String getContentType() {
  48         return mContentType;
  49     }
  50
  51     @Override
  52     public boolean isRepeatable() {
  53         return true;
  54     }
  55
  56     public void addOnDatatransferProgressListener(OnDatatransferProgressListener listener) {
  57         mDataTransferListeners.add(listener);
  58     }
  59
  60     public void addOnDatatransferProgressListeners(Collection<OnDatatransferProgressListener> listeners) {
  61         mDataTransferListeners.addAll(listeners);
  62     }
  63
  64     public void removeOnDatatransferProgressListener(OnDatatransferProgressListener listener) {
  65         mDataTransferListeners.remove(listener);
  66     }
  67
  68
  69     @Override
  70     public void writeRequest(final OutputStream out) throws IOException {
  71         //byte[] tmp = new byte[4096];
  72         ByteBuffer tmp = ByteBuffer.allocate(4096);
  73         int readResult = 0;
  74
  75         // TODO(bprzybylski): each mem allocation can throw OutOfMemoryError we need to handle it
  76         //                    globally in some fashionable manner
  77         RandomAccessFile raf = new RandomAccessFile(mFile, "rw");
  78         FileChannel channel = raf.getChannel();
  79         FileLock lock = channel.tryLock();
  80         Iterator<OnDatatransferProgressListener> it = null;
  81         long transferred = 0;
  82         long size = mFile.length();
  83         if (size == 0) size = -1;
  84         try {
  85             while ((readResult = channel.read(tmp)) >= 0) {
  86                 out.write(tmp.array(), 0, readResult);
  87                 tmp.clear();
  88                 transferred += readResult;
  89                 it = mDataTransferListeners.iterator();
  90                 while (it.hasNext()) {
  91                     it.next().onTransferProgress(readResult, transferred, size, mFile.getName());
  92                 }
  93             }
  94
  95         } catch (IOException io) {
  96             Log.e("FileRequestException", io.getMessage());
  97             throw new RuntimeException("Ugly solution to workaround the default policy of retries when the server falls while uploading ; temporal fix; really", io);
  98
  99         } finally {
 100             lock.release();
 101             channel.close();
 102             raf.close();
 103         }
 104     }
 105
 106 }