Java?Socket實現(xiàn)文件發(fā)送和接收功能以及遇到的Bug問題
Java Socket實現(xiàn)文件發(fā)送和接收功能
在Java中,如何用Socket實現(xiàn)文件的發(fā)送和接收功能?
我的第一版代碼
文件發(fā)送:
public void sendFile(String filePath) {//過長、過多的密文信息直接發(fā)送文件
File file = new File(filePath);
try {
DataOutputStream d_out = new DataOutputStream(socket.getOutputStream());
FileInputStream f_in = new FileInputStream(file);
int all = 0;
byte[] buffer = new byte[4096];
int read = 0;
while ((read = (f_in.read(buffer))) > 0) {
d_out.write(buffer, 0, read);
all += read;
}
System.out.println("Send file length: "+all);
d_out.flush();
f_in.close();
d_out.close();//注意這一行
} catch (IOException e) {
e.printStackTrace();
}
}文件接收:
public void receiveFile(String filePath) {//接收文件
try {
DataOutputStream dosOutputStream = new DataOutputStream(new FileOutputStream(filePath));
byte[] buf = new byte[4096];
int len = 0;
System.out.println("開始接收文件!");
d_in = new DataInputStream(sock.getInputStream());
while((len = d_in.read(buf)) != -1) {
dosOutputStream.write(buf, 0, len);
}
dosOutputStream.flush();
System.out.println("文件接收結(jié)束!");
//d_in.close();
dosOutputStream.close();
} catch (IOException e) {
e.printStackTrace();
}
}上面的寫法的確實現(xiàn)了文件發(fā)送和接收的功能(Socket初始化這里沒有給出,大家可以自行補充這部分,是可以運行的),但是這個實現(xiàn)方式存在一個很致命的問題,就是只能完成一次文件的發(fā)送和接收。
在此之后如果你想再調(diào)用文件的發(fā)送和接收方法,就會遇到如下"Socket is closed"這個問題:

為什么會遇到這一問題
首先肯定是因為Socket被我關(guān)閉了,但我并沒有寫"socket.close();"這樣的代碼呀,為什么還是被關(guān)閉了呢?
我們把注意力放到上面提到的文件發(fā)送的代碼上,注意這一行:
d_out.close();
這里原本的目的是把DataOutputStream給關(guān)閉掉,結(jié)束我們的文件發(fā)送輸出流。
但是當我們關(guān)閉DataOutputStream時,Socket也會隨之關(guān)閉,這便有了后面想再次執(zhí)行sendFile方法時,出現(xiàn)的"Socket is closed"問題。
所以為了能連續(xù)多次地發(fā)送、接收不同的文件,這一行代碼肯定是不能要了。
把這行代碼注釋掉,修改后的代碼為
文件發(fā)送:
public void sendFile(String filePath) {//過長、過多的密文信息直接發(fā)送文件
File file = new File(filePath);
try {
DataOutputStream d_out = new DataOutputStream(socket.getOutputStream());
FileInputStream f_in = new FileInputStream(file);
int all = 0;
byte[] buffer = new byte[4096];
int read = 0;
while ((read = (f_in.read(buffer))) > 0) {
d_out.write(buffer, 0, read);
all += read;
}
System.out.println("Send file length: "+all);
d_out.flush();
f_in.close();
} catch (IOException e) {
e.printStackTrace();
}
}文件接收:
public void receiveFile(String filePath) {//接收文件
try {
DataOutputStream dosOutputStream = new DataOutputStream(new FileOutputStream(filePath));
byte[] buf = new byte[4096];
int len = 0;
System.out.println("開始接收文件!");
d_in = new DataInputStream(sock.getInputStream());
while((len = d_in.read(buf)) != -1) {
dosOutputStream.write(buf, 0, len);
}
dosOutputStream.flush();
System.out.println("文件接收結(jié)束!");
//d_in.close();
dosOutputStream.close();
} catch (IOException e) {
e.printStackTrace();
}
}這樣修改后又遇到了新的問題
那就是在接收文件時,發(fā)生了阻塞的現(xiàn)象:

控制臺就一直卡在“開始接收文件!”這行輸出這里,顯然是因為文件接收的while循環(huán)被卡住了,但剛剛明明沒有這個問題啊,為什么注釋掉“sendFile”中的d_out.close();會使得“receiveFile”的while循環(huán)被卡住呢?
while((len = d_in.read(buf)) != -1) {
dosOutputStream.write(buf, 0, len);
}這篇文章給出了比較好的解答:http://www.fzitv.net/program/325734b3l.htm
原因如下
“只要客戶端的DataOutputStream不close掉,那么服務端的DataInputStream read就永遠不等于-1。即使文件的數(shù)據(jù)已經(jīng)傳完了,DataInputStream依舊會等著客戶端DataOutputStream再傳數(shù)據(jù)過來。最后只能通過判斷文件的的大小來確認文件是否已經(jīng)傳輸完成。”
這便是為什么在d_out.close();沒有被注釋前是沒有這個Bug的:因為當我們將DataOutputStream close掉時,接收方的while循環(huán)也就結(jié)束了。
所以我們現(xiàn)在可以采用len的實際長度來判斷是否已經(jīng)傳輸完成,修改后的代碼如下:
while((len = d_in.read(buf)) != -1) {
dosOutputStream.write(buf, 0, len);
if(len < buf.length) break;
}即:判斷buf是否被填滿,沒有填滿(len<buf.length)則代表已經(jīng)接收完畢
經(jīng)過測試,問題解決
最終代碼如下:
文件發(fā)送:
?
public void sendFile(String filePath) {//發(fā)送文件
File file = new File(filePath);
try {
DataOutputStream d_out = new DataOutputStream(socket.getOutputStream());
FileInputStream f_in = new FileInputStream(file);
int all = 0;
byte[] buffer = new byte[4096];
int read = 0;
while ((read = (f_in.read(buffer))) > 0) {
d_out.write(buffer, 0, read);
all += read;
}
System.out.println("Send file length: "+all);
d_out.flush();
f_in.close();
} catch (IOException e) {
e.printStackTrace();
}
}
?文件接收:
public void receiveFile(String filePath) {//接收文件
try {
DataOutputStream dosOutputStream = new DataOutputStream(new FileOutputStream(filePath));
byte[] buf = new byte[4096];
int len = 0;
System.out.println("開始接收文件!");
d_in = new DataInputStream(sock.getInputStream());
while((len = d_in.read(buf)) != -1) {
dosOutputStream.write(buf, 0, len);
if(len < buf.length) break;
}
dosOutputStream.flush();
System.out.println("文件接收結(jié)束!");
dosOutputStream.close();
} catch (IOException e) {
e.printStackTrace();
}
}總結(jié)
以上為個人經(jīng)驗,希望能給大家一個參考,也希望大家多多支持腳本之家。
相關(guān)文章
詳談HashMap和ConcurrentHashMap的區(qū)別(HashMap的底層源碼)
下面小編就為大家?guī)硪黄斦凥ashMap和ConcurrentHashMap的區(qū)別(HashMap的底層源碼)。小編覺得挺不錯的,現(xiàn)在就分享給大家,也給大家做個參考。一起跟隨小編過來看看吧2017-08-08
Java SpringMVC實現(xiàn)國際化整合案例分析(i18n)
本篇文章主要介紹了Java SpringMVC實現(xiàn)國際化整合案例分析(i18n),具有一定的參考價值,感興趣的小伙伴們可以參考一下2017-05-05
SpringBoot配置數(shù)據(jù)庫密碼加密的方法
由于系統(tǒng)安全的考慮,配置文件中不能出現(xiàn)明文密碼的問題,本文就給大家詳細介紹下springboot配置數(shù)據(jù)庫密碼加密的方法,下面話不多說了,來一起看看詳細的介紹吧,需要的朋友可以參考下2023-08-08

