Add MongoModelStorage.Configuration

Author: colemancda

Sources/MongoDBModel/MongoDatabase.swift

@@ -8,44 +8,53 @@ public struct MongoModelStorage: ModelStorage {
 
     public let model: Model
 
+    public let options: Configuration
+    
     public init(
         database: MongoDatabase,
-        model: Model
+        model: Model,
+        options: Configuration = Configuration()
     ) {
         self.database = database
         self.model = model
+        self.options = options
     }
 
     /// Fetch managed object.
     public func fetch(_ entityName: EntityName, for id: ObjectID) async throws -> ModelData? {
         let entity = try model(for: entityName)
-        return try await database.fetch(entity, for: id)
+        let options = options.collections[entity.id]
+        return try await database.fetch(entity, for: id, options: options)
     }
 
     /// Fetch managed objects.
     public func fetch(_ fetchRequest: FetchRequest) async throws -> [ModelData] {
         let entity = try model(for: fetchRequest.entity)
-        return try await database.fetch(fetchRequest, entity: entity)
+        let options = options.collections[entity.id]
+        return try await database.fetch(fetchRequest, entity: entity, options: options)
     }
 
     /// Fetch and return result count.
     public func count(_ fetchRequest: FetchRequest) async throws -> UInt {
-        try await database.count(fetchRequest)
+        let options = options.collections[fetchRequest.entity]
+        return try await database.count(fetchRequest, options: options)
     }
 
     /// Create or edit a managed object.
     public func insert(_ value: ModelData) async throws {
-        try await database.insert(value)
+        let options = options.collections[value.entity]
+        try await database.insert(value, options: options)
     }
 
     /// Create or edit multiple managed objects.
     public func insert(_ values: [ModelData]) async throws {
-        try await database.insert(values)
+        try await database.insert(values, options: options.collections)
     }
 
     /// Delete the specified managed object.
     public func delete(_ entity: EntityName, for id: ObjectID) async throws {
-        try await database.delete(entity, for: id)
+        let options = options.collections[entity]
+        try await database.delete(entity, for: id, options: options)
     }
 
     private func model(for entityName: EntityName) throws -> EntityDescription {
@@ -56,14 +65,27 @@ public struct MongoModelStorage: ModelStorage {
     }
 }
 
-public extension MongoDatabase {
+public extension MongoModelStorage {
+    
+    struct Configuration {
+        
+        public var collections: [EntityName: MongoCollectionOptions]
+        
+        public init(collections: [EntityName : MongoCollectionOptions] = [:]) {
+            self.collections = collections
+        }
+    }
+}
+
+internal extension MongoDatabase {
 
     /// Fetch managed object.
     func fetch(
         _ entity: EntityDescription,
-        for id: ObjectID
+        for id: ObjectID,
+        options: MongoCollectionOptions?
     ) async throws -> ModelData? {
-        guard let document = try await find(entity.id, for: id) else {
+        guard let document = try await find(entity.id, for: id, options: options) else {
             return nil
         }
         return try ModelData(bson: document, model: entity)
@@ -72,37 +94,47 @@ public extension MongoDatabase {
     /// Fetch managed objects.
     func fetch(
         _ fetchRequest: FetchRequest,
-        entity: EntityDescription
+        entity: EntityDescription,
+        options: MongoCollectionOptions?
     ) async throws -> [ModelData] {
-        try await fetchDocuments(fetchRequest)
+        try await fetchDocuments(fetchRequest, options: options)
             .map { try ModelData(bson: $0, model: entity) }
     }
 
     /// Fetch and return result count.
-    func count(_ fetchRequest: FetchRequest) async throws -> UInt {
+    func count(
+        _ fetchRequest: FetchRequest,
+        options: MongoCollectionOptions?
+    ) async throws -> UInt {
         let entityName = fetchRequest.entity
-        let collection = self.collection(entityName)
+        let collection = self.collection(entityName, options: options)
         let filter = fetchRequest.predicate.map { BSONDocument(filter: $0) } ?? [:]
         let options = CountDocumentsOptions(fetchRequest: fetchRequest)
         let count = try await collection.countDocuments(filter, options: options)
         return UInt(count)
     }
 
     /// Create or edit a managed object.
-    func insert(_ value: ModelData) async throws {
+    func insert(
+        _ value: ModelData,
+        options: MongoCollectionOptions?
+    ) async throws {
         let entityName = value.entity
-        let collection = self.collection(entityName)
+        let collection = self.collection(entityName, options: options)
         let document = BSONDocument(model: value)
         try await collection.insertOne(document)
     }
 
-    func insert(_ values: [ModelData]) async throws {
+    func insert(
+        _ values: [ModelData],
+        options: [EntityName: MongoCollectionOptions]
+    ) async throws {
         var collections = [EntityName: [ModelData]]()
         for value in values {
             collections[value.entity, default: []].append(value)
         }
         for (entity, values) in collections {
-            let collection = self.collection(entity)
+            let collection = self.collection(entity, options: options[entity])
             let documents = values.map { BSONDocument(model: $0) }
             try await collection.insertMany(documents)
         }
@@ -111,23 +143,21 @@ public extension MongoDatabase {
     /// Delete the specified managed object.
     func delete(
         _ entity: EntityName,
-        for id: ObjectID
+        for id: ObjectID,
+        options: MongoCollectionOptions?
     ) async throws {
-        let collection = self.collection(entity)
+        let collection = self.collection(entity, options: options)
         let options: DeleteOptions? = nil
         let filter: BSONDocument = [
             BSONDocument.BuiltInProperty.id.rawValue: .string(id.rawValue)
         ]
         try await collection.deleteOne(filter, options: options)
     }
-}
-
-internal extension MongoDatabase {
 
     func find(
         _ entityName: EntityName,
         for id: ObjectID,
-        options: MongoCollectionOptions? = nil
+        options: MongoCollectionOptions?
     ) async throws -> BSONDocument? {
         let collection = self.collection(entityName, options: options)
         let options: FindOneOptions? = nil
@@ -139,7 +169,7 @@ internal extension MongoDatabase {
 
     func fetchDocuments(
         _ fetchRequest: FetchRequest,
-        options: MongoCollectionOptions? = nil
+        options: MongoCollectionOptions?
     ) async throws -> [BSONDocument] {
         let entityName = fetchRequest.entity
         let collection = self.collection(entityName, options: options)
@@ -156,7 +186,7 @@ internal extension MongoDatabase {
     func create(
         _ entityName: EntityName,
         for id: ObjectID,
-        options: MongoCollectionOptions? = nil
+        options: MongoCollectionOptions?
     ) async throws -> BSONDocument {
         let collection = self.collection(entityName, options: options)
         let document: BSONDocument = [
@@ -172,7 +202,7 @@ fileprivate extension MongoDatabase {
 
     func collection(
         _ name: EntityName,
-        options: MongoCollectionOptions? = nil
+        options: MongoCollectionOptions?
     ) -> MongoCollection<BSONDocument> {
         let tableName = name.rawValue.lowercased() + "s"
         return collection(tableName, options: options)