chrisbryan17's picture
Upload folder using huggingface_hub
d2897cd verified
<?php
namespace Mautic\EmailBundle\Stat;
use Mautic\EmailBundle\Entity\Stat;
use Mautic\EmailBundle\Model\EmailStatModel;
use Mautic\EmailBundle\Stat\Exception\StatNotFoundException;
class StatHelper
{
/**
* Just store email ID and lead ID to avoid doctrine RAM issues with entities.
*
* @var Reference[]
*/
private $stats = [];
/**
* @var array
*/
private $deleteUs = [];
public function __construct(
private EmailStatModel $emailStatModel
) {
}
public function storeStat(Stat $stat, $emailAddress): void
{
$this->emailStatModel->saveEntity($stat);
// to avoid Doctrine RAM issues, we're just going to hold onto ID references
$this->stats[$emailAddress] = new Reference($stat);
// clear stat from doctrine memory
$this->emailStatModel->getRepository()->detachEntity($stat);
}
public function deletePending(): void
{
if (count($this->deleteUs)) {
$this->emailStatModel->getRepository()->deleteStats($this->deleteUs);
}
}
public function markForDeletion(Reference $stat): void
{
$this->deleteUs[] = $stat->getStatId();
}
/**
* @return Reference
*
* @throws StatNotFoundException
*/
public function getStat($emailAddress)
{
if (!isset($this->stats[$emailAddress])) {
throw new StatNotFoundException();
}
return $this->stats[$emailAddress];
}
public function reset(): void
{
$this->deleteUs = [];
$this->stats = [];
}
}