Merge pull request #1 from SwiftMN/more-list-functions

add functions removeAll, contains, and first to ThreadSafeList

Author: vlaminck

Sources/ThreadSafeCollections/ThreadSafeList.swift

@@ -60,6 +60,12 @@ public class ThreadSafeList<V> {
     }
   }
 
+  public func removeAll(where shouldBeRemoved: @escaping (V) -> Bool) {
+    itemQueue.async(flags: .barrier) { // safely write
+      self.items.removeAll(where: shouldBeRemoved)
+    }
+  }
+  
   public func getAll() -> [V] {
     var allItems = [V]()
     itemQueue.sync { // safely read
@@ -76,6 +82,22 @@ public class ThreadSafeList<V> {
     return count
   }
 
+  public func contains(where predicate: (V) throws -> Bool) rethrows -> Bool {
+    var doesContain = false
+    try itemQueue.sync { // safely read
+      doesContain = try self.items.contains(where: predicate)
+    }
+    return doesContain
+  }
+  
+  public func first(where predicate: (V) throws -> Bool) rethrows -> V? {
+    var foundItem: V? = nil
+    try itemQueue.sync { // safely read
+      foundItem = try self.items.first(where: predicate)
+    }
+    return foundItem
+  }
+  
   public subscript(index: Int) -> V? {
     get {
       return self.get(index)

Tests/ThreadSafeCollectionsTests/ThreadSafeCollectionsTests.swift

@@ -38,6 +38,11 @@ final class ThreadSafeCollectionsTests: XCTestCase {
     XCTAssertEqual(0, m.count())
     m.append(contentsOf: items)
     XCTAssertEqual(100, m.count())
+    XCTAssertTrue(m.contains(where: { $0 == 5 }))
+    XCTAssertEqual(5, m.first(where: { $0 == 5 }))
+    m.removeAll(where: { $0 == 5 })
+    XCTAssertFalse(m.contains(where: { $0 == 5 }))
+    XCTAssertNil(m.first(where: { $0 == 5 }))
   }
 
 }