Merge pull request #160 from apple/use-temp-alloc-facility

[OrderedCollection] Use standard temp allocation facility, if available
apple · Aug 17, 2022 · 8cc9d22 · 8cc9d22
2 parents c5ea141 + ddb6d14
commit 8cc9d22
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/Sources/OrderedCollections/Utilities/_UnsafeBitset.swift b/Sources/OrderedCollections/Utilities/_UnsafeBitset.swift
@@ -73,6 +73,15 @@ extension _UnsafeBitset {
     run body: (inout _UnsafeBitset) throws -> Void
   ) rethrows {
     let wordCount = _UnsafeBitset.wordCount(forCapacity: capacity)
+#if compiler(>=5.6)
+    return try withUnsafeTemporaryAllocation(
+      of: Word.self, capacity: wordCount
+    ) { words in
+      words.initialize(repeating: .empty)
+      var bitset = Self(words: words, count: 0)
+      return try body(&bitset)
+    }
+#else
     if wordCount <= 2 {
       var buffer: (Word, Word) = (.empty, .empty)
       return try withUnsafeMutablePointer(to: &buffer) { p in
@@ -87,6 +96,7 @@ extension _UnsafeBitset {
     defer { words.deallocate() }
     var bitset = _UnsafeBitset(words: words, count: 0)
     return try body(&bitset)
+#endif
   }
 }