import java.util.concurrent.locks.ReentrantReadWriteLock;
import net.pterodactylus.sone.data.Post;
+import net.pterodactylus.sone.data.Post.EmptyPost;
import net.pterodactylus.sone.database.BookmarkDatabase;
import com.google.common.base.Function;
}
}
- @Override
- public void bookmarkPost(String postId) {
- lock.writeLock().lock();
+ public void stop() {
+ saveBookmarkedPosts();
+ }
+
+ private void saveBookmarkedPosts() {
+ lock.readLock().lock();
try {
- bookmarkedPosts.add(postId);
+ configurationLoader.saveBookmarkedPosts(this.bookmarkedPosts);
} finally {
- lock.writeLock().unlock();
+ lock.readLock().unlock();
}
}
lock.writeLock().lock();
try {
bookmarkedPosts.add(post.getId());
+ saveBookmarkedPosts();
} finally {
lock.writeLock().unlock();
}
lock.writeLock().lock();
try {
bookmarkedPosts.remove(post.getId());
+ saveBookmarkedPosts();
} finally {
lock.writeLock().unlock();
}
public Set<Post> getBookmarkedPosts() {
lock.readLock().lock();
try {
- return from(bookmarkedPosts).transformAndConcat(
- new Function<String, Iterable<Post>>() {
+ return from(bookmarkedPosts).transform(
+ new Function<String, Post>() {
@Override
- public Iterable<Post> apply(String postId) {
- return memoryDatabase.getPost(postId).asSet();
+ public Post apply(String postId) {
+ return memoryDatabase.getPost(postId)
+ .or(new EmptyPost(postId));
}
}).toSet();
} finally {