private final ReadWriteLock lock = new ReentrantReadWriteLock();
private final MemoryDatabase memoryDatabase;
+ private final ConfigurationLoader configurationLoader;
private final Set<String> bookmarkedPosts = new HashSet<String>();
- public MemoryBookmarkDatabase(MemoryDatabase memoryDatabase) {
+ public MemoryBookmarkDatabase(MemoryDatabase memoryDatabase,
+ ConfigurationLoader configurationLoader) {
this.memoryDatabase = memoryDatabase;
+ this.configurationLoader = configurationLoader;
}
- @Override
- public void bookmarkPost(String postId) {
+ public void start() {
+ loadBookmarkedPosts();
+ }
+
+ private void loadBookmarkedPosts() {
+ Set<String> bookmarkedPosts = configurationLoader.loadBookmarkedPosts();
lock.writeLock().lock();
try {
- bookmarkedPosts.add(postId);
+ this.bookmarkedPosts.clear();
+ this.bookmarkedPosts.addAll(bookmarkedPosts);
} finally {
lock.writeLock().unlock();
}
}
+ public void stop() {
+ saveBookmarkedPosts();
+ }
+
+ private void saveBookmarkedPosts() {
+ lock.readLock().lock();
+ try {
+ configurationLoader.saveBookmarkedPosts(this.bookmarkedPosts);
+ } finally {
+ lock.readLock().unlock();
+ }
+ }
+
@Override
public void bookmarkPost(Post post) {
lock.writeLock().lock();
try {
bookmarkedPosts.add(post.getId());
+ saveBookmarkedPosts();
} finally {
lock.writeLock().unlock();
}