csv 文件是以
aaa,bbb,ccc
aaa,bbb,ccc
保存的
這裏的要求是將 List<T> 類型的線性表 轉化成 類似 html 中 table的格式,即第一行是 head 後面是 body
使用註解的效果如下 :
List<User> users=new ArrayList<User>();
users.add(new User("劉夏楠", 23, "男"));
users.add(new User("劉夏楠", 23, "男"));
users.add(new User("劉夏楠", 23, "男"));
writeBeanToCsvFile("D:\\test.csv", users);
csv文件:
package bean;
import annotation.csvField;
public class User {
@csvField("姓名") private String name;
@csvField("年齡")private Integer age;
private String sex;
public User(String name, Integer age, String sex) {
super();
this.name = name;
this.age = age;
this.sex = sex;
}
public String getName() {
return name;
}
public void setName(String name) {
this.name = name;
}
public Integer getAge() {
return age;
}
public void setAge(Integer age) {
this.age = age;
}
public String getSex() {
return sex;
}
public void setSex(String sex) {
this.sex = sex;
}
}
擁有 @csvField 註解的字段纔會被 寫入csv文件
並且 @csvField 的值作爲csv文件的 title 即 第一行
我們使用反射來達到這個效果
bean 轉 List<String[]> 如下:
private static <T> List<String[]> getStringArrayFromBean(List<T> beans) throws NoSuchMethodException, SecurityException, IllegalAccessException, IllegalArgumentException, InvocationTargetException{
if(beans.size()<1)
throw new IllegalArgumentException("長度不能小於1");
List<String[]> result=new ArrayList<String[]>();
Class<? extends Object> cls=beans.get(0).getClass();//獲取泛型類型
Field[] declaredFields=cls.getDeclaredFields();
List<Field> annoFields=new ArrayList<Field>();
for(int i=0;i<declaredFields.length;i++){//篩選出擁有註解的字段
csvField anno=declaredFields[i].getAnnotation(csvField.class);//獲取註解
if(anno!=null)
annoFields.add(declaredFields[i]);
}
String[] title=new String[annoFields.size()];
for(int i=0;i<annoFields.size();i++){
title[i]=declaredFields[i].getAnnotation(csvField.class).value();//獲取註解的值
}
result.add(title);
for(T each:beans){
String[] item=new String[annoFields.size()];
for(int i=0;i<annoFields.size();i++){
String fieldName=annoFields.get(i).getName();
String methodName="get"+fieldName.substring(0,1).toUpperCase()+fieldName.substring(1);
Method method=each.getClass().getMethod(methodName, null);
String val=method.invoke(each, null).toString();
item[i]=val;
}
result.add(item);
}
return result;
}
然後使用 csv 的一個工具包 javacsv.jar來寫入csv文件
public static <T> void writeBeanToCsvFile(String csvFilePath,List<T> beans) throws IOException, NoSuchMethodException, SecurityException, IllegalAccessException, IllegalArgumentException, InvocationTargetException{
File file =new File(csvFilePath);
if(!file.exists()){ //如果文件不存在,創建文件
file.createNewFile();
}
CsvWriter wr =new CsvWriter(csvFilePath,',', Charset.forName("GBK"));
List<String[]> contents=getStringArrayFromBean(beans);
for(String[] each:contents){
wr.writeRecord(each);
}
wr.close();
}