将List转换为InputStream而不创建中间文件

问题描述 投票:0回答:1

我正在使用一个函数,它将输入作为List和className,我将List转换为分隔符文件(中级),然后将文件转换为InputStream作为返回值。

在这里创建一个中间文件和删除是一个繁琐的过程,因为这将增加时间和性能。

那么还有另一种方法可以直接将数据从List<Class>转换为带有标题的分隔符(也许我们可以将它存储在List<Object>中)然后在不创建中间文件的情况下我们可以直接将其转换为InputStream。

<T> InputStream createInputStream(String csvFileName, List<T> listObject, Class className, char csvPreference) {

        ICsvBeanWriter beanWriter = null;
        File file=new File(csvFileName);
        InputStream inputStream=null;
        try {
            CsvPreference DELIMITER = new CsvPreference.Builder('"', csvPreference, "\n").build();

            beanWriter = new CsvBeanWriter(new FileWriter(file), DELIMITER);
            List<String> fieldList = Arrays.stream(className.getDeclaredFields()).map(Field::getName).collect(Collectors.toList());
            CellProcessor[] processors = new CellProcessor[fieldList.size()];
            for (int i = 0; i < fieldList.size(); i++) {
                processors[i] = new Optional();
            }
            String[] header = new String[fieldList.size()];
            fieldList.toArray(header);
            beanWriter.writeHeader(header);

            for (T obj : listObject) {
                beanWriter.write(obj, header, processors);
            }

        } catch (IOException ex) {
            System.err.println("Error writing the CSV file: " + ex);
        } finally {
            if (beanWriter != null) {
                try {
                    beanWriter.close();
                    inputStream=new FileInputStream(file);
                    file.delete();
                } catch (IOException ex) {
                    System.err.println("Error closing the writer: " + ex);
                }
            }
        }
        return inputStream;
    }

更新1:

用户类

public class Users {

    String id;

    String name;

    String gender;

}

在List中添加元素

List<Users> userList=new ArrayList<Users>();
userList.add(new Users("1","AA","Male"));
userList.add(new Users("2","BB","Male"));
userList.add(new Users("3","CC","Female"));

这是我现在打电话的方式

InputStream inputStream=createInputStream("demo.csv",userList,Users.Class,'|');

这里的中级文件(demo.csv)就像,

id|name|gender
1|AA|Male
2|BB|Male
3|CC|Female

从这个文件,我将转换为inputstream

提前致谢, 松鸦

java list inputstream
1个回答
3
投票

你考虑过这样的事吗?

ByteArrayOutputStream baos = new ByteArrayOutputStream();
beanWriter = new CsvBeanWriter(new PrintWriter(baos), DELIMITER);

//write some stuff

byte[] byteArray = baos.toByteArray();

return new ByteArrayInputStream(byteArray);
© www.soinside.com 2019 - 2024. All rights reserved.