package com.fastbee.base.service.impl;
|
|
import com.fastbee.base.service.ISessionStore;
|
import com.fastbee.base.session.Session;
|
import org.springframework.stereotype.Service;
|
|
import java.util.LinkedHashMap;
|
import java.util.Map;
|
import java.util.concurrent.ConcurrentHashMap;
|
import java.util.concurrent.atomic.AtomicInteger;
|
|
/**
|
* 内存存储服务会话
|
*
|
* @author gsb
|
* @date 2022/10/14 14:18
|
*/
|
@Service
|
public class SessionStoreImpl implements ISessionStore {
|
|
/*session存储集合*/
|
private final ConcurrentHashMap<String, Session> sessionMap = new ConcurrentHashMap<>();
|
|
/**
|
* MQTT会话存储
|
*
|
* @param clientId: 客户端标识
|
* @param session: MQTT会话
|
*/
|
@Override
|
public void storeSession(String clientId, Session session) {
|
sessionMap.put(clientId, session);
|
}
|
|
/**
|
* 根据客户端标识获取相应会话
|
*
|
* @param clientId: 客户端标识
|
*/
|
@Override
|
public Session getSession(String clientId) {
|
return sessionMap.get(clientId);
|
}
|
|
/**
|
* 清除历史会话状态
|
*
|
* @param clientId: 客户端标识
|
*/
|
@Override
|
public void cleanSession(String clientId) {
|
sessionMap.remove(clientId);
|
}
|
|
/**
|
* 根据客户端标识查看是否存在该会话
|
*
|
* @param clientId:
|
*/
|
@Override
|
public boolean containsKey(String clientId) {
|
return sessionMap.containsKey(clientId);
|
}
|
|
/**
|
* 获取集合
|
* @return MAP
|
*/
|
@Override
|
public ConcurrentHashMap<String, Session> getSessionMap(){
|
return sessionMap;
|
}
|
|
|
/**
|
* map分页（从1开始）
|
*
|
* @param sourceMap 分页数据
|
* @param pageSize 页面大小
|
* @param currentPage 当前页面
|
*/
|
@Override
|
public Map<String, Session> listPage(Map<String, Session> sourceMap, int pageSize, int currentPage) {
|
Map<String, Session> map = new LinkedHashMap<>();
|
if (sourceMap.size() > 0) {
|
AtomicInteger flag = new AtomicInteger(0);
|
AtomicInteger size = new AtomicInteger(0);
|
int currIdx = (currentPage > 1 ? (currentPage - 1) * pageSize : 0);
|
sourceMap.forEach((ass, list_km) -> {
|
if (flag.get() >= currIdx) {
|
if (size.get() < pageSize) {
|
map.put(ass, list_km);
|
} else {
|
return;
|
}
|
size.getAndIncrement();
|
}
|
flag.getAndIncrement();
|
});
|
|
}
|
return map;
|
}
|
}
|
