Java基于BIO实现文件上传功能

Z时代
2024-01-10
分类：IT

本文实例为大家分享了Java基于BIO实现文件上传功能的具体代码，供大家参考，具体内容如下

客户端


package com.qst.file;
import java.io.BufferedInputStream;
import java.io.BufferedOutputStream;
import java.io.DataOutputStream;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.FileWriter;
import java.io.IOException;
import java.io.OutputStream;
import java.net.Socket;
public class Client {
 public static void main(String[] args) {
  try {
   // 获取输入流
   FileInputStream stream = new FileInputStream("E:\\Java后端开发之路\\IO\\讲义\\a.txt");
   // 请求与服务端进行建立连接
   Socket socket = new Socket("127.0.0.1", 9999);
   // 字节输出流包装成一个数据输出流
   DataOutputStream outputStream = new DataOutputStream(socket.getOutputStream());
   outputStream.writeUTF(".txt");
   int len;
   byte[] b = new byte[1024];
   while ((len = stream.read(b)) > 0) {
    outputStream.write(b, 0, len);
   }
   outputStream.flush();
   // 通知服务端这边的数据发送完毕了
   socket.shutdownOutput();
  } catch (Exception e) {
   // TODO: handle exception
  }
 }
}

服务端


package com.qst.file;
import java.io.IOException;
import java.net.ServerSocket;
import java.net.Socket;
public class Server {
 public static void main(String[] args) throws IOException {
  // 建立服务器
  ServerSocket socket = new ServerSocket(9999);
  Socket accept = socket.accept();
  // 交给一个独立的线程来处理与这个客户端的文件通信需求。
  new ServerThread(accept).start();
 }
}

服务端线程处理


package com.qst.file;
import java.io.DataInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.net.Socket;
import java.util.UUID;
public class ServerThread extends Thread{
 private Socket socket;
 public ServerThread(Socket accept) {
  this.socket = accept;
 }
 @Override
 public void run() {
  try {
   // 得到一个数据输入流读取客户端发送过来的数据
   DataInputStream stream = new DataInputStream(socket.getInputStream());
    // 读取客户端发送过来的文件类型
            String suffix = stream.readUTF();
            System.out.println("服务端已经成功接收到了文件类型：" + suffix);
            // 定义一个字节输出管道负责把客户端发来的文件数据写出去
   FileOutputStream stream2 = new FileOutputStream("E:\\Java后端开发之路\\IO\\讲义\\b.txt"+UUID.randomUUID().toString()+suffix);
   // 从数据输入流中读取文件数据，写出到字节输出流中去
   byte []  b= new byte[1024];
   int len;
   while((len = stream.read(b)) > 0) {
    stream2.write(b, 0, len);
   }
   System.out.println("操作完毕");
   stream2.close();
   stream.close();
  } catch (IOException e) {
   // TODO Auto-generated catch block
   e.printStackTrace();
  }
 }
}