package appinit

import (
	"fmt"
	"gitee.com/aywc_1/aywc_judge/src/configuration"
	_ "github.com/go-sql-driver/mysql"
	"github.com/jmoiron/sqlx"
	"net/url"
)

var db *sqlx.DB
var sdDB *sqlx.DB

func DBInit() error {
	userName := configuration.GetYamlValue([]string{"config", "mysql", "username"}, 0, "root").(string)
	password := configuration.GetYamlValue([]string{"config", "mysql", "password"}, 0, "aywc").(string)
	host := configuration.GetYamlValue([]string{"config", "mysql", "host"}, 0, "47.116.71.194").(string)
	port := configuration.GetYamlValue([]string{"config", "mysql", "port"}, 0, 3306).(int)
	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))
	var err error
	db, err = sqlx.Connect("mysql",
		dsn)
	if err != nil {
		return err
	}
	db.SetMaxOpenConns(10)
	db.SetMaxIdleConns(5)
	return nil
}

func SDDBInit() error {
	userName := configuration.GetYamlValue([]string{"config", "mysql", "username"}, 0, "root").(string)
	password := configuration.GetYamlValue([]string{"config", "mysql", "password"}, 0, "aywc").(string)
	host := configuration.GetYamlValue([]string{"config", "mysql", "host"}, 0, "47.116.71.194").(string)
	port := configuration.GetYamlValue([]string{"config", "mysql", "port"}, 0, 3306).(int)
	timezone := "'Asia/Shanghai'"
	dsn := fmt.Sprintf("%s:%s@tcp(%s:%d)/sdcc25?charset=utf8mb4&parseTime=True&loc=Local&time_zone=%s", userName, password, host, port, url.QueryEscape(timezone))
	var err error
	db, err = sqlx.Connect("mysql",
		dsn)
	if err != nil {
		return err
	}
	db.SetMaxOpenConns(10)
	db.SetMaxIdleConns(5)
	return nil
}
func GetDB() *sqlx.DB {
	return db
}

func GetSdDB() *sqlx.DB {
	return sdDB
}