qnc init

db/gorm.go

@@ -0,0 +1,58 @@
+package db
+
+import (
+	"fmt"
+	"gorm.io/driver/mysql"
+	"gorm.io/gorm"
+	"log"
+	"qnc-server/config"
+	"qnc-server/model/model"
+)
+
+var DB *gorm.DB
+
+func InitGorm() {
+	dsn := fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=utf8mb4&parseTime=True&loc=Local",
+		config.ConfigData.Database.Username,
+		config.ConfigData.Database.Password,
+		config.ConfigData.Database.Host,
+		config.ConfigData.Database.Port,
+		config.ConfigData.Database.DBName,
+	)
+	var err error
+	DB, err = gorm.Open(mysql.Open(dsn), &gorm.Config{})
+	if err != nil {
+		log.Fatalf("Could not connect to the database: %v", err)
+	}
+
+	if err := DB.AutoMigrate(
+		&model.User{},
+		&model.Authentication{},
+		&model.EvaluateMarriage{},
+		&model.Product{},
+		&model.PayOrder{},
+		&model.EntFeature{},
+		&model.Render{},
+		&model.CarInsurance{},
+		&model.CarMaintenance{},
+		&model.CarUnderName{},
+		&model.CarInsuranceInfo{},
+		&model.CarNewEnergyPower{},
+		&model.CarVinCheckInfo{},
+		&model.CarPersonCarVerify{},
+		&model.CarVehicleTransfer{},
+		&model.CarVehicleValuation{},
+		&model.CarRender{},
+		&model.VerifyRender{},
+		&model.VerifyCardName{},
+		&model.VerifyPhoneName{},
+		&model.VerifySkillCert{},
+		&model.VerifyBankCard{},
+		&model.GlobalNotification{},
+		&model.RenderShadow{},
+		&model.Query{},
+	); err != nil {
+		log.Fatalf("Could not migrate the database: %v", err)
+	}
+	InitRedis() // 初始化
+}

db/redis.go

@@ -0,0 +1,27 @@
+package db
+
+import (
+	"context"
+	"fmt"
+	"github.com/redis/go-redis/v9"
+	"log"
+	"qnc-server/config"
+)
+
+var RedisClient redis.UniversalClient
+
+func InitRedis() {
+	var client redis.UniversalClient
+	client = redis.NewClient(&redis.Options{
+		Addr:     config.ConfigData.Redis.Addr,
+		Password: config.ConfigData.Redis.Password,
+		DB:       config.ConfigData.Redis.DB,
+	})
+	pong, err := client.Ping(context.Background()).Result()
+	if err != nil {
+		log.Fatalf("redis connect ping failed, err:", err.Error())
+	} else {
+		fmt.Printf("redis start，redis connect ping response: %s", pong)
+		RedisClient = client
+	}
+}