Maison  >  Article  >  Java  >  Comment SpringBoot implémente la fonction de téléchargement de fichiers

Comment SpringBoot implémente la fonction de téléchargement de fichiers

WBOY
WBOYavant
2023-05-19 08:31:325124parcourir

1. Lisez le fichier dans la mémoire sous la forme d'un flux à la fois et envoyez-le au front-end via le flux de sortie de réponse

/**
 * @param path     想要下载的文件的路径
 * @param response
 * @功能描述 下载文件:
 */
@RequestMapping("/download")
public void download(String path, HttpServletResponse response) {
    try {
        // path是指想要下载的文件的路径
        File file = new File(path);
        log.info(file.getPath());
        // 获取文件名
        String filename = file.getName();
        // 获取文件后缀名
        String ext = filename.substring(filename.lastIndexOf(".") + 1).toLowerCase();
        log.info("文件后缀名:" + ext);

        // 将文件写入输入流
        FileInputStream fileInputStream = new FileInputStream(file);
        InputStream fis = new BufferedInputStream(fileInputStream);
        byte[] buffer = new byte[fis.available()];
        fis.read(buffer);
        fis.close();

        // 清空response
        response.reset();
        // 设置response的Header
        response.setCharacterEncoding("UTF-8");
        //Content-Disposition的作用:告知浏览器以何种方式显示响应返回的文件,用浏览器打开还是以附件的形式下载到本地保存
        //attachment表示以附件方式下载   inline表示在线打开   "Content-Disposition: inline; filename=文件名.mp3"
        // filename表示文件的默认名称,因为网络传输只支持URL编码的相关支付,因此需要将文件名URL编码后进行传输,前端收到后需要反编码才能获取到真正的名称
        response.addHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode(filename, "UTF-8"));
        // 告知浏览器文件的大小
        response.addHeader("Content-Length", "" + file.length());
        OutputStream outputStream = new BufferedOutputStream(response.getOutputStream());
        response.setContentType("application/octet-stream");
        outputStream.write(buffer);
        outputStream.flush();
    } catch (IOException ex) {
        ex.printStackTrace();
    }
}

2 Écrivez les données du flux d'entrée dans le flux de sortie de réponse dans. une boucle au lieu de tout d'un coup Lire dans la mémoire et sortir vers le front-end via le flux de sortie de réponse

/**
* @param path     指想要下载的文件的路径
* @param response
* @功能描述 下载文件:将输入流中的数据循环写入到响应输出流中,而不是一次性读取到内存
*/
@RequestMapping("/downloadLocal")
public void downloadLocal(String path, HttpServletResponse response) throws IOException {
   // 读到流中
   InputStream inputStream = new FileInputStream(path);// 文件的存放路径
   response.reset();
   response.setContentType("application/octet-stream");
   String filename = new File(path).getName();
   response.addHeader("Content-Disposition", "attachment; filename=" + URLEncoder.encode(filename, "UTF-8"));
   ServletOutputStream outputStream = response.getOutputStream();
   byte[] b = new byte[1024];
   int len;
   //从输入流中读取一定数量的字节,并将其存储在缓冲区字节数组中,读到末尾返回-1
   while ((len = inputStream.read(b)) > 0) {
       outputStream.write(b, 0, len);
   }
   inputStream.close();
}

3 Téléchargez le fichier réseau sur le

/**
* @param path       下载后的文件路径和名称
* @param netAddress 文件所在网络地址
* @功能描述 网络文件下载到服务器本地
*/
@RequestMapping("/netDownloadLocal")
public void downloadNet(String netAddress, String path) throws IOException {
	  URL url = new URL(netAddress);
	  URLConnection conn = url.openConnection();
	  InputStream inputStream = conn.getInputStream();
	  FileOutputStream fileOutputStream = new FileOutputStream(path);
	
	  int bytesum = 0;
	  int byteread;
	  byte[] buffer = new byte[1024];
	  while ((byteread = inputStream.read(buffer)) != -1) {
	      bytesum += byteread;
	      System.out.println(bytesum);
	      fileOutputStream.write(buffer, 0, byteread);
	  }
	  fileOutputStream.close();
}

4 local. le serveur, il sera traité par le serveur puis répondu au front-end

/**
 * @param netAddress
 * @param filename
 * @param isOnLine
 * @param response
 * @功能描述 网络文件获取到服务器后,经服务器处理后响应给前端
 */
@RequestMapping("/netDownLoadNet")
public void netDownLoadNet(String netAddress, String filename, boolean isOnLine, HttpServletResponse response) throws Exception {

    URL url = new URL(netAddress);
    URLConnection conn = url.openConnection();
    InputStream inputStream = conn.getInputStream();

    response.reset();
    response.setContentType(conn.getContentType());
    if (isOnLine) {
        // 在线打开方式 文件名应该编码成UTF-8
        response.setHeader("Content-Disposition", "inline; filename=" + URLEncoder.encode(filename, "UTF-8"));
    } else {
        //纯下载方式 文件名应该编码成UTF-8
        response.setHeader("Content-Disposition", "attachment; filename=" + URLEncoder.encode(filename, "UTF-8"));
    }

    byte[] buffer = new byte[1024];
    int len;
    OutputStream outputStream = response.getOutputStream();
    while ((len = inputStream.read(buffer)) > 0) {
        outputStream.write(buffer, 0, len);
    }
    inputStream.close();
}

5. Exceptions et problèmes courants

(1) L'objet de réponse n'a pas besoin d'être renvoyé via return

Comment SpringBoot implémente la fonction de téléchargement de fichiers

Raison : L'objet de réponse n'a pas besoin d'être renvoyé en tant que valeur de retour de méthode. Il a déjà commencé à produire lorsque la méthode est exécutée, et son Impossible de coopérer avec @RestController, revient au front-end au format JSON

Comment SpringBoot implémente la fonction de téléchargement de fichiers

Comment SpringBoot implémente la fonction de téléchargement de fichiers

Solution : Supprimez l'instruction de retour

(2) Le nom de fichier renvoyé au front-end doit être codé en URL

Raison : La transmission réseau ne peut transmettre que des dizaines de caractères spécifiques, des caractères chinois, des caractères spéciaux, etc. doivent être convertis en caractères spécifiques via Base64 et d'autres encodages pour une transmission sans caractères tronqués

URLEncoder.encode(fileName, "UTF-8")

(3) Le flux IO doit être appris

Ce qui suit est la phrase réécrite : La fonction read() est utilisée pour lire l'octet de données suivant du flux d'entrée et renvoyer la valeur entière de l'octet, avec une valeur comprise entre 0 et 255. Si aucun octet n'est disponible car la fin du flux a été atteinte, -1 est renvoyé. Cette méthode se bloque jusqu'à ce que les données d'entrée soient disponibles, que la fin du flux soit détectée ou qu'une exception soit levée.

2 : read(byte[] b) : lit un certain nombre d'octets du flux d'entrée et le stocke dans le tableau tampon b. Renvoie le nombre réel d'octets lus sous forme d'entier. Cette méthode se bloque jusqu'à ce que les données d'entrée soient disponibles, que la fin du fichier soit détectée ou qu'une exception soit levée. Si la longueur de b est 0, aucun octet n'est lu et 0 est renvoyé sinon, une tentative est effectuée pour lire au moins un octet. Si aucun octet n'est disponible car le flux est à la fin du fichier, la valeur de retour est -1

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!

Déclaration:
Cet article est reproduit dans:. en cas de violation, veuillez contacter admin@php.cn Supprimer