Take enum values from on_dump

fuelen · Feb 17, 2022 · e3705e0 · e3705e0
1 parent 73b1ba2
commit e3705e0
Showing 1 changed file with 2 additions and 8 deletions.
diff --git a/lib/ecto/erd/dot.ex b/lib/ecto/erd/dot.ex
@@ -138,14 +138,8 @@ defmodule Ecto.ERD.Dot do
 
   defp format_field(%Field{name: name}, :name), do: inspect(name)
 
-  defp format_field(%Field{type: {:parameterized, Ecto.Enum, %{values: values}}}, :type) do
-    "#Enum<#{inspect(values)}>"
-  end
-
-  defp format_field(%Field{type: {:parameterized, Ecto.Enum, %{mappings: mappings}}}, :type) do
-    keys = mappings |> Enum.into(%{}) |> Map.keys()
-
-    "#Enum<#{inspect(keys)}>"
+  defp format_field(%Field{type: {:parameterized, Ecto.Enum, %{on_dump: on_dump}}}, :type) do
+    "#Enum<#{inspect(Map.keys(on_dump))}>"
   end
 
   defp format_field(