🎨 优化写法

client/mysql.go

@@ -4,6 +4,7 @@ import (
 	"go-wechat/config"
 	"gorm.io/driver/mysql"
 	"gorm.io/gorm"
+	"gorm.io/gorm/logger"
 	"log"
 )
 
@@ -17,7 +18,13 @@ func InitMySQLClient() {
 		DontSupportRenameIndex:  true, // 重命名索引时采用删除并新建的方式
 		DontSupportRenameColumn: true, // 用 `change` 重命名列
 	}
-	conn, err := gorm.Open(mysql.New(mysqlConfig))
+
+	// gorm 配置
+	gormConfig := gorm.Config{
+		Logger: logger.Default.LogMode(logger.Info),
+	}
+
+	conn, err := gorm.Open(mysql.New(mysqlConfig), &gormConfig)
 	if err != nil {
 		log.Panicf("初始化MySQL连接失败, 错误信息: %v", err)
 	} else {

handler/sys_message.go

@@ -6,7 +6,6 @@ import (
 	"go-wechat/entity"
 	"go-wechat/model"
 	"go-wechat/utils"
-	"log"
 )
 
 // handleNewUserJoin
@@ -15,12 +14,8 @@ import (
 func handleNewUserJoin(m model.Message) {
 	// 判断是否开启迎新
 	var count int64
-	err := client.MySQL.Model(&entity.Friend{}).
+	client.MySQL.Model(&entity.Friend{}).Where("enable_welcome IS TRUE").Where("wxid = ?", m.FromUser).Count(&count)
-		Where("enable_welcome IS TRUE").
+	if count < 1 {
-		Where("wxid = ?", m.FromUser).
-		Count(&count).Error
-	if err != nil {
-		log.Printf("查询是否开启迎新失败: %s", err.Error())
 		return
 	}
 	if count < 1 {