feat: 心跳接口改用JWT token获取用户设备信息

- JWT Claims 增加 device_id 和 instance_id 字段
- 登录时将设备信息编码到 token
- 心跳接口从 token 自动获取用户和设备信息
- 心跳路由移到需要 JWT 认证的路由组
- 更新 API 文档

优点:
- 更安全:无法伪造设备信息
- 请求更简洁:心跳无需传递参数
- 符合业界标准做法

Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
2026-06-16 02:25:53 +08:00
parent 5bbae23bca
commit 113c01e89f
6 changed files with 58 additions and 36 deletions
+3 -1
View File
@@ -88,8 +88,10 @@ func JWT() gin.HandlerFunc {
c.Set("user_id", claims.UserID) c.Set("user_id", claims.UserID)
c.Set("username", claims.Username) c.Set("username", claims.Username)
c.Set("role", claims.Role) c.Set("role", claims.Role)
c.Set("device_id", claims.DeviceID)
c.Set("instance_id", claims.InstanceID)
fmt.Printf("JWT中间件: 验证成功, user_id=%d, role=%s\n", claims.UserID, claims.Role) fmt.Printf("JWT中间件: 验证成功, user_id=%d, role=%s, device_id=%s, instance_id=%s\n", claims.UserID, claims.Role, claims.DeviceID, claims.InstanceID)
c.Next() c.Next()
} }
} }
+30 -18
View File
@@ -65,23 +65,40 @@ func handleAppHeartbeat(c *gin.Context) {
return return
} }
var req struct { // 从 JWT token 中获取用户信息
UserID uint `json:"user_id"` userIDInterface, exists := c.Get("user_id")
DeviceID string `json:"device_id"` if !exists {
InstanceID string `json:"instance_id"` response.Error(c, 401, "未授权")
}
if err := c.ShouldBindJSON(&req); err != nil {
response.Error(c, 400, "参数错误")
return return
} }
userID := userIDInterface.(uint)
usernameInterface, _ := c.Get("username")
username := usernameInterface.(string)
// 从 JWT token 中获取设备信息
deviceIDInterface, exists := c.Get("device_id")
deviceID := ""
if exists {
deviceID = deviceIDInterface.(string)
}
instanceIDInterface, exists := c.Get("instance_id")
instanceID := ""
if exists {
instanceID = instanceIDInterface.(string)
}
if instanceID == "" {
instanceID = deviceID
}
var user model.AppUser var user model.AppUser
if err := database.DB.First(&user, req.UserID).Error; err != nil { if err := database.DB.First(&user, userID).Error; err != nil {
response.Error(c, 404, "用户不存在") response.Error(c, 404, "用户不存在")
return return
} }
if blocked, reason := checkRiskControl(c, app.ID, req.DeviceID, user.Username); blocked { if blocked, reason := checkRiskControl(c, app.ID, deviceID, username); blocked {
log.Printf("[DEBUG] Heartbeat blocked by risk control: %s", reason) log.Printf("[DEBUG] Heartbeat blocked by risk control: %s", reason)
response.Error(c, 403, reason) response.Error(c, 403, reason)
return return
@@ -145,16 +162,16 @@ func handleAppHeartbeat(c *gin.Context) {
return return
} }
if req.DeviceID != "" { if deviceID != "" {
var device model.UserDevice var device model.UserDevice
err := database.DB.Where("user_id = ? AND application_id = ? AND device_id = ?", user.ID, app.ID, req.DeviceID).First(&device).Error err := database.DB.Where("user_id = ? AND application_id = ? AND device_id = ?", user.ID, app.ID, deviceID).First(&device).Error
if err != nil { if err != nil {
device = model.UserDevice{ device = model.UserDevice{
UserID: user.ID, UserID: user.ID,
ApplicationID: app.ID, ApplicationID: app.ID,
DeviceID: req.DeviceID, DeviceID: deviceID,
DeviceName: req.DeviceID, DeviceName: deviceID,
Status: "active", Status: "active",
} }
if err := database.DB.Create(&device).Error; err != nil { if err := database.DB.Create(&device).Error; err != nil {
@@ -162,11 +179,6 @@ func handleAppHeartbeat(c *gin.Context) {
} }
} }
instanceID := req.InstanceID
if instanceID == "" {
instanceID = req.DeviceID
}
if device.ID > 0 { if device.ID > 0 {
var session model.DeviceSession var session model.DeviceSession
sessionErr := database.DB.Where("device_id = ? AND instance_id = ?", device.ID, instanceID).First(&session).Error sessionErr := database.DB.Where("device_id = ? AND instance_id = ?", device.ID, instanceID).First(&session).Error
+1 -1
View File
@@ -6,7 +6,6 @@ import (
func SetupRoutes(r *gin.RouterGroup) { func SetupRoutes(r *gin.RouterGroup) {
SetupPaymentRoutes(r) SetupPaymentRoutes(r)
SetupAccountRoutes(r)
SetupDynamicRoutes(r) SetupDynamicRoutes(r)
SetupUserRoutes(r) SetupUserRoutes(r)
SetupTicketRoutes(r) SetupTicketRoutes(r)
@@ -14,4 +13,5 @@ func SetupRoutes(r *gin.RouterGroup) {
func SetupAuthRoutes(r *gin.RouterGroup) { func SetupAuthRoutes(r *gin.RouterGroup) {
SetupDeviceRoutes(r) SetupDeviceRoutes(r)
SetupAccountRoutes(r)
} }
+5 -1
View File
@@ -681,7 +681,11 @@ func handleAppLogin(c *gin.Context) {
} }
} }
token, err := jwt.GenerateToken(user.ID, user.Username, "app_user") tokenInstanceID := req.InstanceID
if tokenInstanceID == "" {
tokenInstanceID = req.DeviceID
}
token, err := jwt.GenerateTokenWithDevice(user.ID, user.Username, "app_user", req.DeviceID, tokenInstanceID)
if err != nil { if err != nil {
log.Printf("[DEBUG] Failed to generate token: %v", err) log.Printf("[DEBUG] Failed to generate token: %v", err)
} }
+17 -8
View File
@@ -10,14 +10,21 @@ import (
// Claims JWT声明 // Claims JWT声明
type Claims struct { type Claims struct {
UserID uint `json:"user_id"` UserID uint `json:"user_id"`
Username string `json:"username"` Username string `json:"username"`
Role string `json:"role"` Role string `json:"role"`
DeviceID string `json:"device_id,omitempty"`
InstanceID string `json:"instance_id,omitempty"`
jwt.RegisteredClaims jwt.RegisteredClaims
} }
// GenerateToken 生成JWT令牌 // GenerateToken 生成JWT令牌
func GenerateToken(userID uint, username, role string) (string, error) { func GenerateToken(userID uint, username, role string) (string, error) {
return GenerateTokenWithDevice(userID, username, role, "", "")
}
// GenerateTokenWithDevice 生成包含设备信息的JWT令牌
func GenerateTokenWithDevice(userID uint, username, role, deviceID, instanceID string) (string, error) {
expireHours := config.GetInt("app.jwt_expire_hours") expireHours := config.GetInt("app.jwt_expire_hours")
if expireHours <= 0 { if expireHours <= 0 {
expireHours = 24 expireHours = 24
@@ -26,9 +33,11 @@ func GenerateToken(userID uint, username, role string) (string, error) {
expiresAt := time.Now().Add(time.Hour * time.Duration(expireHours)) expiresAt := time.Now().Add(time.Hour * time.Duration(expireHours))
claims := Claims{ claims := Claims{
UserID: userID, UserID: userID,
Username: username, Username: username,
Role: role, Role: role,
DeviceID: deviceID,
InstanceID: instanceID,
RegisteredClaims: jwt.RegisteredClaims{ RegisteredClaims: jwt.RegisteredClaims{
ExpiresAt: jwt.NewNumericDate(expiresAt), ExpiresAt: jwt.NewNumericDate(expiresAt),
IssuedAt: jwt.NewNumericDate(time.Now()), IssuedAt: jwt.NewNumericDate(time.Now()),
@@ -70,6 +79,6 @@ func RefreshToken(tokenString string) (string, error) {
return "", errors.New("token is still valid") return "", errors.New("token is still valid")
} }
// 生成新令牌 // 生成新令牌,保留设备信息
return GenerateToken(claims.UserID, claims.Username, claims.Role) return GenerateTokenWithDevice(claims.UserID, claims.Username, claims.Role, claims.DeviceID, claims.InstanceID)
} }
+2 -7
View File
@@ -365,15 +365,10 @@ Content-Type: application/json
``` ```
POST /api/v1/app/{appKey}/heartbeat POST /api/v1/app/{appKey}/heartbeat
Authorization: Bearer {token} Authorization: Bearer {token}
Content-Type: application/json
{
"user_id": 1,
"device_id": "设备ID",
"instance_id": "实例ID"
}
``` ```
> 用户ID、设备ID、实例ID均从 JWT token 中自动获取,无需在请求体中传递。token 在登录时已包含设备信息。
**响应** **响应**
```json ```json
{ {