Play 2.4: перехватывать и изменять тело ответа

Question

Согласно play documentation этого является то, что пользовательские действия должны выглядеть следующим образом:

object CustomAction extends ActionBuilder[Request] { 
    def invokeBlock[A](request: Request[A], block: Request[A] => Future[Result]): Future[Result] = { 
     block(request) 
    } 
} 

Но сказать, если бы я хотел добавить «Foo» для каждого тела ответа, как мне делать что? Очевидно, что ниже не работает:

block.andThen(result => result.map(r => r.body.toString + "foo")).apply(request) 

Любые идеи?

UPDATE: Что-то стоит отметить, что это действие будет в основном используется как асинхронный в контроллере:

def test = CustomAction.async { 
    //... 
} 

Answer 1

Вы должны будете принять Enumerator[Array[Byte]] от Result тела и кормить его к iteratee для фактического потребления тела результата, прежде чем вы сможете его изменить. Таким образом, простой iteratee, который потребляет результирующее тело и преобразует в строку может выглядеть следующим образом:

block.apply(request).flatMap { res => 
    Iteratee.flatten(res.body |>> Iteratee.consume[Array[Byte]]()).run.map { byteArray => 
    val bodyStr = new String(byteArray.map(_.toChar)) 
    Ok(bodyStr + "foo") 
    } 
} 

Я использовал flatMap в результате запуска Iteratee.flatten является Future[T]. Зайдите в https://www.playframework.com/documentation/2.4.x/Enumerators для получения дополнительной информации о том, как работать с Enumerators/Iteratees.