/*
 * Copyright 2012  Research Studios Austria Forschungsges.m.b.H.
 *
 *    Licensed under the Apache License, Version 2.0 (the "License");
 *    you may not use this file except in compliance with the License.
 *    You may obtain a copy of the License at
 *
 *        http://www.apache.org/licenses/LICENSE-2.0
 *
 *    Unless required by applicable law or agreed to in writing, software
 *    distributed under the License is distributed on an "AS IS" BASIS,
 *    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *    See the License for the specific language governing permissions and
 *    limitations under the License.
 */

package won.owner.web.websocket;


import org.springframework.web.socket.WebSocketSession;
import won.owner.model.User;

import java.net.URI;
import java.util.Set;

public class WebSocketSessionService
{
  private WebSocketSessionMapping<Long> userIdToSession = new WebSocketSessionMapping<Long>();
  private WebSocketSessionMapping<URI> needUriToSession = new WebSocketSessionMapping<URI>();

  public void addMapping(User user, WebSocketSession session) {
    this.userIdToSession.addMapping(user.getId(), session);
  }

  public void addMapping(URI needUri, WebSocketSession session) {
    this.needUriToSession.addMapping(needUri, session);
  }

  public void removeMapping(User user, WebSocketSession session) {
    this.userIdToSession.removeMapping(user.getId(), session);
  }

  public void removeMapping(URI needUri, WebSocketSession session) {
    this.needUriToSession.removeMapping(needUri, session);
  }

  public Set<WebSocketSession> getWebSocketSessions(User user) {
    return this.userIdToSession.getWebSocketSessions(user.getId());
  }

  public Set<WebSocketSession> getWebSocketSessions(URI needUri) {
    return this.needUriToSession.getWebSocketSessions(needUri);
  }



}