src/main/java/me/imsonmia/epqapi/messaging/WebSocketHandler.java


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

package me.imsonmia.epqapi.messaging;

import java.io.IOException;
import java.util.HashMap;
import java.util.Set;
import java.util.concurrent.CopyOnWriteArraySet;

import org.slf4j.Logger;

import jakarta.websocket.EncodeException;
import jakarta.websocket.OnClose;
import jakarta.websocket.OnMessage;
import jakarta.websocket.OnOpen;
import jakarta.websocket.Session;
import jakarta.websocket.server.ServerEndpoint;
import me.imsonmia.epqapi.model.ChatConvert;
import me.imsonmia.epqapi.model.ChatMessage;
import me.imsonmia.epqapi.repository.ChatMessageRepository;

// https://www.baeldung.com/java-websockets
@ServerEndpoint(
    value = "/chat/{username}",
    encoders = MessageEncoder.class,
    decoders = MessageDecoder.class
    )
public class WebSocketHandler {
    private Logger logger;
    private ChatMessageRepository chatMessageRepository;
    private Session session;
    private Set<WebSocketHandler> chatEndpoints = new CopyOnWriteArraySet<>();
    private HashMap<String, String> users = new HashMap<>();
    @OnOpen
    public void onOpen(Session session, String username) throws IOException {
        this.session = session;
        chatEndpoints.add(this);
        users.put(session.getId(), username);
        Message message = new Message();
        message.setFrom(username);
        message.setContent("Connected, hello world!");
        broadcast(message);
        ChatMessage cmessage = new ChatConvert().fromMessage(message);
        chatMessageRepository.save(cmessage);
    }
    @OnMessage
    public void onMessage(Session session, Message message) throws IOException {
        message.setFrom(users.get(session.getId()));
        broadcast(message);
    }
    @OnClose
    public void onClose(Session session) throws IOException {
        chatEndpoints.remove(this);
        Message message = new Message();
        message.setFrom(users.get(session.getId()));
        message.setContent("Disconnected!");
        broadcast(message);
    }
    public void broadcast(Message message) {
        chatEndpoints.forEach(endpoint -> {
            try {
                endpoint.session.getBasicRemote().sendObject(message);
            }
            catch (IOException e) {
                logger.info("Error sending message! IOException");
                logger.error(null, e);
            }
            catch (EncodeException e) {
                logger.info("Error sending message! EncodeException");
            }  
        });
    }
}