Download file in csv format for java (2)

上一遍说到了基于注解式文件下载CSV格式,今天咱就聊聊基于函数式的文件下载,函数式接口是java8提到的新名词,感兴趣的可以查看下源码,可以简化开发代码,自定义流式风格,深度学习请移步至Oracle相关文档,我们这次用到的是:java.util.function包.

直接上道理(代码出现的常量请看上一篇,这里不贴了):

	/**
	 * 导出csv文件数据,基于函数式接口实现
	 * @param fileName 文件名
	 * @param datas 数据源
	 * @return
	 */
	public File exportCsvFile(String fileName, List<T> datas) {
		File f = null;
		OutputStreamWriter fileWriter = null;
		try {
			f = File.createTempFile(fileName, FILE_FORMAT);
			fileWriter = new OutputStreamWriter(new FileOutputStream(f), ENCODINT_GBK);
			fileWriter.append(objectToString(datas));
		} catch (Exception e) {
			// do something
		} finally {
			try {
				fileWriter.flush();
				fileWriter.close();
			} catch (IOException e) {
				// do something
			}
		}
		return f;
	}

注意:我用的是创建临时文件方法,File.createTempFile(String perfix,String suffix)中perfix是有长度要求的;源码片段截图如下:

也就是说参数fileName的长度必须大于3,否则会抛IAE;

	/**
	 * 将对象数据处理成字符串
	 * @param datas 对象数据集合
	 * @return
	 */
	protected String objectToString(List<T> datas) {
		StringBuilder sb = new StringBuilder();
                //获取数据提供者中所要导出的所有不为空的字段名
		List<String> list = providers.stream().map(e -> e.getField()).filter(Objects::nonNull)
				.collect(Collectors.toList());
                //导入时必须显式的Filed
		if (!list.isEmpty()) {
			String filedRow = Joiner.on(COMMA_DELIMITER).join(list);
			sb.append(filedRow).append(NEW_LINE_SEPARATOR);
		}
               //获取字段名标题
		String titleRow = Joiner.on(COMMA_DELIMITER)
				.join(providers.stream().map(e -> e.getTitle()).collect(Collectors.toList()));
		sb.append(titleRow).append(NEW_LINE_SEPARATOR);
		for (T data : datas) {
			for (Provider<T> p : providers) {
				Object o = p.getFunc().apply(data);//获取数据
				sb.append(SPECIAL_SYMBOLS);
				sb.append(Objects.isNull(o) ? BLANK_STRING : o);
				sb.append(COMMA_DELIMITER);
			}
			sb.delete(sb.length() - 1, sb.length());
			sb.append(NEW_LINE_SEPARATOR);
		}
		return sb.toString();
	}

信息提供者: 

class Provider<T> {
	private String field;
	private String title;
	private Function<T, Object> func;

	public Provider(String field, String title, Function<T, Object> func) {
		super();
		this.field = field;
		this.title = title;
		this.func = func;
	}

	public Provider(String title, Function<T, Object> func) {
		super();
		this.title = title;
		this.func = func;
	}

	public String getField() {
		return field;
	}

	public void setField(String field) {
		this.field = field;
	}

	public String getTitle() {
		return title;
	}

	public void setTitle(String title) {
		this.title = title;
	}

	public Function<T, Object> getFunc() {
		return func;
	}

	public void setFunc(Function<T, Object> func) {
		this.func = func;
	}
}

注册数据源: 

	/**
	 * 导入导出使用
	 * @param filed 为null时隐去域名,不可导入,同导出使用
	 * @param title 
	 * @param func
	 * @return
	 */
	public ExportCsvUtils<T> register(String filed, String title, Function<T, Object> func) {
		providers.add(new Provider<T>(filed, title, func));
		return this;
	}

	/**
	 * 仅导出使用
	 * @param title
	 * @param func
	 * @return
	 */
	public ExportCsvUtils<T> register(String title, Function<T, Object> func) {
		providers.add(new Provider<T>(title, func));
		return this;
	}

//数据收集器
private List<Provider<T>> providers = Lists.newArrayList();

测试类:

class Test {
	String a;
	Integer b;

	public String getA() {
		return a;
	}

	public void setA(String a) {
		this.a = a;
	}

	public Integer getB() {
		return b;
	}

	public void setB(Integer b) {
		this.b = b;
	}

	public Test(String a, Integer b) {
		super();
		this.a = a;
		this.b = b;
	}

	@Override
	public String toString() {
		return "Test [a=" + a + ", b=" + b + "]";
	}

}

测试使用:

public static void main(String[] args) {
		List<Test> list = Lists.newArrayList(new Test("c", 1), new Test("d", null), new Test("e", 3));
		ExportCsvUtils<Test> utils = new ExportCsvUtils<Test>();
		File file = utils.register("测试a", a -> a.getA()).register("测试b", a -> Objects.isNull(a.getB()) ? 2 : a.getB())
				.exportCsvFile("测试下载文件", list);//注意:文件名长度大于3
		System.err.println(file);
	}

测试结果:

好了数据导出大功告成,但是按实际需求还没完,代码片段中有提到数据导入时Filed必须是显式的,那接下来咱们就得研究下如何导入CSV文件的数据了.

为什么导出数据中会有"`"符号,是因为导出的csv文件用wps表格或excel表格打开时防止出现科学计数法,日期格式变动(原:yyyy-MM-dd打开后:yyyy/MM/dd)等问题,造成数据错误问题,届时对数据的导入麻烦就大了.