websocket群发和指定用户发，用户之间可以互相发消息，工作人员可以给指定用户或者群体发消息
package com.example.websocket.websocketdemo;
import org.springframework.stereotype.Component;
import javax.websocket.*;
import javax.websocket.server.PathParam;
import javax.websocket.server.ServerEndpoint;
import java.io.IOException;
import java.text.DateFormat;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.concurrent.ConcurrentHashMap;
/**
* @ServerEndpoint 注解是一个类层次的注解，它的功能主要是将目前的类定义成一个websocket服务器端,
* 注解的值将被用于监听用户连接的终端访问URL地址,客户端可以通过这个URL来连接到WebSocket服务器端
* @ServerEndpoint 可以把当前类变成websocket服务类
*/
@ServerEndpoint("/websocket/{userno}")
@Component
public class WebSocketTest {
//静态变量，用来记录当前在线连接数。应该把它设计成线程安全的。
private static int onlineCount = 0;
//concurrent包的线程安全Set，用来存放每个客户端对应的MyWebSocket对象。若要实现服务端与单一客户端通信的话，可以使用Map来存放，其中Key可以为用户标识
private static ConcurrentHashMap<String, WebSocketTest> webSocketSet = new ConcurrentHashMap<String, WebSocketTest>();
//与某个客户端的连接会话，需要通过它来给客户端发送数据
private Session session;
//当前发消息的人员编号
private String userno = "";
/**
* 连接建立成功调用的方法
*
* @param session 可选的参数。session为与某个客户端的连接会话，需要通过它来给客户端发送数据
*/
@OnOpen
public void onOpen(@PathParam(value = "userno") String param, Session session, EndpointConfig config) {
System.out.println(param);
userno = param;//接收到发送消息的人员编号
this.session = session;
webSocketSet.put(param, this);//加入map中
addOnlineCount(); //在线数加1
System.out.println("有新连接加入！当前在线人数为" getOnlineCount());
}
/**
* 连接关闭调用的方法
*/
@OnClose
public void onClose() {
if (!userno.equals("")) {
webSocketSet.remove(userno); //从set中删除
subOnlineCount(); //在线数减1
System.out.println("有一连接关闭！当前在线人数为" getOnlineCount());
}
}
/**
* 收到客户端消息后调用的方法
*
* @param message 客户端发送过来的消息
* @param session 可选的参数
*/
@OnMessage
public void onMessage(String message, Session session) {
System.out.println("来自客户端的消息:" message);
// session.get
//群发消息
if (1 < 2) {
sendAll(message);
} else {
//给指定的人发消息
sendToUser(message);
}
}
/**
* 给指定的人发送消息
* @param message
*/
private void sendToUser(String message) {
String sendUserno = message.split("[|]")[1];
String sendMessage = message.split("[|]")[0];
String now = getNowTime();
try {
if (webSocketSet.get(sendUserno) != null) {
webSocketSet.get(sendUserno).sendMessage(now "用户" userno "发来消息：" " <br/> " sendMessage);
} else {
System.out.println("当前用户不在线");
}
} catch (IOException e) {
e.printStackTrace();
}
}
/**
* 给所有人发消息
* @param message
*/
private void sendAll(String message) {
String now = getNowTime();
String sendMessage = message.split("[|]")[0];
//遍历HashMap
for (String key : webSocketSet.keySet()) {
try {
//判断接收用户是否是当前发消息的用户
if (!userno.equals(key)) {
webSocketSet.get(key).sendMessage(now "用户" userno "发来消息：" " <br/> " sendMessage);
System.out.println("key = " key);
}
} catch (IOException e) {
e.printStackTrace();
}
}
}
/**
* 获取当前时间
*
* @return
*/
private String getNowTime() {
Date date = new Date();
DateFormat format = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
String time = format.format(date);
return time;
}
/**
* 发生错误时调用
*
* @param session
* @param error
*/
@OnError
public void onError(Session session, Throwable error) {
System.out.println("发生错误");
error.printStackTrace();
}
/**
* 这个方法与上面几个方法不一样。没有用注解，是根据自己需要添加的方法。
*
* @param message
* @throws IOException
*/
public void sendMessage(String message) throws IOException {
this.session.getBasicRemote().sendText(message);
//this.session.getAsyncRemote().sendText(message);
}
public static synchronized int getOnlineCount() {
return onlineCount;
}
public static synchronized void addOnlineCount() {
WebSocketTest.onlineCount ;
}
public static synchronized void subOnlineCount() {
WebSocketTest.onlineCount--;
}
}
资源文件列表
websocket/.gitignore , 249
websocket/.idea/.name , 13
websocket/.idea/compiler.xml , 630
websocket/.idea/encodings.xml , 172
websocket/.idea/libraries/Maven__ch_qos_logback_logback_classic_1_1_11.xml , 575
websocket/.idea/libraries/Maven__ch_qos_logback_logback_core_1_1_11.xml , 554
websocket/.idea/libraries/Maven__com_fasterxml_classmate_1_3_4.xml , 522
websocket/.idea/libraries/Maven__com_fasterxml_jackson_core_jackson_annotations_2_8_0.xml , 644
websocket/.idea/libraries/Maven__com_fasterxml_jackson_core_jackson_core_2_8_10.xml , 602
websocket/.idea/libraries/Maven__com_fasterxml_jackson_core_jackson_databind_2_8_10.xml , 630
websocket/.idea/libraries/Maven__com_jayway_jsonpath_json_path_2_2_0.xml , 546
websocket/.idea/libraries/Maven__com_vaadin_external_google_android_json_0_0_20131108_vaadin1.xml , 700
websocket/.idea/libraries/Maven__javax_validation_validation_api_1_1_0_Final.xml , 611
websocket/.idea/libraries/Maven__junit_junit_4_12.xml , 455
websocket/.idea/libraries/Maven__net_minidev_accessors_smart_1_1.xml , 542
websocket/.idea/libraries/Maven__net_minidev_json_smart_2_2_1.xml , 521
websocket/.idea/libraries/Maven__org_apache_tomcat_embed_tomcat_embed_core_8_5_23.xml , 625
websocket/.idea/libraries/Maven__org_apache_tomcat_embed_tomcat_embed_el_8_5_23.xml , 611
websocket/.idea/libraries/Maven__org_apache_tomcat_embed_tomcat_embed_websocket_8_5_23.xml , 660
websocket/.idea/libraries/Maven__org_apache_tomcat_tomcat_annotations_api_8_5_23.xml , 636
websocket/.idea/libraries/Maven__org_assertj_assertj_core_2_6_0.xml , 535
websocket/.idea/libraries/Maven__org_hamcrest_hamcrest_core_1_3.xml , 532
websocket/.idea/libraries/Maven__org_hamcrest_hamcrest_library_1_3.xml , 553
websocket/.idea/libraries/Maven__org_hibernate_hibernate_validator_5_3_5_Final.xml , 634
websocket/.idea/libraries/Maven__org_jboss_logging_jboss_logging_3_3_1_Final.xml , 608
websocket/.idea/libraries/Maven__org_mockito_mockito_core_1_10_19.xml , 549
websocket/.idea/libraries/Maven__org_objenesis_objenesis_2_1.xml , 508
websocket/.idea/libraries/Maven__org_ow2_asm_asm_5_0_3.xml , 472
websocket/.idea/libraries/Maven__org_skyscreamer_jsonassert_1_4_0.xml , 537
websocket/.idea/libraries/Maven__org_slf4j_jcl_over_slf4j_1_7_25.xml , 548
websocket/.idea/libraries/Maven__org_slf4j_jul_to_slf4j_1_7_25.xml , 534
websocket/.idea/libraries/Maven__org_slf4j_log4j_over_slf4j_1_7_25.xml , 562
websocket/.idea/libraries/Maven__org_slf4j_slf4j_api_1_7_25.xml , 513
websocket/.idea/libraries/Maven__org_springframework_boot_spring_boot_1_5_8_RELEASE.xml , 636
websocket/.idea/libraries/Maven__org_springframework_boot_spring_boot_autoconfigure_1_5_8_RELEASE.xml , 734
websocket/.idea/libraries/Maven__org_springframework_boot_spring_boot_starter_1_5_8_RELEASE.xml , 692
websocket/.idea/libraries/Maven__org_springframework_boot_spring_boot_starter_logging_1_5_8_RELEASE.xml , 748
websocket/.idea/libraries/Maven__org_springframework_boot_spring_boot_starter_test_1_5_8_RELEASE.xml , 727
websocket/.idea/libraries/Maven__org_springframework_boot_spring_boot_starter_tomcat_1_5_8_RELEASE.xml , 741
websocket/.idea/libraries/Maven__org_springframework_boot_spring_boot_starter_websocket_1_3_5_RELEASE.xml , 762
websocket/.idea/libraries/Maven__org_springframework_boot_spring_boot_starter_web_1_5_8_RELEASE.xml , 720
websocket/.idea/libraries/Maven__org_springframework_boot_spring_boot_test_1_5_8_RELEASE.xml , 671
websocket/.idea/libraries/Maven__org_springframework_boot_spring_boot_test_autoconfigure_1_5_8_RELEASE.xml , 769
websocket/.idea/libraries/Maven__org_springframework_spring_aop_4_3_12_RELEASE.xml , 616
websocket/.idea/libraries/Maven__org_springframework_spring_beans_4_3_12_RELEASE.xml , 630
websocket/.idea/libraries/Maven__org_springframework_spring_context_4_3_12_RELEASE.xml , 644
websocket/.idea/libraries/Maven__org_springframework_spring_core_4_3_12_RELEASE.xml , 623
websocket/.idea/libraries/Maven__org_springframework_spring_expression_4_3_12_RELEASE.xml , 665
websocket/.idea/libraries/Maven__org_springframework_spring_messaging_4_3_12_RELEASE.xml , 658
websocket/.idea/libraries/Maven__org_springframework_spring_test_4_3_12_RELEASE.xml , 623
websocket/.idea/libraries/Maven__org_springframework_spring_webmvc_4_3_12_RELEASE.xml , 637
websocket/.idea/libraries/Maven__org_springframework_spring_websocket_4_3_12_RELEASE.xml , 658
websocket/.idea/libraries/Maven__org_springframework_spring_web_4_3_12_RELEASE.xml , 616
websocket/.idea/libraries/Maven__org_yaml_snakeyaml_1_17.xml , 495
websocket/.idea/misc.xml , 439
websocket/.idea/modules.xml , 258
websocket/.idea/workspace.xml , 43837
websocket/.mvn/wrapper/maven-wrapper.jar , 47610
websocket/.mvn/wrapper/maven-wrapper.properties , 110
websocket/mvnw , 6468
websocket/mvnw.cmd , 4994
websocket/pom.xml , 1632
websocket/src/main/java/com/example/websocket/websocketdemo/WebSocketConfig.java , 423
websocket/src/main/java/com/example/websocket/websocketdemo/WebsocketdemoApplication.java , 813
websocket/src/main/java/com/example/websocket/websocketdemo/WebSocketTest.java , 5648
websocket/src/main/resources/application.properties , 0
websocket/src/main/resources/index.html , 2971
websocket/src/test/java/com/example/websocket/websocketdemo/WebsocketdemoApplicationTests.java , 359
websocket/target/classes/application.properties , 0
websocket/target/classes/com/example/websocket/websocketdemo/Contr.class , 598
websocket/target/classes/com/example/websocket/websocketdemo/WebSocketConfig.class , 744
websocket/target/classes/com/example/websocket/websocketdemo/WebsocketdemoApplication.class , 1286
websocket/target/classes/com/example/websocket/websocketdemo/WebSocketTest.class , 5016
websocket/target/classes/index.html , 1872
websocket/target/test-classes/com/example/websocket/websocketdemo/WebsocketdemoApplicationTests.class , 687
websocket/websocket.iml , 6643