src/com/owncloud/android/operations/ChunkedUploadFileOperation.java

   1 /* ownCloud Android client application
   2  *   Copyright (C) 2012 Bartek Przybylski
   3  *   Copyright (C) 2012-2013 ownCloud Inc.
   4  *
   5  *   This program is free software: you can redistribute it and/or modify
   6  *   it under the terms of the GNU General Public License version 2,
   7  *   as published by the Free Software Foundation.
   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 com.owncloud.android.operations;
  20
  21 import java.io.File;
  22 import java.io.IOException;
  23 import java.io.RandomAccessFile;
  24 import java.nio.channels.FileChannel;
  25 import java.util.Random;
  26
  27 import org.apache.commons.httpclient.HttpException;
  28 import org.apache.commons.httpclient.methods.PutMethod;
  29
  30 import com.owncloud.android.datamodel.OCFile;
  31 import com.owncloud.android.oc_framework.network.ProgressiveDataTransferer;
  32 import com.owncloud.android.oc_framework.network.webdav.ChunkFromFileChannelRequestEntity;
  33 import com.owncloud.android.oc_framework.network.webdav.WebdavClient;
  34 import com.owncloud.android.oc_framework.network.webdav.WebdavUtils;
  35 import com.owncloud.android.utils.Log_OC;
  36
  37
  38 import android.accounts.Account;
  39 import android.content.Context;
  40
  41
  42 public class ChunkedUploadFileOperation extends UploadFileOperation {
  43
  44     private static final long CHUNK_SIZE = 1024000;
  45     private static final String OC_CHUNKED_HEADER = "OC-Chunked";
  46     private static final String TAG = ChunkedUploadFileOperation.class.getSimpleName();
  47
  48     public ChunkedUploadFileOperation(  Account account,
  49                                         OCFile file,
  50                                         boolean isInstant,
  51                                         boolean forceOverwrite,
  52                                         int localBehaviour, Context context) {
  53
  54         super(account, file, isInstant, forceOverwrite, localBehaviour, context);
  55     }
  56
  57     @Override
  58     protected int uploadFile(WebdavClient client) throws HttpException, IOException {
  59         int status = -1;
  60
  61         FileChannel channel = null;
  62         RandomAccessFile raf = null;
  63         try {
  64             File file = new File(getStoragePath());
  65             raf = new RandomAccessFile(file, "r");
  66             channel = raf.getChannel();
  67             mEntity = new ChunkFromFileChannelRequestEntity(channel, getMimeType(), CHUNK_SIZE, file);
  68             ((ProgressiveDataTransferer)mEntity).addDatatransferProgressListeners(getDataTransferListeners());
  69             long offset = 0;
  70             String uriPrefix = client.getBaseUri() + WebdavUtils.encodePath(getRemotePath()) + "-chunking-" + Math.abs((new Random()).nextInt(9000)+1000) + "-" ;
  71             long chunkCount = (long) Math.ceil((double)file.length() / CHUNK_SIZE);
  72             for (int chunkIndex = 0; chunkIndex < chunkCount ; chunkIndex++, offset += CHUNK_SIZE) {
  73                 if (mPutMethod != null) {
  74                     mPutMethod.releaseConnection();    // let the connection available for other methods
  75                 }
  76                 mPutMethod = new PutMethod(uriPrefix + chunkCount + "-" + chunkIndex);
  77                 mPutMethod.addRequestHeader(OC_CHUNKED_HEADER, OC_CHUNKED_HEADER);
  78                 ((ChunkFromFileChannelRequestEntity)mEntity).setOffset(offset);
  79                 mPutMethod.setRequestEntity(mEntity);
  80                 status = client.executeMethod(mPutMethod);
  81                 client.exhaustResponse(mPutMethod.getResponseBodyAsStream());
  82                 Log_OC.d(TAG, "Upload of " + getStoragePath() + " to " + getRemotePath() + ", chunk index " + chunkIndex + ", count " + chunkCount + ", HTTP result status " + status);
  83                 if (!isSuccess(status))
  84                     break;
  85             }
  86
  87         } finally {
  88             if (channel != null)
  89                 channel.close();
  90             if (raf != null)
  91                 raf.close();
  92             if (mPutMethod != null)
  93                 mPutMethod.releaseConnection();    // let the connection available for other methods
  94         }
  95         return status;
  96     }
  97
  98 }