在 go 框架中识别性能瓶颈时，常见的瓶颈包括内存泄漏、数据库连接池不当、goroutine 泄漏、http 请求处理不当和缓存使用不当。针对这些瓶颈，解决方案分别为：使用 defer 关闭文件描述符、使用数据库连接池、使用 waitgroup 或 context.context 等待 goroutine 完成、考虑使用中间件或并发并行处理任务、以及使用缓存机制。

Go 框架中的性能瓶颈识别和解决

简介

在开发使用 Go 框架的高性能应用程序时，识别和解决性能瓶颈至关重要。本文将研究 Go 框架中常见的性能瓶颈，并提供解决这些瓶颈的方法，包括实际示例。

立即学习“go语言免费学习笔记（深入）”；

常见性能瓶颈

• 内存泄漏：未释放不再使用的分配的内存，最终导致内存不足。
• 数据库连接池不当：当应用程序创建过多的数据库连接时，导致数据库资源紧张。
• Goroutine 泄漏：Go 例程在完成其任务后未能关闭，导致系统资源消耗过多。
• HTTP 请求处理不当：HTTP 请求处理程序处理请求的方式不高效，导致响应时间变长。
• 缓存使用不当：未能有效使用缓存来减少数据库查询或其他昂贵操作的调用次数。

实战案例

内存泄漏

// 编写代码时忘记关闭文件描述符
func processFile(file *os.File) {
  // ...
}

解决方案：使用 defer 确保在函数返回时关闭文件。

Golang Uber Dig

使用 uber-go/dig 在 Golang 中实现依赖注入——基于反射容器，提供 Provide/Invoke、dig.In/dig.Out 参数与结果对象，支持命名依赖等

下载

func processFile(file *os.File) {
  defer file.Close()
  // ...
}

数据库连接池不当

// 在处理每个请求时创建新的数据库连接
func handler(w http.ResponseWriter, r *http.Request) {
  db, err := sql.Open("...")
  if err != nil {
    // ...
  }
  // ...
}

解决方案：使用 database/sql 包创建数据库连接池并复用连接。

var db *sql.DB

func init() {
  db, err := sql.Open("...")
  if err != nil {
    // ...
  }
}

func handler(w http.ResponseWriter, r *http.Request) {
  rows, err := db.Query("...")
  if err != nil {
    // ...
  }
  // ...
}

Goroutine 泄漏

// 创建 Goroutine 但忘记等待其完成
func handleRequest(w http.ResponseWriter, r *http.Request) {
  go func() {
    // 处理请求
  }()
}

解决方案：使用 WaitGroup 或 context.Context 来等待 Goroutine 完成。

func handleRequest(w http.ResponseWriter, r *http.Request) {
  var wg sync.WaitGroup
  wg.Add(1)

  go func() {
    defer wg.Done()
    // 处理请求
  }()

  wg.Wait()
}

HTTP 请求处理不当

// 处理请求时进行大量的计算或 IO 操作
func handler(w http.ResponseWriter, r *http.Request) {
  // 耗时的计算
  value := calculateSomething()

  // IO 操作
  bytes, err := ioutil.ReadAll(r.Body)
  if err != nil {
    // ...
  }

  // ...
}

解决方案：考虑使用中间件或并发来并行处理任务。

缓存使用不当

// 未使用缓存
func getData() []byte {
  // 从数据库获取数据
  rows, err := db.Query("...")
  if err != nil {
    // ...
  }
  // ...

  return data
}

解决方案：使用 sync.Map 或外部缓存服务来缓存获取到的数据。

var cache = sync.Map{}

func getData() []byte {
  data, ok := cache.Load(key)
  if ok {
    return data.([]byte)
  }

  // 从数据库获取数据
  rows, err := db.Query("...")
  if err != nil {
    // ...
  }
  // ...

  cache.Store(key, data)

  return data
}