实现用户认证系统，包含注册、登录、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 models
<p>type User struct {
ID uint <pre class="brush:php;toolbar:false;">gorm:"primaryKey"
Username string gorm:"unique;not null"Password string
gorm:"not null"}
初始化数据库连接：
package database
<p>import (
"gorm.io/driver/sqlite"
"gorm.io/gorm"
)</p><p>var DB *gorm.DB</p><p>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{})
}</p>注册与登录接口
在
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
}
<p>hashedPassword, _ := utils.HashPassword(user.Password)
user.Password = hashedPassword</p><p>if err := database.DB.Create(&user).Error; err != nil {
c.JSON(400, gin.H{"error": "username already exists"})
return
}</p><p>c.JSON(201, gin.H{"message": "user created"})
}</p>登录：验证凭据，签发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
}
<p>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
}</p><p>if !utils.CheckPasswordHash(input.Password, user.Password) {
c.JSON(401, gin.H{"error": "invalid credentials"})
return
}</p><p>token, err := utils.GenerateJWT(user.Username)
if err != nil {
c.JSON(500, gin.H{"error": "could not generate token"})
return
}</p><p>c.JSON(200, gin.H{"token": token})
}</p>JWT中间件实现鉴权
在
middleware/auth.go中实现JWT验证中间件。
package middleware
<p>import (
"net/http"
"strings"
"github.com/dgrijalva/jwt-go"
"github.com/gin-gonic/gin"
)</p><p>var jwtKey = []byte("my_secret_key")</p><p>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
}</p><pre class='brush:php;toolbar:false;'>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 utils
<p>import (
"time"
"github.com/dgrijalva/jwt-go"
)</p><p>var jwtKey = []byte("my_secret_key")</p><p>func GenerateJWT(username string) (string, error) {
claims := &jwt.StandardClaims{
Subject: username,
ExpiresAt: time.Now().Add(24 * time.Hour).Unix(),
}</p><p>token := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
return token.SignedString(jwtKey)
}</p>密码哈希使用
bcrypt：
package utils
<p>import (
"golang.org/x/crypto/bcrypt"
)</p><p>func HashPassword(password string) (string, error) {
bytes, err := bcrypt.GenerateFromPassword([]byte(password), 12)
return string(bytes), err
}</p><p>func CheckPasswordHash(password, hash string) bool {
err := bcrypt.CompareHashAndPassword([]byte(hash), []byte(password))
return err == nil
}</p>基本上就这些。运行程序后，先注册用户，再用返回的token请求受保护接口。整个流程简单清晰，适合初学者理解Golang Web开发和认证机制。
