qqserver.java

自己做的一个QQ聊天系统

package chat.server;

import java.io.BufferedReader;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.OutputStream;
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 chat.trans.*;

public class QQServer {
    static final String USERFILE = "users.txt";
    ServerSocket ss;
	public QQServer() throws Exception {
		File f = new File(USERFILE);
		if(!f.exists()) f.createNewFile();
		ss = new ServerSocket(Protocol.SERVER_PORT);
		while(true){
			Socket s = ss.accept();
			new ServerThread(s).start();
		}

	}
	public static void main(String[] args){
		try{
			new QQServer();
		}catch (Exception e){
			e.printStackTrace();
		}
	}

}
class ServerThread extends Thread{
	static Map<String,PrintWriter>users = new HashMap<String,PrintWriter>();
	Socket s;
	String userName;
	PrintWriter out;
	BufferedReader in;
	public ServerThread(Socket s)throws Exception{
		this.s = s;
			 out = new PrintWriter(s.getOutputStream());
			 InputStreamReader ir =
				 new InputStreamReader(
						 s.getInputStream());
			 in = new BufferedReader(ir);
	}
	public void run() {
		try{
			while(true){
			        String msg = in.readLine();
		           	if(msg == null){
		           		users.remove(userName);
		           		sendAll(Protocol.DEL_USER+userName);
		           		s.close();
		           		break;
		           	}
			        System.out.println("client msg:"
			        		+ msg);
			        process(msg);
			}
		}catch (Exception e){
			e.printStackTrace();
		}
  }
	private void sendAll(String msg){
		Iterator it = users.values().iterator();
		while(it.hasNext()){
			PrintWriter o = (PrintWriter) it.next();
			if(o!=out){
				o.println(msg);
			    o.flush();
			}
		}
	}
   private void process(String msg) throws Exception{
	System.out.println("process message...");
	char type = msg.charAt(0);
	System.out.println("switch:" + type);
	switch(type){
	case Protocol.USER_REGISTER_MSG:
		System.out.println("switch...");
		String name = msg.substring(1,
				msg.indexOf(Protocol.SPLIT_CHAR));
		System.out.println("will validate...");
		boolean b = validata(name);
		if(b){
			save(msg.substring(1));
		    userName = name;
		    sendAll(Protocol.ADD_USER + name);
		    users.put(name,out);
		    out.println(msg(Protocol.ONLINE_USER_MSG));
		}
		else{
			out.println(msg(
					Protocol.REGISTER_RESULT));
		}
		   out.flush();
		   System.out.println("send to client!");
		   break;
	case Protocol.USER_LOGIN_MSG:
		   String n = msg.substring(1,
				   msg.indexOf(Protocol.SPLIT_CHAR));
		   if(users.containsKey(n)){
			       out.println(Protocol.LOGIN_RESULT
			    		          +"不能重复登入!");
			       out.flush();
		   }else{
		        String p = msg.substring(
				   msg.indexOf(Protocol.SPLIT_CHAR)+1);
		        if(validata(n,p)){
		        	userName = n;
		        	sendAll(Protocol.ADD_USER+n);
		        	users.put(n,out);
		        	out.println(msg(
		        			Protocol.ONLINE_USER_MSG));
		        	out.flush();
		        }else{
		        	out.println(Protocol.LOGIN_RESULT
		        			+"登入信息错误");
		        	out.flush();
		        }
	}
		   break;
	case Protocol.CHAT_MSG:
		String[] ss = msg.split(",");
		if(ss[1].equals("all")) sendAll(msg);
		else{
			 users.get(ss[1]).println(msg);
			 users.get(ss[1]).flush();
		}
		break;
  }
}
   private String msg(char type){
	   System.out.println("prepare message...");
	   StringBuffer s = new StringBuffer();
	   s.append(type);
	   switch(type){
	   case Protocol.REGISTER_RESULT:
		   s.append("该登入名已被使用!");
	   case Protocol.ONLINE_USER_MSG:
		   Iterator it = users.keySet().iterator();
		   while(it.hasNext()){			   
		        s.append(it.next());
		        s.append(Protocol.SPLIT_CHAR);
		   }
		   s.deleteCharAt(s.length()-1);
	   }
	   return s.toString();
   }
  private boolean validata(String name,String passwd)
                                   throws Exception{
		  InputStream in = 
			  new FileInputStream(QQServer.USERFILE);
		  InputStreamReader ir = 
			  new InputStreamReader(in);
		  BufferedReader br = new BufferedReader(ir);
		  String s = null;
		  boolean ret = false;
		  while((s=br.readLine())!= null){
			  String[] ss = s.split(Protocol.SPLIT_CHAR+"");
			  if(ss[0].equals(name)&& ss[1].equals(passwd)){
				  ret = true;
				  break;
			  }
	  }
	  br.close();
	  return ret;
    }
  private boolean validata(String name)throws Exception{
              System.out.println("validate loginname");        
	          InputStream in = 
                  new FileInputStream(QQServer.USERFILE);
              InputStreamReader ir = 
                   new InputStreamReader(in);
              BufferedReader br = new BufferedReader(ir);
              String s = null;
              boolean ret = true;
              while((s=br.readLine())!= null){
                     String n = s.substring(0,
                    		      s.indexOf(Protocol.SPLIT_CHAR));
               if(n.equals(name)){
                      ret = false;
                      break;
                    } 
               }
              br.close();
              return ret;
}
   private void save(String user)throws Exception{
	        System.out.println("save user msg");
		   OutputStream o =
			   new FileOutputStream(QQServer.USERFILE,true);
		   PrintWriter pw = new PrintWriter(o);
		   pw.println(user);
		   pw.close();
   }
}