通用CSV读取和加载代码

Question

我想在java中实现以下功能：

一个类，具有根据提供的参数数量读取不同列数的不同csv文件的方法。对于ex：

public boolean readAndLoad(String csvPath,String csvFileName, String csvType);

这里csvPath和csvFileName是自解释的，但csvType在这里是为了获得对应于csv的表名。例如：如果csvType是employee，则表示此csv包含带有员工数据特定标题的员工数据。

现在，我想创建一个通用的泛型方法或类的组合，可以执行此操作，并将不同的csv加载到不同的表中。每个csv中的列数不同。

一种可能的解决方案是在方法中手动检查文件名，然后获取对应于该csv的静态头列表，并在表中插入数据。将有许多if-else检查。还有其他解决方案吗？

Answer 1

只需使用univocity解析器即可完成工作。它有很多功能，速度非常快。如果需要将数据映射到特定类的对象中，可以在其属性上声明带有注释的类，这些注释指示字段名称/位置。如果需要，也可以很好地支持类的层次结构。

如果您只需要为每个可能的输入提供标题，您将发现一个setHeaders方法，该方法允许您为输入分配任何标题列表。您还可以调用selectFields方法来选择要读取的字段和要忽略的字段。

Answer 2

如果csvType应该匹配您的一个实体的名称，我建议首先从其包中识别该实体：

String packageName="com.your.entities";
           ClassLoader classLoader = Thread.currentThread().getContextClassLoader();
            URL packageURL;
            ArrayList<String> classesNames = new ArrayList<String>();;
            packageName = packageName.replace(".", "/");
            packageURL = classLoader.getResource(packageName); 
            URI uri = new URI(packageURL.toString());
            File folder = new File(uri.getPath());
                File[] file = folder.listFiles();
                String className;
                for(File actual: file){
                    className = actual.getName();
                    className = className.substring(0, className.lastIndexOf('.'));
                    classesNames.add(className);}

一旦你得到了你的实体，快速检查csvType是否匹配其中一个，如果是，你应该得到标题的字段数，当然还有字段名：

 packageName = packageName.replace("/", ".");
                   int headerLengh;
                   String header="";
                   for (String name :classesNames)
                   {
                       if(name.equals(csvType))
                       { headerLengh=Class.forName(packageName+"."+name).getDeclaredFields().length;

                       for (int i=0;i<headerLengh;i++)
                       {
                           header=header+Class.forName(packageName+"."+name).getDeclaredFields()[i].getName()+" ";

                       }}}

一旦您的csvType匹配了一个实体，您就有了标题和表，您可以简单地读取它，或者通过您自己的逻辑或方法将其加载到表中（您现在知道了）。