网络通讯---聊天室

Server

package com.wanwan.net;

import java.io.DataInputStream;
import java.io.DataOutputStream;
import java.io.IOException;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.ArrayList;
import java.util.List;

public class Server {
    
    private List<MyChannel> all = new ArrayList<MyChannel>();
    
    public static void main(String[] args) throws IOException{
        new Server().start();
    }
    
    public void start() throws IOException{
        
        ServerSocket server = new ServerSocket(9999);
        
        while(true){
            Socket client = server.accept();
            MyChannel channel = new MyChannel(client);
            all.add(channel);
            new Thread(channel).start();
        }
    }
    
    private class MyChannel implements Runnable{
        
        private DataInputStream dis;
        private DataOutputStream dos;
        private boolean isRunning = true;
        private String name;
        
        public MyChannel(Socket client) {
            // TODO Auto-generated constructor stub
            try {
                dis = new DataInputStream(client.getInputStream());
                dos = new DataOutputStream(client.getOutputStream());
                this.name = dis.readUTF();
                this.send("欢迎进入聊天室");
                this.sendOthers(this.name + "进入了聊天室",true);
            } catch (IOException e) {
                // TODO Auto-generated catch block
                CloseUtil.closeAll(dis,dos);
                isRunning = false;
            }
        }
        
        private String recevie(){
            String msg = "";
            try {
                msg = dis.readUTF();
            } catch (IOException e) {
                // TODO Auto-generated catch block
                CloseUtil.closeAll(dis,dos);
                isRunning = false;
                all.remove(this);
            }
            return msg;
        }
        
        private void send(String msg){
            if(null==msg || msg.equals("")){
                return;
            }
            
            try {
                dos.writeUTF(msg);
                dos.flush();
            } catch (IOException e) {
                // TODO Auto-generated catch block
                CloseUtil.closeAll(dis,dos);
                isRunning = false;
                all.remove(this);
            }
        }
        
        
        private void sendOthers(String msg,boolean sys){
            
            if(msg.startsWith("@") && msg.indexOf(":")>-1){
                
                String name = msg.substring(1, msg.indexOf(":"));
                String content = msg.substring(msg.indexOf(":")+1);
                
                for(MyChannel other:all){
                    if(other.name.equals(name)){
                        other.send(this.name + "对你悄悄的说："+content);
                    }
                }
                
            }else{
                
                for(MyChannel other:all){
                    if(other == this){
                        continue;
                    }
                    if(sys){
                        other.send("系统信息:"+msg);
                    }else{
                        other.send(this.name+"对所有人说:"+msg);
                    }
                    
                }
            }    
        }
        
        
        @Override
        public void run() {
            // TODO Auto-generated method stub
            while(isRunning){
                sendOthers(recevie(),false);
            }
        }
    }
}

Client

package com.wanwan.net;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.net.Socket;
import java.net.UnknownHostException;

public class Client {

    public static void main(String[] args) throws UnknownHostException, IOException{
    
        System.out.println("请输入名称:");
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String name = br.readLine();
        if(name.equals("")){
            return;
        }
        
        Socket client = new Socket("localhost", 9999);
        
        new Thread(new Send(client,name)).start();
        new Thread(new Receive(client)).start();
        
    }
    
}

Send

package com.wanwan.net;

import java.io.BufferedReader;
import java.io.DataOutputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.net.Socket;

public class Send implements Runnable{

    private BufferedReader console;
    private DataOutputStream dos;
    private boolean isRunning = true;
    private String name;
    
    public Send(){
        console = new BufferedReader(new InputStreamReader(System.in));
    }
    
    public Send(Socket client,String name){
        this();
        try {
            dos = new DataOutputStream(client.getOutputStream());
            this.name = name;
            sendM(this.name);
        } catch (IOException e) {
            // TODO Auto-generated catch block
            isRunning = false;
            CloseUtil.closeAll(dos,console);
        }
    }
    
    private String getMsgFromConsole(){
        try {
            return console.readLine();
        } catch (IOException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
        return "";
    }
    
    public void sendM(String msg){
        if(null != msg && !msg.equals("")){
            try {
                dos.writeUTF(msg);
                dos.flush();
            } catch (IOException e) {
                // TODO Auto-generated catch block
                isRunning = false;
                CloseUtil.closeAll(dos,console);
            }
        }
    }
    
    @Override
    public void run() {
        // TODO Auto-generated method stub
        while(isRunning){
            sendM(getMsgFromConsole());
        }
    }
    
}

Receive

package com.wanwan.net;

import java.io.DataInputStream;
import java.io.IOException;
import java.net.Socket;

public class Receive implements Runnable{
    
    private DataInputStream dis;
    private boolean isRuning = true;
    
    public Receive() {
        // TODO Auto-generated constructor stub
    }
    
    public Receive(Socket client){
        try {
            dis = new DataInputStream(client.getInputStream());
        } catch (IOException e) {
            // TODO Auto-generated catch block
            isRuning = false;
            CloseUtil.closeAll(dis);
        }
    }
    
    public String receiveM(){
        String msg = "";
        try {
            msg = dis.readUTF();
        } catch (IOException e) {
            // TODO Auto-generated catch block
            isRuning = false;
            CloseUtil.closeAll(dis);
        }
        return msg;
    }
    
    @Override
    public void run() {
        // TODO Auto-generated method stub
        while(isRuning){
            System.out.println(receiveM());
        }
    }

}

CloseUtil

package com.wanwan.net;

import java.io.Closeable;

public class CloseUtil {

    public static void closeAll(Closeable... io){
        for(Closeable temp:io)
        {
            try{
                if(null != temp){
                    temp.close();
                }
            }catch (Exception e) {
                // TODO: handle exception
            }
        }
    }
    
}