SpringBoot的WebSocket初体验- -

引自我的gitbook：

1.POM配置

<dependency>

<groupId>org.springframework.boot</groupId>

<artifactId>spring-boot-starter-websocket</artifactId>

<exclusions>

<exclusion>

<groupId>org.springframework.boot</groupId>

<artifactId>spring-boot-starter-tomcat</artifactId>

</exclusion>

</exclusions>

</dependency>

<dependency>

<groupId>org.springframework.boot</groupId>

<artifactId>spring-boot-starter-undertow</artifactId>

</dependency>

</dependencies>

2.下面是直接搭建服务器。。

package cn.zhenyang.socket.server.controller.socket;

​

import cn.zhenyang.socket.server.config.GetHttpSessionConfigurator;

import cn.zhenyang.socket.server.config.ServerDecoder;

import cn.zhenyang.socket.server.config.ServerEncoder;

import cn.zhenyang.socket.server.constant.Params;

import org.springframework.stereotype.Component;

import org.springframework.web.context.request.RequestContextHolder;

import org.springframework.web.context.request.ServletRequestAttributes;

​

import javax.servlet.http.HttpServletRequest;

import javax.servlet.http.HttpSession;

import javax.websocket.*;

import javax.websocket.server.ServerEndpoint;

import java.io.IOException;

import java.util.concurrent.CopyOnWriteArraySet;

​

/**

* @Author ZhenYang

* @Date Created in 2018/2/19 3:58

* @Description

*/

​

@ServerEndpoint(value = "/websocket",configurator = GetHttpSessionConfigurator.class/*,decoders = ServerDecoder.class,encoders = ServerEncoder.class*/)

@Component

public class WebSocketServer {

//静态变量，用来记录当前在线连接数。应该把它设计成线程安全的。

public static int onlineCount = 0;

//concurrent包的线程安全Set，用来存放每个客户端对应的MyWebSocket对象。

public static CopyOnWriteArraySet<WebSocketServer> webSocketSet = new CopyOnWriteArraySet<WebSocketServer>();

//与某个客户端的连接会话，需要通过它来给客户端发送数据

public Session session;

​

​

​

​

/**

* 连接建立成功调用的方法

*/

@OnOpen

public void onOpen(Session session,EndpointConfig config) {

this.session = session;

​

System.out.println("the session is : "+session.getId());

​

System.out.println("session content is : "+session.getId());

webSocketSet.add(this); //加入set中

addOnlineCount(); //在线数加1

System.out.println("有新连接加入！当前在线人数为" + getOnlineCount());

// ServletActionContext servletActionContext ;

try {

//sendMessage(CommonConstant.CURRENT_WANGING_NUMBER.toString());

// System.out.println(Params.httpSession.getAttribute("xxx"));

sendMessage("22222222");

} catch (IOException e) {

System.out.println("IO异常");

}

}

​

/**

* 连接关闭调用的方法

*/

@OnClose

public void onClose() {

System.out.println(this.session.getId()+":这个断开");

webSocketSet.remove(this); //从set中删除

subOnlineCount(); //在线数减1

System.out.println("有一连接关闭！当前在线人数为" + getOnlineCount());

}

​

/**

* 收到客户端消息后调用的方法

*

* @param message 客户端发送过来的消息

*/

@OnMessage

public void onMessage(String message, Session session) {

// HttpServletRequest request =

// System.out.println(httpSession.getAttribute("xxx"));

System.out.println("来自客户端的消息:" + message);

System.out.println("onMessage sessionId is : "+session.getId());

//群发消息

for (WebSocketServer item : webSocketSet) {

try {

item.sendMessage(message);

} catch (IOException e) {

e.printStackTrace();

}

}

}

​

/**

* 发生错误时调用

*/

@OnError

public void onError(Session session, Throwable error) {

System.out.println(error.getMessage());

}

​

/**

* 给当前用户发送消息

* @param message

* @throws IOException

*/

public void sendMessage(String message) throws IOException {

System.out.println(Params.httpSession.getAttribute("xxx"));

this.session.getBasicRemote().sendText(message+" : sendMessage id is : "+this.session.getId());

//this.session.getAsyncRemote().sendText(message);

}

​

/**

* 群发自定义消息

*/

public static void sendInfo(String message) throws IOException {

for (WebSocketServer item : webSocketSet) {

try {

item.sendMessage(message);

} catch (IOException e) {

continue;

}

}

}

​

public static synchronized int getOnlineCount() {

return onlineCount;

}

​

public static synchronized void addOnlineCount() {

WebSocketServer.onlineCount++;

}

​

public static synchronized void subOnlineCount() {

WebSocketServer.onlineCount--;

}

}

另声明了一个静态会话。

public static HttpSession httpSession;

相关配置：

@Component

public class GetHttpSessionConfigurator extends ServerEndpointConfig.Configurator {

@Override

public void modifyHandshake(ServerEndpointConfig sec, HandshakeRequest request, HandshakeResponse response) {

HttpSession httpSession = (HttpSession) request.getHttpSession();

//解决httpSession为null的情况

if (httpSession == null) {

HttpServletRequest request2 = ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes()).getRequest();

httpSession = request2.getSession();

// httpSession = Params.httpSession;

}

sec.getUserProperties().put(HttpSession.class.getName(), httpSession);

}

}

​

/**

*

* @Author ZhenYang

* @Date Created in 2018/1/31 9:13

* @Description

* 使用@ServerEndpoint创立websocket endpoint

* 首先要注入 ServerEndpointExporter，

* 这个bean会自动注册使用了 @ServerEndpoint 注

* 解声明的 Web Socket endpoint。

* 要注意，如果使用独立的 Servlet 容器，

* 而不是直接使用 Spring Boot 的内置容器，

* 就不要注入 ServerEndpointExporter，

* 因为它将由容器自己提供和管理

*/

@Configuration

//@EnableWebSocket

//public class WebSocketConfig implements WebSocketConfigurer {

public class WebSocketConfig {

​

@Bean

public ServerEndpointExporter serverEndpointExporter(){

return new ServerEndpointExporter();

}

//@Override

//public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {

// registry.addHandler(myHandler(),"/websocket")

// .addInterceptors(new HttpSessionHandshakeInterceptor())

// ;

//}

// @Bean

// public WebSocketHandler myHandler(){

// return new MyHandler();

// }

}

3.搭建一下客户端

客户端就这一个类

package cn.zhenyang.socket.client.controller.rest;

​

import cn.zhenyang.common.websocket.constant.BasicParameter;

import cn.zhenyang.common.websocket.utils.WebSocketClientUtil;

import cn.zhenyang.socket.client.entity.User;

import com.alibaba.fastjson.JSON;

import org.java_websocket.client.WebSocketClient;

import org.springframework.web.bind.annotation.PostMapping;

import org.springframework.web.bind.annotation.RequestBody;

import org.springframework.web.bind.annotation.RequestMapping;

import org.springframework.web.bind.annotation.RestController;

import org.springframework.web.util.NestedServletException;

​

/**

* @Author ZhenYang

* @Date Created in 2018/2/19 23:48

* @Description

*/

@RestController

@RequestMapping("/")

public class WebsocketClientRest {

public static WebSocketClient client;

​

@RequestMapping("init")

public void initConnect() {

client = WebSocketClientUtil.getWebSocketClient(BasicParameter.WebSocketServerURI);

System.out.println(client.getDraft());

// while (!client.getReadyState().equals(WebSocket.READYSTATE.OPEN)) {

// System.out.println("还没有打开");

// }

// System.out.println("打开了");

}

​

@PostMapping("send")

public void send() {

// String str = JSON.toJSONString(user);

try {

client.send("asdfsadfdsfas");

}catch (Exception e){

System.out.println("连接失败");

}

​

}

@RequestMapping("closeSocket")

public void close(){

client.close();

}

}

也涉及一个静态变量，用于指定服务器地址

public static final String WebSocketServerURI = "ws://localhost:8081/websocket";

​