golang函数性能优化与并发编程

为了提升 go 语言函数性能，优先使用经过优化的 go 标准库函数；避免过度分配，预分配变量或使用缓存。并发编程中，使用 goroutine 实现并发；通过通道在 goroutine 之间安全通信；使用原子操作确保并发访问共享变量时的安全性。

Go 语言函数性能优化与并发编程

性能优化

1. 使用标准库

优先使用 Go 标准库中的函数，因为它们经过优化和广泛测试。例如，使用 sort.Sort() 而不是自己实现排序算法。

package main

import "sort"

func main() {
    s := []int{3, 1, 2}
    sort.Ints(s)
    _ = s // 使用 s 以防止编译器优化为常量
}

2. 避免分配

过度分配会影响性能。通过预分配变量或使用缓存来减少分配。

package main

import (
    "bufio"
    "os"
)

func main() {
    f, err := os.Open("file.txt")
    if err != nil {
        // 处理错误
    }

    // 使用 bufio 包预分配空间
    scanner := bufio.NewScanner(f)
    for scanner.Scan() {
        // 处理扫描的行
    }
}

并发编程

1. Goroutine

Goroutine 轻量级线程，可用于实现并发。使用 go 关键字创建 goroutine。

package main

func main() {
    go func() {
        // 并发执行的代码
    }()
}

2. 通道

通道用于在 goroutine 之间安全地通信。一个 goroutine 从通道发送数据，另一个 goroutine 从通道接收数据。

package main

import "sync"

func main() {
    // 创建通道
    ch := make(chan int)

    // Goroutine 发送数据
    go func() {
        ch <- 1
    }()

    // Goroutine 接收数据
    var wg sync.WaitGroup
    wg.Add(1)
    go func() {
        defer wg.Done()
        v := <-ch
        _ = v // 使用 v 以防止编译器优化为常量
    }()
    wg.Wait()
}

3. 原子操作

原子操作可确保并发访问共享变量时的安全性。使用 sync/atomic 包中提供的函数，例如 atomic.AddInt64()。

package main

import "sync/atomic"

func main() {
    var counter int64

    // 多个 goroutine 并发更新计数器
    for i := 0; i < 1000; i++ {
        go func() {
            atomic.AddInt64(&counter, 1)
        }()
    }

    // 等待所有 goroutine 完成
    // ...

    _ = counter // 使用 counter 以防止编译器优化为常量
}