5. gin集成Gorm

文章目录

代码地址:https://gitee.com/lymgoforIT/golang-trick/tree/master/24-gin-learning

本节主要介绍如何使用gin集成gorm,并完成用户的创建、修改、删除、查询等功能

本节主要是做最基本的连接和增删改查演示,具体使用细节可以查看官方文档:https://gorm.io/zh_CN/docs/

一:连接DB

引入gormmysql驱动

go 复制代码
go get gorm.io/gorm
go get gorm.io/driver/mysql

定义modelrequst

go 复制代码
type User struct {
	gorm.Model
	Name  string `json:"name" binding:"required"`
	Email string `json:"email" binding:"required"`
}

type UpdateUserInput struct {
	Name  string `json:"name"`
	Email string `json:"email"`
}

注:

  • binding taggin中绑定参数时的校验,将入参绑定到user对象时,对应字段为必传项
  • 上面UserUpdateUserInput看上去拥有的字段差不多,为什么还定义了两个结构体呢?原因是User为与DB交互的modelUpdateUserInput为接口的request,实际工作中modelrequest会有很多相似的字段,但一般不会完全一样,如request中可能会带有size,page等用于分页查询的字段。

连接DB

go 复制代码
package main

import (
	"github.com/gin-gonic/gin"
	"gorm.io/driver/mysql"
	"gorm.io/gorm"
	"net/http"
)

type User struct {
	gorm.Model
	Name  string `json:"name" binding:"required"`
	Email string `json:"email" binding:"required"`
}

type UpdateUserInput struct {
	Name  string `json:"name"`
	Email string `json:"email"`
}

func main() {

	dsn := "root:root@(127.0.0.1:3306)/test?charset=utf8mb4&parseTime=true&loc=Local"
	db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
	if err != nil {
		panic(any("failed to connect to db"))
	}

	// 建表
	err = db.AutoMigrate(&User{})
	if err != nil {
		panic(any("create table err"))
	}
}

运行上述代码后,可看到DB中创建了users表,当model没有实现Table方法时,表名默认为在model结构体名字后面加s,因此此处model名为User,所以表名是users

二:增删改查

代码如:

go 复制代码
package main

import (
	"github.com/gin-gonic/gin"
	"gorm.io/driver/mysql"
	"gorm.io/gorm"
	"log"
	"net/http"
)

type User struct {
	gorm.Model
	Name  string `json:"name" binding:"required"`
	Email string `json:"email" binding:"required"`
}

type UpdateUserInput struct {
	Name  string `json:"name"`
	Email string `json:"email"`
}

func main() {

	dsn := "root:root@(127.0.0.1:3306)/test?charset=utf8mb4&parseTime=true&loc=Local"
	db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
	if err != nil {
		panic(any("failed to connect to db"))
	}

	// 建表
	err = db.AutoMigrate(&User{})
	if err != nil {
		panic(any("create table err"))
	}

	r := gin.Default()

	r.GET("/ping", func(c *gin.Context) {
		c.JSON(http.StatusOK, gin.H{
			"message": "pong",
		})
	})

	// Get /users
	r.GET("/users", func(c *gin.Context) {
		var users []User
		err := db.Find(&users).Error
		if err != nil {
			c.AbortWithStatusJSON(http.StatusInternalServerError, gin.H{
				"error": "Failed to find users",
			})
			log.Println("Failed to find users")
			return
		}
		c.JSON(http.StatusOK, gin.H{"data": users})
	})

	// Post /users
	r.POST("/users", func(c *gin.Context) {
		var input User
		if err := c.ShouldBindJSON(&input); err != nil {
			c.AbortWithStatusJSON(http.StatusInternalServerError, gin.H{
				"error": err.Error(),
			})
			return
		}

		if err := db.Create(&input).Error; err != nil {
			c.AbortWithStatusJSON(http.StatusInternalServerError, gin.H{
				"error": "Failed to create users",
			})
			log.Println("Failed to create users")
			return
		}
		c.JSON(http.StatusOK, gin.H{"data": input})
	})

	// Put /users
	r.PUT("/users/:id", func(c *gin.Context) {
		var input UpdateUserInput
		if err := c.ShouldBindJSON(&input); err != nil {
			c.AbortWithStatusJSON(http.StatusInternalServerError, gin.H{
				"error": "invalid params",
			})
			log.Println("invalid params")
			return
		}

		id := c.Param("id")
		var user User
		if err := db.Where("id = ?", id).Find(&user).Error; err != nil {
			c.AbortWithStatusJSON(http.StatusInternalServerError, gin.H{
				"error": "Failed to Find user",
			})
			log.Println("Failed to Find user")
			return
		}
		if input.Name != "" {
			user.Name = input.Name
		}
		if input.Email != "" {
			user.Email = input.Email
		}

		if err := db.Save(&user).Error; err != nil {
			c.AbortWithStatusJSON(http.StatusInternalServerError, gin.H{
				"error": "Failed to save user",
			})
			log.Println("Failed to save user")
			return
		}
		c.JSON(http.StatusOK, gin.H{"data": user})
	})

	// Delete /users
	r.DELETE("/users/:id", func(c *gin.Context) {
		id := c.Param("id")
		var user User
		if err := db.First(&user, id).Error; err != nil {
			c.AbortWithStatusJSON(http.StatusInternalServerError, gin.H{
				"error": "Failed to Find user",
			})
			log.Println("Failed to Find user")
			return
		}
		if err := db.Delete(&user, id).Error; err != nil {
			c.AbortWithStatusJSON(http.StatusInternalServerError, gin.H{
				"error": "Failed to delete user",
			})
			log.Println("Failed to delete user")
			return
		}

		c.JSON(http.StatusOK, gin.H{"data": "Delete user " + id})
	})

	r.Run(":8080")
}

三:使用PostMan测试

Get

因为表中还没有数据,所以查询结果是空的

Post
post请求,因为我们想用raw请求体,所以需要先设置ContentType请求头

构造请求体,发送请求

可以多创建几个用户,然后查看DB以及使用get请求查询所有用户

查看DB

使用Get请求查

Put

同样要注意设置ContentType请求头

id1的用户的邮箱改掉

Delete

同样要注意设置ContentType请求头

id1的用户删除

下图是点了两次发送后的结果,第一次就删除了,所以第二次删除显示找不到用户了

查询看下

继续删除id2的用户,然后查看DB,可以看到删除是软删除

相关推荐
神奇侠20242 天前
快速开发-基于gin的中间件web项目开发
中间件·gin
Zyxalia6 天前
gin + es 实践 03
elasticsearch·jenkins·gin
Zyxalia6 天前
gin + es 实践 01
大数据·elasticsearch·gin
柯南二号6 天前
Go使用Gin写一个对MySQL的增删改查服务
mysql·golang·gin
Zyxalia7 天前
gin + es 实践 05
大数据·elasticsearch·gin
Zyxalia7 天前
gin + es 实践 07
网络·elasticsearch·gin
pedestrian_h10 天前
gRPC学习笔记记录以及整合gin开发
笔记·学习·golang·gin·grpc
liuyunshengsir13 天前
Gin 集成 prometheus 客户端实现注册和暴露指标
prometheus·gin
唐僧洗头爱飘柔952714 天前
(Go Gin)Gin学习笔记(三)数据解析和绑定:结构体分析,包括JSON解析、form解析、URL解析,区分绑定的Bind方法
golang·gin·数据绑定·数据解析·web开发框架·数据结构体分析
唐僧洗头爱飘柔952714 天前
(Go Gin)Gin学习笔记(二):路由配置、基本路由、表单参数、上传单个文件、上传多个文件、浅扒路由原理
学习·golang·gin·路由参数·路由配置·web开发框架·路由组