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);
                        StringBuilder sb = new StringBuilder();
                        
                        for(String n : sessionNicks.values() ) {                                
                                sb.append(n).append(" ");
                        }
                        session.getBasicRemote().sendText( "*** On-line guests: " + sb.toString() );
                        
                        
                } 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) {
                
                String nick = sessionNicks.get( session.getId() );
                
                sessions.remove(session.getId());
                sessionNicks.remove(session.getId());
                
                broadcastMessage( "*** " + nick + " has left the chat");
        }
        

        @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	StringBuilder sb = new StringBuilder();
33
34	for(String n : sessionNicks.values() ) {
35	sb.append(n).append(" ");
36	}
37	session.getBasicRemote().sendText( "*** On-line guests: " + sb.toString() );
38
39
40	} catch (IOException e) {
41	System.out.println("Error sending welcome messages " + e.getMessage() );
42	}
43
44
45	sessionNicks.put(session.getId(), guestNick);
46	sessions.put(session.getId(), session);
47
48	}
49
50	@OnError
51	public void error(Session session, Throwable error) {
52	}
53
54	@OnClose
55	public void close(Session session, CloseReason reason) {
56
57	String nick = sessionNicks.get( session.getId() );
58
59	sessions.remove(session.getId());
60	sessionNicks.remove(session.getId());
61
62	broadcastMessage( "*** " + nick + " has left the chat");
63	}
64
65
66	@OnMessage
67	public void onMessage(Session session, String msg) {
68	String parts[] = msg.split("#", 2);
69	String cmd = parts[0];
70
71
72	switch (cmd) {
73	case "CHAT":
74	String messagePart = parts[1];
75	String nick = sessionNicks.get(session.getId() );
76	broadcastMessage(nick + "> " + messagePart);
77	break;
78
79	case "NICK":
80	String oldNick = sessionNicks.get( session.getId() );
81	String newNick = parts[1];
82
83	sessionNicks.put(session.getId(), newNick);
84
85	broadcastMessage( "*** " + oldNick + " switched nick to " + newNick);
86	break;
87	}
88	}
89
90	private void broadcastMessage(String msg) {
91	try {
92	for(Session ses: sessions.values() ) {
93	ses.getBasicRemote().sendText( msg);
94	}
95	} catch (IOException e) {
96	System.out.println("Error broadcasting message " + e.getMessage() );
97	}
98	}
99
100
101	}