Go 语言性能优化技巧

简介: 在Go语言中优化性能涉及数字字符串转换(如用`strconv.Itoa()`代替`fmt.Sprintf()`)、避免不必要的字符串到字节切片转换、预分配切片容量、使用`strings.Builder`拼接、有效利用并发(`goroutine`和`sync.WaitGroup`)、减少内存分配、对象重用(`sync.Pool`)、无锁编程、I/O缓冲、正则预编译和选择高效的序列化方法。这些策略能显著提升代码执行效率和系统资源利用率。

在软件开发中,性能往往是我们需要特别关注的方面之一。对于使用 Go 语言的开发者而言,如何编写高性能的代码是一个重要的考虑点。

今天,我将分享一些在 Go 语言开发中可以采取的性能优化策略,希望能帮助大家写出更高效的程序。

数字与字符串的转换

在处理数字和字符串的转换时,strconv.Itoa() 通常比 fmt.Sprintf() 更加高效。

示例代码

import (
    "fmt"
    "strconv"
)

// 使用 strconv.Itoa() 进行转换
num := 123
str := strconv.Itoa(num)
fmt.Println(str)

// 使用 fmt.Sprintf() 进行转换
str2 := fmt.Sprintf("%d", num)
fmt.Println(str2)

strconv.Itoa() 的速度大约是 fmt.Sprintf() 的两倍,因为后者需要解析格式字符串,而前者直接进行转换。

字符串与字节切片的转换

尽可能避免将 string 转换为 []byte。这个操作会增加额外的内存拷贝,从而影响性能。

处理切片时的性能优化

在使用 for-loopSlice 进行 append 操作时,请先分配足够的容量。

示例代码

// 不推荐的做法
var nums []int
for i := 0; i < 1000; i++ {
   
    nums = append(nums, i)
}

// 推荐的做法
nums := make([]int, 0, 1000) // 预分配足够的容量
for i := 0; i < 1000; i++ {
   
    nums = append(nums, i)
}

通过预先分配足够的容量,可以避免在 append 过程中发生内存重新分配。

字符串的拼接

使用 strings.Builder 对字符串进行拼接,性能远高于使用 ++=

示例代码

import "strings"

// 使用 strings.Builder 进行字符串拼接
var builder strings.Builder
for i := 0; i < 100; i++ {
   
    builder.WriteString("str")
}
result := builder.String()
fmt.Println(result)

并发编程

利用 Go 语言的强大并发特性。使用并发的 goroutine 并配合 sync.WaitGroup 进行同步,可以显著提升程序的执行效率。

示例代码

import (
    "sync"
)

var wg sync.WaitGroup

for i := 0; i < 10; i++ {
   
    wg.Add(1)
    go func(i int) {
   
        defer wg.Done()
        // 你的代码逻辑
    }(i)
}
wg.Wait()

慎用内存分配

避免在性能敏感的代码(热代码)中频繁进行内存分配,以减少垃圾回收的压力。

对象重用

使用 sync.Pool 来重用对象,可以有效降低内存分配的频率和垃圾回收的压力。

pool := sync.Pool{
   
    New: func() interface{
   } {
   
        return new(MyStruct)
    },
}
myVar := pool.Get().(*MyStruct)
// 使用 myVar 后,记得放回池中
pool.Put(myVar)

无锁编程

尽可能采用无锁操作,比如使用 sync/atomic 包中的原子操作,以避免锁的开销。

// 使用 Atomic 包进行原子操作
var counter int64
sync/atomic.AddInt64(&counter, 1)

I/O 缓冲

I/O 操作是非常慢的,使用缓冲 I/O(如 bufio.NewWriter()bufio.NewReader())可以显著提升 I/O 性能。

// 使用 bufio 包进行 I/O 缓冲
writer := bufio.NewWriter(file)
defer writer.Flush()
writer.WriteString("Hello, World")

正则表达式优化

在循环内部使用正则表达式时,应先用 regexp.Compile() 将其编译,以提升性能。

re := regexp.MustCompile("some-regex-pattern")
matches := re.FindAllString("some string", -1)

序列化性能选择

如果对性能有高要求,考虑使用 protobufmsgp 等序列化方案,而不是 json,因为 json 序列化涉及反射,性能较低。

Map 使用技巧

在使用 map 时,整型作为键的性能会比字符串高,这是因为整型的比较操作比字符串比较要快。

// 使用整型作为 Map 的 key
mapInt := map[int]string{
   1: "one", 2: "two"}

总结

  • 如果需要把数字转换成宇符串,使用 strconv.ltoa()fmt.Sprintf() 要快一倍左右。
  • 尽可能避免把 String 转成 []Byte,这个转换会导致性能下降。
  • 如果在 for-loop 里对某个 Slice 使用 append(),请先把 Slice 的容量扩充到位,这样可以避免内存重新分配以及系统自动按 2 的 N 次方幂进行扩展但又用不到的情况,从而避免浪费内存。
  • 使用 StringBuffer 或是 StringBuild 来拼接字符串,性能会比使用 ++= 高三到四个数量级。
  • 尽可能使用并发的 goroutine 然后使用 sync.WaitGroup 来同步分片操作。
  • 避免在热代码中进行内存分配,这样会导致 gc 很忙。
  • 尽可能使用 sync.Pool 来重用对象。
  • 使用 lock-free 的操作,避免使用 mutex,尽可能使用 sync/Atomic 包。
  • 使用 I/O 缓冲,I/O 是个非常非常慢的操作,使用 bufio.NewWrite()bufio.NewReader() 可以带来更高的性能。
  • 对于在 for-loop 里的固定的正则表达式,一定要使用 regexp.Compile() 编译正则表达式。性能会提升两个数量级。
  • 如果你需要更高性能的协议,就要考虑使用 protobufmsgp 而不是 json,因为 json 的序列化和反序列化里使用了反射。
  • 你在使用 Map 的时候,使用整型的 key 会比字符串的要快,因为整型比较比字符串比较要快。

通过上述的技巧,我们可以在编写 Go 程序时更加注重性能。从字符串处理到并发控制,再到内存管理,每一个环节都有提升效率的空间。希望这些技巧能够帮助你在开发过程中写出更高效、更优化的代码。

相关文章
|
19天前
|
存储 Go 索引
go语言中数组和切片
go语言中数组和切片
31 7
|
18天前
|
Go 开发工具
百炼-千问模型通过openai接口构建assistant 等 go语言
由于阿里百炼平台通义千问大模型没有完善的go语言兼容openapi示例,并且官方答复assistant是不兼容openapi sdk的。 实际使用中发现是能够支持的,所以自己写了一个demo test示例,给大家做一个参考。
|
19天前
|
程序员 Go
go语言中结构体(Struct)
go语言中结构体(Struct)
93 71
|
18天前
|
存储 Go 索引
go语言中的数组(Array)
go语言中的数组(Array)
100 67
|
19天前
|
存储 Go
go语言中映射
go语言中映射
32 11
|
21天前
|
Go
go语言for遍历映射(map)
go语言for遍历映射(map)
31 12
|
20天前
|
Go 索引
go语言使用索引遍历
go语言使用索引遍历
28 9
|
20天前
|
Go 索引
go语言使用range关键字
go语言使用range关键字
25 7
|
20天前
|
Go 索引
go语言修改元素
go语言修改元素
26 6
|
10天前
|
Go 数据安全/隐私保护 UED
优化Go语言中的网络连接:设置代理超时参数
优化Go语言中的网络连接:设置代理超时参数