Delete newsletter data from woocommerce purchase stats using a repository

[MAILPOET-5845]

mailpoet/lib/Newsletter/NewslettersRepository.php

@@ -21,7 +21,6 @@ use MailPoet\Entities\SendingQueueEntity;
 use MailPoet\Entities\StatisticsClickEntity;
 use MailPoet\Entities\StatisticsNewsletterEntity;
 use MailPoet\Entities\StatisticsOpenEntity;
-use MailPoet\Entities\StatisticsWooCommercePurchaseEntity;
 use MailPoet\Entities\StatsNotificationEntity;
 use MailPoet\Logging\LoggerFactory;
 use MailPoet\Newsletter\Options\NewsletterOptionsRepository;
@@ -32,6 +31,7 @@ use MailPoet\Newsletter\Sending\SendingQueuesRepository;
 use MailPoet\Statistics\StatisticsClicksRepository;
 use MailPoet\Statistics\StatisticsNewslettersRepository;
 use MailPoet\Statistics\StatisticsOpensRepository;
+use MailPoet\Statistics\StatisticsWooCommercePurchasesRepository;
 use MailPoet\Util\Helpers;
 use MailPoet\WP\Functions as WPFunctions;
 use MailPoetVendor\Carbon\Carbon;
@@ -54,6 +54,7 @@ class NewslettersRepository extends Repository {
   private StatisticsClicksRepository $statisticsClicksRepository;
   private StatisticsNewslettersRepository $statisticsNewslettersRepository;
   private StatisticsOpensRepository $statisticsOpensRepository;
+  private StatisticsWooCommercePurchasesRepository $statisticsWooCommercePurchasesRepository;
   private WPFunctions $wp;
 
   public function __construct(
@@ -68,6 +69,7 @@ class NewslettersRepository extends Repository {
     StatisticsClicksRepository $statisticsClicksRepository,
     StatisticsNewslettersRepository $statisticsNewslettersRepository,
     StatisticsOpensRepository $statisticsOpensRepository,
+    StatisticsWooCommercePurchasesRepository $statisticsWooCommercePurchasesRepository,
     WPFunctions $wp
   ) {
     parent::__construct($entityManager);
@@ -82,6 +84,7 @@ class NewslettersRepository extends Repository {
     $this->statisticsClicksRepository = $statisticsClicksRepository;
     $this->statisticsNewslettersRepository = $statisticsNewslettersRepository;
     $this->statisticsOpensRepository = $statisticsOpensRepository;
+    $this->statisticsWooCommercePurchasesRepository = $statisticsWooCommercePurchasesRepository;
     $this->wp = $wp;
   }
 
@@ -419,11 +422,7 @@ class NewslettersRepository extends Repository {
       $this->statisticsClicksRepository->deleteByNewsletterIds($ids);
 
       // Update WooCommerce statistics and remove newsletter and click id
-      $statisticsPurchasesTable = $entityManager->getClassMetadata(StatisticsWooCommercePurchaseEntity::class)->getTableName();
+      $this->statisticsWooCommercePurchasesRepository->removeNewsletterDataByNewsletterIds($ids);
-      $entityManager->getConnection()->executeStatement("
-         UPDATE $statisticsPurchasesTable s
-         SET s.`newsletter_id` = 0 WHERE s.`newsletter_id` IN (:ids)
-      ", ['ids' => $ids], ['ids' => Connection::PARAM_INT_ARRAY]);
 
       // Delete newsletter posts, options, links, and segments
       $this->newsletterPostsRepository->deleteByNewsletterIds($ids);

mailpoet/lib/Statistics/StatisticsWooCommercePurchasesRepository.php

@@ -112,4 +112,21 @@ class StatisticsWooCommercePurchasesRepository extends Repository {
     }, $data);
     return $data;
   }
+
+  /** @param int[] $ids */
+  public function removeNewsletterDataByNewsletterIds(array $ids): void {
+    $this->entityManager->createQueryBuilder()
+      ->update(StatisticsWooCommercePurchaseEntity::class, 'swp')
+      ->set('swp.newsletter', 0)
+      ->where('swp.newsletter IN (:ids)')
+      ->setParameter('ids', $ids)
+      ->getQuery()
+      ->execute();
+
+    // update was done via DQL, make sure the entities are also refreshed in the entity manager
+    $this->refreshAll(function (StatisticsWooCommercePurchaseEntity $entity) use ($ids) {
+      $newsletter = $entity->getNewsletter();
+      return $newsletter && in_array($newsletter->getId(), $ids, true);
+    });
+  }
 }