Java实现文件上传功能
文件上传功能是互联网中非常常见的应用,例如修改用户头像时需要上传照片,把文件上传到云端网盘等等。
其网络传输协议可以使用TCP协议,在Java中可以通过Socket
和ServerSocket
实现。其本质就是把本地客户端的文件复制到服务器端,对文件复制还不熟悉的小伙伴可以再回顾一下我的这篇博客:用Java实现文件复制
下面我将以上传图片文件photo.jpg
为例,总结文件上传的几个步骤。
客户端
目的:读取本地文件,上传到服务器,读取服务器回写的数据
实现步骤:
- 创建一个本地字节输入流
FileInputStream
对象,构造方法中绑定要读取的数据源 - 创建一个客户端
Socket
对象,构造方法中绑定服务器的IP地址和端口号 - 使用
Socket
中的方法getOutputStream
,获取网络字节输出流OutputStream
对象 - 使用本地字节输入流
FileInputStream
对象中的方法read
,读取本地文件 - 使用网络字节输出流
OutputStream
对象中的方法write
,把读取到的文件上传到服务器 - 使用
Socket
中的方法getInputStream
,获取网络字节输入流InputStream
对象 - 使用网络字节输入流
InputStream
对象中的方法read
读取服务回写的数据 - 释放资源(
FileInputStream
,Socket
)
注:本例中我使用的IP地址为本地地址127.0.0.1,端口号为8000,这些可以根据需要修改即可。
代码实现如下:
public class FileUploadClient {
public static void main(String[] args) throws IOException {
//1.创建一个本地字节输入流FileInputStream对象,构造方法中绑定要读取的数据源
FileInputStream fis = new FileInputStream("C:\\photo.jpg");
//2.创建一个客户端Socket对象,构造方法中绑定服务器的IP地址和端口号
Socket socket = new Socket("127.0.0.1", 8000);
//3.使用Socket中的方法getOutputStream,获取网络字节输出流OutputStream对象
OutputStream os = socket.getOutputStream();
//4.使用本地字节输入流FileInputStream对象中的方法read,读取本地文件
byte[] bytes = new byte[1024];
int len = 0;
while((len = fis.read(bytes)) != -1){
//5.使用网络字节输出流OutputStream对象中的方法write,把读取到的文件上传到服务器
os.write(bytes, 0, len);
}
//关闭此套接字的输出流
socket.shutdownOutput();
//6.使用Socket中的方法getInputStream,获取网络字节输入流InputStream对象
InputStream is = socket.getInputStream();
//7.使用网络字节输入流InputStream对象中的方法read读取服务回写的数据
while((len = is.read(bytes)) != -1){
System.out.println(new String(bytes, 0, len));
}
//8.释放资源(FileInputStream,Socket)
fis.close();
socket.close();
}
}
服务器端
目的:读取客户端上传的文件,保存到服务器的硬盘,给客户端回写”上传成功”
注:本例中文件上传的目的地:服务器的硬盘 d:\upload文件夹下
实现步骤:
- 创建一个服务器
ServerSocket
对象,和系统要指定的端口号 - 使用
ServerSocket
对象中的方法accep
t,获取到请求的客户端Socket对象 - 使用
Socket
对象中的方法getInputStream
,获取到网络字节输入流InputStream
对象 - 判断
d:\\upload
文件夹是否存在,不存在则创建 - 创建一个本地字节输出流
FileOutputStream
对象,构造方法中绑定要输出的目的地 - 使用网络字节输入流
InputStream
对象中的方法read
,读取客户端上传的文件 - 使用本地字节输出流
FileOutputStream
对象中的方法write
,把读取到的文件保存到服务器的硬盘上 - 使用
Socket
对象中的方法getOutputStream
,获取到网络字节输出流OutputStream
对象 - 使用网络字节输出流
OutputStream
对象中的方法write
,给客户端回写”上传成功” - 释放资源(
FileOutputStream
,Socket
)
代码实现如下:
public class FileUploadServer {
public static void main(String[] args) throws IOException {
//1.创建一个服务器ServerSocket对象,和系统要指定的端口号
ServerSocket server = new ServerSocket(8000);
//让服务器一直处于监听状态
while (true){
//2.使用ServerSocket对象中的方法accept,获取到请求的客户端Socket对象
Socket socket = server.accept();
//使用多线程技术,提高程序的效率
new Thread(new Runnable() {
@Override
public void run() {
try{
//3.使用Socket对象中的方法getInputStream,获取到网络字节输入流InputStream对象
InputStream is = socket.getInputStream();
//4.判断d:\\upload文件夹是否存在,不存在则创建
File file = new File("D:\\upload");
if(!file.exists()){
file.mkdir();
}
//为文件随机创建一个文件名,确保不重复
String fileName = System.currentTimeMillis() + new Random().nextInt(9000) + ".txt";
//5.创建一个本地字节输出流FileOutputStream对象,构造方法中绑定要输出的目的地
FileOutputStream fos = new FileOutputStream(file + File.separator + fileName);
//6.使用网络字节输入流InputStream对象中的方法read,读取客户端上传的文件
byte[] bytes = new byte[1024];
int len = 0;
while((len = is.read(bytes)) != -1){
//7.使用本地字节输出流FileOutputStream对象中的方法write,把读取到的文件保存到服务器的硬盘上
fos.write(bytes, 0, len);
}
//8.使用Socket对象中的方法getOutputStream,获取到网络字节输出流OutputStream对象
OutputStream os = socket.getOutputStream();
//9.使用网络字节输出流OutputStream对象中的方法write,给客户端回写"上传成功"
os.write("upload success".getBytes());
//10.释放资源(FileOutputStream,Socket)
fos.close();
socket.close();
}catch (IOException e){
e.printStackTrace();
}
}
}).start();
}
}
}