WebFlux 中 WebSocket 实现通信

Author: liqiangqiang

springboot-webflux-8-websocket/pom.xml

@@ -0,0 +1,45 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
+         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
+  <modelVersion>4.0.0</modelVersion>
+
+  <groupId>springboot</groupId>
+  <artifactId>springboot-webflux-8-websocket</artifactId>
+  <version>0.0.1-SNAPSHOT</version>
+  <name>springboot-webflux-8-websocket :: Spring Boot WebFlux 中 WebSocket 实现通信</name>
+
+  <parent>
+    <groupId>org.springframework.boot</groupId>
+    <artifactId>spring-boot-starter-parent</artifactId>
+    <version>2.0.1.RELEASE</version>
+    <relativePath/> <!-- lookup parent from repository -->
+  </parent>
+
+  <properties>
+    <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
+    <project.reporting.outputEncoding>UTF-8</project.reporting.outputEncoding>
+    <java.version>1.8</java.version>
+  </properties>
+
+  <dependencies>
+    <dependency>
+      <groupId>org.springframework.boot</groupId>
+      <artifactId>spring-boot-starter-webflux</artifactId>
+    </dependency>
+
+    <dependency>
+      <groupId>org.springframework.boot</groupId>
+      <artifactId>spring-boot-starter-test</artifactId>
+      <scope>test</scope>
+    </dependency>
+  </dependencies>
+
+  <build>
+    <plugins>
+      <plugin>
+        <groupId>org.springframework.boot</groupId>
+        <artifactId>spring-boot-maven-plugin</artifactId>
+      </plugin>
+    </plugins>
+  </build>
+</project>

springboot-webflux-8-websocket/src/main/java/org/spring/springboot/Application.java

@@ -0,0 +1,21 @@
+package org.spring.springboot;
+
+import org.springframework.boot.SpringApplication;
+import org.springframework.boot.autoconfigure.SpringBootApplication;
+
+/**
+ * Spring Boot 应用启动类
+ *
+ * Created by bysocket on 09/29/2017.
+ */
+// Spring Boot 应用的标识
+@SpringBootApplication
+public class Application {
+
+    public static void main(String[] args) {
+        // 程序启动入口
+        // 启动嵌入式的 Tomcat 并初始化 Spring 环境及其各 Spring 组件
+        SpringApplication.run(Application.class,args);
+
+    }
+}

springboot-webflux-8-websocket/src/main/java/org/spring/springboot/config/WebSocketConfiguration.java

@@ -0,0 +1,35 @@
+package org.spring.springboot.config;
+
+import org.spring.springboot.handler.EchoHandler;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.core.Ordered;
+import org.springframework.web.reactive.HandlerMapping;
+import org.springframework.web.reactive.handler.SimpleUrlHandlerMapping;
+import org.springframework.web.reactive.socket.WebSocketHandler;
+import org.springframework.web.reactive.socket.server.support.WebSocketHandlerAdapter;
+
+import java.util.HashMap;
+import java.util.Map;
+
+@Configuration
+public class WebSocketConfiguration {
+
+    @Autowired
+    @Bean
+    public HandlerMapping webSocketMapping(final EchoHandler echoHandler) {
+        final Map<String, WebSocketHandler> map = new HashMap<>();
+        map.put("/echo", echoHandler);
+
+        final SimpleUrlHandlerMapping mapping = new SimpleUrlHandlerMapping();
+        mapping.setOrder(Ordered.HIGHEST_PRECEDENCE);
+        mapping.setUrlMap(map);
+        return mapping;
+    }
+
+    @Bean
+    public WebSocketHandlerAdapter handlerAdapter() {
+        return new WebSocketHandlerAdapter();
+    }
+}

springboot-webflux-8-websocket/src/main/java/org/spring/springboot/handler/EchoHandler.java

@@ -0,0 +1,17 @@
+
+package org.spring.springboot.handler;
+
+import org.springframework.stereotype.Component;
+import org.springframework.web.reactive.socket.WebSocketHandler;
+import org.springframework.web.reactive.socket.WebSocketSession;
+import reactor.core.publisher.Mono;
+
+@Component
+public class EchoHandler implements WebSocketHandler {
+    @Override
+    public Mono<Void> handle(final WebSocketSession session) {
+        return session.send(
+                session.receive()
+                        .map(msg -> session.textMessage("服务端返回：小明， -> " + msg.getPayloadAsText())));
+    }
+}

springboot-webflux-8-websocket/src/main/resources/application.properties

@@ -0,0 +1,28 @@
+<!DOCTYPE html>
+<html lang="en">
+<head>
+  <meta charset="UTF-8">
+  <title>Client WebSocket</title>
+</head>
+<body>
+
+<div class="events"></div>
+<script>
+  var clientWebSocket = new WebSocket("ws://localhost:8080/echo");
+  clientWebSocket.onopen = function () {
+    console.log("clientWebSocket.onopen", clientWebSocket);
+    console.log("clientWebSocket.readyState", "websocketstatus");
+    clientWebSocket.send("你好");
+  }
+
+  clientWebSocket.onclose = function (error) {
+    console.log("clientWebSocket.onclose", clientWebSocket, error);
+    events("再见");
+  }
+
+  function events(responseEvent) {
+    document.querySelector(".events").innerHTML += responseEvent + "<br>";
+  }
+</script>
+</body>
+</html>

springboot-webflux-8-websocket/src/main/resources/websocket-client.html

@@ -0,0 +1,19 @@
+import org.springframework.web.reactive.socket.WebSocketMessage;
+import org.springframework.web.reactive.socket.client.ReactorNettyWebSocketClient;
+import org.springframework.web.reactive.socket.client.WebSocketClient;
+import reactor.core.publisher.Flux;
+
+import java.net.URI;
+import java.time.Duration;
+
+public class WSClient {
+    public static void main(final String[] args) {
+        final WebSocketClient client = new ReactorNettyWebSocketClient();
+        client.execute(URI.create("ws://localhost:8080/echo"), session ->
+                session.send(Flux.just(session.textMessage("你好")))
+                        .thenMany(session.receive().take(1).map(WebSocketMessage::getPayloadAsText))
+                        .doOnNext(System.out::println)
+                        .then())
+                .block(Duration.ofMillis(5000));
+    }
+}