Golang defer 与 MySQL 连接关闭的陷阱 (database is closed)

在 golang 某些 orm 中，你经常会看到这种用法：

func main() {
  db, err := gorm.Open("sqlite3", "test.db")
  if err != nil {
    panic("failed to connect database")
  }
  defer db.Close()

  // db 操作
}

在主逻辑中使用 defer 最后关闭连接是没有问题的。

如果是在模块化开发，我们把一些数据库连接的操作封装到函数中，就不能带有 defer，比如：

func Connect() *gorm.DB {
    db, err := gorm.Open("mysql", "root:123456@(xx.xx.xx.xx:3306)/dbname?charset=utf8mb4&parseTime=True&loc=Local")
    if err != nil {
        panic("MySql Connect error")
    }
    defer db.Close()

    return db
}

由于 defer 是在逻辑的最后执行的，实际上到 return 的时候，返回的已经是 null。

继续操作会提示：sql: database is closed

那么我们可以把 defer db.Close() 放在具体的 repository 逻辑中。

合理使用 defer 释放资源，可以有效减少资源句柄的占用。