简单的文件操作_strutil.slash-优快云博客

本文链接：https://blog.youkuaiyun.com/qq_19342829/article/details/141018992

简单的文件操作

1. 需求

文件生成获取删除案例
想在项目跟目录生成一个文件，然后对文件进行操作，要在window以及linux服务都可以进行操作，这时候需要使用System.getProperty("user.dir")获取项目根目录，方便项目进行文件操作
文件存储数据以及读取
项目中有bpmn流程文件，需要进行记录解析与存储，当前存储到数据库，后续还要给前端进行使用。

2. 文件生成获取删除案例

整体代码部分相对简单，主要包含文件创建，文件内容获取，文件删除，以及获取resource目录下文件内容

项目整体结构如下图
在这里插入图片描述

2.1 controller主要代码实现

package cn.git.controller;

import cn.hutool.core.util.StrUtil;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import java.io.*;
import java.nio.charset.StandardCharsets;

/**
 * @description: 文件路径测试
 * @program: bank-credit-sy
 * @author: lixuchun
 * @create: 2023-08-08
 */
@RestController
@RequestMapping("/path")
public class PathTestController {

    /**
     * 生成一个文件到resource目录下
     *
     * @return
     */
    @GetMapping("/makeFile")
    public String makeFile() {
        // 在服务根目录创建一个文件
        String bathPath = System.getProperty("user.dir").concat(StrUtil.SLASH).concat("test.txt");
        File file = new File(bathPath);
        try {
            if (!file.exists()) {
                file.createNewFile();
            }
            FileWriter fileWriter = new FileWriter(file);
            fileWriter.write("123");
            fileWriter.flush();
            fileWriter.close();
        } catch (Exception e) {
            e.printStackTrace();
        }
        return file.getAbsolutePath();
    }

    /**
     * 读取文件内容
     *
     * @return
     * @throws IOException
     */
    @GetMapping("/getFile")
    public String getFile() throws IOException {
        String bathPath = System.getProperty("user.dir").concat(StrUtil.SLASH).concat("test.txt");
        File file = new File(bathPath);
        // 读取txt内容信息
        FileReader fileReader = new FileReader(file);
        char[] chars = new char[1024];
        int len = fileReader.read(chars);
        fileReader.close();
        return new String(chars, 0, len);
    }

    /**
     * 删除文件
     *
     * @return
     */
    @GetMapping("/dropFile")
    public String dropFile() {
        String bathPath = System.getProperty("user.dir").concat(StrUtil.SLASH).concat("test.txt");
        File file = new File(bathPath);
        if (file.exists()) {
            file.delete();
        }
        return "删除成功";
    }

    /**
     * 读取resource目录下的文件
     *
     * @return
     * @throws IOException
     */
    @GetMapping("/getResourceFile")
    public String getResourceFile() throws IOException {
        // 通过classLoader获取inputStream
        InputStream inputStream = this.getClass().getClassLoader().getResourceAsStream("name.txt");
        StringBuilder content = new StringBuilder();
        try {
            assert inputStream != null;
            try (BufferedReader reader = new BufferedReader(new InputStreamReader(inputStream, StandardCharsets.UTF_8))) {
                String line;
                while ((line = reader.readLine()) != null) {
                    content.append(line).append("\n");
                }
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
        return content.toString();
    }

}

2.2 ftp下载压缩文件并下载

此段简化代码为通过ftp下载文件到服务器，然后下载服务器文件到本地，简单记录一下代码过程

	private HttpServletResponse response;
	
	Ftp ftp = new Ftp(ftpServerIp, FTP_SERVER_PORT, ftpUsername, ftpPassword);
	String localFilePath = System.getProperty(USER_DIR).concat(StrUtil.SLASH).concat(fileName).concat(RAR);
	localFile = new File(localFilePath);
	ftp.download(DOWNLOAD_PATH.concat(remoteFileName), localFile);
	
	// 本地生成文件输入流
	if (localFile.exists() && localFile.length() > NewCoreProperties.FLAG_INT_0) {
	    // 获取文件inputStream
	    fileInputStream = new FileInputStream(localFilePath);
	
	    // 最终下载文件名称
	    fileName = fileName.concat(formatDate).concat(RAR);
	    response.setHeader(CONTENT_DISPOSITION, ATTACH_FLAG.concat(new String(fileName.getBytes(GBK), ISO8859_1)));
	    outputStream = response.getOutputStream();
	    byte[] buffer = new byte[BYTE_LENGTH];
	    int length;
	    while ((length = fileInputStream.read(buffer)) != NewCoreProperties.FLAG_MINUS_1) {
	        outputStream.write(buffer, NewCoreProperties.FLAG_INT_0, length);
	    }
	} else {
	    throw new ServiceException("当前下载数据为空!");
	}

2.3 测试

上传jar包，并且java -jar简单启动服务
在这里插入图片描述
生成文件
获取文件内容

在这里插入图片描述
查看生成文件以及文件内容

获取resource目录下文件信息

3. 文件存储数据以及读取

此处使用原有DragRule拖拽规则部分简化实现代码，不进行测试了，具体代码如下

文件上传数据库

    /**
     * 添加链信息
     *
     * @param dragRuleChain 规则链
     * @param uri
     * @param file
     */
    @Override
    public void addChain(String dragRuleChain, String uri, MultipartFile file) throws IOException {
        // 文件名称，以及base64编码字符串
        String fileName = file.getOriginalFilename();
        String fileBase64Str = Base64.getEncoder().encodeToString(file.getBytes());

        // 获取缓存信息,作为最终返回参数
        Object dragRuleChainObj = redisUtil.hget(CommonDragRuleConstant.DRAG_RULE_CHAIN_FLAG, uri);
        if (ObjectUtil.isNull(dragRuleChainObj)) {
            throw new ServiceException(StrUtil.format("当前缓存中没有uri[{}]规则链信息，请确认是否缓存中没有数据!", uri));
        }
        DragChainDTO cacheDragChainDTO = JSONObject.parseObject(dragRuleChainObj.toString(), DragChainDTO.class);

        // 校验当前规则链参数是否合法，满足所有规则入参出参
        commonDragRuleUtil.checkDragRuleWithParamLegal(cacheDragChainDTO);

        // 开始添加规则链信息
        TbSysDragChain tbSysDragChain = dragRuleConvert.dragChainDTOToTbSysDragChain(cacheDragChainDTO);
        tbSysDragChain.setDragRuleChain(dragRuleChain);
        tbSysDragChain.setChainFileName(fileName);
        tbSysDragChain.setChainFile(fileBase64Str);
        tbSysDragChainMapper.insert(tbSysDragChain);

        // 修改规则链缓存信息
        cacheDragChainDTO.setIfLoadDB(RuleConstant.NUM_1_STR);
        redisUtil.hset(CommonDragRuleConstant.DRAG_RULE_CHAIN_FLAG,
                cacheDragChainDTO.getUri(), JSONObject.toJSONString(cacheDragChainDTO));
    }

文件下载

    /**
     * 获取规则文件信息
     *
     * @param uri
     * @throws IOException
     */
    @Override
    public void downloadFile(String uri) throws IOException {
        QueryWrapper<TbSysDragChain> chainQueryWrapper = new QueryWrapper<>();
        chainQueryWrapper.lambda().eq(TbSysDragChain::getUri, uri);
        TbSysDragChain tbSysDragRuleChain = tbSysDragChainMapper.selectOne(chainQueryWrapper);
        if (ObjectUtil.isNull(tbSysDragRuleChain) || StrUtil.isBlank(tbSysDragRuleChain.getChainFile())) {
            throw new ServiceException(StrUtil.format("通过uri[{}]获取规则链文件信息为空!", uri));
        }

        // 获取文件信息
        String base64FileStr = tbSysDragRuleChain.getChainFile();
        String fileName = tbSysDragRuleChain.getChainFileName();
        byte[] decodedBytes = Base64.getDecoder().decode(base64FileStr);
        // 设置响应头信息，比如Content-Type和Content-Disposition
        response.setContentType(CommonDragRuleConstant.APPLICATION_XML);
        // 根据实际文件类型设置MIME类型，这里是XML文件
        response.setHeader(CommonDragRuleConstant.CONTENT_DISPOSITION, "attachment;filename=\"" + fileName + "\"");
        try (OutputStream out = response.getOutputStream()) {
            out.write(decodedBytes);
        } catch (IOException e) {
            // 处理IO异常
            String errorMessage = LogUtil.getStackTraceInfo(e);
            throw new ServiceException(StrUtil.format("下载文件异常，异常信息为[{}]", errorMessage));
        } finally {
            response.flushBuffer();
        }
    }

获取文件XML信息

    /**
     * 获取规则文件信息
     *
     * @param uri
     */
    @Override
    public String getXMLStr(String uri) {
        QueryWrapper<TbSysDragChain> chainQueryWrapper = new QueryWrapper<>();
        chainQueryWrapper.lambda().eq(TbSysDragChain::getUri, uri);
        TbSysDragChain tbSysDragRuleChain = tbSysDragChainMapper.selectOne(chainQueryWrapper);
        if (ObjectUtil.isNull(tbSysDragRuleChain) || StrUtil.isBlank(tbSysDragRuleChain.getChainFile())) {
            throw new ServiceException(StrUtil.format("通过uri[{}]获取规则链文件信息为空!", uri));
        }

        // 获取文件信息
        String base64FileStr = tbSysDragRuleChain.getChainFile();
        byte[] decodedBytes = Base64.getDecoder().decode(base64FileStr);
        return new String(decodedBytes, StandardCharsets.UTF_8);
    }