spring boot 项目上线获取文件路径处理

偏执的太偏执、 2022-02-20 00:23 507阅读 0赞

**获取绝对路径**

**Class的getResource()方法是从当前.class文件路径查找资源,ClassLoader则是从jar包根目录查找**

String path = this.getClass().getResource("/application.yml").getPath();
             //String path = this.getClass().getClassLoader().getResource("/application.yml").getPath();
                //显示为前缀例如 /C:/a.txt 取后面,需要判断服务器使用的系统
               String substring = path.substring(1);

**获取相对路径 **

File resourcefile = ResourceUtils.getFile("classpath:application.yml");
    Resource resourcefile = new ClassPathResource("classpath:application.yml");

**上线的时候spring boot 压缩成jar或者war包读取resources的资源读取失败**

**只能使用转换成流对象在服务器创建文件**

new ClassPathResource("application.yml").getInputStream();//上线环境未测试
    InputStream inputStream = this.getClass().getClassLoader().getResourceAsStream();//测试可用

/**
     * 在项目中resources有自己的文件读取需要处理
     */
    public class ProdFileUtils{
        //获取服务器系统
        private static final String SYSTEM = System.getProperty("os.name");
    
        private static final String WINDOWS = "C:\\myFile";
    
        private static final String LINUX = "/myFile";
    
        private ProdFileUtils() {
        }
    
        public static Path getPath() {
            String[] os = SYSTEM.split(" ");
            if (Objects.equals(os[0], "Windows")) {
                return Paths.get(WINDOWS);
            } else {
                return Paths.get(LINUX);
            }
        }
    
    
        /**
         * 获取项目资源文件
         *
         * @param resourceFile 资源路径不能以'/'开头,使用相对路径
         * @return path
         */
        public static Path getFile(String resourceFile) {
            InputStream inputStream = ProdFileUtils.class.getClassLoader().getResourceAsStream(resourceFile);
            try {
                //多系统路径处理
                if (StringUtils.isNotBlank(SYSTEM)) {
                    String[] os = SYSTEM.split(" ");
                    String fileURI;
                    if (Objects.equals(os[0], "Windows")) {
                        fileURI = WINDOWS;
                        if (resourceFile.contains("/")) {
                            resourceFile = resourceFile.replaceAll("/", "\\\\");
                            String substring = resourceFile.substring(0, resourceFile.lastIndexOf("\\"));
                            fileURI = Paths.get(fileURI, substring).toString();
                        }
                    } else {
                        fileURI = LINUX;
                        if (resourceFile.contains("/")) {
                            String substring = resourceFile.substring(0, resourceFile.lastIndexOf("/"));
                            fileURI = Paths.get(fileURI, substring).toString();
                        }
                    }
    
                    if (Files.notExists(Paths.get(fileURI))) {
                        Files.createDirectories(Paths.get(fileURI));
                    }
                    Path path = Paths.get(fileURI, resourceFile.substring(resourceFile.lastIndexOf(File.separator) + 1));
                    if (Files.notExists(path)) {
                        if (Objects.nonNull(inputStream)) {
                            long copy = Files.copy(inputStream, path);
                            if (copy > 0) {
                                return path;
                            }
                        }
                    } else {
                        return path;
                    }
    
                }
            } catch (IOException e) {
                e.printStackTrace();
            } finally {
                if (Objects.nonNull(inputStream)) {
                    try {
                        inputStream.close();
                    } catch (IOException e) {
                        e.printStackTrace();
                    }
                }
            }
            return null;
        }
    
    }