-
Notifications
You must be signed in to change notification settings - Fork 4
Expand file tree
/
Copy pathSTTimeProfiler.swift
More file actions
141 lines (129 loc) · 5.13 KB
/
Copy pathSTTimeProfiler.swift
File metadata and controls
141 lines (129 loc) · 5.13 KB
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
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
//
// STTimeProfiler.swift
// STBaseProject
//
// Created by 寒江孤影 on 2024/01/01.
//
import Foundation
import QuartzCore
public enum STTimeProfiler {
private static let lock = NSLock()
private static var startTimes: [String: CFTimeInterval] = [:]
/// 开始计时
/// - Parameter tag: 计时任务标识，用于区分不同的计时任务
public static func st_start(tag: String = "default") {
self.lock.lock()
defer { self.lock.unlock() }
let startTime = CACurrentMediaTime()
self.startTimes[tag] = startTime
STLog("⏱️ [\(tag)] 开始计时")
}
/// 结束计时并打印耗时
/// - Parameters:
/// - tag: 计时任务标识，默认为 "default"
/// - message: 自定义消息，会显示在耗时信息中
public static func st_end(tag: String = "default", message: String? = nil) {
self.lock.lock()
defer { self.lock.unlock() }
guard let startTime = self.startTimes[tag] else {
STLog("⚠️ [\(tag)] 未找到对应的开始时间，请先调用 st_start(tag:)")
return
}
let endTime = CACurrentMediaTime()
let duration = endTime - startTime
self.startTimes.removeValue(forKey: tag)
let messageText = message.map { " - \($0)" } ?? ""
let durationText = self.st_formatDuration(duration)
STLog("✅ [\(tag)] 耗时: \(durationText)\(messageText)")
}
/// 获取当前耗时（不结束计时）
/// - Parameter tag: 计时任务标识，默认为 "default"
/// - Returns: 耗时（秒），如果未找到开始时间则返回 nil
public static func st_elapsedTime(tag: String = "default") -> Double? {
self.lock.lock()
defer { self.lock.unlock() }
guard let startTime = self.startTimes[tag] else {
return nil
}
let currentTime = CACurrentMediaTime()
return currentTime - startTime
}
/// 打印当前耗时（不结束计时）
/// - Parameters:
/// - tag: 计时任务标识，默认为 "default"
/// - message: 自定义消息
public static func st_logElapsed(tag: String = "default", message: String? = nil) {
guard let elapsed = self.st_elapsedTime(tag: tag) else {
STLog("⚠️ [\(tag)] 未找到对应的开始时间，请先调用 st_start(tag:)")
return
}
let messageText = message.map { " - \($0)" } ?? ""
let durationText = self.st_formatDuration(elapsed)
STLog("⏳ [\(tag)] 当前耗时: \(durationText)\(messageText)")
}
/// 执行代码块并测量耗时
/// - Parameters:
/// - tag: 计时任务标识，默认为 "default"
/// - message: 自定义消息
/// - block: 要执行的代码块
/// - Returns: 代码块的返回值
@discardableResult
public static func st_measure<T>(tag: String = "default", message: String? = nil, block: () throws -> T) rethrows -> T {
self.st_start(tag: tag)
defer {
self.st_end(tag: tag, message: message)
}
return try block()
}
/// 执行异步代码块并测量耗时
/// - Parameters:
/// - tag: 计时任务标识，默认为 "default"
/// - message: 自定义消息
/// - block: 要执行的异步代码块
public static func st_measureAsync(tag: String = "default", message: String? = nil, block: @escaping () async throws -> Void) {
self.st_start(tag: tag)
Task {
do {
try await block()
self.st_end(tag: tag, message: message)
} catch {
self.st_end(tag: tag, message: "\(message ?? "") - 错误: \(error.localizedDescription)")
}
}
}
/// 清除所有计时任务
public static func st_clearAll() {
self.lock.lock()
defer { self.lock.unlock() }
self.startTimes.removeAll()
STLog("🧹 已清除所有计时任务")
}
/// 清除指定标签的计时任务
/// - Parameter tag: 计时任务标识
public static func st_clear(tag: String) {
self.lock.lock()
defer { self.lock.unlock() }
self.startTimes.removeValue(forKey: tag)
STLog("🧹 已清除计时任务: \(tag)")
}
/// 格式化耗时显示
/// - Parameter duration: 耗时（秒）
/// - Returns: 格式化后的字符串
private static func st_formatDuration(_ duration: Double) -> String {
if duration < 0.001 {
// 小于1毫秒，显示微秒
return String(format: "%.2f μs", duration * 1_000_000)
} else if duration < 1.0 {
// 小于1秒，显示毫秒
return String(format: "%.2f ms", duration * 1000)
} else if duration < 60.0 {
// 小于1分钟，显示秒
return String(format: "%.3f s", duration)
} else {
// 大于1分钟，显示分钟和秒
let minutes = Int(duration / 60)
let seconds = duration.truncatingRemainder(dividingBy: 60)
return String(format: "%d m %.2f s", minutes, seconds)
}
}
}