如何读取HTTP分块响应并在Elixir中向客户端发送分块响应?

时间:2017-05-14 16:59:44

标签: logging proxy elixir phoenix-framework chunked-encoding

我正在使用Elixir / Phoenix,我有一个端点,它返回一个分块响应,比如说永不停止的日志行流。但是,日志行来自另一个服务A,它也返回一个分块响应。我希望我的端点从服务A读取分块响应,并以块的形式将它们传递给客户端。实质上,它只是服务A的代理,但我不能让客户端直接连接到服务A,因为我需要执行一些身份验证。

1 个答案:

答案 0 :(得分:5)

以下是使用Phoenix发送块数据的示例:

  def test_1(conn, _params) do
    conn = conn
    |> put_resp_content_type("text/event-stream")
    |> send_chunked(200)

    conn |> chunk("a")
    conn |> chunk("b")
    conn |> chunk("c")

    # send data five times, once per second, to simulate a log
    for n <- 1..5 do
      conn |> chunk(Integer.to_string(n))
      Process.sleep(1000)
    end

    conn
  end

有一些可用于Elixir / Erlang的http库;我喜欢HTTPoison

这是一个从网址中读取数据块的示例,并在它们进入时将其发送出去:

  def test_2(conn, _params) do
    url = "http://localhost:4000/test_1"
    %HTTPoison.AsyncResponse{id: id} = HTTPoison.get!(url, %{}, stream_to: self())

    conn = conn
    |> put_resp_content_type("text/event-stream")
    |> send_chunked(200)

    process_httpoison_chunks(conn, id)
  end

  def process_httpoison_chunks(conn, id) do
    receive do
      %HTTPoison.AsyncStatus{id: ^id} ->
        # TODO handle status
        process_httpoison_chunks(conn, id)
      %HTTPoison.AsyncHeaders{id: ^id, headers: %{"Connection" => "keep-alive"}} ->
        # TODO handle headers
        process_httpoison_chunks(conn, id)
      %HTTPoison.AsyncChunk{id: ^id, chunk: chunk_data} ->
        conn |> chunk(chunk_data)
        process_httpoison_chunks(conn, id)
      %HTTPoison.AsyncEnd{id: ^id} ->
        conn
    end
  end

一些参考文献: