|
| 1 | +<?php |
| 2 | + |
| 3 | +namespace Oro\Bridge\MarketingCRM\Provider; |
| 4 | + |
| 5 | +use Doctrine\Common\Persistence\ManagerRegistry; |
| 6 | +use Doctrine\DBAL\Connection; |
| 7 | +use Doctrine\ORM\EntityRepository; |
| 8 | +use Doctrine\ORM\NoResultException; |
| 9 | +use Doctrine\ORM\QueryBuilder; |
| 10 | +use Oro\Bundle\ConfigBundle\Config\ConfigManager; |
| 11 | +use Oro\Bundle\FeatureToggleBundle\Checker\FeatureCheckerHolderTrait; |
| 12 | +use Oro\Bundle\FeatureToggleBundle\Checker\FeatureToggleableInterface; |
| 13 | +use Oro\Bundle\SecurityBundle\ORM\Walker\AclHelper; |
| 14 | +use Oro\Bundle\TrackingBundle\Entity\UniqueTrackingVisit; |
| 15 | + |
| 16 | +class AbstractPrecalculatedVisitProvider implements FeatureToggleableInterface |
| 17 | +{ |
| 18 | + use FeatureCheckerHolderTrait; |
| 19 | + |
| 20 | + /** |
| 21 | + * @var ManagerRegistry |
| 22 | + */ |
| 23 | + private $registry; |
| 24 | + |
| 25 | + /** |
| 26 | + * @var AclHelper |
| 27 | + */ |
| 28 | + private $aclHelper; |
| 29 | + |
| 30 | + /** |
| 31 | + * @var ConfigManager |
| 32 | + */ |
| 33 | + private $configManager; |
| 34 | + |
| 35 | + /** |
| 36 | + * @param ManagerRegistry $registry |
| 37 | + * @param ConfigManager $configManager |
| 38 | + * @param AclHelper $aclHelper |
| 39 | + */ |
| 40 | + public function __construct( |
| 41 | + ManagerRegistry $registry, |
| 42 | + ConfigManager $configManager, |
| 43 | + AclHelper $aclHelper |
| 44 | + ) { |
| 45 | + $this->registry = $registry; |
| 46 | + $this->aclHelper = $aclHelper; |
| 47 | + $this->configManager = $configManager; |
| 48 | + } |
| 49 | + |
| 50 | + /** |
| 51 | + * @param QueryBuilder $queryBuilder |
| 52 | + * @return int |
| 53 | + */ |
| 54 | + protected function getSingleIntegerResult(QueryBuilder $queryBuilder) |
| 55 | + { |
| 56 | + try { |
| 57 | + return (int)$this->aclHelper->apply($queryBuilder)->getSingleScalarResult(); |
| 58 | + } catch (NoResultException $ex) { |
| 59 | + return 0; |
| 60 | + } |
| 61 | + } |
| 62 | + |
| 63 | + /** |
| 64 | + * @param QueryBuilder $queryBuilder |
| 65 | + * @param \DateTime $from |
| 66 | + * @param \DateTime $to |
| 67 | + */ |
| 68 | + protected function applyDateLimit(QueryBuilder $queryBuilder, \DateTime $from, \DateTime $to) |
| 69 | + { |
| 70 | + if ($from && $to && $this->getDate($from) === $this->getDate($to)) { |
| 71 | + $queryBuilder->andWhere($queryBuilder->expr()->eq('t.actionDate', ':date')) |
| 72 | + ->setParameter('date', $this->getDate($from)); |
| 73 | + } else { |
| 74 | + if ($from) { |
| 75 | + $queryBuilder |
| 76 | + ->andWhere($queryBuilder->expr()->gte('t.actionDate', ':from')) |
| 77 | + ->setParameter('from', $this->getDate($from)); |
| 78 | + } |
| 79 | + if ($to) { |
| 80 | + $queryBuilder |
| 81 | + ->andWhere($queryBuilder->expr()->lte('t.actionDate', ':to')) |
| 82 | + ->setParameter('to', $this->getDate($to)); |
| 83 | + } |
| 84 | + } |
| 85 | + } |
| 86 | + |
| 87 | + /** |
| 88 | + * @return QueryBuilder |
| 89 | + */ |
| 90 | + protected function createUniqueVisitQueryBuilder() |
| 91 | + { |
| 92 | + $queryBuilder = $this |
| 93 | + ->getUniqueTrackingVisitRepository() |
| 94 | + ->createQueryBuilder('t'); |
| 95 | + |
| 96 | + return $queryBuilder; |
| 97 | + } |
| 98 | + |
| 99 | + /** |
| 100 | + * @return bool |
| 101 | + */ |
| 102 | + protected function isPrecalculatedStatisticEnabled() |
| 103 | + { |
| 104 | + return $this->configManager->get('oro_tracking.precalculated_statistic_enabled'); |
| 105 | + } |
| 106 | + |
| 107 | + /** |
| 108 | + * @return EntityRepository |
| 109 | + */ |
| 110 | + private function getUniqueTrackingVisitRepository() |
| 111 | + { |
| 112 | + return $this->registry->getManagerForClass(UniqueTrackingVisit::class) |
| 113 | + ->getRepository(UniqueTrackingVisit::class); |
| 114 | + } |
| 115 | + |
| 116 | + /** |
| 117 | + * @param \DateTime $dateTime |
| 118 | + * @return string |
| 119 | + */ |
| 120 | + private function getDate(\DateTime $dateTime) |
| 121 | + { |
| 122 | + /** @var Connection $connection */ |
| 123 | + $connection = $this->registry->getConnection(); |
| 124 | + $dateFormat = $connection->getDatabasePlatform()->getDateFormatString(); |
| 125 | + |
| 126 | + return $dateTime->format($dateFormat); |
| 127 | + } |
| 128 | +} |
0 commit comments