Load friends before modifying them.

[Sone.git] / src / main / java / net / pterodactylus / sone / database / memory / MemoryFriendDatabase.java

diff --git a/src/main/java/net/pterodactylus/sone/database/memory/MemoryFriendDatabase.java b/src/main/java/net/pterodactylus/sone/database/memory/MemoryFriendDatabase.java
index 1802843..0be8738 100644 (file)
--- a/src/main/java/net/pterodactylus/sone/database/memory/MemoryFriendDatabase.java
+++ b/src/main/java/net/pterodactylus/sone/database/memory/MemoryFriendDatabase.java
@@ -43,6 +43,7 @@ class MemoryFriendDatabase {
        }
 
        void addFriend(String localSoneId, String friendSoneId) {
+               loadFriends(localSoneId);
                lock.writeLock().lock();
                try {
                        if (soneFriends.put(localSoneId, friendSoneId)) {
@@ -54,6 +55,7 @@ class MemoryFriendDatabase {
        }
 
        void removeFriend(String localSoneId, String friendSoneId) {
+               loadFriends(localSoneId);
                lock.writeLock().lock();
                try {
                        if (soneFriends.remove(localSoneId, friendSoneId)) {