java socket 實現多個客戶端通過服務器一對一聊天並實現文件傳輸

服務器端：

package cn.com.test09;

import java.io.DataInputStream;
import java.io.DataOutputStream;
import java.io.IOException;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.HashMap;
import java.util.Set;

public class t12 {

    public static void main(String[] args) {
        // TODO Auto-generated method stub
       new ServerOb(8080);
    }

}
class ServerOb{
    private int port;
    ServerSocket ss;
    public ServerOb(int p){
        this.port=p;
        
        try {
            ss = new ServerSocket(port);
        } catch (IOException e1) {
            // TODO Auto-generated catch block
            e1.printStackTrace();
        }
        while(true){
            try {
                
                Socket soc = ss.accept();
                new Thread(new clientO(soc)).start();
                
            } catch (IOException e) {
                // TODO Auto-generated catch block
                System.out.println(" ServerOb 構造函數  出錯");
                e.printStackTrace();
            }
        }
    }
}
class clientO implements Runnable{
    public static HashMap<String, Socket> map= new HashMap<String, Socket>();
    private  String number = null;
    private Socket soc;
    private  DataInputStream in;
    private DataOutputStream out;
     public clientO(Socket s){
      this.soc=s;
      try {
        this.in=new DataInputStream(soc.getInputStream());
        this.out=new DataOutputStream(soc.getOutputStream());
    } catch (IOException e) {
        // TODO Auto-generated catch block
        e.printStackTrace();
    }
    
     getKey();
     }
     public String getNumber(){
        
         try {
             number=in.readUTF();//1
        } catch (IOException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
        return number;
     }
     public void  getKey(){
         map.put(getNumber(), soc);//得到帳號保存Map
         Set<String> ke = map.keySet();
        if(ke!=null)
         try {
             out.writeUTF(ke.toString());//2
             out.flush();
         } catch (IOException e) {
             // TODO Auto-generated catch block
             e.printStackTrace();
         }
     }
    
    @Override
    public void run() {
        // TODO Auto-generated method stub
        String oth;
        try {
            
            
                while(true){
                    oth = in.readUTF();
                    
                    if(oth.equalsIgnoreCase("l")){
                        getOtherNumber();
                    }
                    if(oth.equalsIgnoreCase("w")){
                        getFileransmission();
                    }
                    if(oth.equalsIgnoreCase("t")){
                        in.close();
                        out.close();
                        map.remove(number);
                        soc.close();
                        break;
                    }
                    
            }}
                 catch (IOException e) {
                    // TODO Auto-generated catch block
                    e.printStackTrace();
                    System.out.println("run  中出錯");
                }
        }
    public  void getFileransmission(){
        while(true){
            try {
                String oth = in.readUTF();//傳輸對象編號
                if(oth.equalsIgnoreCase("bye")){
                    return;
                }
                Socket other = map.get(oth);//取得傳輸對象socket
                String fileName = in.readUTF();//傳輸文件名
                
                DataOutputStream asd = new DataOutputStream(other.getOutputStream());
                asd.writeUTF("w");
                asd.flush();
                asd.writeUTF(number+"====="+fileName);
                asd.flush();
                // 文件字節傳輸
                byte[] b= new byte[1024];
                int i=0;
                while((i=in.read(b))!=-1){
                    asd.write(b, 0, i);
                    asd.flush();
                }
                
                
                
            } catch (IOException e) {
                // TODO Auto-generated catch block
                e.printStackTrace();
                System.out.println(number+"文件傳輸出錯");
            }
        }
    }
    public void  getOtherNumber(){
        while(true){
        try {    
            System.out.println("hhhhhhhhhhhhhhhhhh");
            String oth = in.readUTF();//4
            System.out.println(oth);
            Socket other = map.get(oth);
           DataOutputStream asd = new DataOutputStream(other.getOutputStream());
                
        
            
            if(oth.equalsIgnoreCase("bye")){
                return;
            }
            String value = in.readUTF();//5
            
            String str=number +"對你說：\r\n"+value;
            
            asd.writeUTF("l");
            asd.flush();
            asd.writeUTF(str);
            asd.flush();
            if(value.equalsIgnoreCase("bye")){
                return;
            }
        } catch (IOException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
            System.out.println(number+"聊天傳輸出錯");
        }
        }
    }
    
    
}

 

客戶端：

package cn.com.test09;

import java.io.BufferedReader;
import java.io.DataInputStream;
import java.io.DataOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.OutputStream;
import java.net.Socket;
import java.net.UnknownHostException;
import java.util.Scanner;

public class t13 {

    public static void main(String[] args) {
        
    new ClientOO("192.168.0.105",8080,"a");//服務器ip  服務器端口    自身編號
    }
}
class  ClientOO{
    private String path;
    private int port;
    private String name;
    private Scanner sca= new Scanner(System.in);
    private  DataInputStream in;
    private DataOutputStream out;
    public ClientOO(String path,int port,String name){
        this.path=path;
        this.port=port;
        this.name=name;
        try {
            Socket s=new Socket(path,port);
            this.in=new DataInputStream(s.getInputStream());
            this.out=new DataOutputStream(s.getOutputStream());
        } catch (UnknownHostException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        } catch (IOException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
        send();
        
        hhhhh();
    }
    public void send(){
        String s = null;
        try {
            out.writeUTF(name);//  1
            out.flush();
            
            
            s=in.readUTF();//2
        } catch (IOException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
        System.out.println("在線列表："+s);
    }
    public  void hhhhh(){
        while(true){
        System.out.println("聊天請輸入 l  文件傳輸請輸入   w  退出 請輸入  t ");
        String str=sca.next();
        if(str.equalsIgnoreCase("l")){
            try {
                out.writeUTF(str);
                out.flush();
            } catch (IOException e) {
                // TODO Auto-generated catch block
                e.printStackTrace();
            }
            System.out.println("====================");
            MyServerReader1 hhhh = new  MyServerReader1(in);
            MyServerWriter1 wwwwww = new MyServerWriter1(out,str);
            hhhh.start();
            wwwwww.start();
            try {
                wwwwww.join();
            } catch (InterruptedException e) {
                // TODO Auto-generated catch block
                e.printStackTrace();
            }
        }
        if(str.equalsIgnoreCase("w")){
            try {
                out.writeUTF(str);
                out.flush();
            } catch (IOException e) {
                // TODO Auto-generated catch block
                e.printStackTrace();
            }
            MyServerReader1 hhhh = new  MyServerReader1(in);
            MyServerWriter1 wwwwww = new MyServerWriter1(out,str);
            hhhh.start();
            wwwwww.start();
            try {
                wwwwww.join();
            } catch (InterruptedException e) {
                // TODO Auto-generated catch block
                e.printStackTrace();
            }
            
        }
        if(str.equalsIgnoreCase("t")){
            try {
                out.writeUTF(str);
                out.flush();
            } catch (IOException e) {
                // TODO Auto-generated catch block
                e.printStackTrace();
            }
            System.out.println("你退出系統");
            System.exit(0);
        }
        
        }
    }
}
class MyServerReader1 extends Thread {
    private DataInputStream dis;
    private String str;
    public MyServerReader1(DataInputStream dis) {
        this.dis = dis;
        
    }

    public void run() {
        try {
                str=dis.readUTF();
                if(str.equalsIgnoreCase("l")){
                    liaotian();
                }
                if(str.equalsIgnoreCase("w")){
                    wenjian();
                }
            
            
        } catch (IOException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
       
    }
    public void wenjian (){
        
        String fileName;
        try {
            String[] hhh = dis.readUTF().split("=====");
            fileName =hhh[1] ;
            File  f= new File("F:\\"+fileName);
            OutputStream os= new FileOutputStream(f);
            byte[] b= new byte[1024];
            int i=0;
            while((i=dis.read(b))!=-1){
                os.write(b, 0, i);
                
            }
            os.close();
            System.out.println(hhh[0]+"給你傳輸了文件："+fileName);
        } catch (IOException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
        
        
    }
    public  void liaotian(){
        String info;
        try {
           while (true) {
               // 如果對方，即客戶端沒有說話，那么就會阻塞到這里，
                // 這里的阻塞並不會影響到其他線程
               info = dis.readUTF();
                // 如果狀態有阻塞變為非阻塞，那么就打印接受到的信息
               System.out.println("對方說: " + info);
                if (info.equals("bye")) {
                    System.out.println("對方下線，選着其他好友!");
                  break;
                }
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}

// 從鍵盤獲得輸入流並寫入信息到客戶端
class MyServerWriter1 extends Thread {
   private DataOutputStream dos;
   private Scanner sc;
   private String str;
    public MyServerWriter1(DataOutputStream dos,String s) {
        this.dos = dos;
        this.str=s;
        this.sc=new Scanner(System.in);
    }

    public void run() {
        
       if(str.equalsIgnoreCase("l")){
           liaotian();
       }if(str.equalsIgnoreCase("w")){
           wenjian();
       }
    }
    public void wenjian(){
        System.out.println("=======進入文件傳輸=======");
        String s= sc.next();
        try {
            dos.writeUTF(s);
            dos.flush();
           System.out.println("輸入本地文件路徑：");
            String filepath= sc.next();
            String filename=filepath.substring(filepath.lastIndexOf("\\")+1);
            dos.writeUTF(filename);
            dos.flush();
            File f= new File(filepath);
            InputStream in= new FileInputStream(f);
            byte[] b= new byte[1024];
            int i=0;
            while((i=in.read(b))!=-1){
                dos.write(b, 0, i);
                
            }
            dos.flush();
            in.close();
            System.out.println("對"+s+"文件傳輸完成");
        } catch (IOException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
        
    }
    public void liaotian (){
        System.out.println("=======進入聊天=======");
         
        String info;
        try {
            while (true) {
                info = sc.next();//4
                if (info.equals("bye")) {
                    System.out.println("結束與對方聊天!");
                    break;
                }
              
                dos.writeUTF(info);
                dos.flush();
                info = sc.next();///5
                dos.writeUTF(info);
                dos.flush();
                if (info.equals("bye")) {
                    System.out.println("結束與對方聊天!");
                    break;
                }
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}