From 3c00bfae896f920cdfffb0c4b140f91afe6529c2 Mon Sep 17 00:00:00 2001
From: Kavin <20838718+FireMasterK@users.noreply.github.com>
Date: Thu, 6 Oct 2022 11:04:51 +0100
Subject: [PATCH] Query all videos at once and batch update.

---
 .../me/kavin/piped/utils/DatabaseHelper.java  |  9 +++++
 .../me/kavin/piped/utils/DownloaderImpl.java  |  1 +
 .../me/kavin/piped/utils/ResponseHelper.java  | 38 +++++++++++++++++--
 3 files changed, 45 insertions(+), 3 deletions(-)

diff --git a/src/main/java/me/kavin/piped/utils/DatabaseHelper.java b/src/main/java/me/kavin/piped/utils/DatabaseHelper.java
index a0b5b8f..56b14a1 100644
--- a/src/main/java/me/kavin/piped/utils/DatabaseHelper.java
+++ b/src/main/java/me/kavin/piped/utils/DatabaseHelper.java
@@ -74,6 +74,15 @@ public class DatabaseHelper {
         return s.createQuery(cr).uniqueResult();
     }
 
+    public static List<Video> getVideosFromIds(SharedSessionContract s, Collection<String> ids) {
+        CriteriaBuilder cb = s.getCriteriaBuilder();
+        CriteriaQuery<Video> cr = cb.createQuery(Video.class);
+        Root<Video> root = cr.from(Video.class);
+        cr.select(root).where(root.get("id").in(ids));
+
+        return s.createQuery(cr).list();
+    }
+
     public static Video getVideoFromId(String id) {
         try (StatelessSession s = DatabaseSessionFactory.createStatelessSession()) {
             return getVideoFromId(s, id);
diff --git a/src/main/java/me/kavin/piped/utils/DownloaderImpl.java b/src/main/java/me/kavin/piped/utils/DownloaderImpl.java
index 0be70f9..c850a1c 100644
--- a/src/main/java/me/kavin/piped/utils/DownloaderImpl.java
+++ b/src/main/java/me/kavin/piped/utils/DownloaderImpl.java
@@ -30,6 +30,7 @@ public class DownloaderImpl extends Downloader {
     final AsyncLoadingCache<Request, Response> responseCache = Caffeine.newBuilder()
             .expireAfterWrite(1, TimeUnit.MINUTES)
             .scheduler(Scheduler.systemScheduler())
+            .executor(Multithreading.getCachedExecutor())
             .maximumSize(100).buildAsync(this::executeRequest);
 
     @Override
diff --git a/src/main/java/me/kavin/piped/utils/ResponseHelper.java b/src/main/java/me/kavin/piped/utils/ResponseHelper.java
index 2cb6e1c..64c8c70 100644
--- a/src/main/java/me/kavin/piped/utils/ResponseHelper.java
+++ b/src/main/java/me/kavin/piped/utils/ResponseHelper.java
@@ -37,6 +37,7 @@ import org.schabi.newpipe.extractor.channel.ChannelInfoItem;
 import org.schabi.newpipe.extractor.comments.CommentsInfo;
 import org.schabi.newpipe.extractor.comments.CommentsInfoItem;
 import org.schabi.newpipe.extractor.exceptions.ExtractionException;
+import org.schabi.newpipe.extractor.exceptions.ParsingException;
 import org.schabi.newpipe.extractor.kiosk.KioskExtractor;
 import org.schabi.newpipe.extractor.kiosk.KioskInfo;
 import org.schabi.newpipe.extractor.kiosk.KioskList;
@@ -231,7 +232,7 @@ public class ResponseHelper {
 
             me.kavin.piped.utils.obj.db.Channel channel = DatabaseHelper.getChannelFromId(info.getId());
 
-            try (Session s = DatabaseSessionFactory.createSession()) {
+            try (StatelessSession s = DatabaseSessionFactory.createStatelessSession()) {
 
                 if (channel != null) {
                     if (channel.isVerified() != info.isVerified()
@@ -239,9 +240,29 @@ public class ResponseHelper {
                         channel.setVerified(info.isVerified());
                         channel.setUploaderAvatar(info.getAvatarUrl());
                         var tr = s.beginTransaction();
-                        s.merge(channel);
+                        s.update(channel);
                         tr.commit();
                     }
+
+                    Set<String> ids = info.getRelatedItems()
+                            .stream()
+                            .filter(item -> {
+                                long time = item.getUploadDate() != null
+                                        ? item.getUploadDate().offsetDateTime().toInstant().toEpochMilli()
+                                        : System.currentTimeMillis();
+                                return System.currentTimeMillis() - time < TimeUnit.DAYS.toMillis(Constants.FEED_RETENTION);
+                            })
+                            .map(item -> {
+                                try {
+                                    return YOUTUBE_SERVICE.getStreamLHFactory().getId(item.getUrl());
+                                } catch (ParsingException e) {
+                                    throw new RuntimeException(e);
+                                }
+                            })
+                            .collect(Collectors.toUnmodifiableSet());
+
+                    List<Video> videos = DatabaseHelper.getVideosFromIds(s, ids);
+
                     for (StreamInfoItem item : info.getRelatedItems()) {
                         long time = item.getUploadDate() != null
                                 ? item.getUploadDate().offsetDateTime().toInstant().toEpochMilli()
@@ -249,7 +270,14 @@ public class ResponseHelper {
                         if (System.currentTimeMillis() - time < TimeUnit.DAYS.toMillis(Constants.FEED_RETENTION))
                             try {
                                 String id = YOUTUBE_SERVICE.getStreamLHFactory().getId(item.getUrl());
-                                updateVideo(id, item, time, true);
+                                var video = videos.stream()
+                                        .filter(v -> v.getId().equals(id))
+                                        .findFirst();
+                                if (video.isPresent()) {
+                                    updateVideo(s, video.get(), item);
+                                } else {
+                                    handleNewVideo("https://youtube.com/watch?v=" + id, time, channel);
+                                }
                             } catch (Exception e) {
                                 ExceptionHandler.handle(e);
                             }
@@ -1624,6 +1652,10 @@ public class ResponseHelper {
         });
     }
 
+    private static void updateVideo(StatelessSession s, Video video, StreamInfoItem item) {
+        updateVideo(s, video, item.getViewCount(), item.getDuration(), item.getName());
+    }
+
     private static void updateVideo(StatelessSession s, Video video, long views, long duration, String title) {
 
         boolean changed = false;