解決Java中socket使用getInputStream()阻塞問(wèn)題

更新時(shí)間：2021年12月13日 15:35:03 作者：godelgnis

這篇文章主要介紹了解決Java中socket使用getInputStream()阻塞問(wèn)題，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教

socket使用getInputStream()阻塞

今天用socket進(jìn)行編程練習(xí)時(shí)，發(fā)現(xiàn)程序到了getInputStream(）這里就進(jìn)行不下去了

Socket socket = new Socket("127.0.0.1", 800);
ObjectInputStream reader = new ObjectInputStream(socket.getInputStream());
System.out.println("a");
ObjectOutputStream writer = new ObjectOutputStream(socket.getOutputStream());

就這樣的一個(gè)測(cè)試代碼，a不會(huì)打印出來(lái)

后來(lái)發(fā)現(xiàn)是getInputStream(）會(huì)一直阻塞在那里阻塞

我把兩行代碼調(diào)了一下就好了，還不太清楚原因，先記下來(lái)

Socket socket = new Socket("127.0.0.1", 800);
ObjectOutputStream writer = new ObjectOutputStream(socket.getOutputStream());
System.out.println("a");
ObjectInputStream reader = new ObjectInputStream(socket.getInputStream());

用線程解決Socket的getInputStream阻塞

1.背景

在Socket通信中，當(dāng)我們希望傳輸對(duì)象時(shí)，往往會(huì)用到輸入/輸出對(duì)象流。

ObjectInputStream in=new ObjectInputStream(socket.getInputStream());
ObjectOutputStream out=new ObjectOutputStream(socket.getOutputStream());

2.問(wèn)題

當(dāng)程序調(diào)用socket.getInputStream()程序被被卡住。

3.原因

socket.getInputStream()方法會(huì)導(dǎo)致程序阻塞，直到inputStream收到對(duì)方發(fā)過(guò)來(lái)的報(bào)文消息，程序才會(huì)繼續(xù)往下執(zhí)行。

public ObjectInputStream(InputStream in) throws IOException的官方API顯示：

Creates an ObjectInputStream that reads from the specified InputStream. A serialization stream header is read from the stream and verified. This constructor will block until the corresponding ObjectOutputStream has written and flushed the header. [1]

4.解決辦法

用線程的方式處理輸入流。以下為示例代碼：

//===============客戶端代碼 SocketClient.java=====================

import java.io.IOException;
import java.io.ObjectInputStream;
import java.io.ObjectOutputStream;
import java.net.Socket;
import java.net.UnknownHostException; 
 
public class SocketClient {
	private Socket socket;
	private ObjectOutputStream out;
	private ObjectInputStream in;	
	public SocketClient(){
		try {
			socket=new Socket("localhost",8081);
			out=new ObjectOutputStream(socket.getOutputStream());
			ReadThread readThread=new ReadThread();
			readThread.start();
		} catch (UnknownHostException e) {
			e.printStackTrace();
		} catch (IOException e) {
			e.printStackTrace();
		}
	}
	
	public void sendMessage(String msg){
		System.out.println("send message:"+msg);
		try {
			out.writeObject(msg);
			out.flush();
		} catch (IOException e) {
			e.printStackTrace();
		}
	}
	
	class ReadThread extends Thread{
		boolean runFlag=true;
		public void run(){
			try {
				in=new ObjectInputStream(socket.getInputStream());
			} catch (IOException e1) {
				e1.printStackTrace();
			}
			while(runFlag){
				if(socket.isClosed()){
					return;
				}
				try {
					Object obj=in.readObject();
					if(obj instanceof String){
						System.out.println("Client recive:"+obj);
					}
				} 
				catch (IOException e) {
					e.printStackTrace();
				} 
				catch (ClassNotFoundException e) {
					e.printStackTrace();
				}
			}
		}
		
		public void exit(){
			runFlag=false;
		}
	}
	
	public static void main(String[] args) {
		SocketClient socketClient=new SocketClient();
		System.out.println("build socketClient");
		try {
			Thread.sleep(1000);
		} catch (InterruptedException e) {
			e.printStackTrace();
		}
		socketClient.sendMessage("Hello first.");
		try {
			Thread.sleep(1000);
		} catch (InterruptedException e) {
			e.printStackTrace();
		}
		socketClient.sendMessage("Hello second.");
	} 
}

//============服務(wù)器端代碼 SocketService.java===========

import java.io.IOException;
import java.io.ObjectInputStream;
import java.io.ObjectOutputStream;
import java.net.ServerSocket;
import java.net.Socket;
import java.net.SocketException;
import java.util.Date;
 
public class SocketService {
	ServerSocket serverSocket;	
	public SocketService(){
		try {
			serverSocket=new ServerSocket(8081);
			while(true){
				Socket socket=serverSocket.accept();
				SocketServiceThread sst=new SocketServiceThread(socket);
				sst.start();
			}
		} catch (IOException e) {
			e.printStackTrace();
		}
	}
	
	class SocketServiceThread extends Thread{
		Socket socket;
		ObjectInputStream in;
		ObjectOutputStream out;
		boolean runFlag=true;
		public SocketServiceThread(Socket socket){
			if(null==socket){
				runFlag=false;
				return;
			}
			this.socket=socket;
			try {
				out=new ObjectOutputStream(socket.getOutputStream());
			} catch (IOException e) {
				e.printStackTrace();
			}
		}
		
		public void run(){
			if(null==socket){
				System.out.println("socket is null");
				return;
			}
			try {
				in=new ObjectInputStream(socket.getInputStream());
				while(runFlag){
					if(socket.isClosed()){
						System.out.println("socket is closed");
						return;
					}
					try {
						String obj=(String)in.readObject();
						if(obj instanceof String){
							System.out.println("Server recive:"+obj);
							Date date=new Date();
							out.writeObject("["+date+"]"+obj);
							out.flush();
						}
						else{
							System.out.println("Server recive:"+obj);
						}
					} 
					catch (ClassNotFoundException e) {
						e.printStackTrace();
					}
					catch (SocketException e){
						e.printStackTrace();
						return;
					}
					catch (IOException e){
						e.printStackTrace();
					}
				}
			} catch (IOException e1) {
				e1.printStackTrace();
				return;
			} catch (Exception e){
				return;
			}
		}
		
		public void exit(){
			runFlag=false;
		}
	}
	
	public static void main(String[] args) {
		System.out.println("===============start service===============");
		new SocketService();
	} 
}