dfsf

[pub/Android/ownCloud.git] / src / eu / alefzero / webdav / ChunkFromFileChannelRequestEntity.java

diff --git a/src/eu/alefzero/webdav/ChunkFromFileChannelRequestEntity.java b/src/eu/alefzero/webdav/ChunkFromFileChannelRequestEntity.java
index c3c9a69..13c1104 100644 (file)
--- a/src/eu/alefzero/webdav/ChunkFromFileChannelRequestEntity.java
+++ b/src/eu/alefzero/webdav/ChunkFromFileChannelRequestEntity.java
@@ -1,9 +1,9 @@
 /* ownCloud Android client application
 /* ownCloud Android client application

- *   Copyright (C) 2012 Bartek Przybylski
+ *   Copyright (C) 2012-2013 ownCloud Inc.

  *
  *   This program is free software: you can redistribute it and/or modify
  *   it under the terms of the GNU General Public License as published by
  *
  *   This program is free software: you can redistribute it and/or modify
  *   it under the terms of the GNU General Public License as published by

- *   the Free Software Foundation, either version 3 of the License, or
+ *   the Free Software Foundation, either version 2 of the License, or

  *   (at your option) any later version.
  *
  *   This program is distributed in the hope that it will be useful,
  *   (at your option) any later version.
  *
  *   This program is distributed in the hope that it will be useful,


@@ -18,13 +18,20 @@
 
 package eu.alefzero.webdav;
 
 
 package eu.alefzero.webdav;
 

+import java.io.File;

 import java.io.IOException;
 import java.io.OutputStream;
 import java.nio.ByteBuffer;
 import java.nio.channels.FileChannel;
 import java.io.IOException;
 import java.io.OutputStream;
 import java.nio.ByteBuffer;
 import java.nio.channels.FileChannel;

+import java.util.Collection;
+import java.util.HashSet;
+import java.util.Iterator;
+import java.util.Set;

 
 import org.apache.commons.httpclient.methods.RequestEntity;
 
 
 import org.apache.commons.httpclient.methods.RequestEntity;
 

+import com.owncloud.android.Log_OC;
+

 import eu.alefzero.webdav.OnDatatransferProgressListener;
 
 import android.util.Log;
 import eu.alefzero.webdav.OnDatatransferProgressListener;
 
 import android.util.Log;


@@ -42,23 +49,27 @@ public class ChunkFromFileChannelRequestEntity implements RequestEntity {
     //private final File mFile;
     private final FileChannel mChannel;
     private final String mContentType;
     //private final File mFile;
     private final FileChannel mChannel;
     private final String mContentType;

-    private final long mSize;
+    private final long mChunkSize;
+    private final File mFile;

     private long mOffset;
     private long mOffset;

-    private OnDatatransferProgressListener mListener;
+    private long mTransferred;
+    Set<OnDatatransferProgressListener> mDataTransferListeners = new HashSet<OnDatatransferProgressListener>();

     private ByteBuffer mBuffer = ByteBuffer.allocate(4096);
 
     private ByteBuffer mBuffer = ByteBuffer.allocate(4096);
 

-    public ChunkFromFileChannelRequestEntity(final FileChannel channel, final String contentType, long size) {
+    public ChunkFromFileChannelRequestEntity(final FileChannel channel, final String contentType, long chunkSize, final File file) {

         super();
         if (channel == null) {
             throw new IllegalArgumentException("File may not be null");
         }
         super();
         if (channel == null) {
             throw new IllegalArgumentException("File may not be null");
         }

-        if (size <= 0) {
-            throw new IllegalArgumentException("Size must be greater than zero");
+        if (chunkSize <= 0) {
+            throw new IllegalArgumentException("Chunk size must be greater than zero");

         }
         mChannel = channel;
         mContentType = contentType;
         }
         mChannel = channel;
         mContentType = contentType;

-        mSize = size;
+        mChunkSize = chunkSize;
+        mFile = file;

         mOffset = 0;
         mOffset = 0;

+        mTransferred = 0;

     }
     
     public void setOffset(long offset) {
     }
     
     public void setOffset(long offset) {


@@ -67,9 +78,9 @@ public class ChunkFromFileChannelRequestEntity implements RequestEntity {
     
     public long getContentLength() {
         try {
     
     public long getContentLength() {
         try {

-            return Math.min(mSize, mChannel.size() - mChannel.position());
+            return Math.min(mChunkSize, mChannel.size() - mChannel.position());

         } catch (IOException e) {
         } catch (IOException e) {

-            return mSize;
+            return mChunkSize;

         }
     }
 
         }
     }
 


@@ -81,25 +92,40 @@ public class ChunkFromFileChannelRequestEntity implements RequestEntity {
         return true;
     }
     
         return true;
     }
     

-    public void setOnDatatransferProgressListener(OnDatatransferProgressListener listener) {
-        mListener = listener;
+    public void addOnDatatransferProgressListener(OnDatatransferProgressListener listener) {
+        mDataTransferListeners.add(listener);
+    }
+    
+    public void addOnDatatransferProgressListeners(Collection<OnDatatransferProgressListener> listeners) {
+        mDataTransferListeners.addAll(listeners);

     }
     
     }
     

+    public void removeOnDatatransferProgressListener(OnDatatransferProgressListener listener) {
+        mDataTransferListeners.remove(listener);
+    }
+    
+    

     public void writeRequest(final OutputStream out) throws IOException {
         int readCount = 0;
     public void writeRequest(final OutputStream out) throws IOException {
         int readCount = 0;

+        Iterator<OnDatatransferProgressListener> it = null;

         
        try {
             mChannel.position(mOffset);
         
        try {
             mChannel.position(mOffset);

-            while (mChannel.position() < mOffset + mSize && mChannel.position() < mChannel.size()) {
+            long size = mFile.length();
+            if (size == 0) size = -1;
+            while (mChannel.position() < mOffset + mChunkSize && mChannel.position() < mChannel.size()) {

                 readCount = mChannel.read(mBuffer);
                 out.write(mBuffer.array(), 0, readCount);
                 mBuffer.clear();
                 readCount = mChannel.read(mBuffer);
                 out.write(mBuffer.array(), 0, readCount);
                 mBuffer.clear();

-                if (mListener != null) 
-                    mListener.transferProgress(readCount);
+                mTransferred += readCount;
+                it = mDataTransferListeners.iterator();
+                while (it.hasNext()) {
+                    it.next().onTransferProgress(readCount, mTransferred, size, mFile.getName());
+                }

             }
             
         } catch (IOException io) {
             }
             
         } catch (IOException io) {

-            Log.e(TAG, io.getMessage());
+            Log_OC.e(TAG, io.getMessage());

             throw new RuntimeException("Ugly solution to workaround the default policy of retries when the server falls while uploading ; temporal fix; really", io);   
             
         }
             throw new RuntimeException("Ugly solution to workaround the default policy of retries when the server falls while uploading ; temporal fix; really", io);   
             
         }