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")