Просмотр исходного кода

增加注册自动生成初始令牌。

GuoRuqiang 1 год назад
Родитель
Сommit
c97e2875b4
1 измененных файлов с 31 добавлено и 0 удалено
  1. 31 0
      controller/user.go

+ 31 - 0
controller/user.go

@@ -186,6 +186,37 @@ func Register(c *gin.Context) {
 		})
 		})
 		return
 		return
 	}
 	}
+	
+	// 获取插入后的用户ID
+	var insertedUser model.User
+	if err := model.DB.Where("username = ?", cleanUser.Username).First(&insertedUser).Error; err != nil {
+		c.JSON(http.StatusOK, gin.H{
+			"success": false,
+			"message": "用户注册失败或用户ID获取失败",
+		})
+		return
+	}
+	// 生成默认令牌
+	// tokenName := cleanUser.Username + "的初始令牌"
+	token := model.Token{
+		UserId:             insertedUser.Id, // 使用插入后的用户ID
+		Name:               cleanUser.Username + "的初始令牌",  
+		Key:                common.GenerateKey(),
+		CreatedTime:        common.GetTimestamp(),
+		AccessedTime:       common.GetTimestamp(),
+		ExpiredTime:        -1,  // 永不过期
+		RemainQuota:        500000,  // 示例额度
+		UnlimitedQuota:     true,
+		ModelLimitsEnabled: false,
+	}
+	if err := token.Insert(); err != nil {
+		c.JSON(http.StatusOK, gin.H{
+			"success": false,
+			"message": "创建默认令牌失败",
+		})
+		return
+	}
+
 	c.JSON(http.StatusOK, gin.H{
 	c.JSON(http.StatusOK, gin.H{
 		"success": true,
 		"success": true,
 		"message": "",
 		"message": "",