mirror of https://github.com/OpenVidu/openvidu.git
Introduced SessionStorage which is an extract of the SessionManager
Thomas Joußen
parent
6effca3389
commit
6a1a8ea172
|
|
@ -0,0 +1,307 @@
|
||||||
|
package io.openvidu.server.core;
|
||||||
|
|
||||||
|
import io.openvidu.client.OpenViduException;
|
||||||
|
import io.openvidu.client.internal.ProtocolElements;
|
||||||
|
import io.openvidu.java.client.SessionProperties;
|
||||||
|
import io.openvidu.server.OpenViduServer;
|
||||||
|
import io.openvidu.server.coturn.CoturnCredentialsService;
|
||||||
|
import io.openvidu.server.coturn.TurnCredentials;
|
||||||
|
import org.slf4j.Logger;
|
||||||
|
import org.slf4j.LoggerFactory;
|
||||||
|
import org.springframework.beans.factory.annotation.Autowired;
|
||||||
|
|
||||||
|
import java.util.Collection;
|
||||||
|
import java.util.HashSet;
|
||||||
|
import java.util.Set;
|
||||||
|
import java.util.concurrent.ConcurrentHashMap;
|
||||||
|
import java.util.concurrent.ConcurrentMap;
|
||||||
|
|
||||||
|
public class SessionStorage {
|
||||||
|
|
||||||
|
private static final Logger log = LoggerFactory.getLogger(SessionStorage.class);
|
||||||
|
|
||||||
|
@Autowired
|
||||||
|
private Utils utils;
|
||||||
|
|
||||||
|
@Autowired
|
||||||
|
private CoturnCredentialsService coturnCredentialsService;
|
||||||
|
|
||||||
|
protected ConcurrentMap<String, Session> sessions = new ConcurrentHashMap<>();
|
||||||
|
protected ConcurrentMap<String, SessionProperties> sessionProperties = new ConcurrentHashMap<>();
|
||||||
|
protected ConcurrentMap<String, ConcurrentHashMap<String, Participant>> sessionidParticipantpublicidParticipant = new ConcurrentHashMap<>();
|
||||||
|
protected ConcurrentMap<String, Boolean> insecureUsers = new ConcurrentHashMap<>();
|
||||||
|
protected ConcurrentMap<String, ConcurrentHashMap<String, Token>> sessionidTokenTokenobj = new ConcurrentHashMap<>();
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns a Session given its id
|
||||||
|
*
|
||||||
|
* @return Session
|
||||||
|
*/
|
||||||
|
public Session getSession(String sessionId) {
|
||||||
|
return sessions.get(sessionId);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns all currently active (opened) sessions.
|
||||||
|
*
|
||||||
|
* @return set of the session's identifiers
|
||||||
|
*/
|
||||||
|
public Set<String> getSessions() {
|
||||||
|
return new HashSet<String>(sessions.keySet());
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns all currently active (opened) sessions.
|
||||||
|
*
|
||||||
|
* @return set of the session's identifiers
|
||||||
|
*/
|
||||||
|
public Collection<Session> getSessionObjects() {
|
||||||
|
return sessions.values();
|
||||||
|
}
|
||||||
|
|
||||||
|
public Session putSessionIfAbsent(String sessionId, Session session) { return sessions.putIfAbsent(sessionId, session); }
|
||||||
|
|
||||||
|
public SessionProperties getSessionProperties(String sessionId) { return this.sessionProperties.get(sessionId); }
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns all the participants inside a session.
|
||||||
|
*
|
||||||
|
* @param sessionId
|
||||||
|
* identifier of the session
|
||||||
|
* @return set of {@link Participant}
|
||||||
|
* @throws OpenViduException
|
||||||
|
* in case the session doesn't exist
|
||||||
|
*/
|
||||||
|
public Set<Participant> getParticipants(String sessionId) throws OpenViduException {
|
||||||
|
Session session = sessions.get(sessionId);
|
||||||
|
if (session == null) {
|
||||||
|
throw new OpenViduException(OpenViduException.Code.ROOM_NOT_FOUND_ERROR_CODE, "Session '" + sessionId + "' not found");
|
||||||
|
}
|
||||||
|
Set<Participant> participants = session.getParticipants();
|
||||||
|
participants.removeIf(p -> p.isClosed());
|
||||||
|
return participants;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns a participant in a session
|
||||||
|
*
|
||||||
|
* @param sessionId
|
||||||
|
* identifier of the session
|
||||||
|
* @param participantPrivateId
|
||||||
|
* private identifier of the participant
|
||||||
|
* @return {@link Participant}
|
||||||
|
* @throws OpenViduException
|
||||||
|
* in case the session doesn't exist or the participant doesn't
|
||||||
|
* belong to it
|
||||||
|
*/
|
||||||
|
public Participant getParticipant(String sessionId, String participantPrivateId) throws OpenViduException {
|
||||||
|
Session session = sessions.get(sessionId);
|
||||||
|
if (session == null) {
|
||||||
|
throw new OpenViduException(OpenViduException.Code.ROOM_NOT_FOUND_ERROR_CODE, "Session '" + sessionId + "' not found");
|
||||||
|
}
|
||||||
|
Participant participant = session.getParticipantByPrivateId(participantPrivateId);
|
||||||
|
if (participant == null) {
|
||||||
|
throw new OpenViduException(OpenViduException.Code.USER_NOT_FOUND_ERROR_CODE,
|
||||||
|
"Participant '" + participantPrivateId + "' not found in session '" + sessionId + "'");
|
||||||
|
}
|
||||||
|
return participant;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns a participant
|
||||||
|
*
|
||||||
|
* @param participantPrivateId
|
||||||
|
* private identifier of the participant
|
||||||
|
* @return {@link Participant}
|
||||||
|
* @throws OpenViduException
|
||||||
|
* in case the participant doesn't exist
|
||||||
|
*/
|
||||||
|
public Participant getParticipant(String participantPrivateId) throws OpenViduException {
|
||||||
|
for (Session session : sessions.values()) {
|
||||||
|
if (!session.isClosed()) {
|
||||||
|
if (session.getParticipantByPrivateId(participantPrivateId) != null) {
|
||||||
|
return session.getParticipantByPrivateId(participantPrivateId);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
throw new OpenViduException(OpenViduException.Code.USER_NOT_FOUND_ERROR_CODE,
|
||||||
|
"No participant with private id '" + participantPrivateId + "' was found");
|
||||||
|
}
|
||||||
|
|
||||||
|
public void storeSessionId(String sessionId, SessionProperties sessionProperties) {
|
||||||
|
this.sessionidParticipantpublicidParticipant.putIfAbsent(sessionId, new ConcurrentHashMap<>());
|
||||||
|
this.sessionProperties.putIfAbsent(sessionId, sessionProperties);
|
||||||
|
showTokens();
|
||||||
|
}
|
||||||
|
|
||||||
|
public ConcurrentHashMap<String, Token> putTokenObject(String sessionId, ConcurrentHashMap<String, Token> map) {
|
||||||
|
return this.sessionidTokenTokenobj.putIfAbsent(sessionId, map);
|
||||||
|
}
|
||||||
|
|
||||||
|
public void removeTokenObject(String sessionId) {
|
||||||
|
this.sessionidTokenTokenobj.remove(sessionId);
|
||||||
|
}
|
||||||
|
|
||||||
|
public ConcurrentHashMap<String, Token> getTokenObject(String sessionId) {
|
||||||
|
return this.sessionidTokenTokenobj.get(sessionId);
|
||||||
|
}
|
||||||
|
|
||||||
|
public String newToken(String sessionId, ParticipantRole role, String serverMetadata) throws OpenViduException {
|
||||||
|
ConcurrentHashMap<String, Token> map = this.putTokenObject(sessionId, new ConcurrentHashMap<>());
|
||||||
|
|
||||||
|
if (map != null) {
|
||||||
|
|
||||||
|
if (!utils.isMetadataFormatCorrect(serverMetadata)) {
|
||||||
|
log.error("Data invalid format. Max length allowed is 10000 chars");
|
||||||
|
throw new OpenViduException(OpenViduException.Code.GENERIC_ERROR_CODE,
|
||||||
|
"Data invalid format. Max length allowed is 10000 chars");
|
||||||
|
}
|
||||||
|
|
||||||
|
String token = OpenViduServer.publicUrl;
|
||||||
|
token += "?sessionId=" + sessionId;
|
||||||
|
token += "&token=" + utils.generateRandomChain();
|
||||||
|
token += "&role=" + role.name();
|
||||||
|
TurnCredentials turnCredentials = null;
|
||||||
|
if (this.coturnCredentialsService.isCoturnAvailable()) {
|
||||||
|
turnCredentials = coturnCredentialsService.createUser();
|
||||||
|
if (turnCredentials != null) {
|
||||||
|
token += "&turnUsername=" + turnCredentials.getUsername();
|
||||||
|
token += "&turnCredential=" + turnCredentials.getCredential();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Token t = new Token(token, role, serverMetadata, turnCredentials);
|
||||||
|
|
||||||
|
map.putIfAbsent(token, t);
|
||||||
|
this.showTokens();
|
||||||
|
return token;
|
||||||
|
|
||||||
|
} else {
|
||||||
|
this.removeTokenObject(sessionId);
|
||||||
|
log.error("sessionId [" + sessionId + "] is not valid");
|
||||||
|
throw new OpenViduException(OpenViduException.Code.ROOM_NOT_FOUND_ERROR_CODE, "sessionId [" + sessionId + "] not found");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public boolean isTokenValidInSession(String token, String sessionId, String participanPrivatetId) {
|
||||||
|
if (!this.isInsecureParticipant(participanPrivatetId)) {
|
||||||
|
if (this.sessionidTokenTokenobj.get(sessionId) != null) {
|
||||||
|
return this.sessionidTokenTokenobj.get(sessionId).containsKey(token);
|
||||||
|
} else {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
this.sessionidParticipantpublicidParticipant.putIfAbsent(sessionId, new ConcurrentHashMap<>());
|
||||||
|
this.sessionidTokenTokenobj.putIfAbsent(sessionId, new ConcurrentHashMap<>());
|
||||||
|
this.sessionidTokenTokenobj.get(sessionId).putIfAbsent(token,
|
||||||
|
new Token(token, ParticipantRole.PUBLISHER, "",
|
||||||
|
this.coturnCredentialsService.isCoturnAvailable()
|
||||||
|
? this.coturnCredentialsService.createUser()
|
||||||
|
: null));
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public boolean isParticipantInSession(String sessionId, Participant participant) {
|
||||||
|
Session session = this.sessions.get(sessionId);
|
||||||
|
if (session != null) {
|
||||||
|
return (session.getParticipantByPrivateId(participant.getParticipantPrivateId()) != null);
|
||||||
|
} else {
|
||||||
|
throw new OpenViduException(OpenViduException.Code.ROOM_NOT_FOUND_ERROR_CODE, "[" + sessionId + "] is not a valid sessionId");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public boolean isPublisherInSession(String sessionId, Participant participant) {
|
||||||
|
if (!this.isInsecureParticipant(participant.getParticipantPrivateId())) {
|
||||||
|
if (this.sessionidParticipantpublicidParticipant.get(sessionId) != null) {
|
||||||
|
return (ParticipantRole.PUBLISHER.equals(participant.getToken().getRole())
|
||||||
|
|| ParticipantRole.MODERATOR.equals(participant.getToken().getRole()));
|
||||||
|
} else {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public boolean isInsecureParticipant(String participantPrivateId) {
|
||||||
|
if (this.insecureUsers.containsKey(participantPrivateId)) {
|
||||||
|
log.info("The user with private id {} is an INSECURE user", participantPrivateId);
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void newInsecureParticipant(String participantPrivateId) {
|
||||||
|
this.insecureUsers.put(participantPrivateId, true);
|
||||||
|
}
|
||||||
|
|
||||||
|
public void removeInsecureParticipant(String participantPrivateId) {
|
||||||
|
this.insecureUsers.remove(participantPrivateId);
|
||||||
|
}
|
||||||
|
|
||||||
|
public ConcurrentHashMap<String, Participant> getPublicIdParticipantMap(String sessionId) {
|
||||||
|
return this.sessionidParticipantpublicidParticipant.get(sessionId);
|
||||||
|
}
|
||||||
|
|
||||||
|
public Participant newParticipant(String sessionId, String participantPrivatetId, Token token,
|
||||||
|
String clientMetadata) {
|
||||||
|
if (this.sessionidParticipantpublicidParticipant.get(sessionId) != null) {
|
||||||
|
String participantPublicId = utils.generateRandomChain();
|
||||||
|
Participant p = new Participant(participantPrivatetId, participantPublicId, token, clientMetadata);
|
||||||
|
while (this.sessionidParticipantpublicidParticipant.get(sessionId).putIfAbsent(participantPublicId,
|
||||||
|
p) != null) {
|
||||||
|
participantPublicId = utils.generateRandomChain();
|
||||||
|
p.setParticipantPublicId(participantPublicId);
|
||||||
|
}
|
||||||
|
return p;
|
||||||
|
} else {
|
||||||
|
throw new OpenViduException(OpenViduException.Code.ROOM_NOT_FOUND_ERROR_CODE, sessionId);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public Participant newRecorderParticipant(String sessionId, String participantPrivatetId, Token token,
|
||||||
|
String clientMetadata) {
|
||||||
|
if (this.sessionidParticipantpublicidParticipant.get(sessionId) != null) {
|
||||||
|
Participant p = new Participant(participantPrivatetId, ProtocolElements.RECORDER_PARTICIPANT_PUBLICID,
|
||||||
|
token, clientMetadata);
|
||||||
|
this.sessionidParticipantpublicidParticipant.get(sessionId)
|
||||||
|
.put(ProtocolElements.RECORDER_PARTICIPANT_PUBLICID, p);
|
||||||
|
return p;
|
||||||
|
} else {
|
||||||
|
throw new OpenViduException(OpenViduException.Code.ROOM_NOT_FOUND_ERROR_CODE, sessionId);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public Token consumeToken(String sessionId, String participantPrivateId, String token) {
|
||||||
|
if (this.sessionidTokenTokenobj.get(sessionId) != null) {
|
||||||
|
Token t = this.sessionidTokenTokenobj.get(sessionId).remove(token);
|
||||||
|
if (t != null) {
|
||||||
|
return t;
|
||||||
|
} else {
|
||||||
|
throw new OpenViduException(OpenViduException.Code.TOKEN_CANNOT_BE_CREATED_ERROR_CODE, sessionId);
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
throw new OpenViduException(OpenViduException.Code.ROOM_NOT_FOUND_ERROR_CODE, sessionId);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public void emptyCollections(Session session) {
|
||||||
|
sessions.remove(session.getSessionId());
|
||||||
|
|
||||||
|
sessionProperties.remove(session.getSessionId());
|
||||||
|
sessionidParticipantpublicidParticipant.remove(session.getSessionId());
|
||||||
|
sessionidTokenTokenobj.remove(session.getSessionId());
|
||||||
|
|
||||||
|
log.warn("Session '{}' removed and closed", session.getSessionId());
|
||||||
|
}
|
||||||
|
|
||||||
|
public void showTokens() { log.info("<SESSIONID, TOKENS>: {}", this.sessionidTokenTokenobj.toString()); }
|
||||||
|
|
||||||
|
public void showInsecureParticipants() {
|
||||||
|
log.info("<INSECURE_PARTICIPANTS>: {}", this.insecureUsers.toString());
|
||||||
|
}
|
||||||
|
|
||||||
|
public void showAllParticipants() {
|
||||||
|
log.info("<SESSIONID, PARTICIPANTS>: {}", this.sessionidParticipantpublicidParticipant.toString());
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue