package main

import (
	"fmt"
	"log"
	"os"
	"time"

	"gorm.io/driver/mysql"
	"gorm.io/gorm"
	gormLogger "gorm.io/gorm/logger"
)

//ConnV2 gorm v2版本的连接
func DBConn() (db *gorm.DB) {
	//
	newLogger := gormLogger.New(
		log.New(os.Stdout, "\r\n", log.LstdFlags), // io writer
		gormLogger.Config{
			SlowThreshold: 2 * time.Second,   // 慢 SQL 阈值
			LogLevel:      gormLogger.Silent, // Log level
			Colorful:      true,              // 禁用彩色打印
		},
	)
	//
	connStr := fmt.Sprintf("%s:%s@tcp(%s)/%s?charset=utf8mb4&parseTime=True&loc=Local&timeout=5s", "rhythm", "lGoGOICMOKX50qn0sd5r", "10.21.33.7:3306", "rhythm_backend")
	// 设置 insert into时，自动批量10000的插入
	db, err := gorm.Open(mysql.Open(connStr), &gorm.Config{Logger: newLogger, CreateBatchSize: 10000})
	if err != nil {
		fmt.Println("db conn err", err)
		panic(err)
	}
	return
}