From eef212d72ae5511a36b7f6bc90cdff93ecd374aa Mon Sep 17 00:00:00 2001
From: kwkam <kwkam@users.noreply.github.com>
Date: Wed, 8 Nov 2017 15:17:47 +0800
Subject: [PATCH] UnknownLengthPieceStorage: return piece length

show something in console status when downloading items
with unknown content length
---
 src/UnknownLengthPieceStorage.cc | 9 +++++++++
 src/UnknownLengthPieceStorage.h  | 6 +-----
 2 files changed, 10 insertions(+), 5 deletions(-)

diff --git a/src/UnknownLengthPieceStorage.cc b/src/UnknownLengthPieceStorage.cc
index b7e1ce84..4ea14115 100644
--- a/src/UnknownLengthPieceStorage.cc
+++ b/src/UnknownLengthPieceStorage.cc
@@ -210,6 +210,15 @@ bool UnknownLengthPieceStorage::isPieceUsed(size_t index)
   }
 }
 
+int64_t UnknownLengthPieceStorage::getCompletedLength()
+{
+  // TODO we have to return actual completed length here?
+  if (piece_) {
+    return piece_->getLength();
+  }
+  return totalLength_;
+}
+
 std::shared_ptr<DiskAdaptor> UnknownLengthPieceStorage::getDiskAdaptor()
 {
   return diskAdaptor_;
diff --git a/src/UnknownLengthPieceStorage.h b/src/UnknownLengthPieceStorage.h
index 9c3845dd..2b05918c 100644
--- a/src/UnknownLengthPieceStorage.h
+++ b/src/UnknownLengthPieceStorage.h
@@ -162,11 +162,7 @@ public:
     return totalLength_;
   }
 
-  virtual int64_t getCompletedLength() CXX11_OVERRIDE
-  {
-    // TODO we have to return actual completed length here?
-    return totalLength_;
-  }
+  virtual int64_t getCompletedLength() CXX11_OVERRIDE;
 
   virtual int64_t getFilteredCompletedLength() CXX11_OVERRIDE
   {