将地图列表写入CSV

Question

我有一个独立的应用程序，该应用程序连接到SQL数据库并将ResultSet保存在Map列表中。 这是我到目前为止的内容：

List<Map<String, Object>> rows;
stmt = conn.createStatement();
Resultset rs = stmt.executeQuery(queryString);
ResultSetMetaData rsmd; //Properties of Resultset object and column count

while(rs.next){
  Map<String, Object> rowResult = new HashMap<String, Object>(columnCount);
   for(int i =1; i <=columnCount; i++){
     rowResult.put(rsmd.getColumnName(i), rs.getObject(i));
   }
  rows.add(rowResult);
}

//WRITE TO CSV  
String csv = "C:\\Temp\\data.csv";
CSVWriter writer = new CSVWriter(new FileWriter(csv));

//Write the record to file
writer.writeNext(rows);

//close the writer
writer.close();

如何将列表的“行”添加到带有列的csv中？ 任何线索和建议。 感谢您的帮助。

Answer 1

由于每条记录将以相同的顺序具有相同的列，因此我只对行使用List<List<Object>> 。

对于标题，您不需要在每一行上都获取它们。 只需一次获得它们，如下所示：

List<String> headers = new ArrayList<>();
for (int i = 1; i <= columnCount; i++ )
{
    String colName = rsmd.getColumnName(i);
    headers.add(colName);
}

接下来，您可以获取以下行：

List<List<Object>> rows = new ArrayList<>();

while(rs != null && rs.next)
{
   List<Object> row = new ArrayList<>();
   for(int i =1; i <=columnCount; i++)
   {
       row.add(rs.getObject(i));
   }
   rows.add(row);
}

最后，要创建CSV文件，您可以执行以下操作：

// create the CSVWriter
String csv = "C:\\Temp\\data.csv";
CSVWriter writer = new CSVWriter(new FileWriter(csv));

// write the header line
for (String colName : headers)
{
    writer.write(colName);
}
writer.endRecord();

// write the data records
for (List<Object> row : rows)
{
    for (Object o : row)
    {
        // handle nulls how you wish here
        String val = (o == null) ? "null" : o.toString();
        writer.write(val);
    }
    writer.endRecord();
}

// you should close the CSVWriter in a finally block or use a 
// try-with-resources Statement
writer.close;

注意 ：在我的代码示例中，我正在使用类型推断

请参阅 ： Try-With-Resources语句 。

Answer 2

老实说，我建议您在CSVWriter中使用writeAll方法并传递ResultSet 。

writer.writeAll(rs, true);

第二个参数是include列名，因此csv文件中的第一行将是列名。 然后，当您读取文件时，可以根据需要将其转换回Map中（尽管它将是所有字符串，除非您在读取文件时知道类型是什么）。

希望能有所帮助。

斯科特:)