X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fdatabase%2Fmemory%2FMemoryDatabase.java;h=8d60e0f4b548f4f0c579ad7aca52cda000d76911;hb=4d62892729cbce31c4fb2171ad2c2174db1d44b4;hp=eed00eab1e3607a08b69a2fdb047f7230b9d72cc;hpb=60fda3f6fd8cd72151338c831f509dd8d9d0f9ff;p=Sone.git diff --git a/src/main/java/net/pterodactylus/sone/database/memory/MemoryDatabase.java b/src/main/java/net/pterodactylus/sone/database/memory/MemoryDatabase.java index eed00ea..8d60e0f 100644 --- a/src/main/java/net/pterodactylus/sone/database/memory/MemoryDatabase.java +++ b/src/main/java/net/pterodactylus/sone/database/memory/MemoryDatabase.java @@ -1,5 +1,5 @@ /* - * Sone - MemoryPostDatabase.java - Copyright © 2013 David Roden + * Sone - MemoryDatabase.java - Copyright © 2013 David Roden * * This program is free software: you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by @@ -17,210 +17,482 @@ package net.pterodactylus.sone.database.memory; +import static com.google.common.base.Optional.fromNullable; import static com.google.common.base.Preconditions.checkNotNull; +import static com.google.common.base.Predicates.not; +import static com.google.common.collect.FluentIterable.from; +import static java.util.Collections.emptyList; +import static java.util.logging.Logger.getLogger; +import static net.pterodactylus.sone.data.Sone.LOCAL_SONE_FILTER; +import java.util.ArrayList; import java.util.Collection; import java.util.Collections; +import java.util.Comparator; import java.util.HashMap; import java.util.HashSet; +import java.util.List; import java.util.Map; import java.util.Set; -import java.util.UUID; import java.util.concurrent.locks.ReadWriteLock; import java.util.concurrent.locks.ReentrantReadWriteLock; +import java.util.logging.Level; +import java.util.logging.Logger; +import net.pterodactylus.sone.data.Album; +import net.pterodactylus.sone.data.Image; import net.pterodactylus.sone.data.Post; +import net.pterodactylus.sone.data.PostReply; import net.pterodactylus.sone.data.Sone; -import net.pterodactylus.sone.data.impl.AbstractPostBuilder; -import net.pterodactylus.sone.database.PostBuilder; -import net.pterodactylus.sone.database.PostDatabase; -import net.pterodactylus.sone.database.SoneProvider; +import net.pterodactylus.sone.data.impl.DefaultSoneBuilder; +import net.pterodactylus.sone.database.Database; +import net.pterodactylus.sone.database.DatabaseException; +import net.pterodactylus.sone.database.SoneBuilder; +import net.pterodactylus.sone.freenet.wot.Identity; import net.pterodactylus.util.config.Configuration; import net.pterodactylus.util.config.ConfigurationException; +import com.google.common.base.Function; import com.google.common.base.Optional; +import com.google.common.collect.ArrayListMultimap; +import com.google.common.collect.HashMultimap; +import com.google.common.collect.ListMultimap; +import com.google.common.collect.Maps; +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; import com.google.inject.Inject; /** - * Memory-based {@link PostDatabase} implementation. + * Memory-based {@link Database} implementation. * * @author David ‘Bombe’ Roden */ -public class MemoryPostDatabase implements PostDatabase { +public class MemoryDatabase extends AbstractService implements Database { + + private static final Logger logger = getLogger(MemoryDatabase.class.getName()); /** The lock. */ private final ReadWriteLock lock = new ReentrantReadWriteLock(); - /** The Sone provider. */ - private final SoneProvider soneProvider; + /** The configuration. */ + private final Configuration configuration; + + private final Map identities = Maps.newHashMap(); + private final Map sones = new HashMap(); + private final MemoryPostDatabase memoryPostDatabase; - /** All posts by their ID. */ - private final Map allPosts = new HashMap(); + /** All post replies by their ID. */ + private final Map allPostReplies = new HashMap(); + private final SetMultimap likedPostRepliesBySone = HashMultimap.create(); + private final SetMultimap postReplyLikingSones = HashMultimap.create(); - /** All posts by their Sones. */ - private final Map> sonePosts = new HashMap>(); + /** Replies sorted by Sone. */ + private final SortedSetMultimap sonePostReplies = TreeMultimap.create(new Comparator() { - /** All posts by their recipient. */ - private final Map> recipientPosts = new HashMap>(); + @Override + public int compare(String leftString, String rightString) { + return leftString.compareTo(rightString); + } + }, PostReply.TIME_COMPARATOR); - /** Whether posts are known. */ - private final Set knownPosts = new HashSet(); + /** Replies by post. */ + private final SortedSetMultimap postReplies = TreeMultimap.create(new Comparator() { + + @Override + public int compare(String leftString, String rightString) { + return leftString.compareTo(rightString); + } + }, PostReply.TIME_COMPARATOR); + + /** Whether post replies are known. */ + private final Set knownPostReplies = new HashSet(); + + private final Map allAlbums = new HashMap(); + private final ListMultimap albumChildren = ArrayListMultimap.create(); + private final ListMultimap albumImages = ArrayListMultimap.create(); + + private final Map allImages = new HashMap(); /** * Creates a new memory database. * - * @param soneProvider - * The Sone provider + * @param configuration + * The configuration for loading and saving elements */ @Inject - public MemoryPostDatabase(SoneProvider soneProvider) { - this.soneProvider = soneProvider; + public MemoryDatabase(Configuration configuration) { + this.configuration = configuration; + memoryPostDatabase = new MemoryPostDatabase(this, lock, configuration); } // - // POSTPROVIDER METHODS + // SERVICE METHODS // - /** - * {@inheritDocs} - */ @Override - public Optional getPost(String postId) { + protected void doStart() { + memoryPostDatabase.start(); + loadKnownPostReplies(); + notifyStarted(); + } + + @Override + protected void doStop() { + try { + memoryPostDatabase.stop(); + } catch (DatabaseException de1) { + logger.log(Level.WARNING, "Could not stop post database!", de1); + } + notifyStopped(); + } + + @Override + public Optional getIdentity(String identityId) { lock.readLock().lock(); try { - return Optional.fromNullable(allPosts.get(postId)); + return fromNullable(identities.get(identityId)); } finally { lock.readLock().unlock(); } } - /** - * {@inheritDocs} - */ @Override - public Collection getPosts(String soneId) { - return new HashSet(getPostsFrom(soneId)); + public void storeIdentity(Identity identitiy) { + lock.writeLock().lock(); + try { + identities.put(identitiy.getId(), identitiy); + } finally { + lock.writeLock().unlock(); + } } - /** - * {@inheritDocs} - */ @Override - public Collection getDirectedPosts(String recipientId) { + public Function> getSone() { + return new Function>() { + @Override + public Optional apply(String soneId) { + return (soneId == null) ? Optional.absent() : getSone(soneId); + } + }; + } + + @Override + public Optional getSone(String soneId) { lock.readLock().lock(); try { - Collection posts = recipientPosts.get(recipientId); - return (posts == null) ? Collections. emptySet() : new HashSet(posts); + return fromNullable(sones.get(soneId)); } finally { lock.readLock().unlock(); } } + @Override + public Collection getSones() { + lock.readLock().lock(); + try { + return Collections.unmodifiableCollection(sones.values()); + } finally { + lock.readLock().unlock(); + } + } + + @Override + public Collection getLocalSones() { + lock.readLock().lock(); + try { + return from(getSones()).filter(LOCAL_SONE_FILTER).toSet(); + } finally { + lock.readLock().unlock(); + } + } + + @Override + public Collection getRemoteSones() { + lock.readLock().lock(); + try { + return from(getSones()).filter(not(LOCAL_SONE_FILTER)).toSet(); + } finally { + lock.readLock().unlock(); + } + } + + @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) throws IllegalStateException { + Sone sone = super.build(soneCreated); + lock.writeLock().lock(); + try { + sones.put(sone.getId(), sone); + } finally { + lock.writeLock().unlock(); + } + return sone; + } + }; + } + // - // POSTBUILDERFACTORY METHODS + // POSTPROVIDER METHODS // + @Override + public Function> getPost() { + return memoryPostDatabase.getPost(); + } + + @Override + public Optional getPost(String postId) { + return memoryPostDatabase.getPost(postId); + } + + @Override + public Collection getPosts(String soneId) { + return memoryPostDatabase.getPosts(soneId); + } + + @Override + public Collection getDirectedPosts(String recipientId) { + return memoryPostDatabase.getDirectedPosts(recipientId); + } + /** - * {@inheritDocs} + * Returns whether the given post is known. + * + * @param post + * The post + * @return {@code true} if the post is known, {@code false} otherwise */ @Override - public PostBuilder newPostBuilder() { - return new MemoryPostBuilder(soneProvider); + public boolean isPostKnown(Post post) { + return memoryPostDatabase.isPostKnown(post); + } + + /** + * Sets whether the given post is known. + * + * @param post + * The post + */ + @Override + public void setPostKnown(Post post) { + memoryPostDatabase.setPostKnown(post); + } + + @Override + public void likePost(Post post, Sone localSone) { + memoryPostDatabase.likePost(post, localSone); + } + + @Override + public void unlikePost(Post post, Sone localSone) { + memoryPostDatabase.unlikePost(post, localSone); + } + + public boolean isLiked(Post post, Sone sone) { + return memoryPostDatabase.isLiked(post, sone); + } + + @Override + public Set getLikes(Post post) { + return memoryPostDatabase.getLikes(post); } // // POSTSTORE METHODS // - /** - * {@inheritDocs} - */ @Override public void storePost(Post post) { - checkNotNull(post, "post must not be null"); - lock.writeLock().lock(); + memoryPostDatabase.storePost(post); + } + + @Override + public void removePost(Post post) { + memoryPostDatabase.removePost(post); + } + + @Override + public void storePosts(Sone sone, Collection posts) throws IllegalArgumentException { + /* verify that all posts are from the same Sone. */ + + memoryPostDatabase.storePosts(sone, posts); + } + + @Override + public void removePosts(Sone sone) { + memoryPostDatabase.removePosts(sone); + } + + // + // POSTREPLYPROVIDER METHODS + // + + @Override + public Optional getPostReply(String id) { + lock.readLock().lock(); + try { + return fromNullable(allPostReplies.get(id)); + } finally { + lock.readLock().unlock(); + } + } + + @Override + public List getReplies(String postId) { + lock.readLock().lock(); try { - allPosts.put(post.getId(), post); - getPostsFrom(post.getSone().getId()).add(post); - if (post.getRecipientId().isPresent()) { - getPostsTo(post.getRecipientId().get()).add(post); + if (!postReplies.containsKey(postId)) { + return emptyList(); } + return new ArrayList(postReplies.get(postId)); + } finally { + lock.readLock().unlock(); + } + } + + @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(); } } - /** - * {@inheritDocs} - */ @Override - public void removePost(Post post) { - checkNotNull(post, "post must not be null"); + public void unlikePostReply(PostReply postReply, Sone localSone) { lock.writeLock().lock(); try { - allPosts.remove(post.getId()); - getPostsFrom(post.getSone().getId()).remove(post); - if (post.getRecipientId().isPresent()) { - getPostsTo(post.getRecipientId().get()).remove(post); - } - post.getSone().removePost(post); + 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 getLikes(PostReply postReply) { + lock.readLock().lock(); + try { + return from(postReplyLikingSones.get(postReply.getId())).transform(getSone()).transformAndConcat(this.unwrap()).toSet(); + } finally { + lock.readLock().unlock(); + } + } + + // + // POSTREPLYSTORE METHODS + // + /** - * {@inheritDocs} + * Returns whether the given post reply is known. + * + * @param postReply + * The post reply + * @return {@code true} if the given post reply is known, {@code false} + * otherwise */ + public boolean isPostReplyKnown(PostReply postReply) { + lock.readLock().lock(); + try { + return knownPostReplies.contains(postReply.getId()); + } finally { + lock.readLock().unlock(); + } + } + @Override - public void storePosts(Sone sone, Collection posts) throws IllegalArgumentException { + public void setPostReplyKnown(PostReply postReply) { + lock.writeLock().lock(); + try { + knownPostReplies.add(postReply.getId()); + } finally { + lock.writeLock().unlock(); + } + } + + @Override + public void storePostReply(PostReply postReply) { + lock.writeLock().lock(); + try { + allPostReplies.put(postReply.getId(), postReply); + postReplies.put(postReply.getPostId(), postReply); + } finally { + lock.writeLock().unlock(); + } + } + + @Override + public void storePostReplies(Sone sone, Collection postReplies) { checkNotNull(sone, "sone must not be null"); /* verify that all posts are from the same Sone. */ - for (Post post : posts) { - if (!sone.equals(post.getSone())) { - throw new IllegalArgumentException(String.format("Post from different Sone found: %s", post)); + for (PostReply postReply : postReplies) { + if (!sone.equals(postReply.getSone())) { + throw new IllegalArgumentException(String.format("PostReply from different Sone found: %s", postReply)); } } lock.writeLock().lock(); try { - /* remove all posts by the Sone. */ - getPostsFrom(sone.getId()).clear(); - for (Post post : posts) { - allPosts.remove(post.getId()); - if (post.getRecipientId().isPresent()) { - getPostsTo(post.getRecipientId().get()).remove(post); - } + /* remove all post replies of the Sone. */ + for (PostReply postReply : getRepliesFrom(sone.getId())) { + removePostReply(postReply); } - - /* add new posts. */ - getPostsFrom(sone.getId()).addAll(posts); - for (Post post : posts) { - allPosts.put(post.getId(), post); - if (post.getRecipientId().isPresent()) { - getPostsTo(post.getRecipientId().get()).add(post); - } + for (PostReply postReply : postReplies) { + allPostReplies.put(postReply.getId(), postReply); + sonePostReplies.put(postReply.getSone().getId(), postReply); + this.postReplies.put(postReply.getPostId(), postReply); } } finally { lock.writeLock().unlock(); } } - /** - * {@inheritDocs} - */ @Override - public void removePosts(Sone sone) { + public void removePostReply(PostReply postReply) { + lock.writeLock().lock(); + try { + allPostReplies.remove(postReply.getId()); + postReplies.remove(postReply.getPostId(), postReply); + } finally { + lock.writeLock().unlock(); + } + } + + @Override + public void removePostReplies(Sone sone) { checkNotNull(sone, "sone must not be null"); + lock.writeLock().lock(); try { - /* remove all posts by the Sone. */ - getPostsFrom(sone.getId()).clear(); - for (Post post : sone.getPosts()) { - allPosts.remove(post.getId()); - if (post.getRecipientId().isPresent()) { - getPostsTo(post.getRecipientId().get()).remove(post); - } + for (PostReply postReply : sone.getReplies()) { + removePostReply(postReply); } } finally { lock.writeLock().unlock(); @@ -228,181 +500,257 @@ public class MemoryPostDatabase implements PostDatabase { } // - // POSTDATABASE METHODS + // ALBUMPROVDER METHODS // - /** - * {@inheritDocs} - */ @Override - public void loadKnownPosts(Configuration configuration, String prefix) { + public Optional getAlbum(String albumId) { + lock.readLock().lock(); + try { + return fromNullable(allAlbums.get(albumId)); + } finally { + lock.readLock().unlock(); + } + } + + @Override + public List getAlbums(Album parent) { + lock.readLock().lock(); + try { + return from(albumChildren.get(parent.getId())).transformAndConcat(getAlbum()).toList(); + } finally { + lock.readLock().unlock(); + } + } + + @Override + public void moveUp(Album album) { lock.writeLock().lock(); try { - int postCounter = 0; - while (true) { - String knownPostId = configuration.getStringValue(prefix + postCounter++ + "/ID").getValue(null); - if (knownPostId == null) { - break; - } - knownPosts.add(knownPostId); + List albums = albumChildren.get(album.getParent().getId()); + int currentIndex = albums.indexOf(album.getId()); + if (currentIndex == 0) { + return; } + albums.remove(album.getId()); + albums.add(currentIndex - 1, album.getId()); } finally { lock.writeLock().unlock(); } } - /** - * {@inheritDocs} - */ @Override - public void saveKnownPosts(Configuration configuration, String prefix) throws ConfigurationException { - lock.readLock().lock(); + public void moveDown(Album album) { + lock.writeLock().lock(); try { - int postCounter = 0; - for (String knownPostId : knownPosts) { - configuration.getStringValue(prefix + postCounter++ + "/ID").setValue(knownPostId); + List albums = albumChildren.get(album.getParent().getId()); + int currentIndex = albums.indexOf(album.getId()); + if (currentIndex == (albums.size() - 1)) { + return; } - configuration.getStringValue(prefix + postCounter + "/ID").setValue(null); + albums.remove(album.getId()); + albums.add(currentIndex + 1, album.getId()); } finally { - lock.readLock().unlock(); + lock.writeLock().unlock(); } } // - // PACKAGE-PRIVATE METHODS + // ALBUMSTORE METHODS // - /** - * Returns whether the given post is known. - * - * @param post - * The post - * @return {@code true} if the post is known, {@code false} otherwise - */ - boolean isPostKnown(Post post) { - lock.readLock().lock(); + @Override + public void storeAlbum(Album album) { + lock.writeLock().lock(); try { - return knownPosts.contains(post.getId()); + allAlbums.put(album.getId(), album); + if (!album.isRoot()) { + albumChildren.put(album.getParent().getId(), album.getId()); + } } finally { - lock.readLock().unlock(); + lock.writeLock().unlock(); } } - /** - * Sets whether the given post is known. - * - * @param post - * The post - * @param known - * {@code true} if the post is known, {@code false} otherwise - */ - void setPostKnown(Post post, boolean known) { + @Override + public void removeAlbum(Album album) { lock.writeLock().lock(); try { - if (known) { - knownPosts.add(post.getId()); - } else { - knownPosts.remove(post.getId()); - } + allAlbums.remove(album.getId()); + albumChildren.remove(album.getParent().getId(), album.getId()); } finally { lock.writeLock().unlock(); } } // - // PRIVATE METHODS + // IMAGEPROVIDER METHODS // - /** - * Gets all posts for the given Sone, creating a new collection if there is - * none yet. - * - * @param soneId - * The ID of the Sone to get the posts for - * @return All posts - */ - private Collection getPostsFrom(String soneId) { - Collection posts = null; + @Override + public Optional getImage(String imageId) { lock.readLock().lock(); try { - posts = sonePosts.get(soneId); + return fromNullable(allImages.get(imageId)); } finally { lock.readLock().unlock(); } - if (posts != null) { - return posts; + } + + @Override + public List getImages(Album parent) { + lock.readLock().lock(); + try { + return from(albumImages.get(parent.getId())).transformAndConcat(getImage()).toList(); + } finally { + lock.readLock().unlock(); } + } - posts = new HashSet(); + @Override + public void moveUp(Image image) { lock.writeLock().lock(); try { - sonePosts.put(soneId, posts); + List images = albumImages.get(image.getAlbum().getId()); + int currentIndex = images.indexOf(image.getId()); + if (currentIndex == 0) { + return; + } + images.remove(image.getId()); + images.add(currentIndex - 1, image.getId()); } finally { lock.writeLock().unlock(); } + } - return posts; + @Override + public void moveDown(Image image) { + lock.writeLock().lock(); + try { + List images = albumChildren.get(image.getAlbum().getId()); + int currentIndex = images.indexOf(image.getId()); + if (currentIndex == (images.size() - 1)) { + return; + } + images.remove(image.getId()); + images.add(currentIndex + 1, image.getId()); + } finally { + lock.writeLock().unlock(); + } } + // + // IMAGESTORE METHODS + // + + @Override + public void storeImage(Image image) { + lock.writeLock().lock(); + try { + allImages.put(image.getId(), image); + albumImages.put(image.getAlbum().getId(), image.getId()); + } finally { + lock.writeLock().unlock(); + } + } + + @Override + public void removeImage(Image image) { + lock.writeLock().lock(); + try { + allImages.remove(image.getId()); + albumImages.remove(image.getAlbum().getId(), image.getId()); + } finally { + lock.writeLock().unlock(); + } + } + + // + // PRIVATE METHODS + // + /** - * Gets all posts that are directed the given Sone, creating a new - * collection if there is none yet. + * Returns all replies by the given Sone. * - * @param recipientId - * The ID of the Sone to get the posts for - * @return All posts + * @param id + * The ID of the Sone + * @return The post replies of the Sone, sorted by time (newest first) */ - private Collection getPostsTo(String recipientId) { - Collection posts = null; + private Collection getRepliesFrom(String id) { lock.readLock().lock(); try { - posts = recipientPosts.get(recipientId); + if (sonePostReplies.containsKey(id)) { + return Collections.unmodifiableCollection(sonePostReplies.get(id)); + } + return Collections.emptySet(); } finally { lock.readLock().unlock(); } - if (posts != null) { - return posts; - } + } - posts = new HashSet(); + /** Loads the known post replies. */ + private void loadKnownPostReplies() { lock.writeLock().lock(); try { - recipientPosts.put(recipientId, posts); + int replyCounter = 0; + while (true) { + String knownReplyId = configuration.getStringValue("KnownReplies/" + replyCounter++ + "/ID").getValue(null); + if (knownReplyId == null) { + break; + } + knownPostReplies.add(knownReplyId); + } } finally { lock.writeLock().unlock(); } - - return posts; } /** - * {@link PostBuilder} implementation that creates a {@link MemoryPost}. + * Saves the known post replies to the configuration. * - * @author David ‘Bombe’ Roden + * @throws DatabaseException + * if a configuration error occurs */ - private class MemoryPostBuilder extends AbstractPostBuilder { - - /** - * Creates a new memory post builder. - * - * @param soneProvider - * The Sone provider - */ - public MemoryPostBuilder(SoneProvider soneProvider) { - super(soneProvider); + private void saveKnownPostReplies() throws DatabaseException { + lock.readLock().lock(); + try { + int replyCounter = 0; + for (String knownReplyId : knownPostReplies) { + configuration.getStringValue("KnownReplies/" + replyCounter++ + "/ID").setValue(knownReplyId); + } + configuration.getStringValue("KnownReplies/" + replyCounter + "/ID").setValue(null); + } catch (ConfigurationException ce1) { + throw new DatabaseException("Could not save database.", ce1); + } finally { + lock.readLock().unlock(); } + } - /** - * {@inheritDocs} - */ - @Override - public Post build() throws IllegalStateException { - validate(); - Post post = new MemoryPost(MemoryPostDatabase.this, soneProvider, randomId ? UUID.randomUUID().toString() : id, senderId, recipientId, currentTime ? System.currentTimeMillis() : time, text); - post.setKnown(isPostKnown(post)); - return post; - } + private Function> getAlbum() { + return new Function>() { + @Override + public Iterable apply(String input) { + return (input == null) ? Collections.emptyList() : getAlbum(input).asSet(); + } + }; + } + private Function> getImage() { + return new Function>() { + @Override + public Iterable apply(String input) { + return (input == null) ? Collections.emptyList() : getImage(input).asSet(); + } + }; + } + + static Function, Iterable> unwrap() { + return new Function, Iterable>() { + @Override + public Iterable apply(Optional input) { + return (input == null) ? Collections.emptyList() : input.asSet(); + } + }; } }