1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
|
//===----------------------------------------------------------------------===//
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 2014 - 2023 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
//
//===----------------------------------------------------------------------===//
import ArgumentParser
import Foundation
import InstructionCounter
import SwiftParser
import SwiftSyntax
struct PerformanceTest: ParsableCommand {
static var configuration = CommandConfiguration(
commandName: "performance-test",
abstract:
"Parse all .swift files in '--directory' and its subdirectories '--iteration' times and output the average time (in milliseconds) one iteration took."
)
@Flag(name: .long, help: "Parse files incrementally")
var incrementalParse: Bool = false
@Option(help: "The directory in which all .swift files should be parsed")
var directory: String
@Option(help: "How many times should the directory be parsed?")
var iterations: Int
func run() throws {
let sourceDir = URL(fileURLWithPath: self.directory)
let files = try FileManager.default
.enumerator(at: sourceDir, includingPropertiesForKeys: nil)!
.compactMap({ $0 as? URL })
.filter { $0.pathExtension == "swift" }
.map { try Data(contentsOf: $0) }
var fileTransition: [Data: IncrementalParseTransition] = [:]
if self.incrementalParse {
/// The initial parse for incremental parsing
for file in files {
file.withUnsafeBytes { buf in
let parseResult = Parser.parseIncrementally(
source: buf.bindMemory(to: UInt8.self),
parseTransition: nil
)
fileTransition[file] = IncrementalParseTransition(
previousIncrementalParseResult: parseResult,
edits: ConcurrentEdits(fromSequential: [])
)
}
}
}
let start = Date()
let startInstructions = getInstructionsExecuted()
for _ in 0..<self.iterations {
for file in files {
file.withUnsafeBytes { buf in
_ = Parser.parseIncrementally(
source: buf.bindMemory(to: UInt8.self),
parseTransition: fileTransition[file]
)
}
}
}
let endInstructions = getInstructionsExecuted()
let endDate = Date()
print("Time: \(endDate.timeIntervalSince(start) / Double(self.iterations) * 1000)ms")
if endInstructions != startInstructions {
// endInstructions == startInstructions only happens if we are on non-macOS
// platforms that don't support `getInstructionsExecuted`. Don't display anything.
print("Instructions: \(Double(endInstructions - startInstructions) / Double(self.iterations))")
}
}
}
|
