TeamPiped-mirror/Piped-Backend
1
0
Fork 0
mirror of https://github.com/TeamPiped/Piped-Backend.git synced 2024-08-14 23:51:41 +00:00
Code Issues Projects Releases Packages Wiki Activity

Use delete query for better performance

Browse source
This commit is contained in:
theanonymousexyz 2022-04-26 18:34:35 +02:00
parent 756d652d16
commit a5420d48d9
No known key found for this signature in database
GPG key ID: 35EE09F5481049BB
1 changed files with 18 additions and 28 deletions
Download patch file Download diff file Expand all files Collapse all files

46
src/main/java/me/kavin/piped/utils/ResponseHelper.java
View file

@ -11,7 +11,6 @@ import it.unimi.dsi.fastutil.objects.ObjectArrayList;
import me.kavin.piped.consts.Constants; import me.kavin.piped.consts.Constants;
import me.kavin.piped.ipfs.IPFS; import me.kavin.piped.ipfs.IPFS;
import me.kavin.piped.utils.obj.*; import me.kavin.piped.utils.obj.*;
import me.kavin.piped.utils.obj.db.Channel;
import me.kavin.piped.utils.obj.db.PlaylistVideo; import me.kavin.piped.utils.obj.db.PlaylistVideo;
import me.kavin.piped.utils.obj.db.PubSub; import me.kavin.piped.utils.obj.db.PubSub;
import me.kavin.piped.utils.obj.db.User; import me.kavin.piped.utils.obj.db.User;
@ -48,9 +47,12 @@ import org.springframework.security.crypto.argon2.Argon2PasswordEncoder;
import org.springframework.security.crypto.bcrypt.BCryptPasswordEncoder; import org.springframework.security.crypto.bcrypt.BCryptPasswordEncoder;
import javax.persistence.criteria.CriteriaBuilder; import javax.persistence.criteria.CriteriaBuilder;
import javax.persistence.criteria.CriteriaDelete;
import javax.persistence.criteria.CriteriaQuery; import javax.persistence.criteria.CriteriaQuery;
import javax.persistence.criteria.JoinType; import javax.persistence.criteria.JoinType;
import javax.persistence.criteria.Root; import javax.persistence.criteria.Root;
import javax.persistence.criteria.Subquery;
import java.io.IOException; import java.io.IOException;
import java.util.*; import java.util.*;
import java.util.concurrent.TimeUnit; import java.util.concurrent.TimeUnit;
@ -662,7 +664,7 @@ public class ResponseHelper {
s.getTransaction().begin(); s.getTransaction().begin();
s.getTransaction().commit(); s.getTransaction().commit();
Multithreading.runAsync(() -> pruneUnusedPlaylistVideos()); Multithreading.runAsync(() -> pruneUnusedPlaylistVideos());
} catch (Exception e) { } catch (Exception e) {
return Constants.mapper.writeValueAsBytes(new ErrorResponse(ExceptionUtils.getStackTrace(e), e.getMessage())); return Constants.mapper.writeValueAsBytes(new ErrorResponse(ExceptionUtils.getStackTrace(e), e.getMessage()));
@ -954,11 +956,11 @@ public class ResponseHelper {
Multithreading.runAsync(() -> { Multithreading.runAsync(() -> {
try (Session s = DatabaseSessionFactory.createSession()) { try (Session s = DatabaseSessionFactory.createSession()) {
List<Channel> channels = DatabaseHelper.getChannelsFromIds(s, Arrays.asList(channelIds)); var channels = DatabaseHelper.getChannelsFromIds(s, Arrays.asList(channelIds));
outer: outer:
for (String channelId : channelIds) { for (String channelId : channelIds) {
for (Channel channel : channels) for (var channel : channels)
if (channel.getUploaderId().equals(channelId)) if (channel.getUploaderId().equals(channelId))
continue outer; continue outer;
Multithreading.runAsyncLimited(() -> saveChannel(channelId)); Multithreading.runAsyncLimited(() -> saveChannel(channelId));
@ -1216,34 +1218,22 @@ public class ResponseHelper {
} }
private static void pruneUnusedPlaylistVideos() { private static void pruneUnusedPlaylistVideos() {
try (Session s = DatabaseSessionFactory.createSession()) { try (Session s = DatabaseSessionFactory.createSession()) {
CriteriaQuery<me.kavin.piped.utils.obj.db.Playlist> plQuery = CriteriaBuilder cb = s.getCriteriaBuilder();
s.getCriteriaBuilder().createQuery(me.kavin.piped.utils.obj.db.Playlist.class);
plQuery.select(plQuery.from(me.kavin.piped.utils.obj.db.Playlist.class)); CriteriaDelete<PlaylistVideo> pvQuery = cb.createCriteriaDelete(PlaylistVideo.class);
List<me.kavin.piped.utils.obj.db.Playlist> playlists = s.createQuery(plQuery).getResultList(); Root<PlaylistVideo> pvRoot = pvQuery.from(PlaylistVideo.class);
CriteriaQuery<PlaylistVideo> pvQuery = s.getCriteriaBuilder().createQuery(PlaylistVideo.class); Subquery<String> subQuery = pvQuery.subquery(String.class);
pvQuery.select(pvQuery.from(PlaylistVideo.class)); Root<me.kavin.piped.utils.obj.db.Playlist> subRoot =
List<PlaylistVideo> playlistVideos = s.createQuery(pvQuery).getResultList(); subQuery.from(me.kavin.piped.utils.obj.db.Playlist.class);
Iterator<PlaylistVideo> pvIter = playlistVideos.iterator(); subQuery.select(subRoot.join("videos").get("id"));
pvQuery.where(cb.not(pvRoot.get("id").in(subQuery)));
outer:
while (pvIter.hasNext()) {
PlaylistVideo pv = pvIter.next();
for (me.kavin.piped.utils.obj.db.Playlist pl : playlists) {
for (PlaylistVideo v : pl.getVideos()) {
if (v.getId().equals(pv.getId())) {
continue outer;
}
}
}
s.delete(pv);
}
s.getTransaction().begin(); s.getTransaction().begin();
s.createQuery(pvQuery).executeUpdate();
s.getTransaction().commit(); s.getTransaction().commit();
} }
} }