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

通过llama.cpp与羊驼聊天的网页界面- 详解 Serge 的启动使用

 

在 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 释放资源,可以有效减少资源句柄的占用。

 

Refer:Golang_defer陷阱

Link:https://www.cnblogs.com/farwish/p/12721687.html

posted on 2020-04-17 18:26  ercom  阅读(6190)  评论(0编辑  收藏  举报