好久没更新了,忘是没忘记更新这个事,只是手里没什么素材也没什么时间。前几天在某网站上看到一个自己制作单机游戏的dalao,他为了方便玩家自制游戏mod,开放出了这么一个独特的方式来存放数据(我也不知道对你们来说算不算独特,反正对我来说挺独特的了),就是使用excel文档来存放自己游戏中的部分数据。这样的话就涉及到了对excel的文档操作问题了。于是乎自己去稍微试了一下,来和大家分享下这个测试结果。
总的代码不长,大概150+行吧。但是感觉期间遇到的问题比较多,可能会很话痨,如果自己感觉太多的话会分成两期来做,这样又可以水一篇文章了。
首先呢,我使用的是POI中的HSSF系列的类,就是HSSFWorkbook、HSSFSheet、HSSFCell这些的。相比于XSSF系列,HSSF操作的是.xls的excel文件,而XSSF操作的是.xlsx的文件。体感上觉得HSSF比XSSF稍稍好用些(可能是我姿势不对吧)。所以综上,还是使用.xls来测试。
代码方面,先是创建文件的代码,先用File判断文件是否存在,如果不存在的话,new一个HSSFWorkbook的对象。然后一定要注意,同时要新建一个标签页,因为默认在新建excel文档时会有三个标签页,但是使用Java创建出来的文档是不会有标签页的,所以打开的时候会有报错的现象。然后用字节流输出就好了,最后的workbook.write()是将创建好的workbook写到文件字节流里面去。
/**
* 创建excel文件
* @param object
* @throws IOException
*/
public static void createFile(String url) throws IOException {
File file = new File(url);
if (!file.exists()) {
//创建excel文件
HSSFWorkbook workbook = new HSSFWorkbook();
//创建excel的标签页,xls
HSSFSheet sheet = workbook.createSheet("用户表一");
FileOutputStream out = new FileOutputStream(url);
workbook.write(out);
out.close();
}
}
然后是数据导入部分,除了需要建立输入通道以外,还要建立输出通道,为了保存导入的数据。解析已有的excel之后,获取已经创建的标签页,然后创建行和单元格,可以用HSSFStyle设置单元格样式,这里就不多说设置样式的事情了。然后在导入数据时,为了把方法写得灵活一些,用了Object类,然后用反射取得字段名和数据。不过这么做的话会破坏掉原来使用get方法取数的结构,还要禁用原来的private权限,慎用。如果你们有什么好的方法,也可以评论告诉我下啦。最后记得用flush()和workbook.write(fos)输出。
/**
* 将数据添加到excel中
* @param object
* @param url
* @throws IOException
*/
public static void insertFile(Object object, String url) throws IOException {
//打开excel文件
FileInputStream fis = new FileInputStream(url);
//解析excel文件
POIFSFileSystem fs = new POIFSFileSystem(fis);
HSSFWorkbook workbook = new HSSFWorkbook(fs);
//得到已经创建的标签页
HSSFSheet sheet = workbook.getSheet("用户表一");
//创建输出流
FileOutputStream fos = new FileOutputStream(url);
//利用反射机制得到类属性名
Class clazz = object.getClass();
Field[] fields = clazz.getDeclaredFields();
//创建样式
HSSFCellStyle style = workbook.createCellStyle();
//设置样式
style.setAlignment(XSSFCellStyle.ALIGN_CENTER);
//创建行
if (sheet.getLastRowNum() != 0) {
HSSFRow row = sheet.createRow(sheet.getLastRowNum() + 1);
try {
for (int i = 0; i < fields.length; i++) {
HSSFCell cell = row.createCell(i);
Field field = fields[i];
field.setAccessible(true);
Object obj = field.get(object);
if (obj != null) {
cell.setCellValue(field.get(object).toString());
cell.setCellStyle(style);
}
}
}catch(IllegalAccessException e){
}finally{
fos.flush();
workbook.write(fos);
fos.close();
}
}else {
HSSFRow row = sheet.createRow(0);
for (int i = 0; i < fields.length; i++) {
//创建单元格
HSSFCell cell = row.createCell(i);
//设置单元格的值
cell.setCellValue(fields[i].getName());
//设置单元格的样式
cell.setCellStyle(style);
}
row = sheet.createRow(1);
try {
for (int i = 0; i < fields.length; i++) {
HSSFCell cell = row.createCell(i);
Field field = fields[i];
field.setAccessible(true);
Object obj = field.get(object);
if (obj != null) {
cell.setCellValue(field.get(object).toString());
cell.setCellStyle(style);
}
}
}catch(IllegalAccessException e){
}finally{
fos.flush();
workbook.write(fos);
fos.close();
}
}
fos.flush();
workbook.write(fos);
fos.close();
}
然后是数据导出,这个就没什么好说的了,比数据导入要来得简单,只是有个问题要注意下:就是如果当前的单元格为空的话,那么row.getCell()是取不到东西的,也就是null,所以这个地方要记得判断下。然后可选的取数据的方法有很多,我统一处理成String类型,再做的转换。
/**
* 数据导出
* @param url 文件路径
* @return 数据列表
*/
public static List<User> export(String url) throws IOException {
FileInputStream fis = new FileInputStream(url);
POIFSFileSystem fs = new POIFSFileSystem(fis);
HSSFWorkbook workbook = new HSSFWorkbook(fs);
HSSFSheet sheet = workbook.getSheet("用户表一");
List<User> list = new ArrayList<>();
for (int i=1; i<=sheet.getLastRowNum(); i++) {
HSSFRow row = sheet.getRow(i);
User user = new User();
user.setId(row.getCell(0)==null? null : Integer.parseInt(row.getCell(0).getStringCellValue()));
user.setIdcard(row.getCell(1)==null? null : row.getCell(1).getStringCellValue());
user.setUsername(row.getCell(2)==null? null : row.getCell(2).getStringCellValue());
user.setPassword(row.getCell(3)==null? null : row.getCell(3).getStringCellValue());
user.setSex(row.getCell(4)==null? null : Integer.parseInt(row.getCell(4).getStringCellValue()));
user.setBirthday(row.getCell(5)==null? null : row.getCell(5).getStringCellValue());
user.setTel(row.getCell(6)==null? null : row.getCell(6).getStringCellValue());
user.setEmail(row.getCell(7)==null? null : row.getCell(7).getStringCellValue());
user.setCity(row.getCell(8)==null? null : row.getCell(8).getStringCellValue());
user.setAddress(row.getCell(9)==null? null : row.getCell(9).getStringCellValue());
user.setCreate_time(row.getCell(10)==null? null : row.getCell(10).getStringCellValue());
user.setUpdate_time(row.getCell(11)==null? null : row.getCell(11).getStringCellValue());
list.add(user);
}
return list;
}
最后提醒一下,所有的文件操作均要在文件关闭状态下进行。