From 1418d7273e47991e6fbe6b7324a308a047906cde Mon Sep 17 00:00:00 2001
From: junleea <354425203@qq.com>
Date: Sun, 9 Jun 2024 14:22:54 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BD=BF=E7=94=A8video=5Ft2=E6=95=B0=E6=8D=AE?=
 =?UTF-8?q?=E5=BA=93=E6=B5=8B=E8=AF=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 dao/db.go | 12 +++++++++---
 main.go   |  6 ++++++
 2 files changed, 15 insertions(+), 3 deletions(-)

diff --git a/dao/db.go b/dao/db.go
index 5eb3fec..825d3b6 100644
--- a/dao/db.go
+++ b/dao/db.go
@@ -9,7 +9,7 @@ import (
 var DB *gorm.DB
 
 func Init() {
-	dsn := "video_t:SDssrzALGiidPcjE@tcp(127.0.0.1:3306)/video_t?charset=utf8mb4&parseTime=True&loc=Local"
+	dsn := "video_t2:2t2SKHmWEYj2xFKF@tcp(127.0.0.1:3306)/video_t?charset=utf8mb4&parseTime=True&loc=Local"
 
 	db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
 	if err != nil {
@@ -19,8 +19,14 @@ func Init() {
 	if err != nil {
 		fmt.Println("user table:", err)
 	} // 自动迁移，创建表，如果表已经存在，会自动更新表结构，不会删除表,只会创建不存在的表
-	//db.Migrate(&Video{})
-	//db.Migrate(&Device{})
+	err = db.AutoMigrate(&Video{})
+	if err != nil {
+		fmt.Println("video table:", err)
+	} // 自动迁移，创建表，如果表已经存在，会自动更新表结构，不会删除表,只会创建不存在的表
+	err = db.AutoMigrate(&Device{})
+	if err != nil {
+		fmt.Println("device table:", err)
+	} // 自动迁移，创建表，如果表已经存在，会自动更新表结构，不会删除表,只会创建不存在的表
 	DB = db
 }
 
diff --git a/main.go b/main.go
index 9318aea..221577f 100644
--- a/main.go
+++ b/main.go
@@ -1,6 +1,7 @@
 package main
 
 import (
+	"fmt"
 	"github.com/gin-gonic/gin"
 	"github.com/golang-jwt/jwt"
 	"strings"
@@ -26,8 +27,13 @@ func main() {
 	defer worker.CloseRedis()
 }
 
+func preDoRequest(c *gin.Context) {
+	fmt.Println(c.ClientIP())
+}
+
 func JWTAuthMiddleware() gin.HandlerFunc {
 	return func(c *gin.Context) {
+		preDoRequest(c)
 		// 从请求头中获取 JWT 令牌
 		tokenString := c.Request.Header.Get("token")