thoerup/websocket/AnnotatedSocket.java

package dk.thoerup.websocket;

import java.io.IOException;
import java.util.HashMap;
import java.util.Random;

import javax.websocket.CloseReason;
import javax.websocket.EndpointConfig;
import javax.websocket.OnClose;
import javax.websocket.OnError;
import javax.websocket.OnMessage;
import javax.websocket.OnOpen;
import javax.websocket.Session;
import javax.websocket.server.ServerEndpoint;


@ServerEndpoint("/wstest")
public class AnnotatedSocket {
        static private HashMap<String, Session> sessions = new HashMap<String,Session>();
        static private HashMap<String, String> sessionNicks = new HashMap<String,String>();
        
        private static Random rand = new Random();
        
        
        @OnOpen
        public void open(Session session, EndpointConfig conf) {
                String guestNick = "Guest-" + rand.nextInt();
                
                broadcastMessage(guestNick + " has joined the chat");
                try {
                        session.getBasicRemote().sendText( "Welcome " + guestNick);
                } catch (IOException e) {
                        System.out.println("Error sending welcome messages " + e.getMessage() );
                }
                
                
                sessionNicks.put(session.getId(), guestNick);
                sessions.put(session.getId(), session);         
                 
        }
        
        @OnError
        public void error(Session session, Throwable error) {           
        }
        
        @OnClose
        public void close(Session session, CloseReason reason) {
                sessions.remove(session.getId());
                sessionNicks.remove(session.getId());
        }
        

        @OnMessage
        public void onMessage(Session session, String msg) {
                String parts[] = msg.split("#", 2);
                String cmd = parts[0];
                
                
                switch (cmd) {
                case "CHAT":
                        String messagePart = parts[1];
                        String nick = sessionNicks.get(session.getId() );
                        broadcastMessage(nick + "> " + messagePart);
                        break;

                case "NICK":
                        String oldNick = sessionNicks.get( session.getId() );
                        String newNick = parts[1];
                        
                        sessionNicks.put(session.getId(), newNick);
                        
                        broadcastMessage( "*** " + oldNick + " switched nick to " + newNick);
                        break;
                }
        }
        
        private void broadcastMessage(String msg) {
                try {
                        for(Session ses: sessions.values() ) {
                                ses.getBasicRemote().sendText( msg);                            
                        }
                } catch (IOException e) {
                        System.out.println("Error broadcasting message " + e.getMessage() );
                }               
        }
        
        
}
1	package dk.thoerup.websocket;
2
3	import java.io.IOException;
4	import java.util.HashMap;
5	import java.util.Random;
6
7	import javax.websocket.CloseReason;
8	import javax.websocket.EndpointConfig;
9	import javax.websocket.OnClose;
10	import javax.websocket.OnError;
11	import javax.websocket.OnMessage;
12	import javax.websocket.OnOpen;
13	import javax.websocket.Session;
14	import javax.websocket.server.ServerEndpoint;
15
16
17	@ServerEndpoint("/wstest")
18	public class AnnotatedSocket {
19	static private HashMap<String, Session> sessions = new HashMap<String,Session>();
20	static private HashMap<String, String> sessionNicks = new HashMap<String,String>();
21
22	private static Random rand = new Random();
23
24
25	@OnOpen
26	public void open(Session session, EndpointConfig conf) {
27	String guestNick = "Guest-" + rand.nextInt();
28
29	broadcastMessage(guestNick + " has joined the chat");
30	try {
31	session.getBasicRemote().sendText( "Welcome " + guestNick);
32	} catch (IOException e) {
33	System.out.println("Error sending welcome messages " + e.getMessage() );
34	}
35
36
37	sessionNicks.put(session.getId(), guestNick);
38	sessions.put(session.getId(), session);
39
40	}
41
42	@OnError
43	public void error(Session session, Throwable error) {
44	}
45
46	@OnClose
47	public void close(Session session, CloseReason reason) {
48	sessions.remove(session.getId());
49	sessionNicks.remove(session.getId());
50	}
51
52
53	@OnMessage
54	public void onMessage(Session session, String msg) {
55	String parts[] = msg.split("#", 2);
56	String cmd = parts[0];
57
58
59	switch (cmd) {
60	case "CHAT":
61	String messagePart = parts[1];
62	String nick = sessionNicks.get(session.getId() );
63	broadcastMessage(nick + "> " + messagePart);
64	break;
65
66	case "NICK":
67	String oldNick = sessionNicks.get( session.getId() );
68	String newNick = parts[1];
69
70	sessionNicks.put(session.getId(), newNick);
71
72	broadcastMessage( "*** " + oldNick + " switched nick to " + newNick);
73	break;
74	}
75	}
76
77	private void broadcastMessage(String msg) {
78	try {
79	for(Session ses: sessions.values() ) {
80	ses.getBasicRemote().sendText( msg);
81	}
82	} catch (IOException e) {
83	System.out.println("Error broadcasting message " + e.getMessage() );
84	}
85	}
86
87
88	}