Setting *sql.DB without using Open

[yulrizka]

I've design my function to accept *sql.DB as the parameter. This way it's general enough if in the feature we want to use basic driver or switch between ORM.

The way I do this right now is something like this.

func Users(source *sql.DB) (users []User) {
    db := initGorm(source)
    db.Table("user").Find(&users)
}

func initGorm(source *sql.DB) *gorm.DB {
    result := gorm.SetDB("mysql", db)
    return &result
}

// on gorm/main.go
func SetDB(driver string, dbSource *sql.DB) (db DB) {
   db.db = dbSource
   db.dialect = dialect.New(driver)
   db.tagIdentifier = "sql"
   db.parent = &db
   return
}

Is this the right way to do this ? or gorm already have functionality that support this ?

[jinzhu]

For me, I think this is not the best way to do it, I am doing it like this:

// xxx/db.go
var DB gorm.DB
func init() {
  DB = gorm.Open("mysql", "xxxxxx")
}

// xxx/user.go
import . "xxx/db.go"

func Users() (users []User) {
  DB.Table("user").Find(users)
}

Then if you want to change to any other ORM, you just need to change the db package if the syntax is same.