Correctly handle nil with update

OpenKitten · Oct 3, 2018 · b928ae3 · b928ae3
1 parent 3d39ece
commit b928ae3
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/Sources/Meow/Context.swift b/Sources/Meow/Context.swift
@@ -284,8 +284,15 @@ public final class Context {
 
             var set = [String: Primitive?]()
 
+            var unsetKeys = Set(fields)
+
             for key in document.keys where fields.contains(key) {
                 set[key] = document[key]
+                unsetKeys.remove(key)
+            }
+
+            for key in unsetKeys {
+                set[key] = .some(nil)
             }
 
             return self.manager.collection(for: M.self)