This commit is contained in:
maxwell 2024-09-25 16:52:59 +08:00
parent 38f44d0556
commit 6f8a173de6
28 changed files with 7 additions and 1033 deletions

View File

@ -1,29 +0,0 @@
package config
import (
"github.com/gofiber/storage/redis/v3"
)
type Config struct {
DBConnString string
RedisAddr string
JWTSecret string
}
func New() *Config {
return &Config{
DBConnString: "root:password@tcp(localhost:3306)/testfb?charset=utf8mb4&parseTime=True&loc=Local",
RedisAddr: "localhost:6379",
JWTSecret: "your-secret-key",
}
}
func InitRedis(addr string) *redis.Storage {
return redis.New(redis.Config{
Host: addr,
Port: 6379,
Username: "",
Password: "",
Database: 0,
})
}

View File

@ -1,31 +0,0 @@
package config
import (
"github.com/gofiber/storage/redis/v3"
)
type Config struct {
DBConnString string
RedisAddr string
JWTSecret string
ErrInvalidPassword string
}
func New() *Config {
return &Config{
DBConnString: "root:password@tcp(localhost:3306)/testfb?charset=utf8mb4&parseTime=True&loc=Local",
RedisAddr: "localhost:6379",
JWTSecret: "your-secret-key",
ErrInvalidPassword: "Invalid password",
}
}
func InitRedis(addr string) *redis.Storage {
return redis.New(redis.Config{
Host: addr,
Port: 6379,
Username: "",
Password: "",
Database: 0,
})
}

View File

@ -1,31 +0,0 @@
package config
import (
"github.com/gofiber/storage/redis/v3"
)
type Config struct {
DBConnString string
RedisAddr string
JWTSecret string
ErrInvalidPassword string
}
func New() *Config {
return &Config{
DBConnString: "root:password@tcp(localhost:3306)/testfb?charset=utf8mb4&parseTime=True&loc=Local",
RedisAddr: "localhost:6379",
JWTSecret: "your-secret-key",
ErrInvalidPassword: "Invalid password",
}
}
func InitRedis(addr string) *redis.Storage {
return redis.New(redis.Config{
Host: addr,
Port: 6379,
Username: "",
Password: "",
Database: 0,
})
}

View File

@ -1,31 +0,0 @@
package config
import (
"github.com/gofiber/storage/redis/v3"
)
const (
ErrInvalidPassword = "invalid password",
)
type Config struct {
DBConnString string
RedisAddr string
JWTSecret string
}
func New() *Config {
return &Config{
DBConnString: "root:password@tcp(localhost:3306)/testfb?charset=utf8mb4&parseTime=True&loc=Local",
RedisAddr: "localhost:6379",
JWTSecret: "your-secret-key",
}
}
func InitRedis(addr string) *redis.Storage {
return redis.New(redis.Config{
Host: addr,
Port: 6379,
Username: "",
Password: "",
Database: 0,
})
}

View File

@ -1,33 +0,0 @@
package config
import (
"github.com/gofiber/storage/redis/v3"
)
const (
ErrInvalidPassword = "invalid password"
)
type Config struct {
DBConnString string
RedisAddr string
JWTSecret string
}
func New() *Config {
return &Config{
DBConnString: "root:password@tcp(localhost:3306)/testfb?charset=utf8mb4&parseTime=True&loc=Local",
RedisAddr: "localhost:6379",
JWTSecret: "your-secret-key",
}
}
func InitRedis(addr string) *redis.Storage {
return redis.New(redis.Config{
Host: addr,
Port: 6379,
Username: "",
Password: "",
Database: 0,
})
}

View File

@ -1,29 +0,0 @@
package config
import (
"github.com/gofiber/storage/redis/v3"
)
type Config struct {
DBConnString string
RedisAddr string
JWTSecret string
}
func New() *Config {
return &Config{
DBConnString: "root:password@tcp(localhost:3306)/testfb?charset=utf8mb4&parseTime=True&loc=Local",
RedisAddr: "localhost:6379",
JWTSecret: "your-secret-key",
}
}
func InitRedis(addr string) *redis.Storage {
return redis.New(redis.Config{
Host: addr,
Port: 6379,
Username: "",
Password: "",
Database: 0,
})
}

View File

@ -1,105 +0,0 @@
package handlers
import (
"strconv"
"time"
"github.com/gofiber/fiber/v2"
"github.com/gofiber/storage/redis/v3"
"github.com/golang-jwt/jwt/v4"
"gorm.io/gorm"
"testfb/models"
)
func Login(db *gorm.DB, redisClient *redis.Storage) fiber.Handler {
return func(c *fiber.Ctx) error {
var input struct {
Username string `json:"username"`
Password string `json:"password"`
}
if err := c.BodyParser(&input); err != nil {
return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{"error": "Invalid input"})
}
var user models.User
if err := db.Where("username = ?", input.Username).First(&user).Error; err != nil {
return c.Status(fiber.StatusUnauthorized).JSON(fiber.Map{"error": "Invalid credentials"})
}
if err := user.ComparePassword(input.Password); err != nil {
return c.Status(fiber.StatusUnauthorized).JSON(fiber.Map{"error": "Invalid credentials"})
}
token := jwt.New(jwt.SigningMethodHS256)
claims := token.Claims.(jwt.MapClaims)
claims["user_id"] = user.ID
claims["exp"] = time.Now().Add(time.Hour * 24).Unix()
t, err := token.SignedString([]byte("your-secret-key"))
if err != nil {
return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{"error": "Could not login"})
}
// Convert user.ID to string before storing in Redis
userIDStr := strconv.FormatUint(uint64(user.ID), 10)
err = redisClient.Set(t, []byte(userIDStr), 24*time.Hour)
if err != nil {
return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{"error": "Could not store token"})
}
return c.JSON(fiber.Map{"token": t})
}
}
func GetCurrentUser(db *gorm.DB) fiber.Handler {
return func(c *fiber.Ctx) error {
userID := c.Locals("user_id").(uint)
var user models.User
if err := db.First(&user, userID).Error; err != nil {
return c.Status(fiber.StatusNotFound).JSON(fiber.Map{"error": "User not found"})
}
return c.JSON(user)
}
}
func UpdateCurrentUser(db *gorm.DB) fiber.Handler {
return func(c *fiber.Ctx) error {
userID := c.Locals("user_id").(uint)
var input models.User
if err := c.BodyParser(&input); err != nil {
return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{"error": "Invalid input"})
}
var user models.User
if err := db.First(&user, userID).Error; err != nil {
return c.Status(fiber.StatusNotFound).JSON(fiber.Map{"error": "User not found"})
}
user.Email = input.Email
user.Phone = input.Phone
if err := db.Save(&user).Error; err != nil {
return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{"error": "Could not update user"})
}
return c.JSON(user)
}
}
func GetUserByID(db *gorm.DB) fiber.Handler {
return func(c *fiber.Ctx) error {
id := c.Params("id")
var user models.User
if err := db.First(&user, id).Error; err != nil {
return c.Status(fiber.StatusNotFound).JSON(fiber.Map{"error": "User not found"})
}
return c.JSON(user)
}
}

View File

@ -1,106 +0,0 @@
package handlers
import (
"strconv"
"time"
"github.com/gofiber/fiber/v2"
"github.com/gofiber/storage/redis/v3"
"github.com/golang-jwt/jwt/v4"
"gorm.io/gorm"
"testfb/models"
)
func Login(db *gorm.DB, redisClient *redis.Storage) fiber.Handler {
return func(c *fiber.Ctx) error {
var input struct {
Username string `json:"username"`
Password string `json:"password"`
}
if err := c.BodyParser(&input); err != nil {
return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{"error": "Invalid input"})
}
var user models.User
if err := db.Where("username = ?", input.Username).First(&user).Error; err != nil {
return c.Status(fiber.StatusUnauthorized).JSON(fiber.Map{"error": "Invalid credentials"})
}
// if err := user.ComparePassword(input.Password); err != nil {
if err := user.ComparePassword(input.Password); err != nil {
return c.Status(fiber.StatusUnauthorized).JSON(fiber.Map{"error": "Invalid credentials"})
}
token := jwt.New(jwt.SigningMethodHS256)
claims := token.Claims.(jwt.MapClaims)
claims["user_id"] = user.ID
claims["exp"] = time.Now().Add(time.Hour * 24).Unix()
t, err := token.SignedString([]byte("your-secret-key"))
if err != nil {
return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{"error": "Could not login"})
}
// Convert user.ID to string before storing in Redis
userIDStr := strconv.FormatUint(uint64(user.ID), 10)
err = redisClient.Set(t, []byte(userIDStr), 24*time.Hour)
if err != nil {
return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{"error": "Could not store token"})
}
return c.JSON(fiber.Map{"token": t})
}
}
func GetCurrentUser(db *gorm.DB) fiber.Handler {
return func(c *fiber.Ctx) error {
userID := c.Locals("user_id").(uint)
var user models.User
if err := db.First(&user, userID).Error; err != nil {
return c.Status(fiber.StatusNotFound).JSON(fiber.Map{"error": "User not found"})
}
return c.JSON(user)
}
}
func UpdateCurrentUser(db *gorm.DB) fiber.Handler {
return func(c *fiber.Ctx) error {
userID := c.Locals("user_id").(uint)
var input models.User
if err := c.BodyParser(&input); err != nil {
return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{"error": "Invalid input"})
}
var user models.User
if err := db.First(&user, userID).Error; err != nil {
return c.Status(fiber.StatusNotFound).JSON(fiber.Map{"error": "User not found"})
}
user.Email = input.Email
user.Phone = input.Phone
if err := db.Save(&user).Error; err != nil {
return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{"error": "Could not update user"})
}
return c.JSON(user)
}
}
func GetUserByID(db *gorm.DB) fiber.Handler {
return func(c *fiber.Ctx) error {
id := c.Params("id")
var user models.User
if err := db.First(&user, id).Error; err != nil {
return c.Status(fiber.StatusNotFound).JSON(fiber.Map{"error": "User not found"})
}
return c.JSON(user)
}
}

View File

@ -1,31 +0,0 @@
package middleware
import (
"strconv"
"github.com/gofiber/fiber/v2"
"github.com/gofiber/storage/redis/v3"
)
func AuthMiddleware(redisClient *redis.Storage) fiber.Handler {
return func(c *fiber.Ctx) error {
token := c.Get("Authorization")
if token == "" {
return c.Status(fiber.StatusUnauthorized).JSON(fiber.Map{"error": "Missing authorization token"})
}
userIDBytes, err := redisClient.Get(token)
if err != nil {
return c.Status(fiber.StatusUnauthorized).JSON(fiber.Map{"error": "Invalid or expired token"})
}
userIDStr := string(userIDBytes)
userID, err := strconv.ParseUint(userIDStr, 10, 64)
if err != nil {
return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{"error": "Invalid user ID format"})
}
c.Locals("user_id", uint(userID))
return c.Next()
}
}

View File

@ -1,31 +0,0 @@
package middleware
import (
"strconv"
"github.com/gofiber/fiber/v2"
"github.com/gofiber/storage/redis/v3"
)
func AuthMiddleware(redisClient *redis.Storage) fiber.Handler {
return func(c *fiber.Ctx) error {
token := c.Get("Authorization")
if token == "" {
return c.Status(fiber.StatusUnauthorized).JSON(fiber.Map{"error": "Missing authorization token"})
}
userIDBytes, err := redisClient.Get(token)
if err != nil {
return c.Status(fiber.StatusUnauthorized).JSON(fiber.Map{"error": "Invalid or expired token"})
}
userIDStr := string(userIDBytes)
userID, err := strconv.ParseUint(userIDStr, 10, 64)
if err != nil {
return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{"error": "Invalid user ID format"})
}
c.Locals("user_id", uint(userID))
return c.Next()
}
}

View File

@ -1,31 +0,0 @@
package models
import (
"time"
"golang.org/x/crypto/bcrypt"
"gorm.io/gorm"
)
type User struct {
ID uint `gorm:"primarykey" json:"id"`
Username string `gorm:"unique" json:"username"`
Password string `json:"-"`
Email string `json:"email"`
Phone string `json:"phone"`
CreatedAt time.Time `json:"created_at"`
UpdatedAt time.Time `json:"updated_at"`
}
func (u *User) BeforeCreate(tx *gorm.DB) error {
hashedPassword, err := bcrypt.GenerateFromPassword([]byte(u.Password), bcrypt.DefaultCost)
if err != nil {
return err
}
u.Password = string(hashedPassword)
return nil
}
func (u *User) ComparePassword(password string) error {
return bcrypt.CompareHashAndPassword([]byte(u.Password), []byte(password))
}

View File

@ -1,36 +0,0 @@
package models
import (
"time"
"golang.org/x/crypto/bcrypt"
"gorm.io/gorm"
)
type User struct {
ID uint `gorm:"primarykey" json:"id"`
Username string `gorm:"unique" json:"username"`
Password string `json:"-"`
Email string `json:"email"`
Phone string `json:"phone"`
CreatedAt time.Time `json:"created_at"`
UpdatedAt time.Time `json:"updated_at"`
}
func (u *User) BeforeCreate(tx *gorm.DB) error {
hashedPassword, err := bcrypt.GenerateFromPassword([]byte(u.Password), bcrypt.DefaultCost)
if err != nil {
return err
}
u.Password = string(hashedPassword)
return nil
}
func (u *User) ComparePassword(password string) error {
//直接比较密码不用bcrypt
if u.Password != password {
return ErrInvalidPassword
}
return nil
// return bcrypt.CompareHashAndPassword([]byte(u.Password), []byte(password))
}

View File

@ -1,36 +0,0 @@
package models
import (
"time"
"golang.org/x/crypto/bcrypt"
"gorm.io/gorm"
)
type User struct {
ID uint `gorm:"primarykey" json:"id"`
Username string `gorm:"unique" json:"username"`
Password string `json:"-"`
Email string `json:"email"`
Phone string `json:"phone"`
CreatedAt time.Time `json:"created_at"`
UpdatedAt time.Time `json:"updated_at"`
}
func (u *User) BeforeCreate(tx *gorm.DB) error {
hashedPassword, err := bcrypt.GenerateFromPassword([]byte(u.Password), bcrypt.DefaultCost)
if err != nil {
return err
}
u.Password = string(hashedPassword)
return nil
}
func (u *User) ComparePassword(password string) error {
//直接比较密码不用bcrypt
if u.Password != password {
return ErrInvalidPassword
}
return nil
// return bcrypt.CompareHashAndPassword([]byte(u.Password), []byte(password))
}

View File

@ -1,36 +0,0 @@
package models
import (
"time"
"golang.org/x/crypto/bcrypt"
"gorm.io/gorm"
)
type User struct {
ID uint `gorm:"primarykey" json:"id"`
Username string `gorm:"unique" json:"username"`
Password string `json:"-"`
Email string `json:"email"`
Phone string `json:"phone"`
CreatedAt time.Time `json:"created_at"`
UpdatedAt time.Time `json:"updated_at"`
}
func (u *User) BeforeCreate(tx *gorm.DB) error {
hashedPassword, err := bcrypt.GenerateFromPassword([]byte(u.Password), bcrypt.DefaultCost)
if err != nil {
return err
}
u.Password = string(hashedPassword)
return nil
}
func (u *User) ComparePassword(password string) error {
//直接比较密码不用bcrypt
if u.Password != password {
return ErrInvalidPassword
}
return nil
// return bcrypt.CompareHashAndPassword([]byte(u.Password), []byte(password))
}

View File

@ -1,36 +0,0 @@
package models
import (
"time"
"golang.org/x/crypto/bcrypt"
"gorm.io/gorm"
)
type User struct {
ID uint `gorm:"primarykey" json:"id"`
Username string `gorm:"unique" json:"username"`
Password string `json:"-"`
Email string `json:"email"`
Phone string `json:"phone"`
CreatedAt time.Time `json:"created_at"`
UpdatedAt time.Time `json:"updated_at"`
}
func (u *User) BeforeCreate(tx *gorm.DB) error {
hashedPassword, err := bcrypt.GenerateFromPassword([]byte(u.Password), bcrypt.DefaultCost)
if err != nil {
return err
}
u.Password = string(hashedPassword)
return nil
}
func (u *User) ComparePassword(password string) error {
//直接比较密码不用bcrypt
if u.Password != password {
return ErrInvalidPassword
}
return nil
// return bcrypt.CompareHashAndPassword([]byte(u.Password), []byte(password))
}

View File

@ -1,36 +0,0 @@
package models
import (
"time"
"golang.org/x/crypto/bcrypt"
"gorm.io/gorm"
)
type User struct {
ID uint `gorm:"primarykey" json:"id"`
Username string `gorm:"unique" json:"username"`
Password string `json:"-"`
Email string `json:"email"`
Phone string `json:"phone"`
CreatedAt time.Time `json:"created_at"`
UpdatedAt time.Time `json:"updated_at"`
}
func (u *User) BeforeCreate(tx *gorm.DB) error {
hashedPassword, err := bcrypt.GenerateFromPassword([]byte(u.Password), bcrypt.DefaultCost)
if err != nil {
return err
}
u.Password = string(hashedPassword)
return nil
}
func (u *User) ComparePassword(password string) error {
//直接比较密码不用bcrypt
if u.Password != password {
return ErrInvalidPassword
}
return nil
// return bcrypt.CompareHashAndPassword([]byte(u.Password), []byte(password))
}

View File

@ -1,36 +0,0 @@
package models
import (
"time"
"golang.org/x/crypto/bcrypt"
"gorm.io/gorm"
)
type User struct {
ID uint `gorm:"primarykey" json:"id"`
Username string `gorm:"unique" json:"username"`
Password string `json:"-"`
Email string `json:"email"`
Phone string `json:"phone"`
CreatedAt time.Time `json:"created_at"`
UpdatedAt time.Time `json:"updated_at"`
}
func (u *User) BeforeCreate(tx *gorm.DB) error {
hashedPassword, err := bcrypt.GenerateFromPassword([]byte(u.Password), bcrypt.DefaultCost)
if err != nil {
return err
}
u.Password = string(hashedPassword)
return nil
}
func (u *User) ComparePassword(password string) error {
//直接比较密码不用bcrypt
if u.Password != password {
return ErrInvalidPassword
}
return nil
// return bcrypt.CompareHashAndPassword([]byte(u.Password), []byte(password))
}

View File

@ -1,36 +0,0 @@
package models
import (
"time"
"golang.org/x/crypto/bcrypt"
"gorm.io/gorm"
)
type User struct {
ID uint `gorm:"primarykey" json:"id"`
Username string `gorm:"unique" json:"username"`
Password string `json:"-"`
Email string `json:"email"`
Phone string `json:"phone"`
CreatedAt time.Time `json:"created_at"`
UpdatedAt time.Time `json:"updated_at"`
}
func (u *User) BeforeCreate(tx *gorm.DB) error {
hashedPassword, err := bcrypt.GenerateFromPassword([]byte(u.Password), bcrypt.DefaultCost)
if err != nil {
return err
}
u.Password = string(hashedPassword)
return nil
}
func (u *User) ComparePassword(password string) error {
//直接比较密码不用bcrypt
if u.Password != password {
return ErrInvalidPassword
}
return nil
// return bcrypt.CompareHashAndPassword([]byte(u.Password), []byte(password))
}

View File

@ -1,36 +0,0 @@
package models
import (
"time"
"golang.org/x/crypto/bcrypt"
"gorm.io/gorm"
)
type User struct {
ID uint `gorm:"primarykey" json:"id"`
Username string `gorm:"unique" json:"username"`
Password string `json:"-"`
Email string `json:"email"`
Phone string `json:"phone"`
CreatedAt time.Time `json:"created_at"`
UpdatedAt time.Time `json:"updated_at"`
}
func (u *User) BeforeCreate(tx *gorm.DB) error {
hashedPassword, err := bcrypt.GenerateFromPassword([]byte(u.Password), bcrypt.DefaultCost)
if err != nil {
return err
}
u.Password = string(hashedPassword)
return nil
}
func (u *User) ComparePassword(password string) error {
//直接比较密码不用bcrypt
if u.Password != password {
return ErrInvalidPassword
}
return nil
// return bcrypt.CompareHashAndPassword([]byte(u.Password), []byte(password))
}

View File

@ -1,37 +0,0 @@
package models
import (
. "testfb/config"
"time"
"golang.org/x/crypto/bcrypt"
"gorm.io/gorm"
)
type User struct {
ID uint `gorm:"primarykey" json:"id"`
Username string `gorm:"unique" json:"username"`
Password string `json:"-"`
Email string `json:"email"`
Phone string `json:"phone"`
CreatedAt time.Time `json:"created_at"`
UpdatedAt time.Time `json:"updated_at"`
}
func (u *User) BeforeCreate(tx *gorm.DB) error {
hashedPassword, err := bcrypt.GenerateFromPassword([]byte(u.Password), bcrypt.DefaultCost)
if err != nil {
return err
}
u.Password = string(hashedPassword)
return nil
}
func (u *User) ComparePassword(password string) error {
//直接比较密码不用bcrypt
if u.Password != password {
return ErrInvalidPassword
}
return nil
// return bcrypt.CompareHashAndPassword([]byte(u.Password), []byte(password))
}

View File

@ -1,37 +0,0 @@
package models
import (
. "testfb/config"
"time"
"golang.org/x/crypto/bcrypt"
"gorm.io/gorm"
)
type User struct {
ID uint `gorm:"primarykey" json:"id"`
Username string `gorm:"unique" json:"username"`
Password string `json:"-"`
Email string `json:"email"`
Phone string `json:"phone"`
CreatedAt time.Time `json:"created_at"`
UpdatedAt time.Time `json:"updated_at"`
}
func (u *User) BeforeCreate(tx *gorm.DB) error {
hashedPassword, err := bcrypt.GenerateFromPassword([]byte(u.Password), bcrypt.DefaultCost)
if err != nil {
return err
}
u.Password = string(hashedPassword)
return nil
}
func (u *User) ComparePassword(password string) error {
//直接比较密码不用bcrypt
if u.Password != password {
return ErrInvalidPassword
}
return nil
// return bcrypt.CompareHashAndPassword([]byte(u.Password), []byte(password))
}

View File

@ -1,37 +0,0 @@
package models
import (
"time"
"golang.org/x/crypto/bcrypt"
"gorm.io/gorm"
)
type User struct {
ID uint `gorm:"primarykey" json:"id"`
Username string `gorm:"unique" json:"username"`
Password string `json:"-"`
Email string `json:"email"`
Phone string `json:"phone"`
CreatedAt time.Time `json:"created_at"`
UpdatedAt time.Time `json:"updated_at"`
}
func (u *User) BeforeCreate(tx *gorm.DB) error {
hashedPassword, err := bcrypt.GenerateFromPassword([]byte(u.Password), bcrypt.DefaultCost)
if err != nil {
return err
}
u.Password = string(hashedPassword)
return nil
}
func (u *User) ComparePassword(password string) error {
//直接比较密码不用bcrypt
if u.Password != password {
err := "Invalid password"
return err
}
return nil
// return bcrypt.CompareHashAndPassword([]byte(u.Password), []byte(password))
}

View File

@ -1,37 +0,0 @@
package models
import (
"time"
"golang.org/x/crypto/bcrypt"
"gorm.io/gorm"
)
type User struct {
ID uint `gorm:"primarykey" json:"id"`
Username string `gorm:"unique" json:"username"`
Password string `json:"-"`
Email string `json:"email"`
Phone string `json:"phone"`
CreatedAt time.Time `json:"created_at"`
UpdatedAt time.Time `json:"updated_at"`
}
func (u *User) BeforeCreate(tx *gorm.DB) error {
hashedPassword, err := bcrypt.GenerateFromPassword([]byte(u.Password), bcrypt.DefaultCost)
if err != nil {
return err
}
u.Password = string(hashedPassword)
return nil
}
func (u *User) ComparePassword(password string) error {
//直接比较密码不用bcrypt
if u.Password != password {
err := 123
return err
}
return nil
// return bcrypt.CompareHashAndPassword([]byte(u.Password), []byte(password))
}

View File

@ -1,36 +0,0 @@
package models
import (
"time"
"golang.org/x/crypto/bcrypt"
"gorm.io/gorm"
)
type User struct {
ID uint `gorm:"primarykey" json:"id"`
Username string `gorm:"unique" json:"username"`
Password string `json:"-"`
Email string `json:"email"`
Phone string `json:"phone"`
CreatedAt time.Time `json:"created_at"`
UpdatedAt time.Time `json:"updated_at"`
}
func (u *User) BeforeCreate(tx *gorm.DB) error {
hashedPassword, err := bcrypt.GenerateFromPassword([]byte(u.Password), bcrypt.DefaultCost)
if err != nil {
return err
}
u.Password = string(hashedPassword)
return nil
}
func (u *User) ComparePassword(password string) error {
//直接比较密码不用bcrypt
if u.Password == password {
return nil
}
// return bcrypt.CompareHashAndPassword([]byte(u.Password), []byte(password))
}

View File

@ -1,36 +0,0 @@
package models
import (
"time"
"golang.org/x/crypto/bcrypt"
"gorm.io/gorm"
)
type User struct {
ID uint `gorm:"primarykey" json:"id"`
Username string `gorm:"unique" json:"username"`
Password string `json:"-"`
Email string `json:"email"`
Phone string `json:"phone"`
CreatedAt time.Time `json:"created_at"`
UpdatedAt time.Time `json:"updated_at"`
}
func (u *User) BeforeCreate(tx *gorm.DB) error {
hashedPassword, err := bcrypt.GenerateFromPassword([]byte(u.Password), bcrypt.DefaultCost)
if err != nil {
return err
}
u.Password = string(hashedPassword)
return nil
}
func (u *User) ComparePassword(password string) error {
//直接比较密码不用bcrypt
if u.Password == password {
return nil
} else {
return bcrypt.CompareHashAndPassword([]byte(u.Password), []byte(password))
}
}

View File

@ -1,36 +0,0 @@
package models
import (
"time"
"golang.org/x/crypto/bcrypt"
"gorm.io/gorm"
)
type User struct {
ID uint `gorm:"primarykey" json:"id"`
Username string `gorm:"unique" json:"username"`
Password string `json:"-"`
Email string `json:"email"`
Phone string `json:"phone"`
CreatedAt time.Time `json:"created_at"`
UpdatedAt time.Time `json:"updated_at"`
}
func (u *User) BeforeCreate(tx *gorm.DB) error {
hashedPassword, err := bcrypt.GenerateFromPassword([]byte(u.Password), bcrypt.DefaultCost)
if err != nil {
return err
}
u.Password = string(hashedPassword)
return nil
}
func (u *User) ComparePassword(password string) error {
//直接比较密码不用bcrypt
if u.Password == password {
return nil
} else {
return bcrypt.CompareHashAndPassword([]byte(u.Password), []byte(password))
}
}

6
app.log Normal file
View File

@ -0,0 +1,6 @@
[16:24:09] 401 - 193.040931ms POST /login
[16:24:20] 401 - 198.309132ms POST /login
[16:24:25] 401 - 194.157634ms POST /login
[16:24:40] 401 - 3.850798ms POST /login
[16:24:45] 401 - 182.36815ms POST /login
[16:25:04] 401 - 194.986562ms POST /login

View File

@ -13,7 +13,7 @@ type Config struct {
func New() *Config {
return &Config{
DBConnString: "root:password@tcp(localhost:3306)/testfb?charset=utf8mb4&parseTime=True&loc=Local",
RedisAddr: "localhost:6379",
RedisAddr: "localhost",
JWTSecret: "your-secret-key",
}
}