mirror of
https://github.com/mgerb/go-discord-bot
synced 2026-01-09 16:42:48 +00:00
28 lines
471 B
Go
28 lines
471 B
Go
package db
|
|
|
|
import (
|
|
"github.com/jinzhu/gorm"
|
|
|
|
// database driver for sqlite
|
|
_ "github.com/jinzhu/gorm/dialects/sqlite"
|
|
)
|
|
|
|
var conn *gorm.DB
|
|
|
|
// Init - initialize database
|
|
func Init(migrations ...interface{}) {
|
|
var err error
|
|
conn, err = gorm.Open("sqlite3", "data.db")
|
|
|
|
if err != nil {
|
|
panic("failed to connect database")
|
|
}
|
|
conn.DB().SetMaxIdleConns(1)
|
|
conn.AutoMigrate(migrations...)
|
|
}
|
|
|
|
// GetConn - get db connection
|
|
func GetConn() *gorm.DB {
|
|
return conn
|
|
}
|