JAVA实现websocket

@ServerEndpoint注解是一个类层次的注解,它的功能主要是将目前的类定义成一个websocket服务器端,
注解的值将被用于监听用户连接的终端访问URL地址,客户端可以通过这个URL来连接到WebSocket服务器端

首先 maven中加入

        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-websocket</artifactId>
        </dependency>

WebSocket java代码

@Component
@ServerEndpoint(value = "/websocket/{username}")
public class WebSocket {
    private static final Logger log = LoggerFactory.getLogger(WebSocket.class);
    
    public static int onlineCount = 0;  
    public static Map<String, WebSocket> clients = new HashMap<String, WebSocket>();
    public Session session;  
    public String  username;  
    public String getUsername() {
        return username;
    }

    @OnOpen  
    public void onOpen(@PathParam("username") String username, Session session) throws IOException { 
        this.username = username;  
        this.session = session;  
        addOnlineCount();  
        clients.put(session.getId(), this);  
        log.warn("已连接:"+username);  
    }  
  
    @OnClose  
    public void onClose() throws IOException {  
        log.warn("断开"+username);
        clients.remove(session.getId());  
        subOnlineCount();  
    }  
  
    //收到客户端消息后调用的方法
    @OnMessage  
    public void onMessage(String message) throws IOException {   
        if(message.equals("ping")) {
            this.session.getAsyncRemote().sendText("pang");  
        }else if("close".equals(message)) {
            onClose();
            this.session.close();
        }else {
            log.warn(this.username+"收到 "+message);
        }
    }
    @OnError  
    public void onError(Session session, Throwable error) {  
        error.printStackTrace();  
    }
    
    public static void sendMessageTo(JSONObject json) {  
        log.warn("sendMsg:"+json);
        for (WebSocket item : clients.values()) {
            item.session.getAsyncRemote().sendText(json.toJSONString());  
        }
    }
    
    public static void clearMessageFromOpen(String message) {
        try {
            JSONObject json = JSONObject.parseObject(message);
            if(json.get("uuid") != null) {
            }
        } catch (Exception e) {
        }
    }
    
  
    public static synchronized int getOnlineCount() {  
        return onlineCount;  
    }  
  
    public static synchronized void addOnlineCount() {  
        WebSocket.onlineCount++;  
    }  
  
    public static synchronized void subOnlineCount() {  
        WebSocket.onlineCount--;  
    }  
  
    public static synchronized Map<String, WebSocket> getClients() {  
        return clients;  
    } 
    
}

在RunApplication中加入

   /**
     * 会自动注册使用了@ServerEndpoint注解声明的Websocket endpoint
     * 要注意,如果使用独立的servlet容器,
     * 而不是直接使用springboot的内置容器,
     * 就不要注入ServerEndpointExporter,因为它将由容器自己提供和管理。
     */
    @Bean
    public ServerEndpointExporter serverEndpointExporter() {
      return new ServerEndpointExporter();
    }

  • 0
    点赞
  • 16
    收藏
    觉得还不错? 一键收藏
  • 7
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 7
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值