在go语言中，准确地测量函数的执行时间对于性能分析和优化至关重要。go语言提供的defer关键字为我们提供了一种优雅且简洁的方法来实现这一目标。结合time包，我们可以轻松地获取函数的运行时间，并将其精确到毫秒级别。
利用 defer 关键字实现函数耗时统计
defer 语句用于延迟函数的执行，直到周围的函数返回。结合 time 包，我们可以记录函数开始和结束的时间，从而计算出函数的运行时间。以下是一种常见的实现方式：
package main
import (
"log"
"time"
)
func trace(s string) (string, time.Time) {
log.Println("START:", s)
return s, time.Now()
}
func un(s string, startTime time.Time) {
endTime := time.Now()
log.Println(" END:", s, "ElapsedTime in milliseconds:", endTime.Sub(startTime).Milliseconds())
}
func someFunction() {
defer un(trace("someFunction"))
// 模拟一些耗时操作
time.Sleep(100 * time.Millisecond)
}
func main() {
someFunction()
}代码解释:
- trace(s string) (string, time.Time) 函数记录函数的开始时间，并返回函数名和开始时间。
- un(s string, startTime time.Time) 函数计算函数的结束时间，并打印出函数的运行时间（毫秒）。
- 在 someFunction() 函数中，defer un(trace("someFunction")) 语句会在 someFunction() 函数返回之前执行 un 函数，从而计算出 someFunction() 的运行时间。
- time.Sleep(100 * time.Millisecond) 模拟一些耗时操作，以便观察耗时统计效果。
输出结果:
2023/10/27 10:00:00 START: someFunction 2023/10/27 10:00:00 END: someFunction ElapsedTime in milliseconds: 100
从输出结果可以看出，someFunction() 函数的运行时间约为 100 毫秒，与预期的睡眠时间相符。
立即学习“go语言免费学习笔记（深入）”；
注意事项
- 精度问题: 由于日志记录等操作也会消耗时间，因此这种方法测量的函数运行时间并非绝对精确。如果需要更高的精度，可以考虑直接使用 time.Now() 函数记录开始和结束时间，避免使用 log.Println() 函数。
- 并发安全: 在并发环境下，需要注意 trace 和 un 函数的并发安全问题。如果多个 goroutine 同时调用这些函数，可能会导致竞争条件。可以使用互斥锁（sync.Mutex）来保护这些函数。
总结
利用 defer 关键字和 time 包，可以简洁且有效地统计 Go 语言函数的执行耗时。这种方法适用于大多数场景，能够帮助开发者快速定位性能瓶颈，并进行优化。 尽管存在一定的精度问题，但对于大多数应用来说，其提供的精度已经足够。通过结合其他性能分析工具，我们可以更全面地了解程序的性能状况。
