Java實現文件上傳與文件下載的示例代碼

更新時間：2023年07月21日 11:53:05 作者：Xiao5xiao122

在開發(fā)中項目難免會遇到文件上傳和下載的情況，這篇文章主要為大家詳細介紹了Java中實現文件上傳與文件下載的示例代碼，希望對大家有所幫助

文件上傳與文件下載

在開發(fā)中項目難免會遇到文件上傳和下載的情況，如果公司有部署文件服務器是再好不過啦，直接調用文件服務器上傳和下載的接口，保留返回的文件id即可。
但是如果公司沒有文件服務器就比較苦逼了，得手寫文件上傳和下載的接口把文件存儲在服務器的某個位置，以便使用。

直接把代碼貼出來以供記錄筆記。

先controller層

@ApiOperation(value = "文件上傳", consumes = MediaType.MULTIPART_FORM_DATA_VALUE)
    @PostMapping("/file/upload")
    @Transactional(rollbackFor = Exception.class)
    public ApiResult<String> upload(@RequestPart("file") MultipartFile file) {
        try {
            TFileUploadDet fileUpload = FileManageUtil.upload(file);
            return  ApiResult.data(fileUpload.getId());
        } catch (Exception e) {
            log.error(e.getMessage(),e);
            return ApiResult.fail(e.getMessage());
        }
    }

然后服務層

public static TFileUploadDet upload(MultipartFile file) {
        Assert.notNull(file, "文件不能為空");
		//獲取上傳文件名稱
        String originalFilename = file.getOriginalFilename();
        //獲取文件后綴名
        String suffix = originalFilename.substring(originalFilename.lastIndexOf('.'));
        String extName = suffix.substring(1);
        //校驗文件格式是否為規(guī)定
        Assert.isTrue(WjlxEnum.WJLX999.getExtNameList().contains(extName),"文件類型僅支持jpg/mp4/pdf,請確認!");
        String id = SecureUtil.md5(originalFilename + IdUtil.getSnowflakeNextIdStr());
        String fileName = id + suffix;
        String year = DateUtil.format(new Date(), "yyyy");
        String today = DateUtil.format(new Date(), "yyyy-MM-dd");
        //獲取文件存儲路徑
        String fileId = File.separator + year + File.separator + today + File.separator + fileName;
        Path path = Paths.get(filePath(fileId));
        //創(chuàng)建文件夾（存在文件夾就直接存在該文件夾下沒有文件夾就先創(chuàng)建文件夾在保存）
        FileUtil.mkParentDirs(path);
        InputStream inputStream = null;
        OutputStream outputStream = null;
		//文件流上傳文件
        try {
            inputStream = file.getInputStream();
            outputStream = Files.newOutputStream(path);
            IoUtil.copy(inputStream, outputStream);
            //一下為自定義返回參數，視情況而定
            TFileUploadDet fileUpload = new TFileUploadDet();
            fileUpload.setId(id);
            fileUpload.setWjmc(originalFilename);
            fileUpload.setWjlj(path.toString());
            fileUpload.setCreateTime(new Date());
            return fileUpload;
        } catch (IOException e) {
            throw new RuntimeException("文件上傳失敗", e);
        } finally {
            IoUtil.close(outputStream);
            IoUtil.close(inputStream);
        }
    }
/**
     * 獲取完整的文件路徑
     *
     * @param fileId 文件唯一標識
     * @return 文件路徑
     */
    public static String filePath(String fileId) {
        return uploadFilePath + fileId;
    }

注：需要注意的是獲取完整文件路徑方法中的 uploadFilePath ：是文件將要保存在服務器中的位置路徑，可以在配置文件中配置好后引用。最后把返回的信息存在在表中使用。整個文件上傳就完成了。其中用到了hutool的第三方包我使用的版本為5.8.6

<dependency>
                <groupId>cn.hutool</groupId>
                <artifactId>hutool-core</artifactId>
                <version>5.8.6</version>
            </dependency>

文件下載

這個就沒有什么好說的了，比較簡單，就是流的操作而已。傳入的參數為文件上傳是保存的文件的id即可。

@ApiOperation(value = "通過文件id下載文件")
    @PostMapping("/downloadById")
    public void downloadFileById(HttpServletRequest request, HttpServletResponse response, @RequestBody @Valid IdDTO idDTO) {
        String id = idDTO.getId();
        Assert.isTrue(StringUtils.isNotEmpty(id), "主表id不能為空");
        TFile tFile = fileService.getById(id);
        Assert.notNull(tFile, "要下載的文件不存在");
        String fileName = tFile.getFileName();
        request.getSession();
        //獲取頁面輸出流
        try (ServletOutputStream out = response.getOutputStream()) {
            String path = tFile.getFileLink();
            File file = new File(path);
            byte[] bytes = FileUtils.readFileToByteArray(file);
            // 設置響應頭
            response.setHeader("Content-Disposition", "attachment;filename=\"" + fileName + "\"");
            response.addHeader("Pargam", "no-cache");
            response.addHeader("Cache-Control", "no-cache");
            out.write(bytes);
            out.flush();
        } catch (IOException e) {
            log.error(e.getMessage());
            throw new RuntimeException("下載文件出錯");
        }
    }

到此這篇關于Java實現文件上傳與文件下載的示例代碼的文章就介紹到這了,更多相關Java文件上傳下載內容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: