在循环中延迟释放资源的正确方法？

Question

我需要在循环中对数据库进行 SQL 查询：

for rows.Next() {

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

   // do something with `fields`

}

什么会更好：保持原样或在循环后移动defer：

for rows.Next() {

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

   // do something with `fields`
}

defer fields.Close()

或者是其他东西 ？

Answer 1

延迟函数的执行不仅被延迟，延迟到周围函数返回的那一刻，即使封闭函数突然终止，它也会执行，例如恐慌。 Spec: Defer statements:

“延迟”语句调用一个函数，该函数的执行被推迟到周围函数返回的那一刻，或者因为周围函数执行了return statement，到达了它的function body的末尾，或者因为对应的goroutine是panicking.

每当您创建一个提供正确关闭/处置它的方法的值或资源时，您应该始终使用 defer 语句来确保它被释放，即使您的其他代码发生恐慌以防止内存或其他系统资源泄漏。

确实，如果你在循环中分配资源，你不应该简单地使用defer，因为那样就不会释放资源尽早和应该（在每次迭代结束时），仅在 for 语句之后（仅在所有迭代之后）。

你应该做的是，如果你有一个分配此类资源的 sn-p，将其包装在一个函数中——匿名函数或命名函数——，在该函数中你可以使用defer，资源将被释放为一旦不再需要它们，重要的是即使您的代码中存在可能会恐慌的错误。

例子：

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

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

或者如果放入命名函数：

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)
}

此外，如果您想在第一个错误时终止，您可以从 foo() 返回错误：

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() 返回一个错误，当使用 defer 调用时该错误被丢弃。如果我们想检查返回的错误，我们可以使用这样的匿名函数：

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
}

Answer 2

defer 的全部要点是它在函数返回之前不会执行，因此放置它的合适位置应该是在您要关闭的资源打开之后立即执行。但是，由于您是在循环内创建资源，因此根本不应该使用 defer - 否则，在函数退出之前，您不会关闭在循环内创建的任何资源，因此它们会堆积起来直到然后。相反，您应该在每次循环迭代结束时关闭它们，没有defer:

for rows.Next() {

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

   // do something with `fields`

   fields.Close()
}

Answer 3

你可以构造一个局部函数来解决这个问题

    for i := 0; i < 5; i++ {
        func() {
            f, err := os.Open("/path/to/file")
            if err != nil {
                log.Fatal(err)
            } else {
                defer f.Close()
            }
        }()
    }