kavin/Piped-Backend
1
0
Fork 0
mirror of https://github.com/TeamPiped/Piped-Backend.git synced 2025-04-29 16:30:29 +05:30
Code Issues Projects Releases Wiki Activity

Cleanup PubSub query.

Browse Source
This commit is contained in:
FireMasterK 2021-08-21 01:52:39 +05:30
parent a391d11464
commit 15f7eef82c
No known key found for this signature in database
GPG Key ID: 49451E4482CC5BCD
1 changed files with 10 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
src/main/java/me/kavin/piped/Main.java
View File

@ -13,11 +13,11 @@ import org.schabi.newpipe.extractor.NewPipe;
import org.schabi.newpipe.extractor.localization.Localization; import org.schabi.newpipe.extractor.localization.Localization;
import io.activej.inject.Injector; import io.activej.inject.Injector;
import me.kavin.piped.utils.DatabaseHelper;
import me.kavin.piped.utils.DatabaseSessionFactory; import me.kavin.piped.utils.DatabaseSessionFactory;
import me.kavin.piped.utils.DownloaderImpl; import me.kavin.piped.utils.DownloaderImpl;
import me.kavin.piped.utils.Multithreading; import me.kavin.piped.utils.Multithreading;
import me.kavin.piped.utils.ResponseHelper; import me.kavin.piped.utils.ResponseHelper;
import me.kavin.piped.utils.obj.db.PubSub;
public class Main { public class Main {
@ -33,21 +33,22 @@ public class Main {
try { try {
Session s = DatabaseSessionFactory.createSession(); Session s = DatabaseSessionFactory.createSession();
List<String> channels = DatabaseHelper.getGlobalSubscribedChannelIds(s); List<PubSub> pubSubList = s.createNativeQuery(
"select distinct pubsub.* from pubsub inner join users_subscribed on pubsub.id = users_subscribed.channel",
PubSub.class).getResultList();
DatabaseHelper.getPubSubFromIds(s, channels).forEach(pubsub -> { pubSubList.removeIf(pubsub -> {
if (System.currentTimeMillis() - pubsub.getSubbedAt() < TimeUnit.DAYS.toMillis(4)) return System.currentTimeMillis() - pubsub.getSubbedAt() < TimeUnit.DAYS.toMillis(4);
channels.remove(pubsub.getId());
}); });
Collections.shuffle(channels); Collections.shuffle(pubSubList);
for (String channelId : channels) for (PubSub pubsub : pubSubList)
if (channelId != null) if (pubsub != null)
Multithreading.runAsyncLimitedPubSub(() -> { Multithreading.runAsyncLimitedPubSub(() -> {
Session sess = DatabaseSessionFactory.createSession(); Session sess = DatabaseSessionFactory.createSession();
try { try {
ResponseHelper.subscribePubSub(channelId, sess); ResponseHelper.subscribePubSub(pubsub.getId(), sess);
} catch (Exception e) { } catch (Exception e) {
e.printStackTrace(); e.printStackTrace();
} }