import java.util.List;
import java.util.Map;
import java.util.Set;
-import java.util.SortedSet;
-import java.util.TreeSet;
import java.util.concurrent.locks.ReadWriteLock;
import java.util.concurrent.locks.ReentrantReadWriteLock;
import net.pterodactylus.sone.data.Image;
import net.pterodactylus.sone.data.Post;
import net.pterodactylus.sone.data.PostReply;
-import net.pterodactylus.sone.data.Reply;
import net.pterodactylus.sone.data.Sone;
import net.pterodactylus.sone.data.impl.DefaultSoneBuilder;
import net.pterodactylus.sone.database.Database;
import com.google.common.collect.ListMultimap;
import com.google.common.collect.Maps;
import com.google.common.collect.Multimap;
+import com.google.common.collect.SetMultimap;
import com.google.common.collect.SortedSetMultimap;
import com.google.common.collect.TreeMultimap;
import com.google.common.util.concurrent.AbstractService;
/** All posts by their Sones. */
private final Multimap<String, Post> sonePosts = HashMultimap.create();
+ private final SetMultimap<String, String> likedPostsBySone = HashMultimap.create();
+ private final SetMultimap<String, String> postLikingSones = HashMultimap.create();
/** All posts by their recipient. */
private final Multimap<String, Post> recipientPosts = HashMultimap.create();
/** All post replies by their ID. */
private final Map<String, PostReply> allPostReplies = new HashMap<String, PostReply>();
+ private final SetMultimap<String, String> likedPostRepliesBySone = HashMultimap.create();
+ private final SetMultimap<String, String> postReplyLikingSones = HashMultimap.create();
/** Replies sorted by Sone. */
private final SortedSetMultimap<String, PostReply> sonePostReplies = TreeMultimap.create(new Comparator<String>() {
}
@Override
+ public void storeIdentity(Identity identitiy) {
+ lock.writeLock().lock();
+ try {
+ identities.put(identitiy.getId(), identitiy);
+ } finally {
+ lock.writeLock().unlock();
+ }
+ }
+
+ @Override
+ public Function<String, Optional<Sone>> getSone() {
+ return new Function<String, Optional<Sone>>() {
+ @Override
+ public Optional<Sone> apply(String soneId) {
+ return (soneId == null) ? Optional.<Sone>absent() : getSone(soneId);
+ }
+ };
+ }
+
+ @Override
public Optional<Sone> getSone(String soneId) {
lock.readLock().lock();
try {
}
@Override
+ public void storeSone(Sone sone) {
+ lock.writeLock().lock();
+ try {
+ sones.put(sone.getId(), sone);
+ } finally {
+ lock.writeLock().unlock();
+ }
+ }
+
+ @Override
public SoneBuilder newSoneBuilder() {
return new DefaultSoneBuilder(this) {
@Override
public Sone build(Optional<SoneCreated> soneCreated) throws IllegalStateException {
Sone sone = super.build(soneCreated);
- sones.put(sone.getId(), sone);
+ lock.writeLock().lock();
+ try {
+ sones.put(sone.getId(), sone);
+ } finally {
+ lock.writeLock().unlock();
+ }
return sone;
}
};
//
@Override
+ public Function<String, Optional<Post>> getPost() {
+ return new Function<String, Optional<Post>>() {
+ @Override
+ public Optional<Post> apply(String postId) {
+ return (postId == null) ? Optional.<Post>absent() : getPost(postId);
+ }
+ };
+ }
+
+ @Override
public Optional<Post> getPost(String postId) {
lock.readLock().lock();
try {
}
}
+ @Override
+ public void likePost(Post post, Sone localSone) {
+ lock.writeLock().lock();
+ try {
+ likedPostsBySone.put(localSone.getId(), post.getId());
+ postLikingSones.put(post.getId(), localSone.getId());
+ } finally {
+ lock.writeLock().unlock();
+ }
+ }
+
+ @Override
+ public void unlikePost(Post post, Sone localSone) {
+ lock.writeLock().lock();
+ try {
+ likedPostsBySone.remove(localSone.getId(), post.getId());
+ postLikingSones.remove(post.getId(), localSone.getId());
+ } finally {
+ lock.writeLock().unlock();
+ }
+ }
+
+ public boolean isLiked(Post post, Sone sone) {
+ lock.readLock().lock();
+ try {
+ return likedPostsBySone.containsEntry(sone.getId(), post.getId());
+ } finally {
+ lock.readLock().unlock();
+ }
+ }
+
+ @Override
+ public Set<Sone> getLikes(Post post) {
+ lock.readLock().lock();
+ try {
+ return from(postLikingSones.get(post.getId())).transform(getSone()).transformAndConcat(this.<Sone>unwrap()).toSet();
+ } finally {
+ lock.readLock().unlock();
+ }
+ }
+
//
// POSTSTORE METHODS
//
}
}
+ @Override
+ public void likePostReply(PostReply postReply, Sone localSone) {
+ lock.writeLock().lock();
+ try {
+ likedPostRepliesBySone.put(localSone.getId(), postReply.getId());
+ postReplyLikingSones.put(postReply.getId(), localSone.getId());
+ } finally {
+ lock.writeLock().unlock();
+ }
+ }
+
+ @Override
+ public void unlikePostReply(PostReply postReply, Sone localSone) {
+ lock.writeLock().lock();
+ try {
+ likedPostRepliesBySone.remove(localSone.getId(), postReply.getId());
+ postReplyLikingSones.remove(postReply.getId(), localSone.getId());
+ } finally {
+ lock.writeLock().unlock();
+ }
+ }
+
+ @Override
+ public boolean isLiked(PostReply postReply, Sone sone) {
+ lock.readLock().lock();
+ try {
+ return postReplyLikingSones.containsEntry(postReply.getId(), sone.getId());
+ } finally {
+ lock.readLock().unlock();
+ }
+ }
+
+ @Override
+ public Set<Sone> getLikes(PostReply postReply) {
+ lock.readLock().lock();
+ try {
+ return from(postReplyLikingSones.get(postReply.getId())).transform(getSone()).transformAndConcat(this.<Sone>unwrap()).toSet();
+ } finally {
+ lock.readLock().unlock();
+ }
+ }
+
//
// POSTREPLYSTORE METHODS
//
lock.writeLock().lock();
try {
allAlbums.put(album.getId(), album);
- albumChildren.put(album.getParent().getId(), album.getId());
+ if (!album.isRoot()) {
+ albumChildren.put(album.getParent().getId(), album.getId());
+ }
} finally {
lock.writeLock().unlock();
}
};
}
+ private static <T> Function<Optional<T>, Iterable<T>> unwrap() {
+ return new Function<Optional<T>, Iterable<T>>() {
+ @Override
+ public Iterable<T> apply(Optional<T> input) {
+ return (input == null) ? Collections.<T>emptyList() : input.asSet();
+ }
+ };
+ }
+
}