1 files changed, 6 insertions, 10 deletions
diff --git a/src/FileAllocationCommand.h b/src/FileAllocationCommand.h
index 0ca78937..22d77a07 100644
--- a/src/FileAllocationCommand.h
+++ b/src/FileAllocationCommand.h
@@ -35,29 +35,25 @@
 #ifndef _D_FILE_ALLOCATION_COMMAND_H_
 #define _D_FILE_ALLOCATION_COMMAND_H_
 
-#include "Command.h"
+#include "RealtimeCommand.h"
 #include "Request.h"
-#include "RequestGroup.h"
-#include "DownloadEngine.h"
 #include "TimeA2.h"
 #include "FileAllocationEntry.h"
 
-class FileAllocationCommand : public Command {
+class FileAllocationCommand : public RealtimeCommand {
 private:
   RequestHandle _req;
-  RequestGroup* _requestGroup;
-  DownloadEngine* _e;
   FileAllocationEntryHandle _fileAllocationEntry;
   Time _timer;
 public:
   FileAllocationCommand(int cuid, const RequestHandle& req, RequestGroup* requestGroup, DownloadEngine* e, const FileAllocationEntryHandle& fileAllocationEntry):
-    Command(cuid),
+    RealtimeCommand(cuid, requestGroup, e),
     _req(req),
-    _requestGroup(requestGroup),
-    _e(e),
     _fileAllocationEntry(fileAllocationEntry) {}
 
-  virtual bool execute();
+  virtual bool executeInternal();
+
+  virtual bool handleException(Exception* e);
 };
 
 #endif // _D_FILE_ALLOCATION_COMMAND_H_