StateSessionMap.java

package de.vipra.ws;

import java.util.HashMap;
import java.util.stream.Stream;

import javax.websocket.Session;

public class StateSessionMap extends HashMap<StateSession, StateSession> {

	private static final long serialVersionUID = 1L;

	public Stream<Session> stream(final State state) {
		return entrySet().stream().filter(s -> s.getKey().getState() == state).map(s -> s.getKey().getSession());
	}

	public void add(final StateSession session) {
		put(session, session);
	}

}