SpringBoot 读取jar包下resource中的文件夹

前段时间,在基于springboot开发过程中,遇到一个问题:程序需要读取resource下的某个目录的全部文件,而且需要能以File的方式读取。但是, 打包后,springboot项目就成了jar包了,读取文件,会报错: ... cannot be resolved to absolute file path because it does not reside in the file system:jar:file: ....  。

一般情况下,我们读取resource下的某个文件,可以这样通过IO流的方式读取:

Resource resource =new ClassPathResource(fileName);

InputStream is = resource.getInputStream();

当时,有些时候,需要使用File:resource.getFile() ,这时,在jar包下就会报上面的错误。可是,如果是文件夹怎么办呢?

经过网上的一番查询,确定了一个方案:把jar包中的文件,先存到一个临时文件夹下,然后通过临时文件夹,可以实现以File的方式读取,文件读取完成后,删除临时文件目录。

文件复制代码如下:

import lombok.extern.log4j.Log4j2;

import org.springframework.core.io.ClassPathResource;

import org.springframework.core.io.Resource;

import java.io.*;

import java.net.URL;

import java.util.Enumeration;

import java.util.HashMap;

import java.util.Map;

import java.util.zip.ZipEntry;

import java.util.zip.ZipFile;

@Log4j2

public class FileUtils {

    /**

    * 复制文件到目标目录

    * @param resourcePath resource的文件夹路径

    * @param tmpDir 临时目录

    * @param fileType 文件类型

    */

    public static void copyJavaResourceDir2TmpDir(String resourcePath, String tmpDir, FileType fileType) {

        Map<String, Object> fileMap = new HashMap<>();

        if (resourcePath.endsWith("/")) {

            resourcePath = resourcePath.substring(0, resourcePath.lastIndexOf("/"));

        }

        try {

            Enumeration resources = null;

            try {

                resources = Thread.currentThread().getContextClassLoader().getResources(resourcePath);

            } catch (Exception ex) {

                ex.printStackTrace();

            }

            if (resources == null || !resources.hasMoreElements()) {

                resources = FileUtils.class.getClassLoader().getResources(resourcePath);

            }

            while (resources.hasMoreElements()) {

                URL resource = (URL) resources.nextElement();

                if (resource.getProtocol().equals("file")) { // resource是文件

                    continue;

                }

                String[] split = resource.toString().split(":");

                String filepath = split[2];

                if (OperatingSystem.isWindows()) {

                    filepath = filepath + ":" + split[3];

                }

                String[] split2 = filepath.split("!");

                String zipFileName = split2[0];

                ZipFile zipFile = new ZipFile(zipFileName);

                Enumeration entries = zipFile.entries();

                while (entries.hasMoreElements()) {

                    ZipEntry entry = (ZipEntry) entries.nextElement();

                    String entryName = entry.getName();

                    if (entry.isDirectory()) {

                        continue;

                    }

                    if (entryName.contains(resourcePath) && entryName.endsWith(fileType.toString().toLowerCase())) {

                        String dir = entryName.substring(0, entryName.lastIndexOf("/"));

                        if (!dir.endsWith(resourcePath)) { // 目标路径含有子目录

                            dir = dir.substring(dir.indexOf(resourcePath) + resourcePath.length() + 1);

                            if (dir.contains("/")) { // 多级子目录

                                String[] subDir = dir.split("/");

                                Map<String, Object> map = fileMap;

                                for (String d : subDir) {

                                    map = makeMapDir(map, d);

                                }

                                map.putAll(readOneFromJar(zipFile.getInputStream(entry), entryName));

                            } else { //一级子目录

                                if (fileMap.get(dir) == null) {

                                    fileMap.put(dir, new HashMap<String, Object>());

                                }

                                ((Map<String, Object>) fileMap.get(dir))

                                        .putAll(readOneFromJar(zipFile.getInputStream(entry), entryName));

                            }

                        } else { // 目标路径不含子目录

                            fileMap.putAll(readOneFromJar(zipFile.getInputStream(entry), entryName));

                        }

                    }

                }

            }

        } catch (Exception e) {

            log.error("读取resource文件异常:", e);

        } finally {

            try {

                // 写到目标缓存路径

                createFile(fileMap, tmpDir);

            } catch (Exception e) {

                log.error("创建临时文件异常:", e);

            }

        }

    }

    private static void createFile(Map<String, Object> fileMap, String targetDir) {

        fileMap.forEach((key, value) -> {

            if (value instanceof Map) {

                createFile((Map<String, Object>) value, targetDir + File.separator + key);

            } else {

                createNewFile(targetDir + File.separator + key, value.toString());

            }

        });

    }

    public static void createNewFile(String filePath, String value) {

        try {

            File file = new File(filePath);

            if (!file.exists()) {

                File parentDir = file.getParentFile();

                if (!parentDir.exists()) {

                    parentDir.mkdirs();

                }

                file.createNewFile();

            }

            PrintWriter out = new PrintWriter(new BufferedWriter(new OutputStreamWriter(new FileOutputStream(file), "utf-8")));

            out.write(value);

            out.flush();

            out.close();

        } catch (Exception e) {

            log.error("创建文件异常:", e);

        }

    }

    private static Map<String, Object> makeMapDir(Map<String, Object> fileMap, String d) {

        if (fileMap.get(d) == null) {

            fileMap.put(d, new HashMap<String, Object>());

        }

        return (Map<String, Object>) fileMap.get(d);

    }

    public static Map<String, String> readOneFromJar(InputStream inputStream, String entryName) {

        Map<String, String> fileMap = new HashMap<>();

        try (Reader reader = new InputStreamReader(inputStream, "utf-8")) {

            StringBuilder builder = new StringBuilder();

            int ch = 0;

            while ((ch = reader.read()) != -1) {

                builder.append((char) ch);

            }

            String filename = entryName.substring(entryName.lastIndexOf("/") + 1);

            fileMap.put(filename, builder.toString());

        } catch (Exception ex) {

            log.error("读取文件异常:", ex);

        }

        return fileMap;

    }

    public enum FileType {

        XSD, TXT, XLS, XLSX, DOC, DOCX, XML, SQL, PROPERTIES, SH

    }

}   


程序调用代码如下:

String tmpDir = System.getProperty("user.dir") + File.separator +"tmp";

try {

    FileUtils.copyJavaResourceDir2TmpDir("xsd", tmpDir, FileUtils.FileType.XSD);

    loadXsd(new File(tmpDir));

} catch (Exception e) {

    logger.error("加载xsd文件异常:", e);

} finally {

    FileUtils.delete(tmpDir);

}

这样,就不用每次部署实例的时候,都把需要文件和jar一起部署了,方便多了。

OperatingSystem是Filnk中copy过来的代码

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 212,718评论 6 492
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 90,683评论 3 385
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 158,207评论 0 348
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 56,755评论 1 284
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 65,862评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,050评论 1 291
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,136评论 3 410
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,882评论 0 268
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,330评论 1 303
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,651评论 2 327
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,789评论 1 341
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,477评论 4 333
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,135评论 3 317
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,864评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,099评论 1 267
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,598评论 2 362
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,697评论 2 351

推荐阅读更多精彩内容