WebSocketController.java
3.38 KB
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
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
package net.ziemers.swxercise.ui.ws;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import javax.annotation.PostConstruct;
import javax.websocket.CloseReason;
import javax.websocket.EncodeException;
import javax.websocket.OnClose;
import javax.websocket.OnError;
import javax.websocket.OnMessage;
import javax.websocket.OnOpen;
import javax.websocket.Session;
import javax.websocket.server.PathParam;
import javax.websocket.server.ServerEndpoint;
import java.io.IOException;
import java.util.Collections;
import java.util.HashSet;
import java.util.Set;
/**
* Stub für die WebSocket-Unterstützung.
*
* Aufgepasst: Mittels CDI kann nur @ApplicationScoped injiziert werden,
* da während eines WebSocket-Callbacks kein Session-Kontext aktiv ist.
*/
@ServerEndpoint(value = WebSocketController.serverEndpointPath,
encoders = { WebSocketJson.MessageEncoder.class },
decoders = { WebSocketJson.MessageDecoder.class } )
public class WebSocketController {
static final String serverEndpointPath = "/ws/api/v1/anEndpoint/{aParameter}";
// Eine Verknüpfung zwischen der WebSocket-Session und dem SessionContext des per REST
// angemeldeten Benutzers muss irgendwie auch noch realisiert werden. :)
private static Set<Session> peers = Collections.synchronizedSet(new HashSet<>());
private Logger logger;
@PostConstruct
private void init() {
logger = LoggerFactory.getLogger(this.getClass().getSimpleName());
}
/**
* Callback-Methode für das Öffnen einer neuen WebSocket-Verbindung.
*
* @param session das {@link Session}-Objekt der neuen WebSocket-Verbindung
*/
@OnOpen
public void onOpen(Session session) {
logger.info("WebSocket opened with session id #{}", session.getId());
peers.add(session);
}
/**
* Callback-Methode, die den Empfang einer neuen WebSocket-Nachricht signalisiert.
*
* @param json der JSON-strukturierte Inhalt der WebSocket-Nachricht
* @param session das {@link Session}-Objekt der sendenden WebSocket-Verbindung
*/
@OnMessage
public void onMessage(WebSocketJson json,
Session session,
@PathParam("aParameter") String param) throws IOException, EncodeException {
logger.info("WebSocket Message '{}/{}' received by session id #{}",
json.getMessage(), param, session.getId());
try {
// Wir senden die empfangene Nachricht gleich wieder zurück. Das Marshalling geschieht automatisch.
session.getBasicRemote().sendObject(json);
} catch (IOException e) {
e.printStackTrace();
}
}
/**
* Callback-Methode, wenn in der WebSocket ein Problem auftrat.
*
* @param t die Exception
*/
@OnError
public void onError(Throwable t) {
logger.error("WebSocket Error '{}' occured!", t.getMessage());
}
/**
* Callback-Methode für das Schließen einer geöffneten WebSocket-Verbindung.
*
* @param reason die Ursache für das Schließen der WebSocket-Verbindung
* @param session das {@link Session}-Objekt der geschlossenen WebSocket-Verbindung
*/
@OnClose
public void onClose(CloseReason reason, Session session) {
logger.info("Closing WebSocket due to '{}' by session id #{}", reason.getReasonPhrase(), session.getId());
peers.remove(session);
}
}