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

用Golang实现一个基础的用户认证系统,核心功能包括用户注册、登录、JWT鉴权和受保护的路由访问。整个项目结构清晰,适合学习和扩展。
项目结构设计
一个合理的项目结构有助于后期维护和扩展:
main.go:程序入口,路由配置handlers/:处理HTTP请求逻辑models/:定义用户结构体和数据库操作middleware/:JWT鉴权中间件utils/:工具函数,如密码哈希、生成JWTdatabase/:数据库连接初始化
用户模型与数据库连接
使用SQLite或PostgreSQL存储用户信息。这里以SQLite为例,使用
gorm
作为ORM。
定义用户模型:
立即学习“go语言免费学习笔记(深入)”;
package modelstype User struct {ID uint
gorm:"primaryKey"
Username string
gorm:"unique;not null"
Password string
gorm:"not null"
}
初始化数据库连接:
package databaseimport ("gorm.io/driver/sqlite""gorm.io/gorm")
var DB *gorm.DB
func Init() {var err errorDB, err = gorm.Open(sqlite.Open("users.db"), &gorm.Config{})if err != nil {panic("failed to connect database")}DB.AutoMigrate(&User{})}
注册与登录接口
在
handlers/auth.go
中实现注册和登录逻辑。
注册:接收用户名密码,密码哈希后存入数据库。
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.Userif 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开发和认证机制。
以上就是Golang实现基础用户认证系统项目的详细内容,更多请关注创想鸟其它相关文章!
版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 chuangxiangniao@163.com 举报,一经查实,本站将立刻删除。
发布者:程序猿,转转请注明出处:https://www.chuangxiangniao.com/p/1403519.html
微信扫一扫
支付宝扫一扫