diff --git a/src/main/java/org/ntlab/citrusserver/repositories/PublicBookManager.java b/src/main/java/org/ntlab/citrusserver/repositories/PublicBookManager.java index 907e5ee..bde074a 100644 --- a/src/main/java/org/ntlab/citrusserver/repositories/PublicBookManager.java +++ b/src/main/java/org/ntlab/citrusserver/repositories/PublicBookManager.java @@ -11,7 +11,8 @@ @Repository public class PublicBookManager implements IBookListener{ - private final HashMap> publicBooks = new HashMap<>(); + private final HashMap> publicBooks = new HashMap<>(); private final FavoriteManager favoriteManager; @@ -28,6 +29,8 @@ return books; } + + public ArrayList getAllPublicBooks(int sortBy){ return sort(getAllPublicBooks(), sortBy); } @@ -97,7 +100,9 @@ @Override public void bookDeleted(Account account, Book book) { - + if(publicBooks.containsKey(account.getId())){ + publicBooks.get(account.getId()).remove(book); + } } private ArrayList sort(ArrayList books, int sortMode){