Fixed BSONDocument.init(predicate:)

PureSwift · Aug 21, 2023 · a2047b8 · a2047b8
1 parent 1deaf94
commit a2047b8
Showing 1 changed file with 13 additions and 3 deletions.
diff --git a/Sources/MongoDBModel/Predicate.swift b/Sources/MongoDBModel/Predicate.swift
@@ -50,12 +50,22 @@ public extension BSONDocument {
  // { <field>: { $eq: <value> } }
  guard case let .keyPath(keyPath) = predicate.left,
  let comparisonOperator = ComparisonQueryOperator(predicate: predicate.type),
- case let .value(value) = predicate.right,
  predicate.options.isEmpty,
- predicate.modifier == nil,
- let valueBSON = try? BSON(attributeValue: value) else {
+ predicate.modifier == nil else {
  return nil
  }
+ let valueBSON: BSON
+ switch predicate.right {
+ case .keyPath:
+ return nil
+ case let .attribute(value):
+ guard let bson = try? BSON(attributeValue: value) else {
+ return nil
+ }
+ valueBSON = bson
+ case let .relationship(value):
+ valueBSON = BSON(relationship: value)
+ }
  self = [
  keyPath.rawValue: .document([comparisonOperator.rawValue: valueBSON])]
  }