fixed GetUserByToken
This commit is contained in:
parent
8afc68814e
commit
ff0ecce741
|
@ -12,7 +12,6 @@ import (
|
||||||
"golang-test/message"
|
"golang-test/message"
|
||||||
"golang-test/validators"
|
"golang-test/validators"
|
||||||
"golang.org/x/crypto/bcrypt"
|
"golang.org/x/crypto/bcrypt"
|
||||||
"gorm.io/gorm"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
//func tokenGenerator() string {
|
//func tokenGenerator() string {
|
||||||
|
@ -25,8 +24,8 @@ import (
|
||||||
// a := tokenGenerator()
|
// a := tokenGenerator()
|
||||||
// fmt.Println(a)
|
// fmt.Println(a)
|
||||||
// }
|
// }
|
||||||
type Token struct {
|
|
||||||
gorm.Model
|
type TokenStruct struct {
|
||||||
UserID uint
|
UserID uint
|
||||||
token string
|
token string
|
||||||
}
|
}
|
||||||
|
@ -80,19 +79,19 @@ func generateToken() string {
|
||||||
return base64.StdEncoding.EncodeToString(b)
|
return base64.StdEncoding.EncodeToString(b)
|
||||||
}
|
}
|
||||||
|
|
||||||
func GetUserByToken(c *gin.Context, token string) database.User {
|
func GetUserByToken(c *gin.Context, token string) (database.User, error) {
|
||||||
var tokenStruct Token
|
var tokenObject TokenStruct
|
||||||
|
var err error
|
||||||
|
var user database.User
|
||||||
|
|
||||||
db := database.Connector()
|
db := database.Connector()
|
||||||
validate := validators.Validate
|
if err := db.Where("token = ?", token).First(&tokenObject).Error; err != nil {
|
||||||
if err := db.Where("token = ?", token).First(&tokenStruct).Error; err != nil {
|
return user, err
|
||||||
return nil
|
|
||||||
}
|
}
|
||||||
var user database.User
|
if err := db.First(&user, tokenObject.UserID).Error; err != nil {
|
||||||
if err := db.First(&user, tokenStruct.UserID).Error; err != nil {
|
return user, err
|
||||||
return err
|
|
||||||
}
|
}
|
||||||
return user
|
return user, err
|
||||||
}
|
}
|
||||||
|
|
||||||
func EditUser(c *gin.Context, token string, user database.User) {
|
func EditUser(c *gin.Context, token string, user database.User) {
|
||||||
|
|
Loading…
Reference in New Issue