2017-03-09 85 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