【知识点】利用TCP实现文件传输

By | 7月 3, 2022

最近在与其他系统进行交互需要使用TCP进行文件传输,之前一般都是直接SFTP文件传输,所以对此进行了研究。如何利用TCP流传输上传文件我们先了解下Socket。

为什么要了解Socket呢?因为利用TCP实现文件传输的原理是:首先客户端和服务端通过TCP三次握手建立连接,然后利用Socket实现文件的传输。

Socket(套接字)

Socket是对TCP/IP协议的封装,Socket本身并不是协议,而是一个调用接口(API),通过Socket,我们才能使用TCP/IP协议。与HTTP相比较,socket连接是长连接,理论上客户端和服务端一旦建立连接,则不会主动断掉

Socket的工作原理

通过互联网进行通信,至少需要一对套接字,其中一个运行于客户端,我们称之为 Client Socket,另一个运行于服务器端,我们称之为 Server Socket [3] 。根据连接启动的方式以及本地套接字要连接的目标,套接字之间的连接过程可以分为三个步骤 

(1)服务器监听。

(2)客户端请求。

(3)连接确认 

TCP传输文件实现代码

说明:该种文件传输方式需要服务端和客户端均有相应代码处理,服务端接收数据流写入本地,客户端将文件转换为流传输给服务端。

服务端代码:

package com.onezero.web.service.file;

import java.io.*;
import java.net.ServerSocket;
import java.net.Socket;

public class FileServer {

public static void main(String[] args) {
ServerSocket serverSocket = null;
Socket accept = null;
InputStream is = null;
FileOutputStream fos = null;
OutputStream outputStream = null;
try {
//1、创建一个连接,指定端口号
serverSocket = new ServerSocket(10086);
System.out.println("服务端启动,等待客户端建立连接");
//2. 等待着客户端的链接
accept = serverSocket.accept();
//3.读取客户端的消息输入流
is = accept.getInputStream();

//4.文件输出
fos = new FileOutputStream(new File("2.jpg"));
byte[] bytes = new byte[1024];
int len;
while ((len = is.read(bytes)) != -1) {
fos.write(bytes, 0, len);
}
//5.通知客户端接受完毕
outputStream = accept.getOutputStream();
outputStream.write("服务端接收完毕,可关闭连接!".getBytes());
System.out.println("服务端处理完毕!");
} catch (IOException e) {
e.printStackTrace();
} finally {
//6.关闭资源
if (outputStream != null){
try {
outputStream.close();
} catch (IOException e) {
e.printStackTrace();
}
}
if (fos != null) {
try {
fos.close();
} catch (IOException e) {
e.printStackTrace();
}
}
if (is != null) {
try {
//关闭输入流
is.close();
} catch (IOException e) {
e.printStackTrace();
}
}
if (accept != null) {
try {
accept.close();
} catch (IOException e) {
e.printStackTrace();
}
}
if (serverSocket != null) {
try {
serverSocket.close();
} catch (IOException e) {
e.printStackTrace();
}
}

}
}
}

客户端

package com.onezero.web.service.file;

import java.io.*;
import java.net.Socket;

public class FileClient {

public static void main(String[] args) {
Socket socket = null;
OutputStream os = null;
FileInputStream fis = null;
InputStream fis2 = null;
ByteArrayOutputStream baos = null;
try {
//1.获取服务器的地址,端口号
socket = new Socket("127.0.0.1", 10086);
System.out.println("客户端启动,与服务端建立连接");
// 2.创建socket连接获取网络字节输出流
os = socket.getOutputStream();
// 3、读取需要上传的文件
fis = new FileInputStream(new File("1.jpg"));

//4. 将读取的文件写出,使用字节流
byte[] bytes = new byte[1024];
int len;
while ((len = fis.read(bytes)) != -1) {
os.write(bytes, 0, len);
}
//5.通知服务器请求已完成,否则会服务端一直等待
socket.shutdownOutput();
//6.通知服务器接受完毕,断开连接
fis2 = socket.getInputStream();
baos = new ByteArrayOutputStream();
byte[] bytes2 = new byte[1024];
int len2;
while ((len2=fis2.read(bytes2))!= -1){
baos.write(bytes2,0,len2);
}
System.out.println(baos.toString());

} catch (IOException e) {
e.printStackTrace();
} finally {
// 4.关闭资源
if (baos != null){

}
if (fis != null) {
try {
fis.close();
} catch (IOException e) {
e.printStackTrace();
}
}
if (os != null) {
try {
os.close();
} catch (IOException e) {
e.printStackTrace();
}
}
if (socket != null) {
try {
socket.close();
} catch (IOException e) {
e.printStackTrace();
}
}
}
}
}

运行结果(注意先启动服务端再启动客户端)

服务端输出

客户端输出

图片(客户端文件1.jpg–>服务端文件2.jpg)

END

发表评论

您的电子邮箱地址不会被公开。