├── .gitignore
├── README.md
├── deploy.bat
├── deploy.sh
├── pom.xml
└── src
    └── main
        ├── java
            └── com
            │   └── javashitang
            │       ├── WebSocketApplication.java
            │       ├── config
            │           └── WebSocketConfig.java
            │       └── controller
            │           ├── GroupChatController.java
            │           ├── ShowTopController.java
            │           ├── ViewController.java
            │           └── WebSocketServerController.java
        └── resources
            ├── application.yml
            ├── log4j2.xml
            └── templates
                ├── demo.html
                └── index.html


/.gitignore:
--------------------------------------------------------------------------------
1 | /*.iml
2 | /target
3 | /.idea
4 | /.mvn
5 | /server_log
6 | /mvnw
7 | /mvnw.cmd
8 | /logs


--------------------------------------------------------------------------------
/README.md:
--------------------------------------------------------------------------------
 1 | ## WebSocket 使用指南
 2 | 
 3 | 介绍博客：https://blog.csdn.net/zzti_erlie/article/details/101060892
 4 | 
 5 | 在线开房：http://www.javashitang.com:8090/
 6 | 
 7 | 
 8 | 
 9 | 
10 | 


--------------------------------------------------------------------------------
/deploy.bat:
--------------------------------------------------------------------------------
1 | call clean package -DskipTests
2 | cd ./target
3 | scp spring-boot-websocket-1.0.jar root@www.javashitang.com:/opt/application/group-chat
4 | cd ..
5 | @pause
6 | 


--------------------------------------------------------------------------------
/deploy.sh:
--------------------------------------------------------------------------------
1 | mvn clean package -DskipTests
2 | cd ./target
3 | scp spring-boot-websocket-1.0.jar root@www.javashitang.com:/opt/application/group-chat
4 | cd ..
5 | 


--------------------------------------------------------------------------------
/pom.xml:
--------------------------------------------------------------------------------
 1 | <?xml version="1.0" encoding="UTF-8"?>
 2 | <project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
 3 | 	xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
 4 | 	<modelVersion>4.0.0</modelVersion>
 5 | 	<parent>
 6 | 		<groupId>org.springframework.boot</groupId>
 7 | 		<artifactId>spring-boot-starter-parent</artifactId>
 8 | 		<version>2.1.1.RELEASE</version>
 9 | 		<relativePath/> <!-- lookup parent from repository -->
10 | 	</parent>
11 | 
12 | 	<groupId>com.javashitang</groupId>
13 | 	<artifactId>spring-boot-websocket</artifactId>
14 | 	<version>1.0</version>
15 | 	<packaging>jar</packaging>
16 | 	<name>spring-boot-websocket</name>
17 | 	<description>spring-boot-websocket</description>
18 | 
19 | 	<properties>
20 | 		<java.version>1.8</java.version>
21 | 	</properties>
22 | 
23 | 	<dependencies>
24 | 
25 | 		<dependency>
26 | 			<groupId>org.springframework.boot</groupId>
27 | 			<artifactId>spring-boot-starter-web</artifactId>
28 | 			<exclusions>
29 | 				<exclusion>
30 | 					<groupId>org.springframework.boot</groupId>
31 | 					<artifactId>spring-boot-starter-logging</artifactId>
32 | 				</exclusion>
33 | 			</exclusions>
34 | 		</dependency>
35 | 
36 | 		<dependency>
37 | 			<groupId>org.springframework.boot</groupId>
38 | 			<artifactId>spring-boot-starter-log4j2</artifactId>
39 | 		</dependency>
40 | 
41 | 		<dependency>
42 | 			<groupId>org.springframework.boot</groupId>
43 | 			<artifactId>spring-boot-starter-websocket</artifactId>
44 | 		</dependency>
45 | 
46 | 		<dependency>
47 | 			<groupId>org.springframework.boot</groupId>
48 | 			<artifactId>spring-boot-starter-thymeleaf</artifactId>
49 | 		</dependency>
50 | 
51 | 		<dependency>
52 | 			<groupId>org.springframework.boot</groupId>
53 | 			<artifactId>spring-boot-starter-test</artifactId>
54 | 			<scope>test</scope>
55 | 		</dependency>
56 | 
57 | 		<dependency>
58 | 			<groupId>org.projectlombok</groupId>
59 | 			<artifactId>lombok</artifactId>
60 | 			<version>1.16.18</version>
61 | 		</dependency>
62 | 
63 | 		<!-- https://mvnrepository.com/artifact/org.apache.commons/commons-lang3 -->
64 | 		<dependency>
65 | 			<groupId>org.apache.commons</groupId>
66 | 			<artifactId>commons-lang3</artifactId>
67 | 			<version>3.8.1</version>
68 | 		</dependency>
69 | 
70 | 	</dependencies>
71 | 
72 | 	<build>
73 | 		<plugins>
74 | 			<plugin>
75 | 				<groupId>org.springframework.boot</groupId>
76 | 				<artifactId>spring-boot-maven-plugin</artifactId>
77 | 			</plugin>
78 | 		</plugins>
79 | 	</build>
80 | 
81 | 
82 | </project>
83 | 


--------------------------------------------------------------------------------
/src/main/java/com/javashitang/WebSocketApplication.java:
--------------------------------------------------------------------------------
 1 | package com.javashitang;
 2 | 
 3 | import org.springframework.boot.SpringApplication;
 4 | import org.springframework.boot.autoconfigure.SpringBootApplication;
 5 | 
 6 | @SpringBootApplication
 7 | public class WebSocketApplication {
 8 | 
 9 | 	public static void main(String[] args) {
10 | 		SpringApplication.run(WebSocketApplication.class, args);
11 | 	}
12 | 
13 | }
14 | 
15 | 


--------------------------------------------------------------------------------
/src/main/java/com/javashitang/config/WebSocketConfig.java:
--------------------------------------------------------------------------------
 1 | package com.javashitang.config;
 2 | 
 3 | import org.springframework.context.annotation.Bean;
 4 | import org.springframework.context.annotation.Configuration;
 5 | import org.springframework.web.socket.server.standard.ServerEndpointExporter;
 6 | 
 7 | @Configuration
 8 | public class WebSocketConfig {
 9 | 
10 |     @Bean
11 |     public ServerEndpointExporter serverEndpointExporter() {
12 |         return new ServerEndpointExporter();
13 |     }
14 | 
15 | }
16 | 


--------------------------------------------------------------------------------
/src/main/java/com/javashitang/controller/GroupChatController.java:
--------------------------------------------------------------------------------
 1 | package com.javashitang.controller;
 2 | 
 3 | import lombok.extern.slf4j.Slf4j;
 4 | import org.springframework.stereotype.Component;
 5 | 
 6 | import javax.websocket.OnClose;
 7 | import javax.websocket.OnError;
 8 | import javax.websocket.OnMessage;
 9 | import javax.websocket.OnOpen;
10 | import javax.websocket.Session;
11 | import javax.websocket.server.PathParam;
12 | import javax.websocket.server.ServerEndpoint;
13 | import java.io.IOException;
14 | import java.util.ArrayList;
15 | import java.util.List;
16 | import java.util.concurrent.ConcurrentHashMap;
17 | 
18 | @Slf4j
19 | @Component
20 | @ServerEndpoint("/groupChat/{sid}/{username}")
21 | public class GroupChatController {
22 | 
23 |     // 保存 组id->组成员 的映射关系
24 |     private static ConcurrentHashMap<String, List<Session>> groupMemberInfoMap = new ConcurrentHashMap<>();
25 | 
26 |     // 收到消息调用的方法，群成员发送消息
27 |     @OnMessage
28 |     public void onMessage(@PathParam("sid") String sid,
29 |                           @PathParam("username") String username, String message) {
30 |         List<Session> sessionList = groupMemberInfoMap.get(sid);
31 |         // 先一个群组内的成员发送消息
32 |         sessionList.forEach(item -> {
33 |             try {
34 |                 String text = username + ": " + message;
35 |                 item.getBasicRemote().sendText(text);
36 |             } catch (IOException e) {
37 |                 e.printStackTrace();
38 |             }
39 |         });
40 |     }
41 | 
42 |     // 建立连接调用的方法，群成员加入
43 |     @OnOpen
44 |     public void onOpen(Session session, @PathParam("sid") String sid) {
45 |         List<Session> sessionList = groupMemberInfoMap.get(sid);
46 |         if (sessionList == null) {
47 |             sessionList = new ArrayList<>();
48 |             groupMemberInfoMap.put(sid,sessionList);
49 |         }
50 |         sessionList.add(session);
51 |         log.info("Connection connected");
52 |         log.info("sid: {}, sessionList size: {}", sid, sessionList.size());
53 |     }
54 | 
55 |     // 关闭连接调用的方法，群成员退出
56 |     @OnClose
57 |     public void onClose(Session session, @PathParam("sid") String sid) {
58 |         List<Session> sessionList = groupMemberInfoMap.get(sid);
59 |         sessionList.remove(session);
60 |         log.info("Connection closed");
61 |         log.info("sid: {}, sessionList size: {}", sid, sessionList.size());
62 |     }
63 | 
64 |     // 传输消息错误调用的方法
65 |     @OnError
66 |     public void OnError(Throwable error) {
67 |         log.info("Connection error");
68 |     }
69 | }
70 | 


--------------------------------------------------------------------------------
/src/main/java/com/javashitang/controller/ShowTopController.java:
--------------------------------------------------------------------------------
 1 | package com.javashitang.controller;
 2 | 
 3 | import org.springframework.web.bind.annotation.RequestMapping;
 4 | import org.springframework.web.bind.annotation.RestController;
 5 | 
 6 | import java.util.concurrent.TimeUnit;
 7 | 
 8 | /**
 9 |  * @author lilimin
10 |  * @since 2021-03-06
11 |  */
12 | @RestController
13 | @RequestMapping("top")
14 | public class ShowTopController {
15 | 
16 |     private Object lock1 = new Object();
17 |     private Object lock2 = new Object();
18 | 
19 |     @RequestMapping("test")
20 |     public String test() {
21 |         return "success";
22 |     }
23 | 
24 |     @RequestMapping("loop")
25 |     public String loop() {
26 |         System.out.println("start");
27 |         while (true) {}
28 |     }
29 | 
30 |     @RequestMapping("deadlock")
31 |     public String deadlock() {
32 |         new Thread(() -> {
33 |             synchronized (lock1) {
34 |                 try{
35 |                     TimeUnit.SECONDS.sleep(1);
36 |                 } catch (Exception e) {}
37 |                 synchronized (lock2) {
38 |                     System.out.println("thread1 over");
39 |                 }
40 |             }
41 |         }).start();
42 |         new Thread(() -> {
43 |             synchronized (lock2) {
44 |                 try{
45 |                     TimeUnit.SECONDS.sleep(1);
46 |                 } catch (Exception e) {}
47 |                 synchronized (lock1) {
48 |                     System.out.println("thread2 over");
49 |                 }
50 |             }
51 |         }).start();
52 |         return "success";
53 |     }
54 | }
55 | 
56 | 


--------------------------------------------------------------------------------
/src/main/java/com/javashitang/controller/ViewController.java:
--------------------------------------------------------------------------------
 1 | package com.javashitang.controller;
 2 | 
 3 | import org.springframework.stereotype.Controller;
 4 | import org.springframework.web.bind.annotation.GetMapping;
 5 | 
 6 | @Controller
 7 | public class ViewController {
 8 | 
 9 |     @GetMapping("index")
10 |     public String index() {
11 |         return "index";
12 |     }
13 | }
14 | 


--------------------------------------------------------------------------------
/src/main/java/com/javashitang/controller/WebSocketServerController.java:
--------------------------------------------------------------------------------
 1 | package com.javashitang.controller;
 2 | 
 3 | import org.springframework.stereotype.Component;
 4 | 
 5 | import javax.websocket.OnClose;
 6 | import javax.websocket.OnError;
 7 | import javax.websocket.OnMessage;
 8 | import javax.websocket.OnOpen;
 9 | import javax.websocket.Session;
10 | import javax.websocket.server.PathParam;
11 | import javax.websocket.server.ServerEndpoint;
12 | import java.io.IOException;
13 | 
14 | @Component
15 | @ServerEndpoint("/forshow/websocket/{sid}")
16 | public class WebSocketServerController {
17 | 
18 |     // 收到消息调用的方法
19 |     @OnMessage
20 |     public void onMessage(Session session, String message) {
21 |         try {
22 |             session.getBasicRemote().sendText(message);
23 |         } catch (IOException e) {
24 |             e.printStackTrace();
25 |         }
26 |     }
27 | 
28 |     // 建立连接调用的方法
29 |     @OnOpen
30 |     public void onOpen(@PathParam("sid") String sid) {
31 |         System.out.println("Client connected");
32 |     }
33 | 
34 |     // 关闭连接调用的方法
35 |     @OnClose
36 |     public void onClose() {
37 |         System.out.println("Connection closed");
38 |     }
39 | 
40 |     // 传输消息错误调用的方法
41 |     @OnError
42 |     public void OnError(Throwable error) {
43 |         System.out.println("Connection error");
44 |     }
45 | }
46 | 


--------------------------------------------------------------------------------
/src/main/resources/application.yml:
--------------------------------------------------------------------------------
1 | server:
2 |   port: 8090


--------------------------------------------------------------------------------
/src/main/resources/log4j2.xml:
--------------------------------------------------------------------------------
 1 | <?xml version="1.0" encoding="UTF-8"?>
 2 | <!-- http://logging.apache.org/log4j/2.x/manual/configuration.html -->
 3 | <!-- prd config -->
 4 | <Configuration monitorInterval="60">
 5 | 
 6 |     <Properties>
 7 |         <property name="stdoutPattern">%d{HH:mm:ss.SSS} |-%5level %logger{20} [%t] |%X{ip} - %msg%n</property>
 8 |         <property name="filePattern">%d{yyyy-MM-dd HH:mm:ss.SSS} |-%5level %logger{20} [%t] %X{ip} - %msg%n</property>
 9 |         <property name="patternNoTime">%msg%n</property>
10 |         <property name="logDir">./logs</property>
11 |         <property name="logLevel">info</property>
12 |     </Properties>
13 | 
14 |     <Appenders>
15 |         <Console name="stdout" target="SYSTEM_OUT">
16 |             <PatternLayout>
17 |                 <pattern>${stdoutPattern}</pattern>
18 |             </PatternLayout>
19 |         </Console>
20 |         <RollingFile name="logfile" fileName="${logDir}/server.log" filePattern="${logDir}/server-%d{yyyyMMddHH}.log.gz">
21 |             <PatternLayout>
22 |                 <pattern>${filePattern}</pattern>
23 |             </PatternLayout>
24 |             <Policies>
25 |                 <TimeBasedTriggeringPolicy interval="24" modulate="true" />
26 | <!--                <SizeBasedTriggeringPolicy size="2048 MB" /> -->
27 |             </Policies>
28 |             <DefaultRolloverStrategy max="30" />
29 |         </RollingFile>
30 |     </Appenders>
31 |     
32 |     <Loggers>
33 |         <root level="${logLevel}">
34 |             <appender-ref ref="stdout" />
35 |             <appender-ref ref="logfile" />
36 |         </root>
37 |     </Loggers>
38 | 
39 | </Configuration>


--------------------------------------------------------------------------------
/src/main/resources/templates/demo.html:
--------------------------------------------------------------------------------
 1 | <!DOCTYPE html>
 2 | <html>
 3 | <head>
 4 | <title>Testing websockets</title>
 5 | </head>
 6 | <body>
 7 |   <div>
 8 | 	<textarea rows="3" cols="20" id="content"></textarea>
 9 | 	<br>
10 |     <input type="submit" value="Start" onclick="start()" />
11 |   </div>
12 |   <div id="messages"></div>
13 |   <script type="text/javascript">
14 |     var webSocket = 
15 |       new WebSocket('ws://localhost:8080/websocket');
16 | 
17 |     webSocket.onerror = function(event) {
18 |       onError(event)
19 |     };
20 | 
21 |     webSocket.onopen = function(event) {
22 |       onOpen(event)
23 |     };
24 | 
25 |     webSocket.onmessage = function(event) {
26 |       onMessage(event)
27 |     };
28 | 	
29 | 	webSocket.onclose = function(event) {
30 |       onClose(event)
31 |     };
32 | 	
33 | 
34 |     function onMessage(event) {
35 |       document.getElementById('messages').innerHTML 
36 |         += '<br />' + event.data;
37 |     }
38 | 
39 |     function onOpen(event) {
40 |       document.getElementById('messages').innerHTML 
41 |         = 'Connection established';
42 |     }
43 | 
44 |     function onError(event) {
45 |       alert(event);
46 |     }
47 | 	
48 | 	function onClose(event) {
49 |       alert(event);
50 |     }
51 | 
52 |     function start() {
53 | 	  var text = document.getElementById('content').value;
54 |       webSocket.send(text);
55 |     }
56 |   </script>
57 | </body>
58 | </html>


--------------------------------------------------------------------------------
/src/main/resources/templates/index.html:
--------------------------------------------------------------------------------
 1 | <!DOCTYPE html>
 2 | <html>
 3 | <head>
 4 |     <title>Testing websockets</title>
 5 | </head>
 6 | <body>
 7 | <div>
 8 |     <input type="text" id="sid" placeholder="请输入房间号"/><br>
 9 |     <input type="text" id="nickname" placeholder="请输入昵称"/><br>
10 |     <input type="submit" value="连接" onclick="connect()" /><br>
11 |     <textarea rows="3" cols="20" id="content"></textarea><br>
12 |     <input type="submit" value="发送" onclick="start()" />
13 |     <br>
14 | </div>
15 | <div id="messages"></div>
16 | <script type="text/javascript">
17 | 
18 |     var webSocket = null;
19 | 
20 |     function onMessage(event) {
21 |         document.getElementById('messages').innerHTML
22 |             += '<br />' + event.data;
23 |     }
24 | 
25 |     function onOpen(event) {
26 |         document.getElementById('messages').innerHTML
27 |             = 'Connection established';
28 |     }
29 | 
30 |     function onError(event) {
31 |         alert("发生错误");
32 |     }
33 | 
34 |     function onClose(event) {
35 |         alert("连接关闭");
36 |     }
37 | 
38 |     function connect() {
39 |         var sid = document.getElementById('sid').value;
40 |         var nickname = document.getElementById('nickname').value;
41 |         if (url == '' || nickname == '') {
42 |             alert("房间号和用户名不能为空");
43 |             return;
44 |         }
45 |         var serverHot =  window.location.host;
46 |         var url = 'ws://' + serverHot + '/groupChat/' + sid + '/' + nickname;
47 |         webSocket = new WebSocket(url);
48 |         // 发生错误
49 |         webSocket.onerror = function(event) {
50 |             onError(event)
51 |         };
52 | 
53 |         // 连接成功
54 |         webSocket.onopen = function(event) {
55 |             onOpen(event)
56 |         };
57 | 
58 |         // 收到消息
59 |         webSocket.onmessage = function(event) {
60 |             onMessage(event)
61 |         };
62 | 
63 |         // 关闭连接
64 |         webSocket.onclose = function(event) {
65 |             onClose(event)
66 |         };
67 |     }
68 | 
69 |     function start() {
70 |         var text = document.getElementById('content').value;
71 |         webSocket.send(text);
72 |         document.getElementById('content').value = '';
73 |     }
74 | 
75 | </script>
76 | </body>
77 | </html>


--------------------------------------------------------------------------------