上传文件至 websocket

websocket/SockJsTool.java

@@ -0,0 +1,103 @@
+package com.example.utils.SockJs;
+
+
+import org.springframework.messaging.converter.StringMessageConverter;
+import org.springframework.messaging.simp.stomp.StompFrameHandler;
+import org.springframework.messaging.simp.stomp.StompHeaders;
+import org.springframework.messaging.simp.stomp.StompSession;
+import org.springframework.messaging.simp.stomp.StompSessionHandlerAdapter;
+import org.springframework.scheduling.concurrent.ThreadPoolTaskScheduler;
+import org.springframework.util.concurrent.ListenableFuture;
+import org.springframework.web.socket.WebSocketHttpHeaders;
+import org.springframework.web.socket.client.standard.StandardWebSocketClient;
+import org.springframework.web.socket.messaging.WebSocketStompClient;
+import org.springframework.web.socket.sockjs.client.SockJsClient;
+import org.springframework.web.socket.sockjs.client.Transport;
+import org.springframework.web.socket.sockjs.client.WebSocketTransport;
+
+import java.util.ArrayList;
+import java.util.List;
+import java.util.concurrent.ExecutionException;
+
+
+/*<dependency>
+            <groupId>org.java-websocket</groupId>
+            <artifactId>Java-WebSocket</artifactId>
+            <version>1.5.2</version>
+        </dependency>*/
+
+/**
+ * 请求工具
+ */
+public class SockJsTool extends StompSessionHandlerAdapter {
+    /**
+     * 请求地址
+     */
+    private String url;
+    /**
+     * 请求 token
+     */
+    private String token;
+
+    /**
+     *
+     */
+    private StompSession session;
+
+
+    public SockJsTool(String url, String token) {
+        this.url = url;
+        this.token = token;
+        this.httpConnect();
+    }
+
+    public SockJsTool(String url) {
+        this.url = url;
+        this.httpConnect();
+    }
+
+
+    public SockJsTool() {
+    }
+
+    public StompSession getSession() {
+        return session;
+    }
+
+
+    /**
+     * http 链接websocket
+     */
+    public void httpConnect(){
+        //判断是否已连接
+        if(session == null||!session.isConnected()){
+            System.out.println("当前处于断开状态,尝试链接");
+
+            List<Transport> transports = new ArrayList<>();
+            transports.add(new WebSocketTransport(new StandardWebSocketClient()));
+            SockJsClient sockJsClient = new SockJsClient(transports);
+            WebSocketStompClient webSocketStompClient=new WebSocketStompClient(sockJsClient);
+            webSocketStompClient.setMessageConverter(new StringMessageConverter());
+            webSocketStompClient.setDefaultHeartbeat(new long[] { 20000, 0 });
+
+            ThreadPoolTaskScheduler taskScheduler = new ThreadPoolTaskScheduler();
+            taskScheduler.afterPropertiesSet();
+            webSocketStompClient.setTaskScheduler(taskScheduler);
+
+            WebSocketHttpHeaders webSocketHttpHeaders = null;
+            StompHeaders stompHeaders = new StompHeaders();
+            stompHeaders.add("token", token);
+            stompHeaders.set("user","1");
+            ListenableFuture<StompSession> future = webSocketStompClient.connect(url, webSocketHttpHeaders, stompHeaders, new SockJsTool());
+            try {
+                this.session = future.get();
+                this.session.setAutoReceipt(true);
+            } catch (InterruptedException e) {
+                throw new RuntimeException(e);
+            } catch (ExecutionException e) {
+                throw new RuntimeException(e);
+            }
+        }
+    }
+
+}

websocket/TEST009.java

@@ -0,0 +1,26 @@
+package com.example.utils.SockJs;
+
+
+import org.springframework.messaging.simp.stomp.StompHeaders;
+import org.springframework.messaging.simp.stomp.StompSession;
+import org.springframework.messaging.simp.stomp.StompSessionHandlerAdapter;
+
+import java.util.concurrent.ExecutionException;
+
+public class TEST009 {
+    public static void main(String[] args) throws ExecutionException, InterruptedException {
+        SockJsTool tool=new SockJsTool("http://localhost:8080/system/websocket");
+
+        StompSession session=tool.getSession();
+        session.subscribe("/user/topic/subscribePrint5",new StompSessionHandlerAdapter(){
+            @Override
+            public void handleFrame(StompHeaders headers, Object payload) {
+                System.out.println(payload);
+            }
+        });
+
+        session.send("/app/downloadLocal","2");
+
+
+    }
+}