Dtc\QueueBundle\Doctrine\BaseRunManager

Complexity

Total Complexity

16

Size/Duplication

Total Lines	126
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	4

Test Coverage

Coverage

86.36%

Importance

Changes

0

12 Methods

Rating	Name	Duplication	Size	Complexity
A	getObjectManager()	0	4	1
A	setObjectManager()	0	4	1
A	getRepository()	0	4	1
A	getRunArchiveClass()	0	4	1
A	setRunArchiveClass()	0	4	1
A	recordJobRun()	0	14	2
A	pruneStalledRuns()	0	19	4
A	deleteOldRuns()	0	14	3
	getOldLiveRuns()	0	1	?
	removeOlderThan()	0	1	?
A	pruneArchivedRuns()	0	7	1
A	pruneJobTimings()	0	7	1

<?php

namespace Dtc\QueueBundle\Doctrine;

use Doctrine\Common\Persistence\ObjectManager;
use Doctrine\Common\Persistence\ObjectRepository;
use Dtc\QueueBundle\Model\Job;
use Dtc\QueueBundle\Model\JobTiming;
use Dtc\QueueBundle\Model\Run;
use Dtc\QueueBundle\Model\RunManager;

abstract class BaseRunManager extends RunManager
{
    /** @var ObjectManager */
    protected $objectManager;

    /** @var string|null */
    protected $runArchiveClass;

    /**
     * @return ObjectManager
     */
    public function getObjectManager()
    {
        return $this->objectManager;
    }

    /**
     * @param ObjectManager $objectManager
     */
    public function setObjectManager(ObjectManager $objectManager)
    {
        $this->objectManager = $objectManager;
    }

    /**
     * @return ObjectRepository
     */
    public function getRepository()
    {
        return $this->objectManager->getRepository($this->getRunClass());
    }

    /**
     * @return null|string
     */
    public function getRunArchiveClass()
    {
        return $this->runArchiveClass;
    }

    /**
     * @param null|string $runArchiveClass
     */
    public function setRunArchiveClass($runArchiveClass)
    {
        $this->runArchiveClass = $runArchiveClass;
    }

    public function recordJobRun(Job $job)
    {
        parent::recordJobRun($job);

        $finishedAt = $job->getFinishedAt();
        if (null === $finishedAt) {
            $finishedAt = new \DateTime();
        }
        /** @var JobTiming $jobTiming */
        $jobTiming = new $this->jobTimingClass();
        $jobTiming->setFinishedAt($finishedAt);
        $this->objectManager->persist($jobTiming);
        $this->objectManager->flush();
    }

    public function pruneStalledRuns()
    {
        $runs = $this->getOldLiveRuns();
        /** @var Run $run */
        $delete = [];

        foreach ($runs as $run) {
            $lastHeartbeat = $run->getLastHeartbeatAt();
            $time = time() - 3600;
            $processTimeout = $run->getProcessTimeout();
            $time -= $processTimeout;
            $oldDate = new \DateTime("@$time");
            if (null === $lastHeartbeat || $oldDate > $lastHeartbeat) {
                $delete[] = $run;
            }
        }

        return $this->deleteOldRuns($delete);
    }

    /**
     * @param array $delete
     *
     * @return int
     */
    protected function deleteOldRuns(array $delete)
    {
        $count = count($delete);
        $objectManager = $this->getObjectManager();
        for ($i = 0; $i < $count; $i += 100) {
            $deleteList = array_slice($delete, $i, 100);
            foreach ($deleteList as $object) {
                $objectManager->remove($object);
            }
            $objectManager->flush();
        }

        return $count;
    }

    /**
     * @return array
     */
    abstract protected function getOldLiveRuns();

    abstract protected function removeOlderThan(ObjectManager $objectManager, $objectName, $field, \DateTime $olderThan);

    public function pruneArchivedRuns(\DateTime $olderThan)
    {
        /** @var ObjectManager $objectManager */
        $objectManager = $this->getObjectManager();

        return $this->removeOlderThan($objectManager, $this->getRunArchiveClass(), 'endedAt', $olderThan);
    }

    public function pruneJobTimings(\DateTime $olderThan)
    {
        /** @var ObjectManager $objectManager */
        $objectManager = $this->getObjectManager();

        return $this->removeOlderThan($objectManager, $this->getJobTimingClass(), 'createdAt', $olderThan);
    }
}