gin+MySQL简单实现数据库查询
利用 gin 项目搭建一个简易的后端系统。
一个简易的 HTTP 响应接口
首先在 go 工作区的终端输入这条指令：
go get -u github.com/gin-gonic/gin
将 gin 项目的相关依赖保存到本地。
在终端生成 go mod 包管理文件：
go mod init
再创建一个 main.go 文件：
package main
import "github.com/gonic-gin/gin"
func main() {
r := gin.Default()
r.GET("/test", func(c *gin.Context) {
c.JSON(200, gin.H{
"message": "test",
})
})
r.Run(":9999") // 运行在9999端口
}
因为我是在 wsl 运行这个项目，所以还需要先获取虚拟机的 ip，然后用 curl 测试：
curl http://<ip>:9999/ping
返回结果：
{"message":"test"}
连接 MySQL
关于 MySQL 操作这部分，一开始想着简单，只学了查询数据库部分，大多数踩得坑都是在查询部分，后面觉得要举一反三，就在原来基础上又写了添加的部分，在添加数据这一块写的不是很详细。
添加
先把基本的框架写出来，这里我连接的数据库结构体如下：
type SqlUser struct {
User_id string `json:"user_id" from:"user_id"`
User_name string `json:"user_name" from:"user_name"`
}
添加单条数据：
Db, _ := sql.Open("mysql", "<username>:<password>@(localhost:3306)/<yourDatabase>")
router.POST("/add", func(c *gin.Context) {
User_id := c.Request.FormValue("User_id")
User_name := c.Request.FormValue("User_name")
result, err := db.SqlDB.Exec("INSERT INTO t_user (user_id,user_name) VALUE(?,?)", u.User_id, u.User_name)
if err != nil {
log.Fatalln(err)
}
id, err := result.LastInsertId()
if err != nil {
log.Fatalln(err)
}
msg := fmt.Sprintf("insert successfully %d", id)
c.JSON(http.StatusOK, gin.H{
"msg": msg,
"user_name": User_name,
})
})
r.Run(":9999")
执行非 query 操作，使用 Exec 方法，使用 curl 进行测试：
curl -X POST http://<ip>:9999/add -d "User_id=2&User_name=aaa"
返回结果如下：
{"msg":"insert successfully 1","user_name":"test"}
查询
单条查询如下
r.GET("/test/:id", func(c *gin.Context) {
id: c.Param("id")
var u SqlUser
err := Db.QueryRow("select user_id,user_name from t_user where user_id =?", id).Scan(&u.user_id, &u.user_name)
if err != nil {
log.Fatalln(err)
c.JSON(http.StatusOK, gin.H{
"user": nil,
})
return
}
c.JSON(http.StatusOK, gin.H{
"user": u.user_name,
})
})
然后用 curl 命令去测试：
curl http://<ip>:9999/test/1
会得到之前在 MySQL 中预先保存的条目，样例结构如下：
{"user":"test"}
然而，最开始我不是这样写的，当时的我，没整理好这个框架，直接在单文件里封装函数，下面是我的错误示范：
var Db *sql.DB
func QueryData(c *gin.Context) {
id := c.Param("id")
var u SqlUser
err := Db.QueryRow("select user_id,user_name from t_user where user_id =?", id).Scan(&u.user_id, &u.user_name)
if err != nil {
log.Fatalln(err)
c.JSON(200, gin.H{
"user": nil,
})
return
}
c.JSON(200, gin.H{
"user": u.user_name,
})
}
func main() {
Db, _ = sql.Open("/*mysql link*/")
// 略去了这个问题相关的代码
r.GET("/test/:id", QueryData)
r.Run(":9999")
}
然后就出现这样的错误：
runtime error: invalid memory address or nil pointer dereference
必应上的结果是：指针声明后未初始化就赋值。
这个 bug 困扰了我一整天，然后就开始了痛苦的 debug，main 函数的 router 应该是没问题的，接口测试正常，那问题应该是出在 QueryData 上，那只能一个个打印变量测试吧，测试到 Db 时，发现这个变量为 nil，然后紧接着抛出刚才提到的错误。好吧，这个错误我在之前也偶尔犯过，虽然在 main 中对其赋了值，但作用域不一样，所以还是无法传值，那只好去考虑下框架了。
修改
修改单条数据：
router.PUT("/person/:id", func(c *gin.Context) {
cid := c.Param("id")
id, err := strconv.Atoi(cid)
person := Person{Id: id}
err = c.Bind(&person)
if err != nil {
log.Fatalln(err)
}
stmt, err := Db.Prepare("UPDATE person SET user_name=? where user_id=?")
if err != nil {
log.Fatalln(err)
}
defer stmt.Close()
rs, err := stmt.Exec(person.user_name, person.user_id)
if err != nil {
log.Fatalln(err)
}
ra, err := rs.RowsAffected()
if err != nil {
log.Fatalln(err)
}
msg := fmt.Sprintf("UPDATE person %d successful %d", person.user_id, ra)
c.JSON(http.StatusOK, gin.H{
"msg": msg,
})
})
urlencode 方式更新：
curl -X PUT http://<ip>:9999/person/1 -d "user_id=1&user_name=temp"
json 方式更新：
curl -X PUT http://<ip>:9999/person/1 -H "Content-Type: application/json" -d '{"user_name":"temp"}'
删除
router.DELETE("person/:id", func(c *gin.Context) {
cid := c.Param("id")
id, err := strconv.Atoi(cid)
if err != nil {
log.Fatalln(err)
}
rs, err := db.Exec("DELETE FROM person from where user_id=?", id)
if err != nil {
log.Fatalln(err)
}
ra, err := rs.RowsAffected()
if err != nil {
log.Fatalln(err)
}
msg := fmt.Sprintf("DELETE person %d successful %d", id, ra)
c.JSON(http.StatusOK, gin.H{
"msg": msg,
})
})
直接使用删除接口
curl -X "DELETE" http://<ip>:9999/person/1
这里的 DELETE 参数必须在双引号内，不然不会被识别为 delete 请求。
搭建框架
参考了大佬的做法 [1]，我也来创建我的目录：
ginExample tree
.
├── api
│ └── query.go
├── database
│ └── mysql.go
├── main.go
├── models
│ └── queryUser.go
├── router.go
├── go.mod
└── go.sum
这里的话以添加、查询功能为例，增删改同理，具体可以查看文末的参考链接。
api 存放 handler 函数，model 存放数据模型。
数据库处理
// mysql.go
package database
import (
"database/sql"
_ "github.com/go-sql-driver/mysql"
"log"
)
var SqlDB *sql.DB
func init() {
var err error
SqlDB, err = sql.Open("mysql", "root:jaydenmysql@(127.0.0.1:3306)/microtest")
if err != nil {
log.Fatalln(err.Error())
}
err = SqlDB.Ping()
if err != nil {
log.Fatalln(err.Error())
}
}
因为在别的包会用到 SqlDB 这个变量，因此必须大写 (golang 只有大写开头变量才是 public 类变量)
数据 model 封装
抽离出 SqlUser 结构体以及对应的方法：
package models
import (
db "ginExample/database"
"log"
)
type SqlUser struct {
User_id string `json:"user_id" from:"user_id"`
User_name string `json:"user_name" from:"user_name"`
}
func (u *SqlUser) GetUser(id string) SqlUser {
err := db.SqlDB.QueryRow("select user_id,user_name from t_user where user_id =?", id).Scan(&u.User_id, &u.User_name)
if err != nil {
log.Println(err)
u.User_name = "nil"
return *u
}
return *u
}
func (u *SqlUser) AddUser() (id int64, err error) {
result, err := db.SqlDB.Exec("INSERT INTO t_user (user_id,user_name) VALUE(?,?)", u.User_id, u.User_name)
if err != nil {
log.Fatalln(err)
return
}
id, err = result.LastInsertId()
if err != nil {
log.Fatalln(err)
return
}
fmt.Println(id)
return id, err
}
func (u *SqlUser) UpdateUser() int64 {
stmt, err := db.SqlDB.Prepare("UPDATE t_user SET user_name=? where user_id=?")
if err != nil {
log.Fatalln(err)
}
rs, err := stmt.Exec(u.User_name, u.User_id)
if err != nil {
log.Fatalln(err)
}
ra, err := rs.RowsAffected()
if err != nil {
log.Fatalln(err)
}
return ra
}
func (u *SqlUser) DeleteUser(id string) int64 {
rs, err := db.SqlDB.Exec("DELETE FROM t_user where user_id=?", id)
if err != nil {
log.Fatalln(err)
}
ra, err := rs.RowsAffected()
if err != nil {
log.Fatalln(err)
}
return ra
}
handler
然后把具体的 handler 封装到 api 中，handler 操作数据库，因此会引用 model 包。
package api
import (
"fmt"
"net/http"
. "ginExample/models"
"github.com/gin-gonic/gin"
)
func IndexApi(c *gin.Context) {
c.String(http.StatusOK, "It works")
}
func GetUserApi(c *gin.Context) {
var u SqlUser
id := c.Param("id")
u = u.GetUser(id)
c.JSON(200, gin.H{
"user": u.User_name,
})
}
func AddUserApi(c *gin.Context) {
User_id := c.Request.FormValue("User_id")
User_name := c.Request.FormValue("User_name")
// 这里的大小写一定要对应上结构体内的变量名
u := models.SqlUser{User_id: User_id, User_name: User_name}
rows, err := u.AddUser()
if err != nil {
log.Fatalln(err)
}
msg := fmt.Sprintf("insert successfully %d\n", rows)
c.JSON(200, gin.H{
"msg": msg,
"user_name": User_name,
})
}
func UpdateUserApi(c *gin.Context) {
User_id := c.Request.FormValue("User_id")
User_name := c.Request.FormValue("User_name")
u := models.SqlUser{User_id: User_id, User_name: User_name}
row := u.UpdateUser()
msg := fmt.Sprintf("update successful %d", row)
c.JSON(200, gin.H{
"msg": msg,
})
}
func DeleteUserApi(c *gin.Context) {
var u models.SqlUser
id := c.Param("id")
row := u.DeleteUser(id)
msg := fmt.Sprintf("DELETE user successful %d", row)
c.JSON(200, gin.H{
"msg": msg,
})
}
router
最后就是把路由抽离出来：
//router.go
package main
import (
"github.com/gin-gonic/gin"
. "ginExample/api"
)
func initRouter() *gin.Engine {
router := gin.Default()
router.GET("/", IndexApi)
router.GET("/query/:id", GetUserApi)
router.POST("/add", api.AddUserApi)
router.PUT("/update/:id", api.UpdateUserApi)
router.DELETE("/delete/:id", api.DeleteUserApi)
return router
}
app 入口
最后就是 main 的 app 入口，将路由导入，同时在 main 即将结束时，关闭全局数据库连接池。
package main
import db "ginExample/database"
func main() {
defer db.SqlDB.Close()
router := initRouter()
router.Run(":9999")
}
这里运行项目的话，不能像之前简单地使用 go run main.go ，因为 main 包含 main.go 和 router.go，因此要运行 go run *.go，如果最终编译二进制项目，则运行 go build -o app。
测试结果和上面是一样的，至此，基本的访问、操作数据库功能实现。
