chatclient.java

一个聊天室的原代码

/*
 * Created on 2004-12-11
 *
 * TODO To change the template for this generated file go to
 * Window - Preferences - Java - Code Style - Code Templates
 */
package chatclient;

import java.awt.Color;
import java.awt.Dimension;
import java.awt.Event;
import java.awt.GridBagConstraints;
import java.awt.GridBagLayout;
import java.awt.GridLayout;
import java.awt.Insets;
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.net.MalformedURLException;
import java.rmi.Naming;
import java.rmi.NotBoundException;
import java.rmi.RemoteException;
import java.util.Hashtable;

import javax.swing.AbstractAction;
import javax.swing.Action;
import javax.swing.ImageIcon;
import javax.swing.JButton;
import javax.swing.JComboBox;
import javax.swing.JFrame;
import javax.swing.JLabel;
import javax.swing.JMenu;
import javax.swing.JMenuBar;
import javax.swing.JMenuItem;
import javax.swing.JOptionPane;
import javax.swing.JPanel;
import javax.swing.JScrollPane;
import javax.swing.JTextArea;
import javax.swing.KeyStroke;
import javax.swing.text.JTextComponent;
import javax.swing.text.Keymap;

import chat.ChatServer;
import chat.Chatter;

/**
 * @author Lance
 *
 * TODO To change the template for this generated type comment go to
 * Window - Preferences - Java - Code Style - Code Templates
 */
public class ChatClient extends JFrame {
	//存储已有聊天用户, key:用户名.value:对应的Chatter对象
	Hashtable hash=new Hashtable();
	//自己的用户名
	String my_name="chatter";
	//服务器地址
	String serverAddr;
	//代表客户端到远程对象
	Chatter chatter;
	//服务器端到远程对象
	ChatServer server;
	JTextArea displayBox;
	JTextArea inputBox;
	JComboBox usersBox;
	JButton sendButton;
	JLabel statusLabel;
	ConnectionAction connectAction=new ConnectionAction();
	//让用户输入用户名和服务器地址到对话框
	ConnectDlg dlg=new ConnectDlg(this);
	
	public static void main(String[] args){
		new ChatClient();
	}
	
	public ChatClient(){
		super("聊天-客户端");
		layoutComponent();
		setupMenu();
		addWindowListener(new WindowAdapter(){
			public void windowClosing(WindowEvent e){
				exit();
			}
		});
		show();
	}
	
	private void setupMenu(){
		JMenuBar menuBar=new JMenuBar();
		JMenuItem conn=new JMenuItem(connectAction);
		JMenuItem exit=new JMenuItem("退出");
		exit.addActionListener(new AbstractAction(){
			public void actionPerformed(ActionEvent evt){
				exit();
			}
		});
		JMenu file=new JMenu("文件");
		file.add(conn);
		menuBar.add(file);
		setJMenuBar(menuBar);
	}
	
	private void exit(){
		destroy();
		System.exit(0);
	}
	
	public void layoutComponent(){
		setSize(400, 400);
		JPanel contentPane=new JPanel();
		GridBagLayout gridbag=new GridBagLayout();
		GridBagConstraints c=new GridBagConstraints();
		contentPane.setLayout(gridbag);
		c.fill=GridBagConstraints.BOTH;
		c.gridwidth=GridBagConstraints.REMAINDER;
		c.gridheight=6;
		c.weightx=100;
		c.weighty=100;
		c.insets.top=5;
		displayBox=new JTextArea();
		displayBox.setLineWrap(true);
		displayBox.setEditable(false);
		displayBox.setMargin(new Insets(5,5,5,5));
		JScrollPane scrollPane=new JScrollPane(displayBox);
		contentPane.add(scrollPane, c);
		c.gridheight=1;
		c.weightx=0;
		c.weighty=0;
		c.insets.top=10;
		JLabel msgLabel=new JLabel("消息");
		contentPane.add(msgLabel, c);
		c.gridheight=6;
		c.insets.top=0;
		c.gridwidth=GridBagConstraints.RELATIVE;
		c.weightx=100;
		inputBox=new JTextArea();
		addKeymapBindings();
		inputBox.setLineWrap(true);
		inputBox.setWrapStyleWord(true);
		JScrollPane inputScrollPane=new JScrollPane(inputBox);
		inputScrollPane.setPreferredSize(new Dimension(250,50));
		inputScrollPane.setMinimumSize(new Dimension(250,50));
		contentPane.add(inputScrollPane, c);
		c.weightx=0;
		c.gridwidth=GridBagConstraints.REMAINDER;
		sendButton=new JButton(
				new ImageIcon(getClass().getResource("images/send.gif")));
		sendButton.setToolTipText("发送消息");
		sendButton.setPreferredSize(new Dimension(50,50));
		sendButton.setMinimumSize(new Dimension(50,50));
		sendButton.addActionListener(new ActionListener(){
			public void actionPerformed(ActionEvent evt){
				sendMessage();
			}
		});
		
		contentPane.add(sendButton, c);
		
		c.weightx=0;
		c.fill=GridBagConstraints.HORIZONTAL;
		c.gridheight=1;
		JLabel sendtoLabel=new JLabel("发送给");
		contentPane.add(sendtoLabel, c);
		usersBox=new JComboBox();
		usersBox.setBackground(Color.WHITE);
		usersBox.addItem("所有用户");
		contentPane.add(usersBox, c);
		JPanel statusPane=new JPanel(new GridLayout(1,1));
		statusLabel=new JLabel("未连接");
		statusPane.add(statusLabel);
		contentPane.add(statusPane, c);
		setContentPane(contentPane);
		try{
			chatter=new ChatterImpl(this);
		}catch(Exception e){
			e.printStackTrace();
			return;
		}
	}
	
	public void destroy(){
		try{
			disconnect();
		}catch(RemoteException ex){
			ex.printStackTrace();
		}
	}
	
	protected void connect()
		throws
			RemoteException,
			MalformedURLException,
			NotBoundException{
		server=(ChatServer)Naming.lookup("//"+serverAddr+"/ChatServer");
		server.login(my_name, chatter);
		
	}
	
	protected void disconnect()throws RemoteException{
		if(server!=null)
			server.logout(my_name);
	}
	
	public void receiveEnter(String name,Chatter chatter,boolean hasEntered){
		if(name!=null&&chatter!=null){
			hash.put(name, chatter);
			if(!name.equals(my_name)){
				//对新加入聊天的用户,在displayBox给出提示
				if(!hasEntered)
					display(name+" 进入聊天室");
				usersBox.addItem(name);
			}
		}
	}
	
	public void receiveExit(String name){
		if(name!=null&&chatter!=null)
			hash.remove(name);
		for(int i=0; i<usersBox.getItemCount(); i++){
			if(name.equals((String)usersBox.getItemAt(i))){
				usersBox.remove(i);
				break;
			}
		}
		display(name+" 离开聊天室");
	}
	
	public void receiveChat(String name, String message){
		display(name+": "+message);
	}
	
	public void receiveWhisper(String name,String message){
		display(name+" 私聊: "+message);
	}
	
	protected void addKeymapBindings(){
		Keymap keymap=
			JTextComponent.addKeymap("MyBindings", inputBox.getKeymap());
		Action action=null;
		KeyStroke key=null;
		//用户在消息输入框中按回车发送消息
		action=new AbstractAction(){
			public void actionPerformed(ActionEvent evt){
				sendMessage();
			}
		};
		key=KeyStroke.getKeyStroke(KeyEvent.VK_ENTER, 0);
		keymap.addActionForKeyStroke(key, action);
		
		//Ctrl+Enter换行
		action=new AbstractAction(){
			public void actionPerformed(ActionEvent evt){
				inputBox.append("\n");
			}
		};
		key=KeyStroke.getKeyStroke(KeyEvent.VK_ENTER, Event.CTRL_MASK);
		keymap.addActionForKeyStroke(key, action);
		
		inputBox.setKeymap(keymap);
	}
	
	private void display(String s){
		if(!s.endsWith("\n")){
			displayBox.append(s+"\n");
		}else{
			displayBox.append(s);
		}
		
		int length=displayBox.getText().length()-1;
		displayBox.select(length, length);
	}
	
	private void sendMessage(){
		String message=inputBox.getText();
		if(message!=null&&message.length()>0){
			inputBox.setText(null);
			inputBox.setCaretPosition(0);
			display(my_name+":"+message);
			if(server!=null){
				if("所有用户".equals(usersBox.getSelectedItem())){//发送给所有用户
					try{
						server.chat(my_name, message);
					}catch(RemoteException ex){
						ex.printStackTrace();
					}
				}else{//私聊,发给所选用户
					String destUserName=(String)usersBox.getSelectedItem();
					Chatter destChatter=(Chatter)hash.get(destUserName);
					try{
						destChatter.receiveWhisper(my_name, message);
					}catch(RemoteException e){
						e.printStackTrace();
					}
				}
			}
		}
		inputBox.requestFocus();
	}
	
	public void serverStop(){
		display("服务器停止");
		server=null;
		hash.clear();
		connectAction.setEnabled(true);
	}
	
	class ConnectionAction extends AbstractAction{
		public ConnectionAction(){
			super("连接");
			putValue(Action.SHORT_DESCRIPTION,"连接到服务器");
			putValue(
				Action.ACCELERATOR_KEY,
				KeyStroke.getKeyStroke("control C"));
		}

		/* (non-Javadoc)
		 * @see java.awt.event.ActionListener#actionPerformed(java.awt.event.ActionEvent)
		 */
		public void actionPerformed(ActionEvent evt) {
			// TODO Auto-generated method stub
			dlg.pack();
			dlg.setLocationRelativeTo(ChatClient.this);
			dlg.setVisible(true);
			if(dlg.getValue()==JOptionPane.OK_OPTION){
				try{
					my_name=dlg.getUserName();
					serverAddr=dlg.getServerAddr();
					connect();
					inputBox.setEditable(true);
					displayBox.setText("");
					statusLabel.setText(my_name+" 已连接");
					this.setEnabled(true);
				}catch(Exception e){
					e.printStackTrace();
					statusLabel.setText("不能连接到服务器");
					return;
				}
			}
		}
	}//End class ConnectionAction

}//END CLASS