websocket 客户端

pom.xml

@@ -71,6 +71,11 @@
             <version>4.1.2</version>
         </dependency>
         <dependency>
+            <groupId>org.java-websocket</groupId>
+            <artifactId>Java-WebSocket</artifactId>
+            <version>1.3.5</version>
+        </dependency>
+        <dependency>
             <groupId>io.jsonwebtoken</groupId>
             <artifactId>jjwt</artifactId>
             <version>0.9.0</version>

src/main/java/com/iflytop/nuclear/NuclearApplication.java

@@ -1,20 +1,14 @@
 package com.iflytop.nuclear;
 
-import com.iflytop.nuclear.websocket.WebSocketClient;
 import org.mybatis.spring.annotation.MapperScan;
 import org.springframework.boot.Banner;
 import org.springframework.boot.ResourceBanner;
-import org.springframework.boot.SpringApplication;
 import org.springframework.boot.autoconfigure.SpringBootApplication;
 import org.springframework.boot.builder.SpringApplicationBuilder;
 import org.springframework.core.io.ClassPathResource;
 
-import javax.websocket.ContainerProvider;
 import javax.websocket.DeploymentException;
-import javax.websocket.Session;
-import javax.websocket.WebSocketContainer;
 import java.io.IOException;
-import java.net.URI;
 
 @SpringBootApplication
 @MapperScan("com.iflytop.nuclear.mapper")

src/main/java/com/iflytop/nuclear/config/WebSocketConfig.java

@@ -0,0 +1,47 @@
+package com.iflytop.nuclear.config;
+
+import lombok.extern.slf4j.Slf4j;
+import org.java_websocket.client.WebSocketClient;
+import org.java_websocket.drafts.Draft_6455;
+import org.java_websocket.handshake.ServerHandshake;
+import org.springframework.context.annotation.Bean;
+import org.springframework.stereotype.Component;
+
+import java.net.URI;
+
+@Component
+@Slf4j
+public class WebSocketConfig {
+    @Bean
+    public WebSocketClient webSocketClient() {
+        try {
+            WebSocketClient webSocketClient = new WebSocketClient(new URI("ws://localhost:12306"), new Draft_6455()) {
+                @Override
+                public void onOpen(ServerHandshake handshakedata) {
+                    log.info("[websocket] 连接成功");
+                }
+
+                @Override
+                public void onMessage(String message) {
+                    log.info("[websocket] 收到消息={}", message);
+                    //  在这里重写对消息的处理分发
+                }
+
+                @Override
+                public void onClose(int code, String reason, boolean remote) {
+                    log.info("[websocket] 退出连接");
+                }
+
+                @Override
+                public void onError(Exception ex) {
+                    log.info("[websocket] 连接错误={}", ex.getMessage());
+                }
+            };
+            webSocketClient.connect();
+            return webSocketClient;
+        } catch (Exception e) {
+            e.printStackTrace();
+        }
+        return null;
+    }
+}

src/main/java/com/iflytop/nuclear/controller/TestController.java

@@ -0,0 +1,24 @@
+package com.iflytop.nuclear.controller;
+
+/**
+ * @author cool
+ * @date 2023/7/6 15:31
+ */
+
+import com.iflytop.nuclear.websocket.SocketClient;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.web.bind.annotation.GetMapping;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RestController;
+@RestController
+@RequestMapping("/websocket")
+public class TestController {
+
+    @Autowired
+    private SocketClient webScoketClient;
+
+    @GetMapping("/sendMessage")
+    public void sendMessage(){
+        webScoketClient.groupSending("123");
+    }
+}

src/main/java/com/iflytop/nuclear/websocket/SocketClient.java

@@ -0,0 +1,15 @@
+package com.iflytop.nuclear.websocket;
+
+import org.java_websocket.client.WebSocketClient;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Component;
+
+@Component
+public class SocketClient {
+    @Autowired
+    private WebSocketClient webSocketClient;
+
+    public void groupSending(String message) {
+        webSocketClient.send(message+"---6666");
+    }
+}

src/main/java/com/iflytop/nuclear/websocket/WebSocketClient.java

@@ -1,68 +0,0 @@
-package com.iflytop.nuclear.websocket;
-
-import com.alibaba.fastjson.JSON;
-import com.alibaba.fastjson.JSONObject;
-import com.iflytop.nuclear.handler.MessageHandler;
-import lombok.extern.slf4j.Slf4j;
-import org.springframework.beans.factory.annotation.Autowired;
-import org.springframework.stereotype.Component;
-import org.springframework.util.StringUtils;
-
-import javax.websocket.*;
-import java.io.IOException;
-import java.net.URI;
-
-/**
- * @author cool
- * @date 2023/7/4 19:12
- */
-@ClientEndpoint
-@Component
-@Slf4j
-public class WebSocketClient {
-
-    @Autowired
-    MessageHandler messageHandler;
-
-    private Session session;
-
-    @OnOpen
-    public void onOpen(Session session) {
-        this.session = session;
-    }
-
-    // WebSocketContainer container = ContainerProvider.getWebSocketContainer();
-    // String uri ="ws://127.0.0.1:8888/websocket/123";
-    // container.connectToServer(WebSocketClient.class, URI.create(uri));
-
-    @OnMessage
-    public void onMessage(String message) throws IOException {
-        if (!StringUtils.isEmpty(message)) {
-            try {
-                JSONObject jsonObject = JSON.parseObject(message);
-                String command = jsonObject.getString("command");
-                // 根据command类型，分发给不同的handler和detection
-                if (!command.isEmpty()) {
-                    messageHandler.dispatcher(command, jsonObject, this.session);
-                }
-            } catch (Exception e) {
-                JSONObject msg = new JSONObject();
-                msg.put("message", "传输数据必须为包含command字段的JSON格式");
-                this.session.getBasicRemote().sendText(msg.toString());
-                e.printStackTrace();
-            }
-        }
-    }
-
-    @OnError
-    public void onError(Throwable t) {
-        t.printStackTrace();
-    }
-
-    public void sendMessage(String message) throws IOException {
-        if(this.session != null) {
-            this.session.getBasicRemote().sendText(message);
-        }
-    }
-
-}