src/eu/alefzero/webdav/FileRequestEntity.java

   1 /* ownCloud Android client application
   2  *   Copyright (C) 2012 Bartek Przybylski
   3  *
   4  *   This program is free software: you can redistribute it and/or modify
   5  *   it under the terms of the GNU General Public License as published by
   6  *   the Free Software Foundation, either version 3 of the License, or
   7  *   (at your option) any later version.
   8  *
   9  *   This program is distributed in the hope that it will be useful,
  10  *   but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  *   GNU General Public License for more details.
  13  *
  14  *   You should have received a copy of the GNU General Public License
  15  *   along with this program.  If not, see <http://www.gnu.org/licenses/>.
  16  *
  17  */
  18
  19 package eu.alefzero.webdav;
  20
  21 import java.io.File;
  22 import java.io.IOException;
  23 import java.io.OutputStream;
  24 import java.io.RandomAccessFile;
  25 import java.nio.ByteBuffer;
  26 import java.nio.channels.FileChannel;
  27 import java.util.Collection;
  28 import java.util.HashSet;
  29 import java.util.Iterator;
  30 import java.util.Set;
  31
  32 import org.apache.commons.httpclient.methods.RequestEntity;
  33
  34 import eu.alefzero.webdav.OnDatatransferProgressListener;
  35
  36 import android.util.Log;
  37
  38
  39 /**
  40  * A RequestEntity that represents a File.
  41  *
  42  */
  43 public class FileRequestEntity implements RequestEntity {
  44
  45     final File mFile;
  46     final String mContentType;
  47     Set<OnDatatransferProgressListener> mDataTransferListeners = new HashSet<OnDatatransferProgressListener>();
  48
  49     public FileRequestEntity(final File file, final String contentType) {
  50         super();
  51         this.mFile = file;
  52         this.mContentType = contentType;
  53         if (file == null) {
  54             throw new IllegalArgumentException("File may not be null");
  55         }
  56     }
  57
  58     @Override
  59     public long getContentLength() {
  60         return mFile.length();
  61     }
  62
  63     @Override
  64     public String getContentType() {
  65         return mContentType;
  66     }
  67
  68     @Override
  69     public boolean isRepeatable() {
  70         return true;
  71     }
  72
  73     public void addOnDatatransferProgressListener(OnDatatransferProgressListener listener) {
  74         mDataTransferListeners.add(listener);
  75     }
  76
  77     public void addOnDatatransferProgressListeners(Collection<OnDatatransferProgressListener> listeners) {
  78         mDataTransferListeners.addAll(listeners);
  79     }
  80
  81     public void removeOnDatatransferProgressListener(OnDatatransferProgressListener listener) {
  82         mDataTransferListeners.remove(listener);
  83     }
  84
  85
  86     @Override
  87     public void writeRequest(final OutputStream out) throws IOException {
  88         //byte[] tmp = new byte[4096];
  89         ByteBuffer tmp = ByteBuffer.allocate(4096);
  90         int readResult = 0;
  91
  92         // TODO(bprzybylski): each mem allocation can throw OutOfMemoryError we need to handle it
  93         //                    globally in some fashionable manner
  94         RandomAccessFile raf = new RandomAccessFile(mFile, "r");
  95         FileChannel channel = raf.getChannel();
  96         Iterator<OnDatatransferProgressListener> it = null;
  97         long transferred = 0;
  98         long size = mFile.length();
  99         if (size == 0) size = -1;
 100         try {
 101             while ((readResult = channel.read(tmp)) >= 0) {
 102                 out.write(tmp.array(), 0, readResult);
 103                 tmp.clear();
 104                 transferred += readResult;
 105                 it = mDataTransferListeners.iterator();
 106                 while (it.hasNext()) {
 107                     it.next().onTransferProgress(readResult, transferred, size, mFile.getName());
 108                 }
 109             }
 110
 111         } catch (IOException io) {
 112             Log.e("FileRequestException", io.getMessage());
 113             throw new RuntimeException("Ugly solution to workaround the default policy of retries when the server falls while uploading ; temporal fix; really", io);
 114
 115         } finally {
 116             channel.close();
 117             raf.close();
 118         }
 119     }
 120
 121 }