public class NewMsgCollector extends ConnectListener implements ServletContextListener {

private static final String CHANNEL = "hello";

public void contextInitialized(ServletContextEvent contextEvent) {

//注册应用的channel

CometContext context = CometContext.getInstance();

context.registChannel(CHANNEL);

//添加监听器

CometEngine engine = CometContext.getInstance().getEngine();

engine.addConnectListener(this);

}

public void contextDestroyed(ServletContextEvent contextEvent) {}

public boolean handleEvent(ConnectEvent connEvent) {

final CometConnection conn = connEvent.getConn();

//建立连接和用户的关系

doCache(conn);

final String connId = conn.getId();

/*模拟业务逻辑*/

Timer timer = new Timer(true);

TimerTask task = new TimerTask() {

public void run() {

CometEngine engine = CometContext.getInstance().getEngine();

//推送到所有客户端

//engine.sendToAll("hello", connId + " - you have " + ((int)(Math.random() * 9) + 1) + " new message 
");

if (CacheManager.getContent(connId).isExpired()) {

doCache(conn);

}

if (simulateService(String.valueOf(CacheManager.getContent(connId).getValue()))) {

//推送到指定的客户端

engine.sendTo(CHANNEL, engine.getConnection(connId), CacheManager.getContent(connId).getValue()

+ " - you have " + ((int) (Math.random() * 9) + 1) + " new message 
");

}

}

};

timer.schedule(task, 10000, (1000 * 5));

return true;

}

private void doCache(final CometConnection conn) {

Object userId = conn.getRequest().getSession().getAttribute("currentUserId");

if (userId != null) {

CacheManager.putContent(conn.getId(), String.valueOf(userId), CacheConstant.EXPIRE_AFTER_ONE_HOUR);

}

}

/**

* 模拟业务

* 返回true,false

* true即表示需要推送消息,false即不需要推送

*/

private boolean simulateService(String id) {

int flag = (int) Math.round(Math.random());

if (flag == 0) {

System.out.println(id + " - no messge...");

return false;

}

System.out.println(id + " - messge is coming...");

return true;

}

}