优化disconnect功能

This commit is contained in:
heavyrain2012
2021-10-08 11:07:08 +08:00
parent 4bcf784a6d
commit 3d7220d79c
5 changed files with 21 additions and 12 deletions
@@ -109,19 +109,19 @@ public class ClientSession {
}
public void disconnect(boolean cleanSession, boolean disableSession) {
public void disconnect(String userId, boolean cleanSession, boolean disableSession) {
if (cleanSession) {
LOG.info("Client disconnected. Removing its subscriptions. ClientId={}", clientID);
// cleanup topic subscriptions
cleanSession();
cleanSession(userId);
} else if(disableSession) {
LOG.info("Client disconnected. disable session. ClientId={}", clientID);
m_sessionsStore.disableSession(this.clientID);
m_sessionsStore.disableSession(userId, this.clientID);
}
}
public void cleanSession() {
m_sessionsStore.cleanSession(this.clientID);
public void cleanSession(String userId) {
m_sessionsStore.cleanSession(userId, this.clientID);
}
@@ -150,8 +150,8 @@ public interface ISessionsStore {
*/
int getSecondPhaseAckPendingMessages(String clientID);
void disableSession(String clientId);
void cleanSession(String clientID);
void disableSession(String userId, String clientId);
void cleanSession(String userId, String clientID);
boolean isMultiEndpointSupported();
@@ -44,12 +44,14 @@ import io.netty.handler.timeout.IdleStateHandler;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import win.liyufan.im.HttpUtils;
import win.liyufan.im.IMTopic;
import win.liyufan.im.Utility;
import java.util.*;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import static cn.wildfirechat.common.ErrorCode.ERROR_CODE_SUCCESS;
import static io.moquette.spi.impl.InternalRepublisher.createPublishForQos;
import static cn.wildfirechat.common.IMExceptionEvent.EventType.EVENT_CALLBACK_Exception;
import static io.moquette.spi.impl.Utils.readBytesAndRewind;
@@ -582,7 +584,7 @@ public class ProtocolProcessor {
//disconnect the session
m_sessionsStore.sessionForClient(clientID).disconnect(isDup, isRetain);
m_sessionsStore.sessionForClient(clientID).disconnect(session.getUsername(), isDup, isRetain);
}