init submit

y_ys79

2024-01-05 e5840972b6b17ec03bfc1069a9cacfe0cef189c6

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
package com.qianwen.mdc.service; 
 
import org.slf4j.Logger; 
import org.slf4j.LoggerFactory; 
import org.springframework.stereotype.Component; 
 
import javax.websocket.*; 
import javax.websocket.server.ServerEndpoint; 
import java.io.IOException; 
import java.util.concurrent.CopyOnWriteArraySet; 
 
/** 
 * @author: tianmingqiang 
 * @date: 2019/11/15 11:54 
 * @description: MdcWebSocketService 
 */ 
@ServerEndpoint("/websocket") 
@Component 
public class MdcWebSocketService { 
    @OnOpen 
    public void onOpen(Session session) { 
        this.session = session; 
        webSocketSet.add(this); 
        logger.debug("New connection"); 
    } 
 
    @OnClose 
    public void onClose() { 
        webSocketSet.remove(this); 
        logger.debug("Connection closed"); 
    } 
 
    @OnMessage 
    public void onMessage(String message, Session session) { 
        logger.debug("Message from client: {}", message); 
        for (MdcWebSocketService item : webSocketSet) { 
            try { 
                if (item.session == session) { 
                    continue; 
                } 
                item.sendMessage(message); 
            } catch (IOException e) { 
                e.printStackTrace(); 
                continue; 
            } 
        } 
    } 
 
    @OnError 
    public void onError(Session session, Throwable error) { 
        logger.debug("Error occur"); 
        error.printStackTrace(); 
    } 
 
    public void sendMessage(String message) throws IOException { 
//        this.session.getBasicRemote().sendText(message); 
 
        synchronized (session) { 
            session.getBasicRemote().sendText(message); 
        } 
    } 
 
    public static void sendAllMessage(String message) { 
        for (MdcWebSocketService item : webSocketSet) { 
            try { 
                item.sendMessage(message); 
            } catch (IOException e) { 
                e.printStackTrace(); 
                continue; 
            } 
        } 
    } 
 
    public static CopyOnWriteArraySet<MdcWebSocketService> getWebSocketSet() { 
        return webSocketSet; 
    } 
 
    private static CopyOnWriteArraySet<MdcWebSocketService> webSocketSet = new CopyOnWriteArraySet<>(); 
    private Session session; 
 
    private static final Logger logger = LoggerFactory.getLogger(MdcWebSocketService.class); 
}