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

Delete from playlist_videos table on user deletion

Browse source
This commit is contained in:
theanonymousexyz 2022-04-22 23:45:15 +02:00
parent c56cb6eec5
commit 4fc05674f3
No known key found for this signature in database
GPG key ID: 35EE09F5481049BB
1 changed files with 23 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -642,17 +642,31 @@ public class ResponseHelper {
return Constants.mapper.writeValueAsBytes(new IncorrectCredentialsResponse());
try {
s.getTransaction().begin();
CriteriaBuilder plCriteria = s.getCriteriaBuilder();
CriteriaQuery<me.kavin.piped.utils.obj.db.Playlist> plQuery =
plCriteria.createQuery(me.kavin.piped.utils.obj.db.Playlist.class);
Root<me.kavin.piped.utils.obj.db.Playlist> plRoot =
plQuery.from(me.kavin.piped.utils.obj.db.Playlist.class);
plQuery.select(plRoot).where(plCriteria.equal(plRoot.get("owner"), user.getId()));
List<me.kavin.piped.utils.obj.db.Playlist> playlists = s.createQuery(plQuery).getResultList();
for (me.kavin.piped.utils.obj.db.Playlist pl : playlists) {
Iterator<PlaylistVideo> pvIter = pl.getVideos().iterator();
s.createNativeQuery("delete from users_subscribed where subscriber = :id")
.setParameter("id", user.getId()).executeUpdate();
s.createNativeQuery("delete from playlists where owner = :ownerId")
.setParameter("ownerId", user.getId()).executeUpdate();
s.createNativeQuery("delete from users where id = :id")
.setParameter("id", user.getId()).executeUpdate();
while (pvIter.hasNext())
s.delete(pvIter.next());
}
Iterator<me.kavin.piped.utils.obj.db.Playlist> iter = playlists.iterator();
while (iter.hasNext())
s.delete(iter.next());
s.delete(user);
s.getTransaction().begin();
s.getTransaction().commit();
return Constants.mapper.writeValueAsBytes(new DeleteUserResponse(user.getUsername()));
} catch (Exception e) {
return Constants.mapper.writeValueAsBytes(new ErrorResponse(ExceptionUtils.getStackTrace(e), e.getMessage()));