From 41aa7275b9b9567a438647fb2893099f3a23ca1f Mon Sep 17 00:00:00 2001
From: junleea <354425203@qq.com>
Date: Sun, 9 Jun 2024 11:22:58 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E7=99=BB=E5=BD=95=E5=AF=BC?=
 =?UTF-8?q?=E8=87=B4key=E9=94=99=E8=AF=AF=E5=81=B6=E5=8F=91=EF=BC=8Cdb?=
 =?UTF-8?q?=E7=9A=84=E8=87=AA=E5=8A=A8=E8=BF=81=E7=A7=BB?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 dao/db.go       | 6 +++++-
 handler/user.go | 4 ++--
 worker/redis.go | 1 +
 3 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/dao/db.go b/dao/db.go
index 3e25f36..5eb3fec 100644
--- a/dao/db.go
+++ b/dao/db.go
@@ -1,6 +1,7 @@
 package dao
 
 import (
+	"fmt"
 	"gorm.io/driver/mysql"
 	"gorm.io/gorm"
 )
@@ -14,7 +15,10 @@ func Init() {
 	if err != nil {
 		panic("failed to connect database")
 	}
-	//db.Migrate(&User{}) // 自动迁移，创建表，如果表已经存在，会自动更新表结构，不会删除表,只会创建不存在的表
+	err = db.AutoMigrate(&User{})
+	if err != nil {
+		fmt.Println("user table:", err)
+	} // 自动迁移，创建表，如果表已经存在，会自动更新表结构，不会删除表,只会创建不存在的表
 	//db.Migrate(&Video{})
 	//db.Migrate(&Device{})
 	DB = db
diff --git a/handler/user.go b/handler/user.go
index a9f34e9..11e25e0 100644
--- a/handler/user.go
+++ b/handler/user.go
@@ -120,7 +120,8 @@ func loginHandler(c *gin.Context) {
 		}
 		user := service.GetUser(req_data.User, req_data.Password, req_data.Password)
 		if user.ID != 0 {
-			redis_token := worker.GetRedis("user_" + user.Name)
+			key := "user_" + user.Name
+			redis_token := worker.GetRedis(string(key))
 			if redis_token == "" {
 				// 生成 JWT 令牌
 				token := jwt.NewWithClaims(jwt.SigningMethodHS256, jwt.MapClaims{
@@ -154,7 +155,6 @@ func loginHandler(c *gin.Context) {
 	} else {
 		c.JSON(200, gin.H{"error": err.Error(), "code": 1, "message": "error"})
 	}
-
 }
 
 func registerHandler(c *gin.Context) {
diff --git a/worker/redis.go b/worker/redis.go
index 75ca6c2..5b8acdc 100644
--- a/worker/redis.go
+++ b/worker/redis.go
@@ -116,6 +116,7 @@ func GetRedis(key string) string {
 	val, err := redisClient.Get(ctx, key).Result() // 从 Redis 读取键值, 如果键不存在则返回空字符串, 如果出现错误则返回错误
 	if err != nil {
 		fmt.Println("Error getting key: %v", err)
+		return ""
 	}
 	return val
 }