目錄
- Java Websocket 01: 原生模式 Websocket 基礎通信
- Java Websocket 02: 原生模式通過 Websocket 傳輸檔案
Websocket 原生模式
原生模式下
- 服務端通過 @ServerEndpoint 實作其對應的 @OnOpen, @OnClose, @OnMessage, @OnError 方法
- 客戶端創建 WebSocketClient 實作對應的 onOpen(), onClose(), onMessage(), one rror()
演示專案
完整示例代碼 https://github.com/MiltonLai/websocket-demos/tree/main/ws-demo01
目錄結構
│ pom.xml
└───src
├───main
│ ├───java
│ │ └───com
│ │ └───rockbb
│ │ └───test
│ │ └───wsdemo
│ │ SocketServer.java
│ │ WebSocketConfig.java
│ │ WsDemo01App.java
│ └───resources
│ application.yml
└───test
└───java
└───com
└───rockbb
└───test
└───wsdemo
SocketClient.java
pom.xml
- 可以用 JDK11, 也可以用 JDK17
- 通過 Spring Boot plugin repackage, 生成 fat jar
- 用 Java-WebSocket 作為 client 的 websocket 實作庫, 當前最新版本為 1.5.3
<?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>com.rockbb.test</groupId>
<artifactId>ws-demo01</artifactId>
<packaging>jar</packaging>
<version>1.0-SNAPSHOT</version>
<name>WS: Demo 01</name>
<properties>
<!-- Global encoding -->
<project.jdk.version>17</project.jdk.version>
<project.source.encoding>UTF-8</project.source.encoding>
<!-- Global dependency versions -->
<spring-boot.version>2.7.11</spring-boot.version>
</properties>
<dependencyManagement>
<dependencies>
<!-- Spring Boot Dependencies -->
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-dependencies</artifactId>
<version>${spring-boot.version}</version>
<type>pom</type>
<scope>import</scope>
</dependency>
</dependencies>
</dependencyManagement>
<dependencies>
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-web</artifactId>
</dependency>
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-test</artifactId>
<scope>test</scope>
</dependency>
<dependency>
<groupId>org.springframework</groupId>
<artifactId>spring-websocket</artifactId>
</dependency>
<dependency>
<groupId>org.springframework</groupId>
<artifactId>spring-messaging</artifactId>
</dependency>
<dependency>
<groupId>org.java-websocket</groupId>
<artifactId>Java-WebSocket</artifactId>
<version>1.5.3</version>
</dependency>
</dependencies>
<build>
<finalName>ws-demo01</finalName>
<plugins>
<plugin>
<groupId>org.apache.maven.plugins</groupId>
<artifactId>maven-compiler-plugin</artifactId>
<version>3.10.1</version>
<configuration>
<source>${project.jdk.version}</source>
<target>${project.jdk.version}</target>
<encoding>${project.source.encoding}</encoding>
</configuration>
</plugin>
<plugin>
<groupId>org.apache.maven.plugins</groupId>
<artifactId>maven-resources-plugin</artifactId>
<version>3.3.0</version>
<configuration>
<encoding>${project.source.encoding}</encoding>
</configuration>
</plugin>
<plugin>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-maven-plugin</artifactId>
<version>${spring-boot.version}</version>
<executions>
<execution>
<goals>
<goal>repackage</goal>
</goals>
</execution>
</executions>
</plugin>
</plugins>
</build>
</project>
application.yml
設定服務埠為 8763
server:
port: 8763
tomcat:
uri-encoding: UTF-8
spring:
application:
name: ws-demo01
WsDemo01App.java
- 將 @RestController 也合并到應用入口了. 和單獨拆開做一個 Controller 類是一樣的
- '/msg' 路徑用于從 server 往 client 發送訊息
@RestController
@SpringBootApplication
public class WsDemo01App {
public static void main(String[] args) {
SpringApplication.run(WsDemo01App.class, args);
}
@RequestMapping("/msg")
public String sendMsg(String sessionId, String msg) throws IOException {
Session session = SocketServer.getSession(sessionId);
SocketServer.sendMessage(session, msg);
return "send " + sessionId + " : " + msg;
}
}
WebSocketConfig.java
必須顯式宣告 ServerEndpointExporter 這個 Bean 才能提供 websocket 服務
@Configuration
public class WebSocketConfig {
@Bean
public ServerEndpointExporter initServerEndpointExporter(){
return new ServerEndpointExporter();
}
}
SocketServer.java
提供 websocket 服務的關鍵類. @ServerEndpoint 的作用類似于 RestController, 這里指定 client 訪問的路徑格式為 ws://host:port/websocket/server/[id],
當 client 訪問使用不同的 id 時, 會對應產生不同的 SocketServer 實體
@Component
@ServerEndpoint("/websocket/server/{sessionId}")
public class SocketServer {
private static final org.slf4j.Logger log = LoggerFactory.getLogger(SocketServer.class);
private static final Map<String, Session> sessionMap = new ConcurrentHashMap<>();
private String sessionId = "";
@OnOpen
public void onOpen(Session session, @PathParam("sessionId") String sessionId) {
this.sessionId = sessionId;
/* Old connection will be kicked by new connection */
sessionMap.put(sessionId, session);
/*
* this: instance id. New instances will be created for each sessionId
* sessionId: assigned from path variable
* session.getId(): the actual session id (start from 0)
*/
log.info("On open: this{} sessionId {}, actual {}", this, sessionId, session.getId());
}
@OnClose
public void onClose() {
sessionMap.remove(sessionId);
log.info("On close: sessionId {}", sessionId);
}
@OnMessage
public void onMessage(String message, Session session) {
log.info("On message: sessionId {}, {}", session.getId(), message);
}
@OnError
public void one rror(Session session, Throwable error) {
log.error("On error: sessionId {}, {}", session.getId(), error.getMessage());
}
public static void sendMessage(Session session, String message) throws IOException {
session.getBasicRemote().sendText(message);
}
public static Session getSession(String sessionId){
return sessionMap.get(sessionId);
}
}
關于會話物件 Session
OnOpen 會注入一個 Session 引數, 這個是實際的 Websocket Session, 其 ID 是全域唯一的, 可以唯一確定一個客戶端連接. 在當前版本的實作中, 這是一個從0開始自增的整數. 如果你需要實作例如單個用戶登錄多個會話, 在通信中, 將訊息轉發給同一個用戶的多個會話, 就要小心記錄這些 Session 的 ID.
@OnOpen
public void onOpen(Session session, @PathParam("sessionId") String sessionId)
關于會話意外關閉
在客戶端意外停止后, 服務端會收到 one rror 訊息, 可以通過這個訊息管理已經關閉的會話
SocketClient.java
client 測驗類, 連接后可以通過命令列向 server 發送訊息
public class SocketClient {
private static final org.slf4j.Logger log = LoggerFactory.getLogger(SocketClient.class);
public static void main(String[] args) throws URISyntaxException {
WebSocketClient wsClient = new WebSocketClient(
new URI("ws://127.0.0.1:8763/websocket/server/10001")) {
@Override
public void onOpen(ServerHandshake serverHandshake) {
log.info("On open: {}, {}", serverHandshake.getHttpStatus(), serverHandshake.getHttpStatusMessage());
}
@Override
public void onMessage(String s) {
log.info("On message: {}", s);
}
@Override
public void onClose(int i, String s, boolean b) {
log.info("On close: {}, {}, {}", i, s, b);
}
@Override
public void one rror(Exception e) {
log.info("On error: {}", e.getMessage());
}
};
wsClient.connect();
log.info("Connecting...");
while (!ReadyState.OPEN.equals(wsClient.getReadyState())) {
try {
Thread.sleep(100);
} catch (InterruptedException e) {
log.error(e.getMessage(), e);
}
}
log.info("Connected");
wsClient.send("hello");
Scanner scanner = new Scanner(System.in);
while (scanner.hasNext()) {
String line = scanner.next();
wsClient.send(line);
}
wsClient.close();
}
}
代碼的執行程序就是新建一個 WebSocketClient 并實作其處理訊息的介面方法, 使用 10001 作為 sessionId 進行連接, 在連接成功后, 不斷讀取鍵盤輸入 (System.in), 將輸入的字串發送給服務端.
運行示例
示例是一個普通的 Spring Boot jar專案, 可以通過mvn clean package進行編譯, 再通過java -jar ws-demo01.jar運行, 啟動后作業在8763埠
然后運行 SocketClient.java, 可以觀察到服務端接收到的訊息.
服務端可以通過瀏覽器訪問 http://127.0.0.1:8763/msg?sessionId=10001&msg=123 向客戶端發送訊息.
結論
以上說明并演示了原生的 Websocket 實作方式, 可以嘗試運行多個 SocketClient, 使用相同或不同的 server sessionId 路徑, 觀察通信的變化情況.
轉載請註明出處，本文鏈接：https://www.uj5u.com/houduan/555503.html
標籤：其他
上一篇：Scala集合
下一篇：返回列表