Cavemanon/piratepoet
8
Fork 0
Code Pull Requests Actions Releases 16 Activity

Use a query for duplication of segment subscribers

Browse Source 
[MAILPOET-3170]
This commit is contained in:
Jan Lysý
2021-01-25 15:34:43 +01:00
committed by Veljko V
parent c579a35980
commit b9474700b8
1 changed files with 13 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
lib/Segments/SegmentSaveController.php
View File

@@ -5,26 +5,20 @@ namespace MailPoet\Segments;
use InvalidArgumentException;
use MailPoet\Entities\SegmentEntity;
use MailPoet\Entities\SubscriberSegmentEntity;
use MailPoet\Subscribers\SubscriberSegmentRepository;
use MailPoetVendor\Doctrine\ORM\EntityManager;
class SegmentSaveController {
/** @var SegmentsRepository */
private $segmentsRepository;
/** @var SubscriberSegmentRepository */
private $subscriberSegmentRepository;
/** @var EntityManager */
private $entityManager;
public function __construct(
SegmentsRepository $segmentsRepository,
SubscriberSegmentRepository $subscriberSegmentRepository,
EntityManager $entityManager
) {
$this->segmentsRepository = $segmentsRepository;
$this->subscriberSegmentRepository = $subscriberSegmentRepository;
$this->entityManager = $entityManager;
}
@@ -44,25 +38,22 @@ class SegmentSaveController {
$this->checkSegmenUniqueName($duplicate->getName(), $duplicate->getId());
$subscriberSegments = $this->subscriberSegmentRepository->findBy(['segment' => $segmentEntity]);
$this->entityManager->transactional(function (EntityManager $entityManager) use ($duplicate, $subscriberSegments) {
$this->entityManager->transactional(function (EntityManager $entityManager) use ($duplicate, $segmentEntity) {
$entityManager->persist($duplicate);
$entityManager->flush();
foreach ($subscriberSegments as $subscriberSegment) {
$subscriber = $subscriberSegment->getSubscriber();
if (!$subscriber) {
continue;
}
$subscriberDuplicate = new SubscriberSegmentEntity(
$duplicate,
$subscriber,
$subscriberSegment->getStatus()
);
$entityManager->persist($subscriberDuplicate);
}
$entityManager->flush();
$subscriberSegmentTable = $entityManager->getClassMetadata(SubscriberSegmentEntity::class)->getTableName();
$conn = $this->entityManager->getConnection();
$stmt = $conn->prepare("
INSERT INTO $subscriberSegmentTable (segment_id, subscriber_id, status, created_at)
SELECT :duplicateId, subscriber_id, status, NOW()
FROM $subscriberSegmentTable
WHERE segment_id = :segmentId
");
$stmt->execute([
'duplicateId' => $duplicate->getId(),
'segmentId' => $segmentEntity->getId(),
]);
});
return $duplicate;