使用Gin和Gorm实现CRUD操作

Shaw0xyz

在现代Web开发中，高效处理数据并提供快速响应的API是非常重要的。在Go语言生态中，Gin和Gorm分别是流行的Web框架和ORM工具。本文将介绍如何使用Gin和Gorm实现基本的CRUD（创建、读取、更新、删除）操作。


环境准备

在开始之前，确保你已经安装了Go并设置了GOPATH。接下来，创建一个新的Go项目并安装Gin和Gorm：

1. go mod init gin-gorm-crud
   
2. go get -u github.com/gin-gonic/gin
   
3. go get -u gorm.io/gorm
   
4. go get -u gorm.io/driver/sqlite

复制代码

这里我们使用SQLite作为数据库，你也可以选择其他数据库驱动（如PostgreSQL、MySQL等）。

项目结构

项目结构如下：

1. gin-gorm-crud/
   
2. │
   
3. ├── main.go
   
4. └── models/
   
5.     └── user.go

复制代码

创建数据模型

在`models/`目录下创建一个`user.go`文件，定义User模型：

1. package models
   
2. 
   
3. import (
   
4.         "gorm.io/gorm"
   
5. )
   
6. 
   
7. type User struct {
   
8.         gorm.Model
   
9.         Name  string `json:"name"`
   
10.         Email string `json:"email"`
    
11. }

复制代码

初始化Gorm和Gin

在`main.go`中初始化Gorm和Gin：

1. package main
   
2. 
   
3. import (
   
4.         "gin-gorm-crud/models"
   
5.         "gorm.io/driver/sqlite"
   
6.         "gorm.io/gorm"
   
7.         "github.com/gin-gonic/gin"
   
8.         "net/http"
   
9. )
   
10. 
    
11. var DB *gorm.DB
    
12. 
    
13. func initDB() {
    
14.         var err error
    
15.         DB, err = gorm.Open(sqlite.Open("test.db"), &gorm.Config{})
    
16.         if err != nil {
    
17.                 panic("failed to connect database")
    
18.         }
    
19. 
    
20.         DB.AutoMigrate(&models.User{})
    
21. }
    
22. 
    
23. func main() {
    
24.         initDB()
    
25. 
    
26.         r := gin.Default()
    
27.         r.GET("/users", getUsers)
    
28.         r.GET("/users/:id", getUser)
    
29.         r.POST("/users", createUser)
    
30.         r.PUT("/users/:id", updateUser)
    
31.         r.DELETE("/users/:id", deleteUser)
    
32.         
    
33.         r.Run()
    
34. }

复制代码

实现CRUD处理函数

接下来，实现CRUD操作的处理函数：

读取用户

1. func getUsers(c *gin.Context) {
   
2.         var users []models.User
   
3.         DB.Find(&users)
   
4.         c.JSON(http.StatusOK, users)
   
5. }
   
6. 
   
7. func getUser(c *gin.Context) {
   
8.         id := c.Param("id")
   
9.         var user models.User
   
10.         if err := DB.First(&user, id).Error; err != nil {
    
11.                 c.JSON(http.StatusNotFound, gin.H{"error": "User not found"})
    
12.                 return
    
13.         }
    
14.         c.JSON(http.StatusOK, user)
    
15. }

复制代码

创建用户

1. func createUser(c *gin.Context) {
   
2.         var user models.User
   
3.         if err := c.ShouldBindJSON(&user); err != nil {
   
4.                 c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
   
5.                 return
   
6.         }
   
7.         DB.Create(&user)
   
8.         c.JSON(http.StatusCreated, user)
   
9. }
   
10. 
    
11. 更新用户
    
12. 
    
13. func updateUser(c *gin.Context) {
    
14.         id := c.Param("id")
    
15.         var user models.User
    
16.         if err := DB.First(&user, id).Error; err != nil {
    
17.                 c.JSON(http.StatusNotFound, gin.H{"error": "User not found"})
    
18.                 return
    
19.         }
    
20.         if err := c.ShouldBindJSON(&user); err != nil {
    
21.                 c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
    
22.                 return
    
23.         }
    
24.         DB.Save(&user)
    
25.         c.JSON(http.StatusOK, user)
    
26. }
    

复制代码

删除用户

1. func deleteUser(c *gin.Context) {
   
2.         id := c.Param("id")
   
3.         var user models.User
   
4.         if err := DB.First(&user, id).Error; err != nil {
   
5.                 c.JSON(http.StatusNotFound, gin.H{"error": "User not found"})
   
6.                 return
   
7.         }
   
8.         DB.Delete(&user)
   
9.         c.JSON(http.StatusOK, gin.H{"message": "User deleted"})
   
10. }

复制代码

运行项目

完成以上步骤后，运行你的项目：

1. go run main.go

复制代码

打开浏览器或使用Postman等工具访问`http://localhost:8080`，进行CRUD操作的测试：

- 创建用户：POST `/users`，请求体包含JSON格式的用户数据。
- 读取所有用户：GET `/users`
- 读取特定用户：GET `/users/:id`
- 更新用户：PUT `/users/:id`，请求体包含JSON格式的用户数据。
- 删除用户：DELETE `/users/:id`

结语

通过本文，你学习了如何使用Gin和Gorm实现基本的CRUD操作。这种组合非常适合Go语言的Web应用开发，既能高效处理HTTP请求，又能方便地操作数据库。