在线画报socket.io服务器端示范

WhiteboardServer.java

/**
 * 在线画报socket.io服务器端示范
 * 
 * @author yongboy
 * @time 2012-3-27
 * @version 1.0
 */
public class WhiteboardServer {
	public static void main(String[] args) {
		MainServer chatServer = new MainServer(new WhiteboardHandler(), 80);
		chatServer.start();
	}
}

class WhiteboardHandler extends IOHandlerAbs {
	private Logger log = Logger.getLogger(this.getClass());
	// 房间的一对多<房间号,List<客户端>>
	private ConcurrentMap<String, Set<IOClient>> roomClients = new ConcurrentHashMap<String, Set<IOClient>>();

	@Override
	public void OnConnect(IOClient client) {
		log.debug("A user connected :: " + client.getSessionID());

		String content = String.format("5:::{\"name\":\"%s\",\"args\":[%s]}",
				"clientId",
				String.format("{\"id\":\"%s\"}", client.getSessionID()));
		client.send(content);
	}

	@Override
	public void OnDisconnect(IOClient client) {
		log.info("A user disconnected :: " + client.getSessionID()
				+ " :: hope it was fun");

		GenericIO genericIO = (GenericIO) client;
		Object roomObj = genericIO.attr.get("room");

		if (roomObj == null) {
			log.info("the roomObj is null!");
			return;
		}

		String roomId = roomObj.toString();

		Set<IOClient> clients = roomClients.get(roomId);
		log.info("clients size is " + clients.size());
		clients.remove(client);
		log.info("removed clients's size is " + clients.size());

		// 通知其它客户端，有人离线
		broadcastRoom(roomId, client, "roomCount", String.format(
				"{\"room\":\"%s\",\"num\":%s}", roomId, clients.size()));
	}

	@Override
	public void OnMessage(IOClient client, String oriMessage) {
		log.debug("Got a message :: " + oriMessage
				+ " :: echoing it back to :: " + client.getSessionID());
		String jsonString = oriMessage.substring(oriMessage.indexOf('{'));
		JSONObject jsonObject = JSON.parseObject(jsonString);
		String eventName = jsonObject.get("name").toString();
		JSONArray argsArray = jsonObject.getJSONArray("args");
		JSONObject obj = argsArray.getJSONObject(0);
		String roomId = obj.getString("room");

		if (eventName.equals("roomNotice")) {
			if (!roomClients.containsKey(roomId)) {
				roomClients.put(roomId, new HashSet<IOClient>());
			}

			GenericIO genericIO = (GenericIO) client;
			genericIO.attr.put("room", roomId);
			roomClients.get(roomId).add(client);

			int clientNums = roomClients.get(roomId).size();
			broadcastRoom(roomId, client, "roomCount", String.format(
					"{\"room\":\"%s\",\"num\":%s}", roomId, clientNums));

			String content = String.format(
					"5:::{\"name\":\"%s\",\"args\":[%s]}", "roomCount", String
							.format("{\"room\":\"%s\",\"num\":%s}", roomId,
									clientNums));
			client.send(content);

			return;
		}

		broadcastRoom(roomId, client, eventName, obj.toJSONString());
	}

	private void broadcastRoom(String roomId, IOClient client,
			String eventName, String jsonString) {
		Set<IOClient> clients = roomClients.get(roomId);
		if (clients == null || clients.isEmpty())
			return;

		String content = String.format("5:::{\"name\":\"%s\",\"args\":[%s]}",
				eventName, jsonString);
		for (IOClient rc : clients) {
			if (rc == null || rc.getSessionID().equals(client.getSessionID())) {
				continue;
			}

			rc.send(content);
		}
	}

	@Override
	public void OnShutdown() {
		log.debug("shutdown now ~~~");
	}
}