TeamPiped-mirror
/
Piped-Backend
mirror of https://github.com/TeamPiped/Piped-Backend.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #550 from TeamPiped/randomization-cleanup 

Cleanup list randomization code for pubsub.
This commit is contained in:
Kavin 2023-03-11 21:02:11 +00:00 committed by GitHub
parent 04bc80f6c8 7c2014fc85
commit f8622cabeb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -2,6 +2,7 @@ package me.kavin.piped;
import io.activej.inject.Injector;
import io.sentry.Sentry;
import it.unimi.dsi.fastutil.objects.ObjectArrayList;
import jakarta.persistence.criteria.CriteriaBuilder;
import me.kavin.piped.consts.Constants;
import me.kavin.piped.server.ServerLauncher;
@ -81,9 +82,7 @@ public class Main {
public void run() {
try (StatelessSession s = DatabaseSessionFactory.createStatelessSession()) {
final Random random = new Random();
s.createNativeQuery("SELECT id FROM pubsub WHERE subbed_at < :subbedTime AND id IN (" +
List<String> channelIds = s.createNativeQuery("SELECT id FROM pubsub WHERE subbed_at < :subbedTime AND id IN (" +
"SELECT DISTINCT channel FROM users_subscribed" +
" UNION " +
"SELECT id FROM unauthenticated_subscriptions WHERE subscribed_at > :unauthSubbed" +
@ -92,9 +91,12 @@ public class Main {
.setParameter("unauthSubbed", System.currentTimeMillis() - TimeUnit.DAYS.toMillis(Constants.SUBSCRIPTIONS_EXPIRY))
.stream()
.filter(Objects::nonNull)
.collect(Collectors.toUnmodifiableSet())
.stream()
.sorted(Comparator.comparingInt(o -> random.nextInt()))
.distinct()
.collect(Collectors.toCollection(ObjectArrayList::new));
Collections.shuffle(channelIds);
channelIds.stream()
.parallel()
.forEach(id -> Multithreading.runAsyncLimitedPubSub(() -> {
try {