springboot整合websocket最基础入门使用教程详解

<dependency>

       <groupid>org.springframework.boot</groupid>

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

</dependency>

<dependency>

       <groupid>org.projectlombok</groupid>

       <artifactid>lombok</artifactid>

       <version> 1.18 . 12 </version>

</dependency>

package cn.huawei.socket_test_1.config;

import org.springframework.context.annotation.bean;

import org.springframework.stereotype测试数据ponent;

import org.springframework.web.socket.server.standard.serverendpointexporter;

@component

public class websocketconfig {

   /**

    * serverendpointexporter 作用

    *

    * 这个bean会自动注册使用@serverendpoint注解声明的websocket endpoint

    *

    * @return

    */

   @bean

   public serverendpointexporter serverendpointexporter() {

     return new serverendpointexporter();

   }

}

package cn.huawei.socket_test_1.websock;

import lombok.extern.slf4j.slf4j;

import org.springframework.stereotype测试数据ponent;

import javax.websocket.*;

import javax.websocket.server.pathparam;

import javax.websocket.server.serverendpoint;

import java.util.concurrent.concurrenthashmap;

/**

  * @serverendpoint 注解的作用

  *

  * @serverendpoint 注解是一个类层次的注解，它的功能主要是将目前的类定义成一个websocket服务器端,

  * 注解的值将被用于监听用户连接的终端访问url地址,客户端可以通过这个url来连接到websocket服务器端

  */

@slf4j

@component

@serverendpoint ( "/websocket/{name}" )

public class websocket {

   /**

    * 与某个客户端的连接对话，需要通过它来给客户端发送消息

    */

   private session session;

   /**

    * 标识当前连接客户端的用户名

    */

   private string name;

   /**

    * 用于存所有的连接服务的客户端，这个对象存储是安全的

    * 注意这里的kv,设计的很巧妙，v刚好是本类 websocket (用来存放每个客户端对应的mywebsocket对象)

    */

   private static concurrenthashmap<string,websocket> websocketset = new concurrenthashmap<>();

   /**

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

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

    */

   @onopen

   public void onopen(session session, @pathparam (value = "name" ) string name){

     log.info( "----------------------------------" );

     this .session = session;

     this .name = name;

     // name是用来表示唯一客户端，如果需要指定发送，需要指定发送通过name来区分

     websocketset.put(name, this );

     log.info( "[websocket] 连接成功，当前连接人数为：={}" ,websocketset.size());

     log.info( "----------------------------------" );

     log.info( "" );

     groupsending(name+ " 来了" );

   }

   /**

    * 连接关闭调用的方法

    */

   @onclose

   public void onclose(){

     websocketset.remove( this .name);

     log.info( "[websocket] 退出成功，当前连接人数为：={}" ,websocketset.size());

     groupsending(name+ " 走了" );

   }

   /**

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

    */

   @onmessage

   public void onmessage(string message_str){

     log.info( "[websocket] 收到消息：{}" ,message_str);

     //判断是否需要指定发送，具体规则自定义

     //message_str的格式 touser:user2;message:aaaaaaaaaaaaaaaaaa;

     if (message_str.indexof( "touser" ) == 0 ){

       //取出 name和message的值

       string[] split = message_str.split( ";" );

       string[] split1 = split[ 0 ].split( ":" );

       string[] split2 = split[ 1 ].split( ":" );

       string name = split1[ 1 ];

       string message = split2[ 1 ];

       //指定发送

       appointsending(name,message);

     } else {

       //群发

       groupsending(message_str);

     }

   }

   /**

    * 发生错误时调用

    * @param session

    * @param error

    */

   @onerror

   public void onerror(session session, throwable error){

     log.info( "发生错误" );

     error.printstacktrace();

   }

   /**

    * 群发

    * @param message

    */

   public void groupsending(string message){

     for (string name : websocketset.keyset()){

       try {

         websocketset.get(name).session.getbasicremote().sendtext(message);

       } catch (exception e){

         e.printstacktrace();

       }

     }

   }

   /**

    * 指定发送

    * @param name

    * @param message

    */

   public void appointsending(string name,string message){

     try {

       websocketset.get(name).session.getbasicremote().sendtext(message);

     } catch (exception e){

       e.printstacktrace();

     }

   }

}