如何使用Java8流将csv转换为地图

Question

我正在尝试创建一个简单的解析util，它转换一个两列CSV文件并将其放入一个映射。

public Map<String, String> getMapFromCSV(final String filePath) throws IOException {
    return Files.lines(Paths.get(filePath))
                .map(line -> line.split(","))
                .collect(Collectors.toMap(line -> line[0], line -> line[1]));
}

如您所见，我正在创建一个字符串流，用逗号分隔每一行，并将其转换为字符串数组，最后将键映射到索引0，将值映射到索引1。

@Test
public void testGetMapFromCSV() throws IOException{
    actual = util.getMapFromCSV(filePath).get("AL");
    expected = "ALABAMA";

    assertEquals(expected, actual);
}

出于某种原因，当我运行这个测试时，实际值为null。我排除了无效的文件路径，因为它在另一个单元测试中运行良好，并且键值出现在CSV中。我已经盯着它看了几个小时了，我想也许有人能指出我的错误。

此外，我对Java8相当陌生，所以如果有人知道更好/更干净的写作方式，我会感谢反馈。

Answer 1

以一种可能更容易阅读的形式使用映射

public static Map<String, String> getMapFromCSV(final String filePath) throws IOException {
  try(Stream<String> lines = Files.lines(Paths.get(filePath))) {
      return lines.collect(
        mapping(s -> s.split(","), toMap(k -> k[0], v -> v[1])));
  }
}

Answer 2

正如fge所评论的那样，使用try-with资源使这变得更容易。

这是你的答案中的代码，但是使用资源试用来自动关闭一个也是AutoCloseable的Stream。还有一些添加了呼吸空间的空间字符。

public static Map<String, String> getMapFromCSV(final String filePath) throws IOException{

    try (
        Stream < String > lines = Files.lines( Paths.get( filePath ) );
    )
    {
        Map < String , String > resultMap = 
                lines.map( 
                         line -> line.split( "," ) 
                     )
                     .collect(
                         Collectors.toMap( line -> line[0] , line -> line[1] )
                     );
        return resultMap;
    }
    // The `Stream` named `lines` is automatically closed at this point by the try-with-resources.
}

当使用资源尝试的{…}块结束时，在try中打开的所有资源都会自动关闭，顺序与打开顺序相反。无论控制流是在完成时优雅地离开块，还是由于异常或返回而突然离开块（如代码中所示），都是如此。请参阅关于这个问题的讨论，尝试使用java中的参考资料和返回语句。

Answer 3

好的，我添加了行。close（）并从csv中删除所有空白，它就可以工作了！奇怪的是，考虑到csv在我的另一种方法中得到了很好的解析。下面是它的样子：

public static Map<String, String> getMapFromCSV(final String filePath) throws IOException{

        Stream<String> lines = Files.lines(Paths.get(filePath));
        Map<String, String> resultMap = 
                lines.map(line -> line.split(","))
                     .collect(Collectors.toMap(line -> line[0], line -> line[1]));

        lines.close();

        return resultMap;
    }