Java实现的聊天室

public class Setting {
public static String SERVER_IP = "127.0.0.1";
public static int SERVER_PORT = 5250;

public static String COMMAND_LOGIN = "login";
public static String COMMAND_LOGIN_SUC = "login_suc";
public static String COMMAND_LOGIN_FAILED = "login_failed";
public static String COMMAND_LOGOUT = "logout";
public static String COMMAND_SENDMSG = "sendmsg";

public static String COMMAND_USERLIST = "userlist";
public static String COMMAND_MSG = "msg";
}

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.io.PrintWriter;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;
import java.util.Map.Entry;

class Server {
private ServerSocket socket_server;
private int port;
private Map<String, Socket> map_client_socket;

public static void main(String args[]){
try {
new Server(Setting.SERVER_PORT).start();
} catch (IOException e) {
System.out.println(e.getMessage());
}
}

public Server(int _port) throws IOException {
this.port = _port;
map_client_socket = new HashMap<String, Socket>();
}

public void start() throws IOException {
try {
socket_server = new ServerSocket(port);
while(true) {
Socket socket_connection = socket_server.accept();
new ListenService(socket_connection).start();
}
} catch (IOException e) {
System.out.println("初始化错误");
throw e;
}
}

class ListenService extends Thread {
private Socket socket;
private String user_name;
private boolean logined = false;
private BufferedReader in = null;

public ListenService(Socket _socket) {
try {
socket = _socket;
in = new BufferedReader(new InputStreamReader(socket.getInputStream()));
} catch (IOException e) {
e.printStackTrace();
}
}

public void run() {
try {
String command = "";
while(true) {
if((command = in.readLine()) != null) {
if(!logined) {
if(command.equals(Setting.COMMAND_LOGIN)) {
user_name = in.readLine();	//读取用户名

//如果用户名存在则发送无法登陆消息并关闭连接
if(map_client_socket.containsKey(user_name)) {
sendMsg(socket, Setting.COMMAND_LOGIN_FAILED);
in.close();
socket.close();
break;
}

//将socket放入map中
map_client_socket.put(user_name, socket);

String msg = Setting.COMMAND_LOGIN_SUC + "\n" + user_name;
sendMsgToAll(msg);
updateUserList();
logined = true;
}
} else {
if(command.equals(Setting.COMMAND_LOGOUT)) {
in.close();
socket.close();
map_client_socket.remove(user_name);

String msg = Setting.COMMAND_LOGOUT + "\n" + user_name;
sendMsgToAll(msg);
updateUserList();
break;
} else if(command.equals(Setting.COMMAND_SENDMSG)) {
String dest_username = in.readLine();	//读取目标用户名
String msg = Setting.COMMAND_MSG + "\n"
+ user_name + "\n" //将发送者的用户名发过去
+ in.readLine();

if(dest_username.equals("所有人")) {
sendMsgToAll(msg);
} else {
sendMsg(dest_username, msg);
}
}
}
}

}
} catch (IOException e) {
e.printStackTrace();
map_client_socket.remove(user_name);
updateUserList();
}
}

public void sendMsg(Socket dest_socket, String message) {
try {
PrintWriter out = new PrintWriter(new BufferedWriter(
new OutputStreamWriter(dest_socket.getOutputStream())),true);
out.println(message);
out.flush();
} catch (Exception e) {
e.printStackTrace();
}
}

public void sendMsg(String dest_username, String message) {
Socket dest_socket = map_client_socket.get(dest_username);
sendMsg(dest_socket, message);
}

public void sendMsgToAll(String message) {
try {
for(Iterator<Entry<String, Socket>> itor = map_client_socket.entrySet().iterator(); itor.hasNext();) {
Entry<String, Socket> entry = (Entry<String, Socket>)itor.next();

Socket dest_socket = entry.getValue();
PrintWriter out = new PrintWriter(new BufferedWriter(
new OutputStreamWriter(dest_socket.getOutputStream())),true);
out.println(message);
out.flush();
}
} catch (Exception e) {
e.printStackTrace();
}
}

public void updateUserList() {
String msg = Setting.COMMAND_USERLIST + "\n";

for(Iterator<Entry<String, Socket>> itor = map_client_socket.entrySet().iterator(); itor.hasNext();) {
Entry<String, Socket> entry = (Entry<String, Socket>)itor.next();
msg += entry.getKey() + "|";
}

sendMsgToAll(msg);
}
}
}

import java.awt.Color;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.awt.event.KeyEvent;
import java.awt.event.WindowAdapter;
import java.awt.event.WindowEvent;
import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.io.PrintWriter;
import java.net.Socket;
import java.text.SimpleDateFormat;
import java.util.Date;

import javax.swing.JButton;
import javax.swing.JComboBox;
import javax.swing.JFrame;
import javax.swing.JLabel;
import javax.swing.JOptionPane;
import javax.swing.JScrollPane;
import javax.swing.JTextArea;
import javax.swing.JTextPane;
import javax.swing.text.BadLocationException;
import javax.swing.text.Document;
import javax.swing.text.SimpleAttributeSet;
import javax.swing.text.StyleConstants;

public class Client extends JFrame implements ActionListener {
private static final long serialVersionUID = 2389760719589425440L;
private String user_name;
private Socket socket;
JTextPane text_history;
Document docs_history;
JTextArea text_in;
JButton btn_send;
JLabel label_destuser;
JComboBox<String> cbbox_destuser;
SimpleAttributeSet attrset_cmd = new SimpleAttributeSet();
SimpleAttributeSet attrset_msg = new SimpleAttributeSet();
PrintWriter out = null;

public static void main(String args[]) {
try {
String name = JOptionPane.showInputDialog("请输入用户名：","user1");
if(name == null || name.length() == 0)
return;

Socket socket = new Socket(Setting.SERVER_IP, Setting.SERVER_PORT);
new Client(name, socket);
} catch (Exception e) {
System.out.println(e.getMessage());
}
}

public Client(String _user_name, Socket _socket) throws IOException {
super("聊天室 - " + _user_name);
this.setLayout(null);
this.setBounds(200, 200, 500, 480);
this.setResizable(false);
this.setDefaultCloseOperation(JFrame.DISPOSE_ON_CLOSE);
this.addWindowListener(new CloseHandler());

user_name = _user_name;
socket = _socket;

text_history = new JTextPane();
text_history.setBounds(5, 5, 485, 290);
text_history.setEditable(false);
JScrollPane scroll_text_history = new JScrollPane(text_history);
scroll_text_history.setBounds(5, 5, 485, 290);
this.add(scroll_text_history);
docs_history = text_history.getDocument();

text_in = new JTextArea();
text_in.setBounds(5, 305, 485, 110);
text_in.setLineWrap(true);
text_in.setEnabled(false);
JScrollPane scroll_text_in = new JScrollPane(text_in);
scroll_text_in.setVerticalScrollBarPolicy(JScrollPane.VERTICAL_SCROLLBAR_AS_NEEDED);
scroll_text_in.setBounds(5, 305, 485, 110);
this.add(scroll_text_in);

btn_send = new JButton("发送(alt+回车)");
btn_send.setBounds(370, 420, 120, 25);
btn_send.addActionListener(this);
btn_send.setEnabled(false);
btn_send.setMnemonic(KeyEvent.VK_ENTER);
this.add(btn_send);

label_destuser = new JLabel("发送给:");
label_destuser.setBounds(215, 420, 50, 25);
this.add(label_destuser);

cbbox_destuser = new JComboBox<String>();
cbbox_destuser.setBounds(265, 420, 100, 25);
this.add(cbbox_destuser);

StyleConstants.setForeground(attrset_cmd, Color.BLUE);
StyleConstants.setBold(attrset_cmd, true);

StyleConstants.setForeground(attrset_msg, Color.BLACK);

this.setVisible(true);
out = new PrintWriter(new BufferedWriter(new OutputStreamWriter(socket.getOutputStream())),true);
new ListenService(user_name, socket).start();
}

class CloseHandler extends WindowAdapter {
public void windowClosing(final WindowEvent ev) {
try {
out.println(Setting.COMMAND_LOGOUT);
out.flush();
out.close();
socket.close();
System.exit(0);
} catch (IOException e) {
e.printStackTrace();
System.exit(0);
}
}
}

@Override
public void actionPerformed(ActionEvent ev) {
if(ev.getActionCommand()=="发送(alt+回车)") {
String dest_user = (String)cbbox_destuser.getSelectedItem();
if(dest_user == null)
return;

String msg = text_in.getText();
insertText("[我] 对 [" + dest_user +  "] 说: (" + getTime() + ")\n", attrset_cmd);
insertText(msg + "\n", attrset_msg);

out.println(Setting.COMMAND_SENDMSG);
out.println(dest_user);
out.println(msg.replaceAll("\\n", "\\\\n"));
out.flush();

text_in.setText("");
}
}

public void insertText(String str, SimpleAttributeSet attrset) {
try {
docs_history.insertString(docs_history.getLength(), str, attrset);
} catch (BadLocationException ble) {
System.out.println("BadLocationException:" + ble);
}
}

public String getTime() {
SimpleDateFormat df = new SimpleDateFormat("hh:mm:ss");
return df.format(new Date());
}

class ListenService extends Thread {
private Socket socket;
private String user_name;
private BufferedReader in = null;
private PrintWriter out = null;

public ListenService(String _user_name, Socket _socket) {
try {
user_name = _user_name;
socket = _socket;
in = new BufferedReader(new InputStreamReader(socket.getInputStream()));
out = new PrintWriter(new BufferedWriter(new OutputStreamWriter(socket.getOutputStream())),true);
} catch (IOException e) {
e.printStackTrace();
}
}

public void run() {
try {
String command = "";

//发送登陆命令
command = Setting.COMMAND_LOGIN + "\n"
+ user_name + "\n";
out.print(command);
out.flush();

while(true) {
if((command = in.readLine()) != null) {
if(command.equals(Setting.COMMAND_MSG)) {
String user_src = in.readLine();
String msg = in.readLine().replaceAll("\\\\n", "\n");
if(!user_src.equals(user_name)) {
insertText("[" + user_src + "] 对 [我] 说: (" + getTime() + ")\n", attrset_cmd);
insertText(msg + "\n", attrset_msg);
}
} else if(command.equals(Setting.COMMAND_USERLIST)) {
String user_list = in.readLine();
String user[] = user_list.split("\\|");
cbbox_destuser.removeAllItems();

cbbox_destuser.addItem("所有人");
for(int i=0; i<user.length; i++) {
if(!user[i].equals(user_name))
cbbox_destuser.addItem(user[i]);
}
} else if(command.equals(Setting.COMMAND_LOGIN_SUC)) {
String user_login = in.readLine();
insertText("[" + user_login + "] 进入了聊天室. (" + getTime() + ")\n", attrset_cmd);
text_in.setEnabled(true);
btn_send.setEnabled(true);
} else if(command.equals(Setting.COMMAND_LOGIN_FAILED)) {
insertText(user_name + "登陆失败\n", attrset_cmd);
} else if(command.equals(Setting.COMMAND_LOGOUT)) {
String user_logout = in.readLine();
insertText("[" + user_logout + "] 退出了聊天室. (" + getTime() + ")\n", attrset_cmd);
}
}
}
} catch (IOException e) {
e.printStackTrace();
}
}
}
}