Springboot实现WebSocket客户端

在前面的博客中 SpringBoot中的WebSocket讲述了如何搭建基于Springboot的websocket服务端，但是测试的时候是直接用浏览器中的在线测试工具测试的，如何直接写代码测试呢，本篇文章就讲一下如何搭建websocket客户端。

1. 新建一个maven项目命名为websocket_client

1.1 完善pom文件

    4.0.0

    com.owner
    websocket_client
    1.0-SNAPSHOT

    
        org.springframework.boot
        spring-boot-starter-parent
        2.0.6.RELEASE
    

    
        
            org.springframework.boot
            spring-boot-starter-web
        

		
        
            org.java-websocket
            Java-WebSocket
            1.3.8
        
    

1.2 新增application.yml文件

server:
  port: 8002
#  设置服务名
spring:
  application:
    name: websocket-client

1.3 新增启动类

@SpringBootApplication
public class WebSocketApplication {
    public static void main(String[] args){
        SpringApplication.run(WebSocketApplication.class, args);
    }
}

2. WebSocket客户端核心代码实现

2.1 新增WebSocketClient实现类

public class MyWebSocketClient extends WebSocketClient {
    private static final Logger LOGGER = LoggerFactory.getLogger(MyWebSocketClient.class);

    public MyWebSocketClient(URI serverUri) {
        super(serverUri);
    }

    @Override
    public void onOpen(ServerHandshake arg0) {
        // TODO Auto-generated method stub
        LOGGER.info("------ MyWebSocket onOpen ------");
    }

    @Override
    public void onClose(int arg0, String arg1, boolean arg2) {
        // TODO Auto-generated method stub
        LOGGER.info("------ MyWebSocket onClose ------{}",arg1);
    }

    @Override
    public void onError(Exception arg0) {
        // TODO Auto-generated method stub
        LOGGER.info("------ MyWebSocket onError ------{}",arg0);
    }

    @Override
    public void onMessage(String arg0) {
        // TODO Auto-generated method stub
        LOGGER.info("-------- 接收到服务端数据： " + arg0 + "--------");
    }
}

2.2 启动类中添加初始化MyWebSocketClient代码，将MyWebSocketClient交给Spring容器管理

@Bean
    public WebSocketClient webSocketClient() {
        try {
            MyWebSocketClient webSocketClient = new MyWebSocketClient(new URI("ws://127.0.0.1:8001/websocket/12/owner"));
            webSocketClient.connect();
            return webSocketClient;
        } catch (URISyntaxException e) {
            e.printStackTrace();
        }
        return null;
    }

2.3 添加一个WebSocketController类，用发送请求的方式控制WebSocket客户端与服务端通讯

@RestController
public class WebSocketController {

    @Autowired
    private WebSocketClient webSocketClient;

    @RequestMapping("subscribe")
    public String subscribe() {
       // webSocketClient.connect();
        webSocketClient.send("hello sever，i want subscribe data A");
        return "发送订阅成功！！！";
    }
}

3. 测试

首先启动websocket_server,确定websocket服务端正常运行，然后启动websocket_client。
在浏览器中访问http://127.0.0.1:8002/subscribe

再来看一下客户端的日志

上面一条日志是客户端启动时连接服务端成功的日志，第二条是发送订阅信息的日志，服务端返回订阅成功

服务端日志

可以看到服务端正常收到了订阅请求