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)
		}
	}
}