[CronJob] Integrate DoctrineEnvelopeEntityReference custom exception

mpoiriert · Dec 13, 2024 · 4a1b6d2 · 4a1b6d2
1 parent bcb00c8
commit 4a1b6d2
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/packages/cron-job/Message/ExecuteCronJobMessage.php b/packages/cron-job/Message/ExecuteCronJobMessage.php
@@ -6,14 +6,15 @@
 
 use Draw\Component\CronJob\Entity\CronJobExecution;
 use Draw\Component\Messenger\AutoStamp\Message\StampingAwareInterface;
+use Draw\Component\Messenger\DoctrineEnvelopeEntityReference\Exception\ObjectNotFoundException;
 use Draw\Component\Messenger\DoctrineEnvelopeEntityReference\Message\DoctrineReferenceAwareInterface;
+use Draw\Component\Messenger\DoctrineEnvelopeEntityReference\Stamp\PropertyReferenceStamp;
 use Symfony\Component\Messenger\Bridge\Amqp\Transport\AmqpStamp;
 use Symfony\Component\Messenger\Envelope;
-use Symfony\Component\Messenger\Exception\UnrecoverableMessageHandlingException;
 
 class ExecuteCronJobMessage implements DoctrineReferenceAwareInterface, StampingAwareInterface
 {
-    private ?CronJobExecution $execution;
+    private PropertyReferenceStamp|CronJobExecution|null $execution;
 
     public function __construct(CronJobExecution $execution)
     {
@@ -22,8 +23,8 @@ public function __construct(CronJobExecution $execution)
 
     public function getExecution(): CronJobExecution
     {
-        if (null === $this->execution) {
-            throw new UnrecoverableMessageHandlingException('CronJobExecution is not set.');
+        if (!$this->execution instanceof CronJobExecution) {
+            throw new ObjectNotFoundException(CronJobExecution::class, $this->execution);
         }
 
         return $this->execution;