From 474bcd62a33f0b0b296c5000c3382c446aa92a16 Mon Sep 17 00:00:00 2001
From: Pablo Ogando Ferreira <pablo.ogando@teltek.es>
Date: Fri, 2 Feb 2024 09:48:25 +0100
Subject: [PATCH] TTK-26792 fix MongoDB save

---
 Command/ImportPlaylistFromYouTubeChannel.php | 3 ++-
 Command/ImportVideosFromYouTubeChannel.php   | 3 ++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/Command/ImportPlaylistFromYouTubeChannel.php b/Command/ImportPlaylistFromYouTubeChannel.php
index 6a1d079..81547c6 100644
--- a/Command/ImportPlaylistFromYouTubeChannel.php
+++ b/Command/ImportPlaylistFromYouTubeChannel.php
@@ -122,7 +122,8 @@ protected function execute(InputInterface $input, OutputInterface $output): int
             }
         } while (null !== $nextPageToken);
 
-        $this->mongoDBFlush($count);
+        $this->documentManager->flush();
+        $this->documentManager->clear();
 
         $progressBar->finish();
         $output->writeln(' ');
diff --git a/Command/ImportVideosFromYouTubeChannel.php b/Command/ImportVideosFromYouTubeChannel.php
index 9a1585c..55f53e8 100644
--- a/Command/ImportVideosFromYouTubeChannel.php
+++ b/Command/ImportVideosFromYouTubeChannel.php
@@ -166,7 +166,8 @@ protected function execute(InputInterface $input, OutputInterface $output): int
             }
         } while (null !== $nextPageToken);
 
-        $this->mongoDBFlush($count);
+        $this->documentManager->flush();
+        $this->documentManager->clear();
 
         $progressBar->finish();
         $output->writeln(' ');