java 用socket製做一個簡易多人聊天室

時間 2020-03-17

原文原文鏈接

代碼：java

服務器端Server數組

import java.io.*;
import java.net.*;
import java.util.ArrayList;
public class Server{
    public static ServerSocket server_socket;
    public static ArrayList<Socket> socketList=new ArrayList<Socket>();  
    public static void main(String []args){
        try{
            server_socket = new ServerSocket(5000);
            while(true){
                Socket socket = server_socket.accept();
                socketList.add(socket); //把sock對象加入sock集合
                ServerBO_Thread st=new ServerBO_Thread(socket,socketList); //初始化多線程
                st.start();//啓動多線程
            }
        }catch(Exception ex){
            ex.printStackTrace();
        }finally{
            try{
                if(server_socket!=null){
                    server_socket.close();
                }
            }catch(Exception ex){
                ex.printStackTrace();
            }
        }
    }
    public void encryptWrite(String src,DataOutputStream output)throws IOException{
        //將一個字符串轉化爲字符數組
        //System.out.println(src);
        char[] char_arr = src.toCharArray();
        //加密操做
        for(int i = 0;i<char_arr.length;i++){
            output.writeChar(char_arr[i]+13);
        }
        //用做結束標誌符
        output.writeChar(2333);
        output.flush();
    }
    //讀取並解密
    public String readDecrypt(DataInputStream input)throws IOException{
        String rtn="";
        while(true){
            int char_src =input.readChar();
            if(char_src!=2333){
                rtn=rtn+(char)(char_src-13);
            }else{
                break;
            }
        }
        return rtn;
    }
}
class ServerBO_Thread extends Thread{
    Socket client = null;
    ArrayList<Socket> clients;
    ServerBO_Thread(Socket s,ArrayList<Socket> ss){//初始化
        client=s;
        clients=ss; 
    }
    public void run(){
        DataInputStream input = null;
        DataOutputStream output =null;
        try{
            input = new DataInputStream(client.getInputStream());
            Server bo = new Server();
            String receive=null;
            String send=null;
            while(true){//監視當前客戶端有沒有發來消息
                if(!client.isClosed()){
                    receive=bo.readDecrypt(input);
                    clients.trimToSize();
                    String[] param = receive.split("&");
                    if(")start".equals(param[1])){    //分析客戶端發來的內容
                        send = param[0]+"進入聊天室";
                    }else{
                        send = param[0]+"說:    "+param[1];
                    }
                    if(!("3333".equals(param[1]))){//3333爲退出聊天室信號
                        for(Socket socket:clients){ //遍歷socke集合 
                            //把讀取到的消息發送給各個客戶端  
                            if(!socket.isClosed()){
                                output = new DataOutputStream(socket.getOutputStream());
                                bo.encryptWrite(send,output);
                            }
                        }  
                    }else{//若是有客戶端退出
                        for(Socket socket:clients){ //遍歷socke集合 
                                if(socket!=client){//告訴其餘人此人退出聊天室
                                    if(!(socket.isClosed())){
                                        output = new DataOutputStream(socket.getOutputStream());
                                        bo.encryptWrite(param[0]+"已退出聊天室",output);
                                    }
                                }
                            }
                        output = new DataOutputStream(client.getOutputStream());
                        bo.encryptWrite("3333",output);//返回信號給要退出的客戶端，而後關閉線程
                        client.close();
                        input.close();
                        output.close();
                    }
                }else{
                    break;
                }
            }
        }catch(Exception ex){
            ex.printStackTrace();
        }
    }
}

客戶端：服務器

import java.io.IOException;
import java.util.Scanner;
import java.net.*;
import java.io.*;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.io.BufferedWriter;
import java.io.BufferedReader;
import java.io.DataOutputStream;


public class People{
//服務端ip
    public  String ip = "127.0.0.1";
    //服務端端口
    public  int port = 5000;
    public  DataOutputStream output = null;
    public  Socket socket = null;
    public  DataInputStream input = null;
    public  Scanner sc =new Scanner (System.in);
    public  String send ;
    public  String receive;
    public  String name;
    public String sd = null;
    public static void main(String[]aa){
        People po = new People();
        po.start();
    }
    public void start(){
        try{
            System.out.println("*******歡迎使用匿名聊天室！**********");
            System.out.println("請輸入你將要使用的暱稱：");
            name=sc.nextLine();//獲取暱稱
            socket = new Socket(ip,port);
            output=new DataOutputStream(socket.getOutputStream());
            input = new DataInputStream(socket.getInputStream());
            send = name+"&)start";//把暱稱發送到server 告訴全部人有新成員加入聊天室
            System.out.println("(若是要退出聊天室請輸入「3333」！)");
            System.out.println("*******成功進入匿名聊天室！**********");
            System.out.println("");
            encryptWrite(send,output);
            Out out=new Out(output,name,input,socket);
            out.start();//啓動發送聊天內容的多線程
            while(true){    
                String receive = readDecrypt(input);
                if("3333".equals(receive)){//若是收到「3333」則退出聊天室
                    System.out.println("*******成功退出匿名聊天室！**********");
                    input.close();
                    output.close();
                    socket.close();
                    System.exit(0);
                }
                System.out.println(receive);
            }
        }catch(Exception ex){
                ex.printStackTrace();
        }finally{
            try{
                if(socket!=null) socket.close();
                input.close();
                output.close();
            }catch(Exception ex){
                ex.printStackTrace();
            }
        }    
    }
    public void encryptWrite(String src,DataOutputStream output)throws IOException{
        //將一個字符串轉化爲字符數組
        char[] char_arr = src.toCharArray();
        //加密操做
        for(int i = 0;i<char_arr.length;i++){
            output.writeChar(char_arr[i]+13);
        }
        //用做結束標誌符
        output.writeChar(2333);
        output.flush();
    }
    //讀取並解密
    public String readDecrypt(DataInputStream input)throws IOException{
        String rtn="";
        while(true){
            int char_src =input.readChar();
            if(char_src!=2333){
                rtn=rtn+(char)(char_src-13);
            }else{
                break;
            }
        }
        return rtn;
    }
}
class Out extends Thread {
    public DataOutputStream output;
    public DataInputStream input;
    public static String name;
    public Socket socket;
    public  Scanner sc =new Scanner (System.in);
    Out(DataOutputStream ot,String n,DataInputStream it,Socket socket){
        output=ot;
        input=it;
        name=n;
    }
    public void run(){
        People po = new People();
        try{
            while(true){
                String send=sc.nextLine();//獲取用戶輸入
                String send2=name+"&"+send;//把聊天內容打包成約定形式
                po.encryptWrite(send2,output);
            }
        }catch(Exception ex){
            ex.printStackTrace();
        }finally{
            System.out.println("sfef");
        }
    }
}

實現效果：多線程

實驗心得：socket

　　一、用於接收信息的字符串每次都要從新定義一個，不能在開頭只定義一次，這樣會致使數組角標異常的錯誤，由於每次接收到的信息的長度都不同的，因此每次用都要從新new一個字符串。函數

　　二、調用多線程的時候必定要把socket或者input output參數傳遞給構造函數初始化，不能用類.output的形式調用，否則會有空指針的錯誤（緣由大概是直接調用的多是還沒初始化的）。加密

　　三、爲了程序的健壯性，咱們須要對當有客戶端退出聊天室的狀況進行處理，若是不處理，當有人強制退出聊天室時會致使服務端崩潰發生空指針異常。在這個程序主中我約定用「3333」爲退出信息，當客戶端發出3333時即退出聊天室，關閉對應socket，此處關鍵點是要理清誰先關閉誰後關閉，否則也會致使socket崩潰，流程應該是發送3333告知服務器端我要退出了，而後服務器返回信號給客戶端我已知曉，你能夠退出了，而後客戶端服務器端才能夠關閉socketspa