rubbishclass-srv/rubbish/database.go

32 lines
447 B
Go

package rubbish
import (
"log"
"time"
_ "github.com/go-sql-driver/mysql"
"github.com/jinzhu/gorm"
)
var (
Addr *string
DB *gorm.DB
)
func Init() {
if *Addr != "" {
var err error
DB, err = gorm.Open("mysql", *Addr)
if err != nil {
log.Fatal(err)
}
DB.DB().SetMaxIdleConns(10)
DB.DB().SetMaxOpenConns(100)
DB.DB().SetConnMaxLifetime(10 * time.Minute)
err = DB.DB().Ping()
if err != nil {
log.Fatal(err)
}
}
}