仿Spring读取配置文件实现方案

时间:2023-03-09 16:21:52
仿Spring读取配置文件实现方案

一、读取config和jar里的配置文件,以流的方式返回

 /**
* 根据路径,获取当前项目中路径(jar包和config文件)下的配置文件
* @param filePath 存放job配置信息文件的相对路径
* @return job.xml文件流
*/
private static List<InputStream> getJobFilesInputStream(String filePath) throws IOException {
List<InputStream> jobFilesInputStreams = new ArrayList<>();
ResourcePatternResolver resolver = new PathMatchingResourcePatternResolver();
Resource[] resources = resolver.getResources("classpath*:"+ filePath +"/*.xml");
Map<String,String> fileMap = new HashMap<>();
for (Resource resource:resources){
String key = resource.getFilename();
if(StringUtils.isBlank(fileMap.get(key))){//检查是否已经读取同文件的流
jobFilesInputStreams.add(resource.getInputStream());
fileMap.put(key,key);
}
}
return jobFilesInputStreams;
}

二、读取config和jar里的配置文件,以文件对象的方式返回

 /**
* 根据路径,获取当前项目中路径(jar包和config文件)下的配置文件
* @param filePath 存放job配置信息文件的相对路径
* @return job.xml文件对象
*/
private static List<File> getJobFiles(String filePath) throws IOException {
List<File> jobFiles = new ArrayList<>();
ResourcePatternResolver resolver = new PathMatchingResourcePatternResolver();
Resource[] resources = resolver.getResources("classpath*:"+ filePath +"/*.xml");
Map<String,String> fileMap = new HashMap<>();
for (Resource resource:resources){
String key = resource.getFilename();
if(StringUtils.isBlank(fileMap.get(key))){//检查是否已经读取同文件的流
jobFiles.add(resource.getFile());
fileMap.put(key,key);
}
}
return jobFiles;
}

三、读取config和jar里的配置文件,以文件路径的方式返回

 /**
* 根据路径,获取当前项目中路径(jar包和config文件)下的配置文件
* @param filePath 存放job配置信息文件的相对路径
* @return job.xml文件路径
*/
private static List<String> getJobFilesPath(String filePath) throws IOException {
List<String> jobFilesPath = new ArrayList<>();
ResourcePatternResolver resolver = new PathMatchingResourcePatternResolver();
Resource[] resources = resolver.getResources("classpath*:"+ filePath +"/*.xml");
Map<String,String> fileMap = new HashMap<>();
for (Resource resource:resources){
String key = resource.getFilename();
if(StringUtils.isBlank(fileMap.get(key))){//检查是否已经读取同文件的流
jobFilesPath.add(resource.getFile().getPath());
fileMap.put(key,key);
}
}
return jobFilesPath;
}

经测试:三种方式,都能读取到文件,无论config文件夹下还是jar包里的配置文件。但是因为其他工具类需要配合使用这些配置文件,第二种和第三种不支持配置文件直接打入jar包,因此第一种是比较通用的做法。

第一种有一个地方需要特别注意:流数据只能被消费一次,消费一次后,流中的数据就没有了。

解决方案:成员变量,接收流数据写入内存,每次要用的时候,将这个成员变量数据再转换成流,供其他调用的工具方法进行消费。