1、MyWebSocket

package org.hxm.webSocket;import java.io.IOException;import java.util.concurrent.CopyOnWriteArraySet;import java.util.concurrent.atomic.AtomicInteger;import javax.websocket.OnClose;import javax.websocket.OnError;import javax.websocket.OnMessage;import javax.websocket.OnOpen;import javax.websocket.Session;import javax.websocket.server.ServerEndpoint;import org.springframework.stereotype.Component;@ServerEndpoint(value = "/websocket")@Componentpublic class MyWebSocket {    // 静态变量，用来记录当前在线连接数。应该把它设计成线程安全的。    private static AtomicInteger onlineCount = new AtomicInteger(0);    // concurrent包的线程安全Set，用来存放每个客户端对应的MyWebSocket对象。    private static CopyOnWriteArraySet
     
       webSocketSet = new CopyOnWriteArraySet
      
       ();    // 与某个客户端的连接会话，需要通过它来给客户端发送数据    private Session session;    /**     * 连接建立成功调用的方法     */    @OnOpen    public void onOpen(Session session) {        this.session = session;        webSocketSet.add(this); // 加入set中        onlineCount.incrementAndGet(); // 在线数加1        System.out.println("有新连接加入！当前在线人数为" + getOnlineCount());        try {            sendMessage("有新连接加入");        } catch (IOException e) {            System.out.println("IO异常");        }    }    /**     * 连接关闭调用的方法     */    @OnClose    public void onClose() {        webSocketSet.remove(this); // 从set中删除        onlineCount.decrementAndGet(); // 在线数减1        System.out.println("有一连接关闭！当前在线人数为" + getOnlineCount());    }    /**     * 收到客户端消息后调用的方法     *     * @param message     *            客户端发送过来的消息     */    @OnMessage    public void onMessage(String message, Session session) {        System.out.println("来自客户端的消息:" + message);        // 群发消息        for (MyWebSocket item : webSocketSet) {            try {                item.sendMessage(message);            } catch (IOException e) {                e.printStackTrace();            }        }    }    /**     * 发生错误时调用     */    @OnError    public void onError(Session session, Throwable error) {        System.out.println("发生错误");        error.printStackTrace();    }    public void sendMessage(String message) throws IOException {        this.session.getBasicRemote().sendText(message);        // this.session.getAsyncRemote().sendText(message);    }    /**     * 群发自定义消息     */    public static void sendInfo(String message) throws IOException {        for (MyWebSocket item : webSocketSet) {            try {                item.sendMessage(message);            } catch (IOException e) {                continue;            }        }    }    public static synchronized int getOnlineCount() {        return onlineCount.getAndIncrement();    }}
      
     

2、WebSocketConfig

package org.hxm.webSocket;import org.springframework.context.annotation.Bean;import org.springframework.context.annotation.Configuration;import org.springframework.web.socket.server.standard.ServerEndpointExporter;@Configurationpublic class WebSocketConfig {     @Bean        public ServerEndpointExporter serverEndpointExporter() {            return new ServerEndpointExporter();        }}

3、Application

package org.hxm.webSocket;import java.io.File;import java.io.FileInputStream;import java.io.FileOutputStream;import java.util.Scanner;import org.springframework.boot.SpringApplication;import org.springframework.boot.autoconfigure.SpringBootApplication;@SpringBootApplicationpublic class Application {    public static void main(String[] args) throws InterruptedException {        SpringApplication.run(Application.class, args);    }

4、Test.html

    Welcome
Send    Close