Weight episodes and listening time to define top podcasts

Automattic · Oct 31, 2023 · 844bd08 · 844bd08
1 parent 0c286c8
commit 844bd08
Showing 1 changed file with 12 additions and 9 deletions.
diff --git a/Modules/DataModel/Sources/PocketCastsDataModel/Private/Managers/EndOfYearDataManager.swift b/Modules/DataModel/Sources/PocketCastsDataModel/Private/Managers/EndOfYearDataManager.swift
@@ -218,15 +218,16 @@ class EndOfYearDataManager {
         dbQueue.inDatabase { db in
             do {
                 let query = """
-                            SELECT DISTINCT \(DataManager.episodeTableName).uuid,
+                            SELECT (totalPlayedTime * 0.4) * (played_episodes * 0.6) as weighted, * FROM
+                            (SELECT DISTINCT \(DataManager.episodeTableName).uuid as episodeUuid,
                                 SUM(playedUpTo) as totalPlayedTime,
                                 COUNT(\(DataManager.episodeTableName).id) as played_episodes,
                                 \(DataManager.podcastTableName).*
                             FROM \(DataManager.episodeTableName), \(DataManager.podcastTableName)
                             WHERE `\(DataManager.podcastTableName)`.uuid = `\(DataManager.episodeTableName)`.podcastUuid and
                                 \(listenedEpisodesThisYear)
-                            GROUP BY podcastUuid
-                            ORDER BY totalPlayedTime DESC
+                            GROUP BY podcastUuid)
+                            ORDER BY weighted DESC
                             LIMIT \(limit)
                             """
                 let resultSet = try db.executeQuery(query, values: nil)
@@ -243,12 +244,14 @@ class EndOfYearDataManager {
         }
 
         // If there's a tie on total played time, check number of played episodes
-        return allPodcasts.sorted(by: {
-            if $0.totalPlayedTime == $1.totalPlayedTime {
-                return $0.numberOfPlayedEpisodes > $1.numberOfPlayedEpisodes
-            }
-            return $0.totalPlayedTime > $1.totalPlayedTime
-        })
+//        return allPodcasts.sorted(by: {
+//            if $0.totalPlayedTime == $1.totalPlayedTime {
+//                return $0.numberOfPlayedEpisodes > $1.numberOfPlayedEpisodes
+//            }
+//            return $0.totalPlayedTime > $1.totalPlayedTime
+//        })
+
+        return allPodcasts
     }
 
     /// Return the longest listened episode