2016-10-27 2 views
0

Я очень новичок в akka-http, и я бы хотел передать поток csv с произвольным количеством строк.Потоковая CSV с akka-http в scala

Например, я хотел бы вернуться:

a,1 
b,2 
c,3 

со следующим кодом

implicit val actorSystem = ActorSystem("system") 
implicit val actorMaterializer = ActorMaterializer() 

val map = new mutable.HashMap[String, Int]() 
map.put("a", 1) 
map.put("b", 2) 
map.put("c", 3) 
val `text/csv` = ContentType(MediaTypes.`text/csv`, `UTF-8`) 
val route = 
    path("test") { 
    complete { 
     HttpEntity(`text/csv`, ??? using map) 
    } 
    } 
Http().bindAndHandle(route,"localhost",8080) 

Спасибо за вашу помощь

EDIT: Благодаря Ramon J Romero у Vigil

package test 


import akka.actor.ActorSystem 
import akka.http.scaladsl.Http 
import akka.http.scaladsl.model.HttpCharsets.`UTF-8` 
import akka.http.scaladsl.model._ 
import akka.http.scaladsl.server.Directives._ 
import akka.stream._ 
import akka.util.ByteString 

import scala.collection.mutable 

object Test{ 

    def main(args: Array[String]) { 

    implicit val actorSystem = ActorSystem("system") 
    implicit val actorMaterializer = ActorMaterializer() 

    val map = new mutable.HashMap[String, Int]() 
    map.put("a", 1) 
    map.put("b", 2) 
    map.put("c", 3) 

    val mapStream = Stream.fromIterator(() => map.toIterator) 
     .map((k: String, v: Int) => s"$k,$v") 
     .map(ByteString.apply) 
    val `text/csv` = ContentType(MediaTypes.`text/csv`, `UTF-8`) 
    val route = 
     path("test") { 
     complete { 
      HttpEntity(`text/csv`, mapStream) 
     } 
     } 
    Http().bindAndHandle(route, "localhost", 8080) 

    } 
} 

С помощью этого кода у меня есть две ошибки компиляции:

Error:(29, 28) value fromIterator is not a member of object scala.collection.immutable.Stream 
val mapStream = Stream.fromIterator(() => map.toIterator) 

Error:(38, 11) overloaded method value apply with alternatives: 
    (contentType: akka.http.scaladsl.model.ContentType,file: java.io.File,chunkSize: Int)akka.http.scaladsl.model.UniversalEntity <and> 
    (contentType: akka.http.scaladsl.model.ContentType,data: akka.stream.scaladsl.Source[akka.util.ByteString,Any])akka.http.scaladsl.model.HttpEntity.Chunked <and> 
    (contentType: akka.http.scaladsl.model.ContentType,data: akka.util.ByteString)akka.http.scaladsl.model.HttpEntity.Strict <and> 
    (contentType: akka.http.scaladsl.model.ContentType,bytes: Array[Byte])akka.http.scaladsl.model.HttpEntity.Strict <and> 
    (contentType: akka.http.scaladsl.model.ContentType.NonBinary,string: String)akka.http.scaladsl.model.HttpEntity.Strict 
cannot be applied to (akka.http.scaladsl.model.ContentType.WithCharset, List[akka.util.ByteString]) 
      HttpEntity(`text/csv`, mapStream) 

Я использовал список кортежей, чтобы получить Arround первого выпуска (Хауэр я не знаю, как поток карты в Scala) Нет идеи для второй Спасибо за помощь.

(я использую Scala 2.11.8)

ответ

3

Используйте функцию apply в HttpEntity, которая принимает в Source[ByteString,Any]. Приложение apply создает объект Chunked. Вы можете читать файл, используя код, основанный на documentation для потокового файла ввода-вывода с помощью потока AKKA Source:

import akka.stream.scaladsl._ 

val file = Paths.get("yourFile.csv") 

val entity = HttpEntity(`txt/csv`, FileIO.fromPath(file)) 

поток будет разбить файл на размеры порций, default is currently set to 8192.

Для потоковой передачи карты, которую вы создали вы можете использовать подобный трюк:

val mapStream = Source.fromIterator(() => map.toIterator) 
         .map((k : String, v : Int) => s"$k,$v") 
         .map(ByteString.apply) 

val mapEntity = HttpEntity(`test/csv`, mapStream) 
+0

Спасибо за ваш ответ, как я должен использовать это для потоковых моей карты, а не файл? – ogen

+0

@ogen см. Обновленный ответ ... –

+0

Спасибо за ваш ответ, но у меня есть ошибка компиляции типа несоответствия, ожидаемая ToResponseMarshallable, актуальная Any. Спасибо за вашу помощь – ogen

Смежные вопросы