Files
swift-mirror/stdlib/public/Concurrency/Deque/UnsafeMutableBufferPointer+Utilities.swift
Franz Busch 8dda2274bc Replace custom Deque implementation with swift-collection Deque
# Motivation
For AsyncStream we created a custom internal Deque implementation to use for the buffering. This implementation was relatively bare-bones compared to the one in swift-collections. Furthermore, it lacked some methods that we need to implement the new `AsyncStream` APIs that support producer backpressure.

# Modification
This PR copies over the Deque implementation of swift-collections and makes it internal and non-inlinable.

# Result
We now have a fully functional Deque.
2023-07-23 08:33:03 +01:00

60 lines
1.8 KiB
Swift

//===----------------------------------------------------------------------===//
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 2021 Apple Inc. and the Swift project authors
// Licensed under Apache License v2.0 with Runtime Library Exception
//
// See https://swift.org/LICENSE.txt for license information
//
//===----------------------------------------------------------------------===//
/// This file is copied from swift-collections and should not be modified here.
/// Rather all changes should be made to swift-collections and copied back.
import Swift
extension Collection {
internal func _rebased<Element>() -> UnsafeBufferPointer<Element>
where Self == UnsafeBufferPointer<Element>.SubSequence {
.init(rebasing: self)
}
}
extension Collection {
internal func _rebased<Element>() -> UnsafeMutableBufferPointer<Element>
where Self == UnsafeMutableBufferPointer<Element>.SubSequence {
.init(rebasing: self)
}
}
extension UnsafeMutableBufferPointer {
internal func _initialize(from source: UnsafeBufferPointer<Element>) {
assert(source.count == count)
guard source.count > 0 else { return }
baseAddress!.initialize(from: source.baseAddress!, count: source.count)
}
internal func _initialize<C: Collection>(
from elements: C
) where C.Element == Element {
assert(elements.count == count)
var (it, copied) = elements._copyContents(initializing: self)
precondition(copied == count)
precondition(it.next() == nil)
}
internal func _deinitializeAll() {
guard count > 0 else { return }
baseAddress!.deinitialize(count: count)
}
internal func _assign<C: Collection>(
from replacement: C
) where C.Element == Element {
guard self.count > 0 else { return }
self[0 ..< count]._rebased()._deinitializeAll()
_initialize(from: replacement)
}
}