From ccdd5b31a3dfa07cd13d6f7f0f5eaa3d8ff8a61e Mon Sep 17 00:00:00 2001
From: Tatsuhiro Tsujikawa <tatsuhiro.t@gmail.com>
Date: Thu, 31 May 2007 15:56:20 +0000
Subject: 2007-06-01  Tatsuhiro Tsujikawa  <tujikawa at rednoah dot com>

	* src/FileAllocationCommand.cc: Derived from RealtimeCommand.
	* src/CheckIntegrityCommand.cc: Derived from RealtimeCommand.

	* src/MetalinkEntry.h
	(checksum): Changed to ChecksumHandle
	* src/MetalinkRequestInfo.cc
	(checksum): Changed to ChecksumHandle

	* src/File.cc
	(mkdirs): OPEN_MODE -> DIR_OPEN_MODE
	* src/common.h
	(DIR_OPEN_MODE): New definition

	* src/RequestGroup.cc
	(prepareForNextAction): Added an argument.

	* src/message.h
	(MSG_GOOD_CHECKSUM): New definition
	(MSG_BAD_CHECKSUM): New definition

	* src/HttpResponseCommand.cc
	(handleDefaultEncoding): Continue download sequence in new
	non-segmented download.

	* src/FileAllocationEntry.h
	(_nextDownloadCommand): New variable.

	* src/DownloadCommand.cc
	(prepareForNextSegment): Create ChecksumCommand if checksum is
	available.

	* src/RealtimeCommand.h, src/RealtimeCommand.cc: New class.

	* src/IteratableChecksumValidator.h,
	src/IteratableChecksumValidator.cc: New class.

	* src/ChecksumCommand.h, src/ChecksumCommand.cc: New class.
---
 src/FileAllocationCommand.cc | 52 ++++++++++++++++++++++++--------------------
 1 file changed, 28 insertions(+), 24 deletions(-)

(limited to 'src/FileAllocationCommand.cc')

diff --git a/src/FileAllocationCommand.cc b/src/FileAllocationCommand.cc
index 92f26b6f..ae9750cf 100644
--- a/src/FileAllocationCommand.cc
+++ b/src/FileAllocationCommand.cc
@@ -35,39 +35,43 @@
 #include "FileAllocationCommand.h"
 #include "InitiateConnectionCommandFactory.h"
 #include "message.h"
+#include "DownloadCommand.h"
 
-bool FileAllocationCommand::execute()
+bool FileAllocationCommand::executeInternal()
 {
-  setStatusRealtime();
-  _e->noWait = true;
-  try {
-    _fileAllocationEntry->allocateChunk();
-
-    if(_fileAllocationEntry->allocationFinished()) {
-      int64_t totalLength = _requestGroup->getSegmentMan()->totalSize;
-      logger->debug("%d seconds to allocate %lld byte(s)",
-		    _timer.difference(), totalLength);
+  _fileAllocationEntry->allocateChunk();
+  
+  if(_fileAllocationEntry->allocationFinished()) {
+    int64_t totalLength = _requestGroup->getSegmentMan()->totalSize;
+    logger->debug("%d seconds to allocate %lld byte(s)",
+		  _timer.difference(), totalLength);
     
-      _e->_fileAllocationMan->markCurrentFileAllocationEntryDone();
-
+    _e->_fileAllocationMan->markCurrentFileAllocationEntryDone();
+    
+    if(_fileAllocationEntry->getNextDownloadCommand()) {
+      _e->commands.push_back(_fileAllocationEntry->getNextDownloadCommand());
+      _fileAllocationEntry->setNextDownloadCommand(0);
+    } else {
       int32_t numCommandsToGenerate = 15;
       Commands commands = _requestGroup->getNextCommand(_e, numCommandsToGenerate);
-
+      
       Command* command = InitiateConnectionCommandFactory::createInitiateConnectionCommand(cuid, _req, _requestGroup, _e);
-
+      
       commands.push_front(command);
-
+    
       _e->addCommand(commands);
-      return true;
-    } else {
-      _e->commands.push_back(this);
-      return false;
     }
-  } catch(Exception* e) {
-    _requestGroup->getSegmentMan()->errors++;
-    logger->error("CUID#%d - Exception caught while allocating file space.", e, cuid);
-    delete e;
-    logger->notice(MSG_DOWNLOAD_NOT_COMPLETE, cuid, _requestGroup->getFilePath().c_str());
     return true;
+  } else {
+    _e->commands.push_back(this);
+    return false;
   }
 }
+
+bool FileAllocationCommand::handleException(Exception* e)
+{
+  logger->error("CUID#%d - Exception caught while allocating file space.", e, cuid);
+  delete e;
+  logger->error(MSG_DOWNLOAD_NOT_COMPLETE, cuid, _requestGroup->getFilePath().c_str());
+  return true;
+}
-- 
cgit