增加环境变量GENERATE_DEFAULT_TOKEN 设置之后将生成初始令牌，默认关闭。

constant/env.go

@@ -44,3 +44,6 @@ func InitEnv() {
 		}
 	}
 }
+
+// 是否生成初始令牌，默认关闭。
+var GenerateDefaultToken = common.GetEnvOrDefaultBool("GENERATE_DEFAULT_TOKEN", false)

controller/user.go

@@ -11,6 +11,7 @@ import (
 
 	"github.com/gin-contrib/sessions"
 	"github.com/gin-gonic/gin"
+	"one-api/constant"
 )
 
 type LoginRequest struct {
@@ -197,7 +198,16 @@ func Register(c *gin.Context) {
 		return
 	}
 	// 生成默认令牌
-	// tokenName := cleanUser.Username + "的初始令牌"
+	if constant.GenerateDefaultToken {
+		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
+		}
+		// 生成默认令牌
 		token := model.Token{
 			UserId:             insertedUser.Id, // 使用插入后的用户ID
 			Name:               cleanUser.Username + "的初始令牌",  
@@ -216,6 +226,7 @@ func Register(c *gin.Context) {
 			})
 			return
 		}
+	}
 
 	c.JSON(http.StatusOK, gin.H{
 		"success": true,