Fix some issues

ryanwinchester · Aug 19, 2024 · 2e808ae · 2e808ae
1 parent 3c8584d
commit 2e808ae
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/lib/data_mapper.ex b/lib/data_mapper.ex
@@ -65,7 +65,10 @@ defmodule DataMapper do
 
           # Applies a function to a field.
           {from_key, {to_key, fun}}, acc when is_function(fun) ->
-            Map.put(acc, to_key, fun.(input[from_key]))
+            case input[from_key] do
+              nil -> acc
+              data -> Map.put(acc, to_key, fun.(data))
+            end
 
           # Nested mapping.
           {from_key, {to_key, sub_mapping}}, acc when is_map(sub_mapping) ->
@@ -77,6 +80,7 @@ defmodule DataMapper do
 
             case input[from_key] do
               data when is_list(data) -> Enum.map(data, &map(&1, sub_mapping, opts))
+              nil -> acc
               data -> map(data, sub_mapping, opts)
             end