fix unread message bug

st-universe · Nov 28, 2023 · 0a8689d · 0a8689d
1 parent 90150e5
commit 0a8689d
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 3 deletions.
diff --git a/src/Module/Message/Lib/PrivateMessageListItem.php b/src/Module/Message/Lib/PrivateMessageListItem.php
@@ -64,7 +64,7 @@ public function isMarkableAsNew(): bool
         }
         $this->message->setNew(false);
 
-        $this->privateMessageRepository->save($this->message);
+        $this->privateMessageRepository->save($this->message, true);
 
         return true;
     }

diff --git a/src/Orm/Repository/PrivateMessageRepository.php b/src/Orm/Repository/PrivateMessageRepository.php
@@ -20,11 +20,15 @@ public function prototype(): PrivateMessageInterface
         return new PrivateMessage();
     }
 
-    public function save(PrivateMessageInterface $post): void
+    public function save(PrivateMessageInterface $post, bool $doFlush = false): void
     {
         $em = $this->getEntityManager();
 
         $em->persist($post);
+
+        if ($doFlush) {
+            $em->flush();
+        }
     }
 
     public function delete(PrivateMessageInterface $post): void

diff --git a/src/Orm/Repository/PrivateMessageRepositoryInterface.php b/src/Orm/Repository/PrivateMessageRepositoryInterface.php
@@ -18,7 +18,7 @@ interface PrivateMessageRepositoryInterface extends ObjectRepository
 {
     public function prototype(): PrivateMessageInterface;
 
-    public function save(PrivateMessageInterface $post): void;
+    public function save(PrivateMessageInterface $post, bool $doFlush = false): void;
 
     public function delete(PrivateMessageInterface $post): void;