mirror of
https://github.com/apple/sourcekit-lsp.git
synced 2026-03-02 18:23:24 +01:00
2877675bd5
Change a l public declarations to the `package` access level, accept for: - The `LanguageServerProtocol` module - The `BuildServerProtocol` module - `InProcessClient.InProcessSourceKitLSPClient` - `LanguageServerProtocolJSONRPC` (I would like to create a more ergonomic API for this like `InProcessSourceKitLSPClient` in the future, but for now, we’ll leave it public) Unfortunately, our pattern of marking functions as `@_spi(Testing) public` no longer works with the `package` access level because declarations at the `package` access level cannot be marked as SPI. I have decided to just mark these functions as `package`. Alternatives would be: - Add an underscore to these functions, like we did for functions exposed for testing before the introduction of `SPI` - Use `@testable` import in the test targets and mark the methods as `internal` Resolves #1315 rdar://128295618
50 lines
2.1 KiB
Swift
50 lines
2.1 KiB
Swift
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This source file is part of the Swift.org open source project
|
|
//
|
|
// Copyright (c) 2014 - 2024 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
|
|
// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
package extension Collection where Index == Int {
|
|
/// Partition the elements of the collection into `numberOfBatches` roughly equally sized batches.
|
|
///
|
|
/// Elements are assigned to the batches round-robin. This ensures that elements that are close to each other in the
|
|
/// original collection end up in different batches. This is important because eg. test files will live close to each
|
|
/// other in the file system and test scanning wants to scan them in different batches so we don't end up with one
|
|
/// batch only containing source files and the other only containing test files.
|
|
func partition(intoNumberOfBatches numberOfBatches: Int) -> [[Element]] {
|
|
var batches: [[Element]] = Array(
|
|
repeating: {
|
|
var batch: [Element] = []
|
|
batch.reserveCapacity(self.count / numberOfBatches)
|
|
return batch
|
|
}(),
|
|
count: numberOfBatches
|
|
)
|
|
|
|
for (index, element) in self.enumerated() {
|
|
batches[index % numberOfBatches].append(element)
|
|
}
|
|
return batches.filter { !$0.isEmpty }
|
|
}
|
|
|
|
/// Partition the collection into batches that have a maximum size of `batchSize`.
|
|
///
|
|
/// The last batch will contain the remainder elements.
|
|
func partition(intoBatchesOfSize batchSize: Int) -> [[Element]] {
|
|
var batches: [[Element]] = []
|
|
batches.reserveCapacity(self.count / batchSize)
|
|
var lastIndex = self.startIndex
|
|
for index in stride(from: self.startIndex, to: self.endIndex, by: batchSize).dropFirst() + [self.endIndex] {
|
|
batches.append(Array(self[lastIndex..<index]))
|
|
lastIndex = index
|
|
}
|
|
return batches
|
|
}
|
|
}
|