init

utils/system_init.go

@@ -0,0 +1,40 @@
+package utils
+
+import (
+	"fmt"
+	"github.com/spf13/viper"
+	"gorm.io/driver/mysql"
+	"gorm.io/gorm"
+	"gorm.io/gorm/logger"
+	"log"
+	"os"
+	"time"
+)
+
+func InitConfig() {
+	viper.SetConfigName("app")
+	viper.AddConfigPath("config")
+	err := viper.ReadInConfig()
+	if err != nil {
+		fmt.Println(err)
+	}
+	fmt.Println("config app init")
+
+}
+func InitMySQL() *gorm.DB {
+	//自定义日子模板 打印SQL语句
+	newLogger := logger.New(
+		log.New(os.Stdout, "\r\n", log.LstdFlags),
+		logger.Config{
+			SlowThreshold: time.Second, //慢sql日志
+			LogLevel:      logger.Info, //级别
+			Colorful:      true,        //颜色
+
+		})
+	db, err := gorm.Open(mysql.Open(viper.GetString("mysql.dns")), &gorm.Config{Logger: newLogger})
+	if err != nil {
+		panic("failed to connect database")
+	}
+	fmt.Println("config mysql init")
+	return db
+}