added array handling

asensei · Mar 19, 2019 · 52fae31 · 52fae31
1 parent c9a3ac7
commit 52fae31
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/Sources/FluentMongo/Document+Nested.swift b/Sources/FluentMongo/Document+Nested.swift
@@ -79,10 +79,12 @@ extension Document {
                 switch document[item.key] {
                 case .some(let value as Document):
                     mutableFilter[ensureNoRootNameSpace(item.key)] = removeKeysPrefix(value)
+                case .some(let value as [Document]):
+                    mutableFilter[ensureNoRootNameSpace(item.key)] = value.map { removeKeysPrefix($0) }
                 case .some(let value):
                     mutableFilter[ensureNoRootNameSpace(item.key)] = value
                 case .none:
-                    break
+                    mutableFilter[ensureNoRootNameSpace(item.key)] = nil
                 }
             }