实现用户认证系统,包含注册、登录、JWT鉴权及受保护路由。项目结构清晰,使用Gin框架与GORM操作SQLite数据库,通过bcrypt哈希密码,JWT实现中间件鉴权,适合学习扩展。

用Golang实现一个基础的用户认证系统,核心功能包括用户注册、登录、JWT鉴权和受保护的路由访问。整个项目结构清晰,适合学习和扩展。
项目结构设计
一个合理的项目结构有助于后期维护和扩展:
- main.go:程序入口,路由配置
- handlers/:处理HTTP请求逻辑
- models/:定义用户结构体和数据库操作
- middleware/:JWT鉴权中间件
- utils/:工具函数,如密码哈希、生成JWT
- database/:数据库连接初始化
用户模型与数据库连接
使用SQLite或PostgreSQL存储用户信息。这里以SQLite为例,使用
gorm作为ORM。
定义用户模型:
立即学习“go语言免费学习笔记(深入)”;
package modelstype User struct { ID uint
gorm:"primaryKey"Username stringgorm:"unique;not null"Password stringgorm:"not null"}
初始化数据库连接:
package databaseimport ( "gorm.io/driver/sqlite" "gorm.io/gorm" )
var DB *gorm.DB
func Init() { var err error DB, err = gorm.Open(sqlite.Open("users.db"), &gorm.Config{}) if err != nil { panic("failed to connect database") } DB.AutoMigrate(&User{}) }
注册与登录接口
在
handlers/auth.go中实现注册和登录逻辑。
注册:接收用户名密码,密码哈希后存入数据库。
新版本程序更新主要体现在:完美整合BBS论坛程序,用户只须注册一个帐号,即可全站通用!采用目前流行的Flash滚动切换广告 变换形式多样,受人喜爱!在原有提供的5种在线支付基础上增加北京云网支付!对留言本重新进行编排,加入留言验证码,后台有留言审核开关对购物系统的前台进行了一处安全更新。在原有文字友情链接基础上,增加LOGO友情链接功能强大的6种在线支付方式可选,自由切换。对新闻列表进行了调整,
func Register(c *gin.Context) {
var user models.User
if err := c.ShouldBindJSON(&user); err != nil {
c.JSON(400, gin.H{"error": "invalid input"})
return
}
hashedPassword, _ := utils.HashPassword(user.Password)
user.Password = hashedPassword
if err := database.DB.Create(&user).Error; err != nil {
c.JSON(400, gin.H{"error": "username already exists"})
return
}
c.JSON(201, gin.H{"message": "user created"})
}
登录:验证凭据,签发JWT。
func Login(c *gin.Context) {
var input models.User
if err := c.ShouldBindJSON(&input); err != nil {
c.JSON(400, gin.H{"error": "invalid input"})
return
}
var user models.User
if err := database.DB.Where("username = ?", input.Username).First(&user).Error; err != nil {
c.JSON(401, gin.H{"error": "invalid credentials"})
return
}
if !utils.CheckPasswordHash(input.Password, user.Password) {
c.JSON(401, gin.H{"error": "invalid credentials"})
return
}
token, err := utils.GenerateJWT(user.Username)
if err != nil {
c.JSON(500, gin.H{"error": "could not generate token"})
return
}
c.JSON(200, gin.H{"token": token})
}
JWT中间件实现鉴权
在
middleware/auth.go中实现JWT验证中间件。
package middlewareimport ( "net/http" "strings" "github.com/dgrijalva/jwt-go" "github.com/gin-gonic/gin" )
var jwtKey = []byte("my_secret_key")
func AuthMiddleware() gin.HandlerFunc { return func(c *gin.Context) { authHeader := c.GetHeader("Authorization") if authHeader == "" { c.JSON(401, gin.H{"error": "missing authorization header"}) c.Abort() return }
tokenString := strings.TrimPrefix(authHeader, "Bearer ") claims := &jwt.StandardClaims{} token, err := jwt.ParseWithClaims(tokenString, claims, func(token *jwt.Token) (interface{}, error) { return jwtKey, nil }) if err != nil || !token.Valid { c.JSON(401, gin.H{"error": "invalid or expired token"}) c.Abort() return } c.Set("username", claims.Subject) c.Next()} }
保护路由示例:
router.GET("/protected", middleware.AuthMiddleware(), func(c *gin.Context) {
username := c.MustGet("username").(string)
c.JSON(200, gin.H{"message": "hello " + username})
})
工具函数封装
在
utils/jwt.go中生成JWT:
package utilsimport ( "time" "github.com/dgrijalva/jwt-go" )
var jwtKey = []byte("my_secret_key")
func GenerateJWT(username string) (string, error) { claims := &jwt.StandardClaims{ Subject: username, ExpiresAt: time.Now().Add(24 * time.Hour).Unix(), }
token := jwt.NewWithClaims(jwt.SigningMethodHS256, claims) return token.SignedString(jwtKey) }
密码哈希使用
bcrypt:
package utilsimport ( "golang.org/x/crypto/bcrypt" )
func HashPassword(password string) (string, error) { bytes, err := bcrypt.GenerateFromPassword([]byte(password), 12) return string(bytes), err }
func CheckPasswordHash(password, hash string) bool { err := bcrypt.CompareHashAndPassword([]byte(hash), []byte(password)) return err == nil }
基本上就这些。运行程序后,先注册用户,再用返回的token请求受保护接口。整个流程简单清晰,适合初学者理解Golang Web开发和认证机制。









