使用 Gorm&Gin&MySQL 创建一个简单的 Go Web 用户管理系统

使用 Gorm 和 Gin 创建一个简单的 Go Web 用户管理系统

介绍

在这篇博客中，我们将使用 Go 语言结合 Gorm 和 Gin 框架，创建一个简单的用户管理系统。这个系统将实现用户的注册、登录、查看信息、更新信息等基本功能。我们将逐步介绍每一个步骤的实现，并解决遇到的问题。

项目初始化

1. 创建项目目录并初始化 Go 模块

首先，创建一个新的 Go 项目目录并初始化模块：

mkdir user-management
cd user-management
go mod init user-management

这将会在当前目录下生成一个 go.mod 文件，用于管理项目的依赖。

安装依赖

接下来，我们将安装项目所需的依赖库，包括 Gorm、Gin 以及用于数据库连接的 MySQL 驱动：

go get -u gorm.io/gorm
go get -u gorm.io/driver/mysql
go get -u github.com/gin-gonic/gin
go get -u github.com/golang-jwt/jwt/v4

这些依赖库将帮助我们处理数据库操作、HTTP 请求和用户认证。

配置数据库连接

在项目根目录下创建 main.go 文件，用于初始化数据库连接和启动服务器。首先，我们需要配置数据库连接。

在 main.go 文件中，添加以下代码：

package main

import (
    "log"
    "user-management/controllers"
    "user-management/models"
    "github.com/gin-gonic/gin"
    "gorm.io/driver/mysql"
    "gorm.io/gorm"
)

var DB *gorm.DB

func initDB() {
   
   
    dsn := "username:password@tcp(127.0.0.1:3306)/dbname?charset=utf8mb4&parseTime=True&loc=Local"
    var err error
    models.DB, err = gorm.Open(mysql.Open(dsn), &gorm.Config{
   
   })
    if err != nil {
   
   
        log.Fatalf("Failed to connect to database: %v", err)
    }

    // 自动迁移，创建用户表
    err = models.DB.AutoMigrate(&models.User{
   
   })
    if err != nil {
   
   
        log.Fatalf("Failed to migrate database: %v", err)
    }

    log.Println("Database connected and migrated successfully!")
}

func main() {
   
   
    initDB()

    r := gin.Default()

    // 用户注册路由
    r.POST("/register", controllers.Register)

    // 启动服务器
    r.Run(":8080")
}

代码解释：
在上面的代码中，记得将 username、password 和 dbname 替换为你 MySQL 数据库的实际用户名、密码和数据库名称。

initDB 函数：该函数负责初始化数据库连接。我们使用 Gorm 连接 MySQL 数据库，并配置了数据库的 DSN（数据源名称）。

自动迁移：在连接成功后，我们调用 AutoMigrate 方法，将 User 模型自动迁移到数据库中，这会在数据库中创建一个 users 表。

Gin 路由设置：我们使用 Gin 框架启动一个 HTTP 服务器，并设置了一个用户注册的 POST 路由。

创建数据库模型

接下来，我们需要定义用户模型。模型是数据库表的抽象表示。我们将在 models 目录下创建一个 user.go 文件。

首先，在项目根目录下创建 models 目录，然后在该目录下创建 user.go 文件，并添加以下代码：

package models

import (
    "gorm.io/gorm"
)

// User 定义了用户的结构体
type User struct {
   
   
    gorm.Model
    Username string `gorm:"unique;not null"`
    Password string `gorm:"not null"`
}

代码解释：

User 结构体：该结构体定义了 users 表的结构。gorm.Model 是 Gorm 提供的一个基础模型，它会自动包含
 ID、创建时间、更新时间等字段。Username 字段被设置为唯一且不能为空，Password 字段同样不能为空。

实现用户注册功能

现在我们来实现用户注册功能。当用户提交注册请求时，我们会将用户信息保存到数据库中。

在项目根目录下创建 controllers 目录，然后在该目录下创建 user.go 文件，并添加以下代码：