Appsignal.Utils.ParamsEncoder.preprocess/1 handles structs

appsignal · Oct 27, 2016 · 8295aba · 8295aba
1 parent ba0b447
commit 8295aba
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/lib/appsignal/utils/params_encoder.ex b/lib/appsignal/utils/params_encoder.ex
@@ -12,6 +12,11 @@ defmodule Appsignal.Utils.ParamsEncoder do
   @doc """
   Makes sure the keys are valid for JSON encoding
   """
+  def preprocess(value = %{__struct__: _}) do
+    value
+    |> Map.from_struct
+    |> preprocess
+  end
   def preprocess(value) when is_map(value) do
     Enum.map(value, fn({k, v}) ->
       {safe_key(k), preprocess(v)}

diff --git a/test/utils/param_encoder_test.exs b/test/utils/param_encoder_test.exs
@@ -1,3 +1,7 @@
+defmodule ExampleStruct do
+  defstruct a: "b"
+end
+
 defmodule Appsignal.Utils.ParamsEncoderTest do
   use ExUnit.Case
 
@@ -8,6 +12,7 @@ defmodule Appsignal.Utils.ParamsEncoderTest do
     assert %{"a" => "b"} == ParamsEncoder.preprocess(%{"a" => "b"})
     assert %{:a => "b"} == ParamsEncoder.preprocess(%{:a => "b"})
     assert %{"{:weird, :key}" => "b"} == ParamsEncoder.preprocess(%{{:weird, :key} => "b"})
+    assert %{:a => "b"} == ParamsEncoder.preprocess(%ExampleStruct{})
   end
 
   test "deep nesting" do