Merge pull request #17 from webdevvie/feature/disable-database-usage

Disable database usage for taskqueue service in use with workers

Author: webdevvie

.gitignore

@@ -0,0 +1,2 @@
+.idea
+.idea/*

Service/TaskQueueService.php

@@ -61,6 +61,11 @@ class TaskQueueService
      */
     private $taskRepo;
 
+    /**
+     * @var boolean
+     */
+    private $databaseDisabled = false;
+
     /**
      * @param EntityManager $entityManager
      * @param PheanstalkConnection $beanstalk
@@ -82,6 +87,16 @@ public function __construct(
         $this->logWorkerOutputOnFailure = $params['log_worker_output_on_failure'];
     }
 
+    /**
+     * Disables the use of the task entity
+     *
+     * @return void
+     */
+    public function disableDatabase()
+    {
+        $this->databaseDisabled = true;
+    }
+
     /**
      * Returns the default tube name
      *
@@ -121,11 +136,13 @@ public function queueTask(TaskDescriptionInterface $task, $tube = null)
         }
         $stringVersion = get_class($task) . '::' . $this->serializer->serialize($task, 'json');
         $taskEntity = new Task($task, $stringVersion, $tube);
-        $this->entityManager->persist($taskEntity);
-        $this->entityManager->flush($taskEntity);
-        $task->setTaskIdentifier($taskEntity->getId());
-        //regenerate it now we have an identifier
-        $stringVersion = get_class($task) . '::' . $this->serializer->serialize($task, 'json');
+        if (!$this->databaseDisabled) {
+            $this->entityManager->persist($taskEntity);
+            $this->entityManager->flush($taskEntity);
+            $task->setTaskIdentifier($taskEntity->getId());
+            //regenerate it now we have an identifier
+            $stringVersion = get_class($task) . '::' . $this->serializer->serialize($task, 'json');
+        }
         $this->beanstalk
             ->useTube($tube)
             ->put($stringVersion);
@@ -228,7 +245,13 @@ public function reserveTask($tube = null)
             $this->beanstalk->delete($inTask);
             throw new TaskQueueServiceException("Invalid data in TaskQueue {$tube}");
         }
-        $taskEntity = $this->taskRepo->find($taskObject->getTaskIdentifier());
+        if (!$this->databaseDisabled) {
+            $taskEntity = $this->taskRepo->find($taskObject->getTaskIdentifier());
+        } else {
+            //remake the task entity
+            $taskEntity = new Task($taskObject, '', $tube);
+        }
+
         if (!($taskEntity instanceof Task)) {
             $this->beanstalk->delete($inTask);
             throw new TaskQueueServiceException(
@@ -287,6 +310,9 @@ public function markFailed(WorkPackage $task, $log)
      */
     private function updateTaskLog(WorkPackage $task, $log)
     {
+        if ($this->databaseDisabled) {
+            return;
+        }
         $taskEntity = $task->getTaskEntity();
         if ($taskEntity instanceof Task) {
             $taskEntity->setLog($log);
@@ -307,6 +333,9 @@ private function updateTaskLog(WorkPackage $task, $log)
      */
     private function updateTaskStatus(WorkPackage $task, $status)
     {
+        if ($this->databaseDisabled) {
+            return;
+        }
         $taskEntity = $task->getTaskEntity();
         if ($taskEntity instanceof Task) {
             $taskEntity->setStatus($status);