2017-03-09 60 views
2

Reading a CSV files using Akka Streams - 基于此问题。 我使用Akka Streams读取CSV。 现在我需要逐行处理它 - 但我也需要知道标题的名称是什么。 任何选项?Akka流:使用标题处理CSV

UPD。澄清一下。

FileIO.fromPath(Paths.get("a.csv)) 
.via(Framing.delimiter(ByteString("\n"), 256, true).map(_.utf8String)) 
.runForeach(println /* header + current line for each line*/) 
+0

不会头在您的CSV的第一行?你能否添加更多关于你面临的具体问题的细节? –

+0

添加了我需要的样本。 – Rumoku

回答

6

您可以使用prefixAndTail采取的第一个元素 - 在这种情况下,头 - 然后用后续元素(行)懒洋洋地结合起来。

下面是具有组合的报头和列的例子,以形成Map[String, String]

val flow: Flow[Seq[String], Map[String, String], NotUsed] = Flow[Seq[String]] 
    .prefixAndTail(1).flatMapConcat { case (headers, rows) => 
    rows.map (row => headers.head.zip(row).toMap) 
    } 

val test: Source[Seq[String], NotUsed] = Source(
    List(Seq("col1", "col2"), Seq("a", "b"), Seq("1", "2"))) 

Await.result(test.via(flow).runForeach(println), 20.seconds) 
// Map(col1 -> a, col2 -> b) 
// Map(col1 -> 1, col2 -> 2) 
1

Alpakka,阿卡流连接器的集合,提供CSV support

Source 
    .single(ByteString("""header1,header2,header3 
         |1,2,3 
         |4,5,6""".stripMargin)) 
    .via(CsvParsing.lineScanner()) 
    .via(CsvToMap.toMap()) 
    .map(_.mapValues(_.utf8String)) 
    .runForeach(println) 

// Map(header1 -> 1, header2 -> 2, header3 -> 3) 
// Map(header1 -> 4, header2 -> 5, header3 -> 6) 
+0

这个图书馆太棒了。 – flybonzai