InformSocketServerManager.java

package com.mycim.webapp.websocket;

import com.mycim.framework.logging.Logger;
import com.mycim.framework.logging.LoggerFactory;
import com.mycim.server.security.service.SecurityService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import java.util.HashMap;
import java.util.Map;

@Component
public class InformSocketServerManager {

    private Logger log = LoggerFactory.getLogger(InformSocketServerManager.class);

    private Map<Long, InformSocket> allServers = new HashMap<>();

    @Autowired
    SecurityService securityService;

    public void addServer(Long userRrn, InformSocket informSocket) {
        allServers.put(userRrn, informSocket);
        log.debug(userRrn + "用户加入连接! 当前连接数:"+allServers.size());
    }

    public void removeServer(Long userRrn) {
        allServers.remove(userRrn);
        log.debug(userRrn + "用户退出连接! 当前连接数:"+allServers.size());
    }

    public Map<Long, String> getInformCacheData(Long userRrn) {
        return securityService.getInformUserAndMsg(userRrn);
    }

}