gorm 获取当前数据库连接

在使用 GORM 框架进行数据库操作时，获取当前数据库连接至关重要。gorm.open 返回的数据库变量即是数据库句柄，可以通过将其作为 *gorm.DB 传递给需要使用它的位置来获取数据库连接。这种方法有助于实现数据库连接的模块化和可重用性，从而简化代码结构并提高代码可维护性。

如何获取当前的数据库连接？

package main

import (
     "github.com/labstack/echo"
      "github.com/jinzhu/gorm"
    _ "github.com/jinzhu/gorm/dialects/mysql"
   )

 func main() {
     // echo instance
    e := echo.new()
    db, _ := gorm.open("mysql", "root:root@/golang")
    defer db.close()

    gorm.automigrate(&user.user{})
    e.logger.fatal(e.start(":4000"))
}

这是我想从数据库获取用户的控制器

package controllers

 import (
  "github.com/labstack/echo"
 )

  func login(c echo.Context) error {
  username := c.QueryParam("username")
  }

我如何获取 db 对象或者我需要再次 make gorm.open 一次？或者为数据库对象创建单例并导入它？

解决方案

gorm.open 返回的数据库变量是您的数据库句柄。您只需将它作为 *gorm.db 传递给任何您想要使用它的地方。

以下是如何将数据库句柄正确传递给用户控制器的示例：

type userrepository interface {
    create(user model.user) error
}

type userrepositorymysql struct {
    db *gorm.db
}

// the repository you pass to your controller to provide
// an interface with the database
func (u *userrepositorymysql) create(user model.user) error {
    u.db.create(&user)
    // handle errors here
    return nil
}

type usercontroller struct {
    users userrepository
}

func (u *usercontroller) create(ctx echo.context) {
    var user model.user

    err := ctx.bind(&user)
    // handle errors

    // validate user
    err := u.users.create(user)
}

func main() {
    e := echo.new()

    db, _ := gorm.open("mysql", "root:root@/golang")
    defer db.close()

    userrepo := repo.userrepositorymysql(db)
    usercontroller := controller.usercontroller(userrepo)
    ...

    e.post("/users", usercontroller.create)
    ...
    e.logger.fatal(e.start(":4000"))
}

我建议使用存储库接口，因为您稍后可能想要集成多个不同的数据库，并且它可以轻松模拟并测试您的控制器。

有两种主要方法传递连接对象。创建一个全局变量并将其作为您想要使用的参数传递到方法中：

var db *sql.db

func initdb(datasourcename string) {
    var err error
    db, err = sql.open("postgres", datasourcename)
    if err != nil {
        log.panic(err)
    }

    if err = db.ping(); err != nil {
        log.panic(err)
    }
}

或者您可以创建一个结构体，并在创建 db 实例后将其用作方法的方法接收器。

type DB struct {
   *sqlx.DB
}

func(db *DB) Get() (error)

但它应该是在初始化其实例一次后在函数内部使用的指针字段。

理论要掌握，实操不能落！以上关于《gorm 获取当前数据库连接》的详细介绍，大家都掌握了吧！如果想要继续提升自己的能力，那么就来关注golang学习网公众号吧！