aidlab/aida_back
7
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge remote-tracking branch 'origin/dev/dev' into dev/dev

Browse Source 
# Conflicts:
#	src/main/java/com/ai/da/service/AccountService.java
This commit is contained in:
shahaibo
2024-11-11 16:06:21 +08:00
parent 8c20b51e13 391a28d659
commit 37ff7fbb91
13 changed files with 129 additions and 43 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/main/java/com/ai/da/common/constant/CommonConstant.java
View File

@@ -71,4 +71,9 @@ public class CommonConstant {
public static final String RCA_WORKSHOP_TAG = "#RCAworkshop_2024";
public static final String PORTFOLIO_DELETED_EN = "Portfolio has been deleted";
public static final String PORTFOLIO_DELETED_CN = "作品已删除";
}

6
src/main/java/com/ai/da/common/task/AccountTask.java
View File

@@ -19,7 +19,7 @@ public class AccountTask {
/**
* 每周日晚上刷新 年付用户、月付用户的积分
*/
@Scheduled(cron = "59 59 23 ? * SUN")
// @Scheduled(cron = "59 59 23 ? * SUN")
// @Scheduled(cron = "59 59 23 * * ?")
public void refreshCreditsMonthly() {
log.info("每周日晚11：59：59刷新付费用户积分为 6000");
@@ -33,7 +33,7 @@ public class AccountTask {
}
// 每天凌晨0点执行一次
@Scheduled(cron = "0 0 0 * * ?")
// @Scheduled(cron = "0 0 0 * * ?")
public void cancelActivityBenefits() {
// 1、查询当前所有参与了活动且过期的用户
List<Account> accountList = accountService.getExpiredUserBySystemUser(4);
@@ -46,7 +46,7 @@ public class AccountTask {
}
// 每天检测正式用户到期情况，每天凌晨0点执行
@Scheduled(cron = "0 0 0 * * ?")
// @Scheduled(cron = "0 0 0 * * ?")
public void paidUserToVisitor() {
// 1、查询当前已过期正式用户或试用用户
List<Account> accountList = accountService.getExpiredUserBySystemUser(1);

52
src/main/java/com/ai/da/common/websocket/NotificationConnection.java
View File

@@ -16,11 +16,14 @@ import java.util.concurrent.ConcurrentHashMap;
@Slf4j
public class NotificationConnection {
//连接超时
public static final long MAX_TIME_OUT = 2 * 60 * 1000;
static Map<String,Session> sessionMap = new ConcurrentHashMap<>();
private Session session;
// 这里用ConcurrentHashMap 因为他是一个线程安全的Map
private static ConcurrentHashMap<Long, NotificationConnection> websockets = new ConcurrentHashMap<>();
private static ConcurrentHashMap<NotificationConnection, Long> websockets = new ConcurrentHashMap<>();
//连接建立时执行的操作
/*@OnOpen
public void onOpen(Session session){
@@ -31,24 +34,43 @@ public class NotificationConnection {
@OnOpen
public void onOpen(Session session, @PathParam("id") String id) { // 接收到前端传来的用户ID
this.session = session;
websockets.put(Long.parseLong(id), this); //将ID作为key，当前的对象作为Value
// this.session.setMaxIdleTimeout(MAX_TIME_OUT);
websockets.put(this, Long.parseLong(id)); //将ID作为key，当前的对象作为Value
log.info("【建立连接】 用户为：{}", this.session);
log.info("【建立连接】 用户Id为：{}", id);
log.info("【建立连接】 总数为：{}", websockets.size());
}
@OnClose
public void onClose() {
websockets.remove(this); // 将当前的对象从集合中删除
log.info("【连接断开】 用户为：{}", this.session);
public void onClose(CloseReason reason) {
if (websockets.containsKey(this)) {
websockets.remove(this); // 将当前的对象从集合中删除
log.info("【连接断开】 用户为：{}, sessionId: {}, 原因为{}", websockets.get(this), this.session.getId(), reason);
}
// log.info("【连接断开】 总数为：{}", websockets.size());
}
/**
* 错误时调用
* @param throwable 异常
*/
@OnError
public void onError(Throwable throwable) {
log.info("【连接异常】 用户为：{} , sessionId: {}", websockets.get(this), this.session.getId(), throwable);
websockets.remove(this); // 将当前的对象从集合中删除
}
//收到了客户端消息执行的操作
@OnMessage
public void onMessage(String text){
public void onMessage(Session session, String text){
log.info("收到了一条消息："+text);
// return "已收到你的消息";
if (text.equals("PING")){
sendMsg("PONG", websockets.get(this));
session.setMaxIdleTimeout(MAX_TIME_OUT);
}
}
/*//连接关闭的时候执行的操作
@OnClose
@@ -63,27 +85,31 @@ public class NotificationConnection {
}
}*/
public void sendMsg(String message, Long userId) throws IOException {
public void sendMsg(String message, Long userId) {
if (userId == null) { // 如果等于null则证明是群发
// 获取当前Map的一个迭代器，遍历Map的方式有很多种，看着来
// 这个就是遍历这个集合的过程....
for (Map.Entry<Long, NotificationConnection> entry : websockets.entrySet()) {
for (Map.Entry<NotificationConnection, Long> entry : websockets.entrySet()) {
// 获取每一个Entry实例
// 获取每一个Value，而这个Value就是WebSocket的实例
NotificationConnection webSocket = entry.getValue();
NotificationConnection webSocket = entry.getKey();
// 接下来就是遍历群发
log.info("广播消息 【给用户】 ：{}发送消息【{}】", webSocket, message);
webSocket.session.getBasicRemote().sendText(message); // 发送！！！！！！！！！
try {
webSocket.session.getBasicRemote().sendText(message); // 发送！！！！！！！！！
} catch (IOException e) {
log.error("Failed to send message to session {}: {}", webSocket.session.getId(), e.getMessage());
}
}
} else { // 如果不是群发，则判断ID，其余步骤一致
// 获取当前Map的一个迭代器，遍历Map的方式有很多种，看着来
// 这个就是遍历这个集合的过程....
for (Map.Entry<Long, NotificationConnection> entry : websockets.entrySet()) {
for (Map.Entry<NotificationConnection, Long> entry : websockets.entrySet()) {
// 获取每一个Entry实例
// 获取每一个Value，而这个Value就是WebSocket的实例
NotificationConnection webSocket = entry.getValue();
NotificationConnection webSocket = entry.getKey();
// 获取每一个Key，这个Key就是用户ID
Long key = entry.getKey();
Long key = entry.getValue();
// 判断用户ID与当前的Key相等
if (userId.equals(key)) {
log.info("私发消息 【给用户】 ：{}发送消息【{}】", key, message); // 打印

2
src/main/java/com/ai/da/common/websocket/config/WebSocketConfig.java
View File

@@ -6,8 +6,6 @@ import org.springframework.web.socket.server.standard.ServerEndpointExporter;
/**
* Configuration of WebSocket
*
* @author db1995
*/
@Configuration
public class WebSocketConfig {