diff --git a/broker/src/main/java/io/moquette/persistence/MemoryMessagesStore.java b/broker/src/main/java/io/moquette/persistence/MemoryMessagesStore.java index f9520787c..4c3e4cee6 100755 --- a/broker/src/main/java/io/moquette/persistence/MemoryMessagesStore.java +++ b/broker/src/main/java/io/moquette/persistence/MemoryMessagesStore.java @@ -3101,27 +3101,16 @@ public ErrorCode saveAddFriendRequest(String userId, WFCMessage.AddFriendRequest } } - MultiMap friendsMap = hzInstance.getMultiMap(USER_FRIENDS); - - FriendData friendData1 = null; - Collection friendDatas = friendsMap.get(userId); - if (friendDatas == null || friendDatas.size() == 0) { - friendDatas = loadFriend(friendsMap, userId); - } - for (FriendData fd : friendDatas) { - if (fd.getFriendUid().equals(request.getTargetUid())) { - friendData1 = fd; - break; - } - } - if (friendData1 != null && friendData1.getBlacked() > 0 && !isAdmin) { - return ErrorCode.ERROR_CODE_IN_BLACK_LIST; - } - + FriendData friendData1 = getFriendData(userId, request.getTargetUid()); if (friendData1 != null && friendData1.getState() == 0) { return ErrorCode.ERROR_CODE_ALREADY_FRIENDS; } + FriendData friendData2 = getFriendData(request.getTargetUid(), userId); + if (friendData2 != null && friendData2.getBlacked() > 0 && !isAdmin) { + return ErrorCode.ERROR_CODE_IN_BLACK_LIST; + } + WFCMessage.FriendRequest newRequest = WFCMessage.FriendRequest .newBuilder() .setFromUid(userId)