1 package net.pterodactylus.sone.database.memory;
3 import static com.google.common.collect.FluentIterable.from;
5 import java.util.HashSet;
7 import java.util.concurrent.locks.ReadWriteLock;
8 import java.util.concurrent.locks.ReentrantReadWriteLock;
10 import net.pterodactylus.sone.data.Post;
11 import net.pterodactylus.sone.database.BookmarkDatabase;
13 import com.google.common.base.Function;
16 * Memory-based {@link BookmarkDatabase} implementation.
18 * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
20 public class MemoryBookmarkDatabase implements BookmarkDatabase {
22 private final ReadWriteLock lock = new ReentrantReadWriteLock();
23 private final MemoryDatabase memoryDatabase;
24 private final ConfigurationLoader configurationLoader;
25 private final Set<String> bookmarkedPosts = new HashSet<String>();
27 public MemoryBookmarkDatabase(MemoryDatabase memoryDatabase,
28 ConfigurationLoader configurationLoader) {
29 this.memoryDatabase = memoryDatabase;
30 this.configurationLoader = configurationLoader;
34 loadBookmarkedPosts();
37 private void loadBookmarkedPosts() {
38 Set<String> bookmarkedPosts = configurationLoader.loadBookmarkedPosts();
39 lock.writeLock().lock();
41 this.bookmarkedPosts.clear();
42 this.bookmarkedPosts.addAll(bookmarkedPosts);
44 lock.writeLock().unlock();
49 public void bookmarkPost(String postId) {
50 lock.writeLock().lock();
52 bookmarkedPosts.add(postId);
54 lock.writeLock().unlock();
59 public void bookmarkPost(Post post) {
60 lock.writeLock().lock();
62 bookmarkedPosts.add(post.getId());
64 lock.writeLock().unlock();
69 public void unbookmarkPost(Post post) {
70 lock.writeLock().lock();
72 bookmarkedPosts.remove(post.getId());
74 lock.writeLock().unlock();
79 public boolean isPostBookmarked(Post post) {
80 lock.readLock().lock();
82 return bookmarkedPosts.contains(post.getId());
84 lock.readLock().unlock();
89 public Set<Post> getBookmarkedPosts() {
90 lock.readLock().lock();
92 return from(bookmarkedPosts).transformAndConcat(
93 new Function<String, Iterable<Post>>() {
95 public Iterable<Post> apply(String postId) {
96 return memoryDatabase.getPost(postId).asSet();
100 lock.readLock().unlock();