package com.megatim.fdxconsultation.core.impl.sockets;
|
|
import com.megatim.fdxconsultation.model.dataproduction.DataProduction;
|
import java.io.IOException;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.concurrent.ConcurrentHashMap;
|
import java.util.logging.Level;
|
import java.util.logging.Logger;
|
import java.util.stream.Collectors;
|
import javax.enterprise.context.ApplicationScoped;
|
import javax.websocket.EncodeException;
|
import javax.websocket.Session;
|
|
/**
|
*
|
* @author Gabuntu
|
*/
|
@ApplicationScoped
|
public class DataProductionSessionHandler {
|
|
private final Map<String, Session> idToSessions = new ConcurrentHashMap<>();
|
|
public void publishMessage(DataProduction dataProduction) {
|
getSessions().forEach(s -> {
|
try {
|
s.getBasicRemote().sendObject(dataProduction);
|
} catch (IOException | EncodeException ex) {
|
Logger.getLogger(getClass().getName()).log(Level.SEVERE, ex.getMessage(), ex);
|
}
|
});
|
}
|
|
public void addSession(Session session) {
|
idToSessions.put(session.getId(), session);
|
}
|
|
public void removeSession(Session session) {
|
idToSessions.remove(session.getId());
|
}
|
|
private List<Session> getSessions() {
|
return idToSessions
|
.entrySet()
|
.stream()
|
.map(e -> e.getValue())
|
.collect(Collectors.toList());
|
}
|
}
|
