1. 程式人生 > >簡單的多執行緒socket

簡單的多執行緒socket

1.socket 伺服器搭建

例項化socket伺服器,迴圈獲取請求

package com.orange.util;

import java.io.IOException;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.Timer;
import java.util.TimerTask;
import java.util.concurrent.Executors;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.TimeUnit;

/** * socket伺服器 * * @author Chengjq * */ public class SocketServer { public static int count = 0; public static void main(String[] args) { // TODO 自動生成的方法存根 int port = 4404; // 首先直接建立serversocket ServerSocket serverSocket = null; Socket socket = null;
try { serverSocket = new ServerSocket(port); System.out.println("啟動socketServer成功,等待客戶端的連線"); while (true) { socket = serverSocket.accept(); System.out.println("有新的客戶端請求連線"); SocketThread st = new SocketThread(socket); st.start(); ChatManager.getChatManager().add(st);
//啟動定時任務,如果10s內沒有程序 /*Runnable runnable = new Runnable() { int clientNum = 0; public void run() { // task to run goes here clientNum = ChatManager.getChatManager().vector.size(); System.out.println("剩餘客戶端數量:"+clientNum); if(clientNum ==0 ){ System.out.println("連線超時,或者無客戶端連線,關閉serverSocket"); //關閉socket //..... } } }; ScheduledExecutorService service = Executors .newSingleThreadScheduledExecutor(); // 第二個引數為首次執行的延時時間,第三個引數為定時執行的間隔時間 service.scheduleAtFixedRate(runnable, 2, 10, TimeUnit.SECONDS); */ } } catch (IOException e) { // TODO Auto-generated catch block e.printStackTrace(); } finally { System.out.println("serverSocket已超時"); try { socket.close(); serverSocket.close(); } catch (IOException e) { // TODO Auto-generated catch block e.printStackTrace(); } } } }

SocketThread類實現多執行緒通訊

package com.orange.util;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.PrintWriter;
import java.net.ServerSocket;
import java.net.Socket;

/**
 * SocketThread實現多執行緒通訊
 * 
 * @author Administrator
 *
 */
public class SocketThread extends Thread {

	ServerSocket serverSocket = null;

	Socket socket = null;

	public SocketThread(ServerSocket serverSocket,Socket socket) {
		super();
		this.serverSocket = serverSocket;
		this.socket = socket;
	}
	
	public SocketThread(Socket socket) {
		super();
		this.socket = socket;
	}
	
	public void out(String out) {
        try {
            socket.getOutputStream().write(out.getBytes("utf-8"));
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
	
	public void publish(String out){
		ChatManager.getChatManager().publish(this, out);
	}
	

	@Override
	public void run() {
		// TODO Auto-generated method stub
		BufferedReader socketIn = null;
		PrintWriter socketOut = null;
		String inMess = null;
		try {
			socketIn = new BufferedReader(new InputStreamReader(socket.getInputStream()));
			socketOut = new PrintWriter(socket.getOutputStream());
			while (true) {
				inMess = socketIn.readLine();
				publish(inMess);
				if("bye".equals(inMess)){
					ChatManager.getChatManager().remove(this);
				}
			}
		} catch (IOException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		} finally {
			try {
				System.out.println("已結束當前會話");
				socketOut.close();
				socketIn.close();
				socket.close();
			} catch (IOException e) {
				// TODO Auto-generated catch block
				e.printStackTrace();
			}
		}

	}

}

  單例Chatmanage,對所有客戶端執行緒管控處理

package com.orange.util;

import java.util.Vector;

public class ChatManager {
    // 實現單例化
    private ChatManager() {
    };

    private static final ChatManager cm = new ChatManager();

    public static ChatManager getChatManager() {// 返回值為ChatManager
        return cm;
    }

    // 單例化完成
    Vector<SocketThread> vector = new Vector<SocketThread>();

    public void add(SocketThread st) {// 為當前集合新增SocketThread物件
        vector.add(st);
    }
    
    public void remove(SocketThread st) {// 當前客戶端關閉連線
        vector.remove(st);
    }
    
    public void removeall() {// 關閉所有連線
        for (int i = 0; i < vector.size(); i++) {// 遍歷所有的執行緒
            SocketThread csChatSocket = vector.get(i);
            if(csChatSocket!=null){
                vector.remove(csChatSocket);
            }
                
        }
    }

    // 某一個執行緒向其他的客戶端傳送資訊
    public void publish(SocketThread st, String out) {
        for (int i = 0; i < vector.size(); i++) {// 遍歷所有的執行緒
            SocketThread csChatSocket = vector.get(i);
            if (csChatSocket != st)// 判斷不是當前執行緒就傳送此訊息
                csChatSocket.out(out + "\n");
        }
    }

    // 向當前執行緒發信息
    public void publish_present(SocketThread st, String out) {
        st.out(out + "\n");
    }
}

至此,伺服器搭建完成

2.客戶端(建立兩個客戶端)

客戶端1(監聽指定伺服器,通過控制檯輸入訊息進行伺服器與客戶端以及客戶端之間的通訊,)

package com.orange;

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

/**
 * 客戶端1
 * @author Chengjq
 *
 */
public class SocketClient1 {

	@SuppressWarnings("static-access")
	public static void main(String[] args) {
		try {
			//初始化客戶端
			Socket socket = new Socket("127.0.0.1", 4404);
			BufferedReader readline = new BufferedReader(new InputStreamReader(System.in));
			//獲取輸出列印流
			PrintWriter socketOut = new PrintWriter(socket.getOutputStream());
			String outTemp = null;
			System.out.println("開始準備向伺服器端發起請求---\n自己:");
			// 已啟動連線socket伺服器,準備實時接收來自其他客戶端的訊息
			GetMess getMess = new GetMess(socket);
			getMess.start();
			// 通過控制檯傳送訊息給其他客戶端,以“bye”為結束語
			while ((outTemp = readline.readLine()) != null) {
				//傳送資訊
				socketOut.println(outTemp);
				socketOut.flush();
				if("bye".equals(outTemp)){
					break;
				}
			}
			getMess.currentThread().interrupt();
			//依次關閉各種流
			readline.close();
			socketOut.close();
			socket.close();
		} catch (UnknownHostException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		} catch (IOException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}

	}
}

  客戶端2

package com.orange;

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

public class SocketClient2 {
    @SuppressWarnings("static-access")
    public static void main(String[] args) {
        
        try {
            //初始化客戶端
            Socket socket = new Socket("127.0.0.1", 4404);
            BufferedReader readline = new BufferedReader(new InputStreamReader(System.in));
            //獲取輸出列印流
            PrintWriter socketOut = new PrintWriter(socket.getOutputStream());
            String outTemp = null;
            System.out.println("開始準備向伺服器端發起請求---\n自己:");
            // 已啟動連線socket伺服器,準備實時接收來自其他客戶端的訊息
            GetMess getMess = new GetMess(socket);
            getMess.start();
            // 通過控制檯傳送訊息給其他客戶端,以“bye”為結束語
            while ((outTemp = readline.readLine()) != null) {
                //傳送資訊
                socketOut.println(outTemp);
                socketOut.flush();
                if("bye".equals(outTemp)){
                    break;
                }
            }
            getMess.currentThread().interrupt();
            //依次關閉各種流
            readline.close();
            socketOut.close();
            socket.close();
        } catch (UnknownHostException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        } catch (IOException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }

        
    }
}

GetMess(多執行緒處理獲取其他客戶端的訊息並且展示)

package com.orange;

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

public class GetMess extends Thread {

    Socket socket = null;
    
    public GetMess(Socket socket) {
        super();
        this.socket = socket;
    }

    @Override
    public void run() {
        // TODO Auto-generated method stub
        BufferedReader socketIn = null;
        try {
            InputStream is = socket.getInputStream();
            InputStreamReader isr = new InputStreamReader(is);
            socketIn = new BufferedReader(isr);
            String inTemp = null;
            while(true){
                inTemp = socketIn.readLine();
                if(inTemp != null && !"bye".equals(inTemp) ){
                    System.out.println("好友:\n"+inTemp);
                }else{
                    System.out.println("好友:\n已下線,關閉當前回話");
                    break;
                } 
            }
        } catch (IOException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }finally {
            try {
                Thread.currentThread().interrupt();
                socketIn.close();
            } catch (IOException e) {
                // TODO Auto-generated catch block
                e.printStackTrace();
            }
            
        }
        
    }

    
    
    
}

ok,簡單的sockte服務與客戶端完成了

先啟動server

在分別啟動client

輸入結束bye,關閉當前會話