diff --git a/lab-25/lab-websocket-25-02/src/main/java/cn/iocoder/springboot/lab25/springwebsocket/handler/AuthMessageHandler.java b/lab-25/lab-websocket-25-02/src/main/java/cn/iocoder/springboot/lab25/springwebsocket/handler/AuthMessageHandler.java
index 6d93efd4..fe97846e 100644
--- a/lab-25/lab-websocket-25-02/src/main/java/cn/iocoder/springboot/lab25/springwebsocket/handler/AuthMessageHandler.java
+++ b/lab-25/lab-websocket-25-02/src/main/java/cn/iocoder/springboot/lab25/springwebsocket/handler/AuthMessageHandler.java
@@ -6,14 +6,13 @@ import cn.iocoder.springboot.lab25.springwebsocket.message.UserJoinNoticeRequest
 import cn.iocoder.springboot.lab25.springwebsocket.util.WebSocketUtil;
 import org.springframework.stereotype.Component;
 import org.springframework.util.StringUtils;
-
-import javax.websocket.Session;
+import org.springframework.web.socket.WebSocketSession;
 
 @Component
 public class AuthMessageHandler implements MessageHandler<AuthRequest> {
 
     @Override
-    public void execute(Session session, AuthRequest message) {
+    public void execute(WebSocketSession session, AuthRequest message) {
         // 如果未传递 accessToken
         if (StringUtils.isEmpty(message.getAccessToken())) {
             WebSocketUtil.send(session, AuthResponse.TYPE,
diff --git a/lab-25/lab-websocket-25-02/src/main/java/cn/iocoder/springboot/lab25/springwebsocket/handler/MessageHandler.java b/lab-25/lab-websocket-25-02/src/main/java/cn/iocoder/springboot/lab25/springwebsocket/handler/MessageHandler.java
index aa89c13c..704f4505 100644
--- a/lab-25/lab-websocket-25-02/src/main/java/cn/iocoder/springboot/lab25/springwebsocket/handler/MessageHandler.java
+++ b/lab-25/lab-websocket-25-02/src/main/java/cn/iocoder/springboot/lab25/springwebsocket/handler/MessageHandler.java
@@ -1,8 +1,7 @@
 package cn.iocoder.springboot.lab25.springwebsocket.handler;
 
 import cn.iocoder.springboot.lab25.springwebsocket.message.Message;
-
-import javax.websocket.Session;
+import org.springframework.web.socket.WebSocketSession;
 
 /**
  * 消息处理器接口
@@ -15,7 +14,7 @@ public interface MessageHandler<T extends Message> {
      * @param session 会话
      * @param message 消息
      */
-    void execute(Session session, T message);
+    void execute(WebSocketSession session, T message);
 
     /**
      * @return 消息类型，即每个 Message 实现类上的 TYPE 静态字段
diff --git a/lab-25/lab-websocket-25-02/src/main/java/cn/iocoder/springboot/lab25/springwebsocket/handler/SendToAllHandler.java b/lab-25/lab-websocket-25-02/src/main/java/cn/iocoder/springboot/lab25/springwebsocket/handler/SendToAllHandler.java
index cda1efac..837080cb 100644
--- a/lab-25/lab-websocket-25-02/src/main/java/cn/iocoder/springboot/lab25/springwebsocket/handler/SendToAllHandler.java
+++ b/lab-25/lab-websocket-25-02/src/main/java/cn/iocoder/springboot/lab25/springwebsocket/handler/SendToAllHandler.java
@@ -5,14 +5,13 @@ import cn.iocoder.springboot.lab25.springwebsocket.message.SendToAllRequest;
 import cn.iocoder.springboot.lab25.springwebsocket.message.SendToUserRequest;
 import cn.iocoder.springboot.lab25.springwebsocket.util.WebSocketUtil;
 import org.springframework.stereotype.Component;
-
-import javax.websocket.Session;
+import org.springframework.web.socket.WebSocketSession;
 
 @Component
 public class SendToAllHandler implements MessageHandler<SendToAllRequest> {
 
     @Override
-    public void execute(Session session, SendToAllRequest message) {
+    public void execute(WebSocketSession session, SendToAllRequest message) {
         // 这里，假装直接成功
         SendResponse sendResponse = new SendResponse().setMsgId(message.getMsgId()).setCode(0);
         WebSocketUtil.send(session, SendResponse.TYPE, sendResponse);
diff --git a/lab-25/lab-websocket-25-02/src/main/java/cn/iocoder/springboot/lab25/springwebsocket/handler/SendToOneHandler.java b/lab-25/lab-websocket-25-02/src/main/java/cn/iocoder/springboot/lab25/springwebsocket/handler/SendToOneHandler.java
index d7437bd4..ab1079da 100644
--- a/lab-25/lab-websocket-25-02/src/main/java/cn/iocoder/springboot/lab25/springwebsocket/handler/SendToOneHandler.java
+++ b/lab-25/lab-websocket-25-02/src/main/java/cn/iocoder/springboot/lab25/springwebsocket/handler/SendToOneHandler.java
@@ -5,14 +5,13 @@ import cn.iocoder.springboot.lab25.springwebsocket.message.SendToOneRequest;
 import cn.iocoder.springboot.lab25.springwebsocket.message.SendToUserRequest;
 import cn.iocoder.springboot.lab25.springwebsocket.util.WebSocketUtil;
 import org.springframework.stereotype.Component;
-
-import javax.websocket.Session;
+import org.springframework.web.socket.WebSocketSession;
 
 @Component
 public class SendToOneHandler implements MessageHandler<SendToOneRequest> {
 
     @Override
-    public void execute(Session session, SendToOneRequest message) {
+    public void execute(WebSocketSession session, SendToOneRequest message) {
         // 这里，假装直接成功
         SendResponse sendResponse = new SendResponse().setMsgId(message.getMsgId()).setCode(0);
         WebSocketUtil.send(session, SendResponse.TYPE, sendResponse);