Go 中应该在循环中的哪里放置 defer 语句才能正确释放资源？

如何在循环中使用 Defer 来释放资源：最佳方法

在循环中涉及数据库查询的上下文中，问题如下：延迟语句的最佳放置以确保正确的资源释放。考虑以下循环：

for rows.Next() {

   fields, err := db.Query(.....)
   if err != nil {
      // ...
   }
   defer fields.Close()

   // do something with `fields`

}

出现延迟放置的两个选项：

1. 在循环体内延迟：

   for rows.Next() {
   
      fields, err := db.Query(.....)
      if err != nil {
         // ...
      }
   
      // do something with `fields`
   }
   
   defer fields.Close()

2. 循环后推迟body:

   for rows.Next() {
   
      fields, err := db.Query(.....)
      if err != nil {
         // ...
      }
      
      // do something with `fields`
      
      defer fields.Close()
   }

了解延迟执行

要选择最佳方法，我们必须首先了解延迟的行为。延迟函数不仅会延迟到周围函数返回，而且即使函数由于异常（例如恐慌）而突然终止也会执行。这是即使在特殊情况下也能确保资源释放的重要机制。

延迟放置的潜在问题

在循环内放置延迟可能会阻碍循环终止时的资源释放。如果循环由于循环内处理的错误而提前退出，则 deferred fields.Close() 调用将不会被执行。

相反，将 defer 放在循环体之后可以保证无论如何释放资源循环退出。但是，这种方法会延迟资源清理，直到整个循环完成之后，这可能并不适合所有场景。

最佳解决方案：匿名或命名函数包装器

至为了解决这两个问题，建议的解决方案是将资源分配和释放封装在匿名或命名函数中。通过这样做，可以在函数内使用 defer 以确保函数返回时释放资源。

例如：

// Anonymous function wrapper
for rows.Next() {
    func() {
        fields, err := db.Query(...)
        if err != nil {
            // Handle error and return
            return
        }
        defer fields.Close()

        // do something with `fields`
    }()
}

// Named function wrapper
func foo(rs *db.Rows) {
    fields, err := db.Query(...)
    if err != nil {
        // Handle error and return
        return
    }
    defer fields.Close()

    // do something with `fields`
}

for rows.Next() {
    foo(rs)
}

这种方法允许在资源不再可用时立即释放即使在例外情况下也是需要的。此外，如果目标是在出现第一个错误时终止循环，则可以从包装函数返回错误并进行相应处理：

func foo(rs *db.Rows) error {
    fields, err := db.Query(...)
    if err != nil {
        return fmt.Errorf("db.Query error: %w", err)
    }
    defer fields.Close()

    // do something with `fields`
    return nil
}

for rows.Next() {
    if err := foo(rs); err != nil {
        // Handle error and return
        return
    }
}

使用 Rows.Close() 进行错误处理

需要注意的是 Rows.Close() 返回错误。要处理此错误，可以使用延迟调用 Rows.Close() 的匿名函数：

func foo(rs *db.Rows) (err error) {
    fields, err := db.Query(...)
    if err != nil {
        return fmt.Errorf("db.Query error: %w", err)
    }
    defer func() {
        if err = fields.Close(); err != nil {
            err = fmt.Errorf("Rows.Close() error: %w", err)
        }
    }()

    // do something with `fields`
    return nil
}

以上是Go 中应该在循环中的哪里放置 defer 语句才能正确释放资源？的详细内容。更多信息请关注PHP中文网其他相关文章！