From 6535848ee32870ec2b0873c1d90ab2157b6908e3 Mon Sep 17 00:00:00 2001
From: whycxzp <glperry@163.com>
Date: 星期五, 13 六月 2025 16:54:50 +0800
Subject: [PATCH] 站点信息统计,电池品牌信息统计
---
src/main/java/com/whyc/webSocket/LeaderHomeSocket.java | 9 ++++-----
1 files changed, 4 insertions(+), 5 deletions(-)
diff --git a/src/main/java/com/whyc/webSocket/LeaderHomeSocket.java b/src/main/java/com/whyc/webSocket/LeaderHomeSocket.java
index 6d7bd0c..b61f47b 100644
--- a/src/main/java/com/whyc/webSocket/LeaderHomeSocket.java
+++ b/src/main/java/com/whyc/webSocket/LeaderHomeSocket.java
@@ -2,8 +2,8 @@
import com.whyc.config.WebSocketConfig;
import com.whyc.dto.Response;
+import com.whyc.pojo.db_user.User;
import com.whyc.service.LeaderHomeService;
-import com.whyc.service.LoginService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
@@ -11,8 +11,6 @@
import javax.websocket.*;
import javax.websocket.server.ServerEndpoint;
import java.io.IOException;
-import java.util.HashMap;
-import java.util.Map;
/**
* 棰嗗灞傞椤�
@@ -36,13 +34,14 @@
public void onOpen(Session session, EndpointConfig config){
this.session = session;
HttpSession httpSession = (HttpSession) config.getUserProperties().get("httpSession");
-
+ User user = (User) httpSession.getAttribute("user");
+ Integer userId = user.getId();
Thread thread = new Thread() {
@Override
public void run() {
try {
while (!currentThread().isInterrupted()) {
- Response response = leaderHomeService.getAll();
+ Response response = leaderHomeService.getAll(userId);
session.getBasicRemote().sendObject(new Response().set(1, response));
sleep(4000);
}
--
Gitblit v1.9.1