Go 语言性能优化

在谈到性能优化之前我们先看看,如何判断程序是否需要优化。

计算函数执行时间

有时候,能够知道一个函数计算执行消耗的时间是非常有意义的,尤其是在对比和基准测试中。最简单的一个办法就是在计算开始之前设置一个起始时间,再在计算结束时获取结束时间,最后取出它们的差值。想要实现这样的做法,可以使用 time 包中的 Now()Sub 函数:

  1. start := time.Now()
  2. longCalculation()
  3. end := time.Now()
  4. delta := end.Sub(start)
  5. fmt.Printf("longCalculation took this amount of time: %s\n", delta)

如果您对一段代码进行了所谓的优化,请务必对它们之间的效率进行对比再做出最后的判断。

通过内存缓存优化

当程序在进行大量的计算时,提升性能最直接有效的一种方式就是避免重复计算。通过在内存中缓存和重复利用相同计算的结果,称之为内存缓存。

最明显的例子就是生成 斐波那契数列 的程序:要计算数列中第 n 个数字,需要先得到之前两个数的值,但很明显绝大多数情况下前两个数的值都是已经计算过的。即每个更后面的数都是基于之前计算结果的重复计算。

而我们要做就是将第 n 个数的值存在数组中索引为 n 的位置,然后在数组中查找是否已经计算过,如果没有找到,再进行计算,这样就减少了程序的计算次数。

下面的例子就是依照这个原则实现:

  1. package main
  2. import (
  3. "fmt"
  4. "time"
  5. )
  6. const LIM = 41
  7. var fibs [LIM]uint64
  8. func main() {
  9. var result uint64 = 0
  10. start := time.Now()
  11. for i := 0; i < LIM; i++ {
  12. result = fibonacci(i)
  13. fmt.Printf("fibonacci(%d) is: %d\n", i, result)
  14. }
  15. end := time.Now()
  16. delta := end.Sub(start)
  17. fmt.Printf("longCalculation took this amount of time: %s\n", delta)
  18. }
  19. func fibonacci(n int) (res uint64) {
  20. // memoization: check if fibonacci(n) is already known in array:
  21. if fibs[n] != 0 {
  22. res = fibs[n]
  23. return
  24. }
  25. if n <= 1 {
  26. res = 1
  27. } else {
  28. res = fibonacci(n-1) + fibonacci(n-2)
  29. }
  30. fibs[n] = res
  31. return
  32. }

下面是计算到第 40 位数字的性能对比,具体对比可以自己去实践:

  1. 普通写法:4.730270
  2. 内存缓存:0.001000

从上面的对比可以看出,内存缓存的优势非常大,而且您还可以将它应用到其它类型的计算中,例如使用 map 而不是数组切片

内存缓存的技术在使用计算成本相对昂贵的函数时非常有用(不仅限于例子中的递归),譬如大量进行相同参数的运算。这种技术还可以应用于纯函数中,即相同输入必定获得相同输出的函数。