首先先创建一个文件路径类,用来放置文件的位置
package com.ruoyi.web.core.config;
import lombok.Data;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.PropertySource;
/**
* fileConfig.properties文件配置信息
*/
@Configuration
@ConfigurationProperties(prefix = "config", ignoreUnknownFields = false)
@PropertySource("classpath:fileConfig.properties")
@Data
public class FileConfigProperties {
/**
* 文件存储位置配置
*/
private String filePath;
}
获取到文件储位置后,就可以进行用文件操作类进行操作,这里可以用若依框架里面自带的,或者使用hutool工具,也可以
在这里我自己写了一个工具类, 可以参考一下,就是把读取的文件用Redis进行缓存,根据文件最后的修改时间和文件名作为Key值进行存储 和查询,这样的话就可以保证查询速率,只有在第一次加载的时候,比较慢,后面的话,如果不改动文件内容的话,之后用的都是redis里面缓存的值.
这个工具类如下:
package com.ruoyi.web.controller.tool;
import com.ruoyi.common.core.redis.RedisCache;
import com.ruoyi.common.utils.poi.ExcelUtil;
import com.ruoyi.system.domain.enumVo.FileNameVo;
import com.ruoyi.web.core.config.FileConfigProperties;
import org.springframework.stereotype.Component;
import org.springframework.util.ResourceUtils;
import javax.annotation.Resource;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.util.Collection;
import java.util.List;
@Component
public class FileDataUtil {
@Resource
FileConfigProperties fileConfigProperties;
@Resource
RedisCache redis;
/**
* 读取Excel文件数据
*
* @param fileNameVo 文件类型
* @return 文件数据
* @throws FileNotFoundException 文件不存在
*/
public List<?> getFileDataList(FileNameVo fileNameVo) throws FileNotFoundException {
String fileName = fileNameVo.getFileName();
File file = ResourceUtils.getFile(fileConfigProperties.getFilePath() + fileName);
String fileLastTime = String.valueOf((file.lastModified()));
String dataListKey = fileName + "Data";
if (!redis.hasKey(fileName) || !redis.getCacheObject(fileName).equals(fileLastTime)) {
redis.setCacheObject(fileName, fileLastTime);
FileInputStream fileInputStream = new FileInputStream(file);
ExcelUtil<?> util = new ExcelUtil<>(fileNameVo.getEntityClass());
List<?> dataList = util.importExcel(fileInputStream);
if (!dataList.isEmpty()) redis.setCacheObject(dataListKey, dataList);
return dataList;
} else {
return redis.getCacheObject(dataListKey);
}
}
/**
* 加载数据
*
* @param dataList 数据集
* @param file 文件
*/
public <T> void loadData(List<T> dataList, FileNameVo file) throws Exception {
if (dataList.isEmpty()) {
try {
dataList.addAll((Collection<? extends T>) getFileDataList(file));
} catch (Exception e) {
throw new FileNotFoundException("加载文件数据时发生错误: " + e.getMessage());
}
if (dataList.isEmpty()) {
throw new FileNotFoundException("文件数据为空");
}
}
}
}
这个工具类可以适用不同的类型,也就是List<?>这种 使用的时候直接传入你需要读取的文件,和相对应接收的 List<自定义实体类> 数据集