Posted on 2008-11-26 11:36
dennis 閱讀(2208)
評論(8) 編輯 收藏 所屬分類:
java
yanf4j簡介,請看
這里。
這里俺要介紹下用yanf4j寫個最簡單的聊天室,可以直接telnet上去聊天。請先下載
common-logging和
yanf4j-0.40-stable-bin.tar.gz。
聊天室嘛,肯定要維護一個聊天者聊表:
List<Session> sessionList = new CopyOnWriteArrayList<Session>();
用戶名讓系統產生,用戶1、2、3.....,最簡單的嘛:
AtomicInteger userId = new AtomicInteger();
private String generateUserName() {
return "user" + userId.incrementAndGet();
}
服務端要實現一個處理handler,來接受用戶連接并轉發聊天消息,嗯,繼承HandlerAdapter,覆寫相應的方法即可。當有人連上來的時候,覆寫onSessionCreated方法:
public void onSessionCreated(Session session) {
sessionList.add(session);
// 給session取名
session.attach(generateUserName());
// 廣播某人進來
broadcast(session, session.attachment() + " enter room");
}
當收到聊天消息的時候,廣播給組內成員:
@Override
public void onReceive(Session session, String t) {
if (t.equals("quit"))
session.close();
else
broadcast(session, "[" + session.attachment() + "] say:" + t);
}
private void broadcast(Session s, String msg) {
for (Session session : this.sessionList) {
session.send(msg);
}
}
廣播我們只是簡單的遍歷sessionList并send而已。完整的ChatServerHandler.java清單如下:
package com.google.code.yanf4j.test.example.chat;
import java.util.List;
import java.util.concurrent.CopyOnWriteArrayList;
import java.util.concurrent.atomic.AtomicInteger;
import com.google.code.yanf4j.nio.Session;
import com.google.code.yanf4j.nio.impl.HandlerAdapter;
public class ChatServerHandler extends HandlerAdapter<String> {
List<Session> sessionList = new CopyOnWriteArrayList<Session>();
AtomicInteger userId = new AtomicInteger();
@Override
public void onReceive(Session session, String t) {
if (t.equals("quit"))
session.close();
else
broadcast(session, "[" + session.attachment() + "] say:" + t);
}
@Override
public void onSessionClosed(Session session) {
sessionList.remove(session);
broadcast(session, session.attachment() + " leave room");
}
private void broadcast(Session s, String msg) {
for (Session session : this.sessionList) {
session.send(msg);
}
}
@Override
public void onSessionCreated(Session session) {
sessionList.add(session);
// 給session取名
session.attach(generateUserName());
// 廣播某人進來
broadcast(session, session.attachment() + " enter room");
}
private String generateUserName() {
return "user" + userId.incrementAndGet();
}
}
接下來我們要實現聊天服務器了,yanf4j實現TCP Server的核心類是TCPController,假設端口設為8080:
package com.google.code.yanf4j.test.example.chat;
import java.io.IOException;
import com.google.code.yanf4j.config.Configuration;
import com.google.code.yanf4j.nio.TCPController;
import com.google.code.yanf4j.nio.impl.StringCodecFactory;
public class ChatServer {
public static void main(String[] args) throws IOException {
TCPController controller = new TCPController(new Configuration(),
new StringCodecFactory());
controller.setHandler(new ChatServerHandler());
controller.setPort(8080); //端口
controller.start(); //啟動
}
}
順利完成,啟動后,telnet到8080端口,看看效果
