Fix token generation

Author: DataEraserC

database.db

@@ -4,6 +4,7 @@ go 1.20
 
 require (
 	github.com/gin-gonic/gin v1.9.1
+	github.com/google/uuid v1.3.0
 	gorm.io/driver/sqlite v1.5.2
 	gorm.io/gorm v1.25.2
 )

go.mod

@@ -26,6 +26,8 @@ github.com/golang/protobuf v1.5.0/go.mod h1:FsONVRAS9T7sI+LIUmWTfcYkHO4aIWwzhcaS
 github.com/google/go-cmp v0.5.5 h1:Khx7svrCpmxxtHBq5j2mp/xVjsi8hQMfNLvJFAlrGgU=
 github.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
 github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg=
+github.com/google/uuid v1.3.0 h1:t6JiXgmwXMjEs8VusXIJk2BXHsn+wx8BZdTaoZ5fu7I=
+github.com/google/uuid v1.3.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
 github.com/jinzhu/inflection v1.0.0 h1:K317FqzuhWc8YvSVlFMCCUb36O/S9MCKRDI7QkRKD/E=
 github.com/jinzhu/inflection v1.0.0/go.mod h1:h+uFLlag+Qp1Va5pdKtLDYj+kHp5pxUVkryuEj+Srlc=
 github.com/jinzhu/now v1.1.5 h1:/o9tlHleP7gOFmsnYNz3RGnqzefHA47wQpKrrdTIwXQ=

go.sum

@@ -4,6 +4,7 @@ import (
 	"strconv"
 
 	"github.com/gin-gonic/gin"
+	"github.com/google/uuid"
 	"gorm.io/driver/sqlite"
 	"gorm.io/gorm"
 )
@@ -91,9 +92,16 @@ func main() {
 		}
 
 		var token Token
-		if err := db.Where("user_id = ?", user.ID).FirstOrCreate(&token).Error; err != nil {
-			c.JSON(400, gin.H{"code": 1, "message": "无法生成token"})
-			return
+		if err := db.Where("user_id = ?", user.ID).First(&token).Error; err != nil {
+			// If token does not exist, create a new token for the user
+			token = Token{
+				UserID: user.ID,
+				Token:  generateToken(user.ID), // Assume there is a function generateToken to generate a unique token
+			}
+			if err := db.Create(&token).Error; err != nil {
+				c.JSON(400, gin.H{"code": 1, "message": "无法生成token"})
+				return
+			}
 		}
 
 		c.JSON(200, gin.H{"code": 0, "message": "登录成功", "token": token.Token})
@@ -217,3 +225,8 @@ func main() {
 
 	r.Run()
 }
+
+func generateToken(userID uint) string {
+	token := uuid.New().String() // Generate a unique token using UUID
+	return token
+}