anna/IM-server-new
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

添加好友请求限频功能

Browse Source
This commit is contained in:
heavyrian2012
2024-09-12 21:34:12 +08:00
parent fc35865478
commit a45de0ecd1
6 changed files with 26 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
broker/src/main/java/io/moquette/persistence/MemoryMessagesStore.java
+16 -1
View File
@@ -155,6 +155,7 @@ public class MemoryMessagesStore implements IMessagesStore {
private long mFriendRequestDuration = 7 * 24 * 60 * 60 * 1000;
private long mFriendRejectDuration = 30 * 24 * 60 * 60 * 1000;
private long mFriendRequestExpiration = 7 * 24 * 60 * 60 * 1000;
private RateLimiter mFriendRateLimiter;
private boolean mFriendRobotAutoAccept = true;
private long mPushExpiredTimes = 604800000L;
@@ -344,6 +345,13 @@ public class MemoryMessagesStore implements IMessagesStore {
printMissConfigLog(FRIEND_Request_Expiration_Duration, mFriendRequestExpiration + "");
}
try {
int friendRequestRateLimit = Integer.parseInt(m_Server.getConfig().getProperty(FRIEND_Request_Rate_Limit, "0"));
if(friendRequestRateLimit > 0) {
mFriendRateLimiter = new RateLimiter(86400, friendRequestRateLimit);
}
} catch (Exception e) {
}
try {
mFriendRobotAutoAccept = Boolean.parseBoolean(m_Server.getConfig().getProperty(BrokerConstants.FRIEND_Request_Robot_Auto_Accept, "true"));
@@ -3353,11 +3361,18 @@ public class MemoryMessagesStore implements IMessagesStore {
}
@Override
public ErrorCode saveAddFriendRequest(String userId, WFCMessage.AddFriendRequest request, long[] head, boolean isAdmin) {
public ErrorCode saveAddFriendRequest(String userId, WFCMessage.AddFriendRequest request, long[] head, boolean isAdmin, boolean isRobot) {
if (!isAdmin && mDisableFriendRequest) {
return ErrorCode.ERROR_CODE_NOT_RIGHT;
}
if(!isRobot && !isAdmin && mFriendRateLimiter != null) {
if(!mFriendRateLimiter.isGranted(userId)) {
LOG.error("User {} request add friend over frequency", userId);
return ErrorCode.ERROR_CODE_OVER_FREQUENCY;
}
}
HazelcastInstance hzInstance = m_Server.getHazelcastInstance();
MultiMap<String, WFCMessage.FriendRequest> requestMap = hzInstance.getMultiMap(USER_FRIENDS_REQUEST);