package configuration

import (
	"fmt"
	"github.com/jinzhu/gorm"
	"log"
	"net/url"
)

type DBConfig struct {
}

func NewDBConfig() *DBConfig {
	return &DBConfig{}
}

func (this *DBConfig) GormDB() *gorm.DB {
	userName := GetYamlValue([]string{"config", "mysql", "username"}, 0, "root").(string)
	password := GetYamlValue([]string{"config", "mysql", "password"}, 0, "aywc").(string)
	host := GetYamlValue([]string{"config", "mysql", "host"}, 0, "47.116.71.194").(string)
	port := GetYamlValue([]string{"config", "mysql", "port"}, 0, 3306).(int)
	//db,err:=gorm.Open("mysql","root:root@tcp(192.168.101.95:3306)/aywc?charset=utf8mb4&parseTime=True&loc=Local")
	//db,err:=gorm.Open("mysql","root:root@tcp(192.168.31.32:3306)/aywc?charset=utf8mb4&parseTime=True&loc=Local")
	timezone := "'Asia/Shanghai'"
	dsn := fmt.Sprintf("%s:%s@tcp(%s:%d)/aywc?charset=utf8mb4&parseTime=True&loc=Local&time_zone=%s", userName, password, host, port, url.QueryEscape(timezone))
	db, err := gorm.Open("mysql", dsn)
	if err != nil {
		log.Fatal(err)
	}
	db.SingularTable(true)
	db.DB().SetMaxIdleConns(50)
	db.DB().SetMaxOpenConns(100)
	db.LogMode(true)
	return db
}