No OneTemporary
Actions

Size

14 KB

Referenced Files

None

Subscribers

None

View Options

	diff --git a/lib/tesla/adapter/gun.ex b/lib/tesla/adapter/gun.ex
	index 6dbcff4..f9ac870 100644
	--- a/lib/tesla/adapter/gun.ex
	+++ b/lib/tesla/adapter/gun.ex
	@@ -1,258 +1,274 @@
	if Code.ensure_loaded?(:gun) do
	defmodule Tesla.Adapter.Gun do
	@moduledoc """
	Adapter for [gun] https://github.com/ninenines/gun

	Remember to add `{:gun, "~> 1.3"}` to dependencies
	Also, you need to recompile tesla after adding `:gun` dependency:
	```
	mix deps.clean tesla
	mix deps.compile tesla
	```
	### Example usage
	```
	# set globally in config/config.exs
	config :tesla, :adapter, Tesla.Adapter.Gun
	# set per module
	defmodule MyClient do
	use Tesla
	adapter Tesla.Adapter.Gun
	end
	```

	- ### Options https://ninenines.eu/docs/en/gun/1.3/manual/gun/:
	+ ### Adapter specific options:
	+
	+ * `timeout` - Time, while process, will wait for gun messages.
	+ * `body_as` - What will be returned in `%Tesla.Env{}` body key. Possible values - `:plain`, `:stream`, `:chunks`. Defaults to `:plain`.
	+ * `:plain` - as binary.
	+ * `:stream` - as stream. If you don't want to close connection (because you want to reuse it later) pass `close_conn: false` in adapter opts.
	+ * `:chunks` - as chunks. You can get response body in chunks using `Tesla.Adapter.Gun.read_chunk/3` function.
	+ Processing of the chunks and checking body size must be done by yourself. Example of processing function
	+ is in `test/tesla/adapter/gun_test.exs` - `read_body/3`. If you don't need connection later don't forget
	+ to close it with `Tesla.Adapter.Gun.close/1`.
	+ * `max_body` - Max response body size in bytes. Works only with `body_as: :plain`, with other settings you need to check response
	+ body size by yourself.
	+ * `conn` - Opened connection pid with gun. Is used for reusing gun connections.
	+ * `close_conn` - Close connection or not after receiving full response body. Works with `body_as: :stream`. Is used for reusing gun connections.
	+ Defaults to `true`.
	+
	+ ### Gun options https://ninenines.eu/docs/en/gun/1.3/manual/gun/:

	* `connect_timeout` - Connection timeout.
	* `http_opts` - Options specific to the HTTP protocol.
	* `http2_opts` - Options specific to the HTTP/2 protocol.
	* `protocols` - Ordered list of preferred protocols. Defaults: [http2, http] - for :tls, [http] - for :tcp.
	* `trace` - Whether to enable dbg tracing of the connection process. Should only be used during debugging. Default: false.
	* `transport` - Whether to use TLS or plain TCP. The default varies depending on the port used. Port 443 defaults to tls.
	All other ports default to tcp.
	* `transport_opts` - Transport options. They are TCP options or TLS options depending on the selected transport. Default: [].
	* `ws_opts` - Options specific to the Websocket protocol. Default: %{}.
	* `compress` - Whether to enable permessage-deflate compression. This does not guarantee that compression will
	be used as it is the server that ultimately decides. Defaults to false.
	* `protocols` - A non-empty list enables Websocket protocol negotiation. The list of protocols will be sent
	in the sec-websocket-protocol request header.
	The handler module interface is currently undocumented and must be set to `gun_ws_h`.
	}

	"""
	@behaviour Tesla.Adapter
	alias Tesla.Multipart

	@gun_keys [
	:connect_timeout,
	:http_opts,
	:http2_opts,
	:protocols,
	:retry,
	:retry_timeout,
	:trace,
	:transport,
	:transport_opts,
	:ws_opts
	]

	@adapter_default_timeout 1_000

	@impl true
	@doc false
	def call(env, opts) do
	with {:ok, status, headers, body} <- request(env, opts) do
	{:ok, %{env \| status: status, headers: format_headers(headers), body: body}}
	end
	end

	defp format_headers(headers) do
	for {key, value} <- headers do
	{String.downcase(to_string(key)), to_string(value)}
	end
	end

	defp format_method(method), do: String.upcase(to_string(method))

	defp format_url(nil, nil), do: ""
	defp format_url(nil, query), do: "?" <> query
	defp format_url(path, nil), do: path
	defp format_url(path, query), do: path <> "?" <> query

	defp request(env, opts) do
	request(
	format_method(env.method),
	Tesla.build_url(env.url, env.query),
	env.headers,
	env.body \|\| "",
	- Tesla.Adapter.opts(env, opts) \|> Enum.into(%{})
	+ Tesla.Adapter.opts([close_conn: true, body_as: :plain], env, opts) \|> Enum.into(%{})
	)
	end

	defp request(method, url, headers, %Stream{} = body, opts),
	do: request_stream(method, url, headers, body, opts)

	defp request(method, url, headers, body, opts) when is_function(body),
	do: request_stream(method, url, headers, body, opts)

	defp request(method, url, headers, %Multipart{} = mp, opts) do
	headers = headers ++ Multipart.headers(mp)
	body = Multipart.body(mp)

	request(method, url, headers, body, opts)
	end

	defp request(method, url, headers, body, opts) do
	with {pid, f_url} <- open_conn(url, opts),
	stream <- open_stream(pid, method, f_url, headers, body, false) do
	read_response(pid, stream, opts)
	end
	end

	defp request_stream(method, url, headers, body, opts) do
	with {pid, f_url} <- open_conn(url, opts),
	stream <- open_stream(pid, method, f_url, headers, body, true) do
	read_response(pid, stream, opts)
	end
	end

	defp open_conn(url, opts) do
	uri = URI.parse(url)
	opts = if uri.scheme == "https", do: Map.put(opts, :transport, :tls), else: opts
	- {:ok, pid} = :gun.open(to_charlist(uri.host), uri.port, Map.take(opts, @gun_keys))
	+
	+ {:ok, pid} =
	+ if opts[:conn],
	+ do: {:ok, opts[:conn]},
	+ else: :gun.open(to_charlist(uri.host), uri.port, Map.take(opts, @gun_keys))
	+
	{pid, format_url(uri.path, uri.query)}
	end

	defp open_stream(pid, method, url, headers, body, true) do
	stream = :gun.request(pid, method, url, headers, "")
	for data <- body, do: :ok = :gun.data(pid, stream, :nofin, data)
	:gun.data(pid, stream, :fin, "")
	stream
	end

	defp open_stream(pid, method, url, headers, body, false),
	do: :gun.request(pid, method, url, headers, body)

	defp read_response(pid, stream, opts) do
	receive do
	{:gun_response, ^pid, ^stream, :fin, status, headers} ->
	- :gun.close(pid)
	+ close(pid)
	{:ok, status, headers, ""}

	{:gun_response, ^pid, ^stream, :nofin, status, headers} ->
	- body_as =
	- cond do
	- opts[:stream_response] -> :stream
	- opts[:chunks_response] -> :chunks
	- true -> :plain
	- end
	-
	- format_response(pid, stream, opts, status, headers, body_as)
	+ format_response(pid, stream, opts, status, headers, opts[:body_as])

	{:error, error} ->
	- :gun.close(pid)
	+ close(pid)
	{:error, error}

	{:gun_up, ^pid, :http} ->
	read_response(pid, stream, opts)

	{:gun_error, ^pid, reason} ->
	- :gun.close(pid)
	+ close(pid)
	{:error, reason}

	{:gun_down, ^pid, _, _, _, _} ->
	read_response(pid, stream, opts)

	{:DOWN, _, _, _, reason} ->
	- :gun.close(pid)
	+ close(pid)
	{:error, reason}
	after
	opts[:timeout] \|\| @adapter_default_timeout ->
	- :ok = :gun.close(pid)
	+ :ok = close(pid)
	{:error, :timeout}
	end
	end

	defp format_response(pid, stream, opts, status, headers, :plain) do
	case read_body(pid, stream, opts) do
	{:ok, body} ->
	- :gun.close(pid)
	+ close(pid)
	{:ok, status, headers, body}

	{:error, error} ->
	- :gun.close(pid)
	+ close(pid)
	{:error, error}
	end
	end

	defp format_response(pid, stream, opts, status, headers, :stream) do
	stream_body =
	Stream.resource(
	fn -> %{pid: pid, stream: stream} end,
	fn
	%{pid: pid, stream: stream} ->
	case read_chunk(pid, stream, opts) do
	{:nofin, part} -> {[part], %{pid: pid, stream: stream}}
	{:fin, body} -> {[body], %{pid: pid, final: :fin}}
	end

	%{pid: pid, final: :fin} ->
	{:halt, %{pid: pid}}
	end,
	- fn %{pid: pid} -> :gun.close(pid) end
	+ fn %{pid: pid} ->
	+ if opts[:close_conn], do: close(pid)
	+ end
	)

	{:ok, status, headers, stream_body}
	end

	defp format_response(pid, stream, opts, status, headers, :chunks) do
	- {:ok, status, headers, %{pid: pid, stream: stream, opts: opts}}
	+ {:ok, status, headers, %{pid: pid, stream: stream, opts: Enum.into(opts, [])}}
	end

	def read_chunk(pid, stream, opts) do
	receive do
	{:gun_data, ^pid, ^stream, :fin, body} ->
	{:fin, body}

	{:gun_data, ^pid, ^stream, :nofin, part} ->
	{:nofin, part}
	after
	opts[:timeout] \|\| @adapter_default_timeout ->
	{:error, :timeout}
	end
	end

	- def close(%{pid: pid}) do
	+ def close(pid) do
	:gun.close(pid)
	end

	defp read_body(pid, stream, opts, acc \\ "") do
	limit = opts[:max_body]

	receive do
	{:gun_data, ^pid, ^stream, :fin, body} ->
	check_body_size(acc, body, limit)

	{:gun_data, ^pid, ^stream, :nofin, part} ->
	case check_body_size(acc, part, limit) do
	{:ok, acc} -> read_body(pid, stream, opts, acc)
	{:error, error} -> {:error, error}
	end
	after
	opts[:timeout] \|\| @adapter_default_timeout ->
	{:error, :timeout}
	end
	end

	defp check_body_size(acc, part, nil), do: {:ok, acc <> part}

	defp check_body_size(acc, part, limit) do
	body = acc <> part

	if limit - byte_size(body) >= 0 do
	{:ok, body}
	else
	{:error, :body_too_large}
	end
	end
	end
	end
	diff --git a/test/tesla/adapter/gun_test.exs b/test/tesla/adapter/gun_test.exs
	index 2d73fa2..6c1066a 100644
	--- a/test/tesla/adapter/gun_test.exs
	+++ b/test/tesla/adapter/gun_test.exs
	@@ -1,89 +1,131 @@
	defmodule Tesla.Adapter.GunTest do
	use ExUnit.Case

	use Tesla.AdapterCase, adapter: Tesla.Adapter.Gun
	use Tesla.AdapterCase.Basic
	use Tesla.AdapterCase.Multipart
	use Tesla.AdapterCase.StreamRequestBody
	use Tesla.AdapterCase.SSL

	test "fallback adapter timeout option" do
	request = %Env{
	method: :get,
	url: "#{@http}/delay/2"
	}

	assert {:error, :timeout} = call(request, timeout: 1_000)
	end

	test "max_body option" do
	request = %Env{
	method: :get,
	url: "#{@http}/get",
	query: [
	message: "Hello world!"
	]
	}

	assert {:error, :body_too_large} = call(request, max_body: 5)
	end

	test "without slash" do
	request = %Env{
	method: :get,
	url: "#{@http}"
	}

	assert {:ok, %Env{} = response} = call(request)
	assert response.status == 400
	end

	test "response stream" do
	request = %Env{
	method: :get,
	- url: "#{@http}/stream/10"
	+ url: "#{@http}/stream-bytes/10"
	}

	assert {:ok, %Env{} = response} = call(request)
	assert response.status == 200
	+ assert byte_size(response.body) == 16
	end

	test "read response body in chunks" do
	request = %Env{
	method: :get,
	- url: "#{@http}/stream/10"
	+ url: "#{@http}/stream-bytes/10"
	}

	- assert {:ok, %Env{} = response} = call(request, chunks_response: true)
	+ assert {:ok, %Env{} = response} = call(request, body_as: :chunks)
	assert response.status == 200
	%{pid: pid, stream: stream, opts: opts} = response.body
	- assert opts[:chunks_response]
	+ assert opts[:body_as] == :chunks
	assert is_pid(pid)
	assert is_reference(stream)

	- assert read_body(pid, stream) != []
	+ assert read_body(pid, stream) \|> byte_size() == 16
	+ refute Process.alive?(pid)
	end

	test "read response body in stream" do
	request = %Env{
	method: :get,
	- url: "#{@http}/stream/10"
	+ url: "#{@http}/stream-bytes/10"
	}

	- assert {:ok, %Env{} = response} = call(request, stream_response: true)
	+ assert {:ok, %Env{} = response} = call(request, body_as: :stream)
	assert response.status == 200
	assert is_function(response.body)
	- assert Enum.to_list(response.body) != []
	+ assert Enum.to_list(response.body) \|> List.to_string() \|> byte_size() == 16
	end

	- defp read_body(pid, stream, acc \\ []) do
	+ test "read response body in stream with opened connection without closing connection" do
	+ uri = URI.parse(@http)
	+ {:ok, conn} = :gun.open(to_charlist(uri.host), uri.port)
	+
	+ request = %Env{
	+ method: :get,
	+ url: "#{@http}/stream-bytes/10"
	+ }
	+
	+ assert {:ok, %Env{} = response} =
	+ call(request, body_as: :stream, conn: conn, close_conn: false)
	+
	+ assert response.status == 200
	+ assert is_function(response.body)
	+ assert Enum.to_list(response.body) \|> List.to_string() \|> byte_size() == 16
	+
	+ assert Process.alive?(conn)
	+
	+ :ok = :gun.close(conn)
	+ refute Process.alive?(conn)
	+ end
	+
	+ test "read response body in stream with opened connection with closing connection" do
	+ uri = URI.parse(@http)
	+ {:ok, conn} = :gun.open(to_charlist(uri.host), uri.port)
	+
	+ request = %Env{
	+ method: :get,
	+ url: "#{@http}/stream-bytes/10"
	+ }
	+
	+ assert {:ok, %Env{} = response} = call(request, body_as: :stream, conn: conn)
	+
	+ assert response.status == 200
	+ assert is_function(response.body)
	+ assert Enum.to_list(response.body) \|> List.to_string() \|> byte_size() == 16
	+
	+ refute Process.alive?(conn)
	+ end
	+
	+ defp read_body(pid, stream, acc \\ "") do
	case Tesla.Adapter.Gun.read_chunk(pid, stream, timeout: 1_000) do
	{:fin, body} ->
	:ok = :gun.close(pid)
	- [body \| acc]
	+ acc <> body

	{:nofin, part} ->
	- read_body(pid, stream, [part \| acc])
	+ read_body(pid, stream, acc <> part)
	end
	end
	end

File Metadata

Mime Type: text/x-diff
Expires: Thu, Nov 28, 12:28 PM (1 d, 20 h)
Storage Engine: blob
Storage Format: Raw Data
Storage Handle: 41037
Default Alt Text: (14 KB)

No OneTemporaryActions

View Options

File Metadata

Event Timeline

No OneTemporary
Actions