go gin学习记录2

环境

环境:mac m1,go version 1.17.2, goland, mysql

昨天学习了安装和基本的参数,路由使用,今天接着学习一下数据库的操作。

建立数据库

测试数据库操作,需要先准备一个测试用的数据库,那我们就来建一个测试用数据库t_gin和数据表user:

mysql> create database t_gin;
Query OK, 1 row affected (0.01 sec)

mysql> use t_gin;
Database changed
mysql> create table user(
    -> id int not null auto_increment,
    -> name varchar(32) not null default "" comment "user name",
    -> birth varchar(32) not null default "" comment "user birthday",
    -> primary key(id)
    -> )engine=innodb comment "user info record";
Query OK, 0 rows affected (0.05 sec)

mysql>

好了,测试用的数据库和表已经建好了,接下来开始撸码。
先从原生SQL开始。

安装mysql驱动

要操作mysql,需要先装一下驱动:

$ go get github.com/go-sql-driver/mysql
go get: added github.com/go-sql-driver/mysql v1.7.0

驱动安装好了。

原生SQL-插入操作

想一下,基础的数据库操作包含插入insert、更新update以及使用最多的查询select操作,而我们要操作的目标数据表是user,所以在controller/目录下我们新建一个user.go文件,用来存放对user表的操作。
代码如下:

package controller

import (
	"github.com/gin-gonic/gin"
)

type UserController struct {
}

type User struct {
	ID    int
	Name  string
	Birth string
}

func (u *UserController) CreateUser(c *gin.Context) {

}

func (u *UserController) GetUserInfo(c *gin.Context) {

}

func (u *UserController) UpdateUserInfo(c *gin.Context) {

}

先把3个基础操作的处理列出来,内容一会儿再具体填充。
handle现在有了,路由我们也要来定义一下,按照昨天学的路由分组,user相关的这几个操作可以都归到一个group下,就叫user好了。

在main.go中,新增一个路由组user,用来测试插入、获取、更新操作,代码如下:

	user := r.Group("/user")
	{
		userCtrl := controller.UserController{}
		user.POST("/createUser", userCtrl.CreateUser)
		user.GET("/getUserInfo", userCtrl.GetUserInfo)
		user.POST("/updateUserInfo", userCtrl.UpdateUserInfo)
	}

好的,代码框架现在有了。
由于是做测试,所以user表中只有3个字段,id\name\age,而id又是自增的主键,那在插入、更新操作中,我们需要操作的就只有name\age两个字段。
先完善插入操作CreateUser:

package controller

import (
	"database/sql"
	"github.com/gin-gonic/gin"
	"log"
	"net/http"
)

type UserController struct {
}

type User struct {
	ID    int
	Name  string
	Birth string
}

func (u *UserController) CreateUser(c *gin.Context) {
	name := c.PostForm("name")
	birth := c.PostForm("birth")

	db, err := sql.Open("mysql", "root:@tcp(127.0.0.1:3306)/t_gin?charset=utf8&parseTime=true")
	if err != nil {
		log.Println(err.Error())
		c.JSON(http.StatusOK, gin.H{
			"code": 1,
			"data": false,
		})
	}

	result, err := db.Exec("insert into user(name,birth) values(?,?)", name, birth)
	if err != nil {
		log.Println(err.Error())
		c.JSON(http.StatusOK, gin.H{
			"code": 1,
			"data": false,
		})
	}

	ret := make(map[string]interface{})
	ret["id"], _ = result.LastInsertId()
	ret["effectRows"], _ = result.RowsAffected()

	c.JSON(http.StatusOK, gin.H{
		"code": 0,
		"data": ret,
	})
}

func (u *UserController) GetUserInfo(c *gin.Context) {

}

func (u *UserController) UpdateUserInfo(c *gin.Context) {

}

好的,插入操作已经完善了,那我们测试看一下
在这里插入图片描述

看这个返回是发生异常了,没有得到预期的结果,看一下终端什么情况

2023/02/14 15:51:26 sql: unknown driver "mysql" (forgotten import?)


2023/02/14 15:51:26 [Recovery] 2023/02/14 - 15:51:26 panic recovered:
POST /user/createUser HTTP/1.1
Host: localhost:8080
Accept: */*
Accept-Encoding: gzip, deflate, br
Connection: keep-alive
Content-Length: 268
Content-Type: multipart/form-data; boundary=--------------------------217154931077337827823686
Postman-Token: 4b35385e-4d83-4fec-8d50-ec6779f2cd53
User-Agent: PostmanRuntime/7.30.1


runtime error: invalid memory address or nil pointer dereference

看样子是没有引入mysql的驱动,加一下

import (
	"database/sql"
	"github.com/gin-gonic/gin"
	_ "github.com/go-sql-driver/mysql"
	"log"
	"net/http"
)

然后我们重新启动,再测试看看
在这里插入图片描述
看返回这次是成功了,那我们看下mysql中是否已经有了数据

mysql> select * from user;
+----+-------+----------+
| id | name  | birth    |
+----+-------+----------+
|  1 | Alice | 2002-2-3 |
+----+-------+----------+
1 row in set (0.03 sec)

mysql>

好的,mysql中也有数据了,插入操作尝试成功,符合预期结果。

原生SQL-获取操作

数据有写入,就会有获取。而在日常工作中,获取数据是使用频次最高的操作,那么接下来就尝试完善查询操作。
完善后的查询操作代码如下:

func (u *UserController) GetUserInfo(c *gin.Context) {
	id := c.Query("id")
	db, err := sql.Open("mysql", "root:@tcp(127.0.0.1:3306)/t_gin?charset=utf8&parseTime=true")
	if err != nil {
		log.Println(err.Error())
		c.JSON(http.StatusOK, gin.H{
			"code": 1,
			"data": false,
		})
	}

	row, err := db.Query("select * from user where id=?", id)
	if err != nil {
		log.Println(err.Error())
	}

	var user User
	row.Scan(&user)

	c.JSON(http.StatusOK, gin.H{
		"code": 0,
		"data": user,
	})
}

那我们跑一下看看是否符合预期
在这里插入图片描述
麻蛋,没获取到数据。
看这个样子,应该是数据没绑定到我们定义的struct上,检查一下,确实是绑定的时候没有按照scan的方式来,把这里调整一下:

	var user User
	err = row.Scan(&user.ID, &user.Birth, &user.Name)
	if err != nil {
		log.Println(err.Error())
	}

再试试看是不是解决了问题
在这里插入图片描述
看这返回应该是解决了个鸟蛋。。。
看终端的输出还有条信息打印:

2023/02/14 16:49:19 sql: Scan called without calling Next
[GIN] 2023/02/14 - 16:49:19 | 200 |     6.09775ms |             ::1 | GET      "/user/getUserInfo?id=1"

看这提示是说scan应该调用next,这单条记录查询也要用next吗?好吧先修改一下看看。
将scan部分的代码调整一下:

	var user User
	for row.Next() {
		err = row.Scan(&user.ID, &user.Birth, &user.Name)
		if err != nil {
			log.Println(err.Error())
		}
	}

来吧,见证奇迹的时刻到了!
在这里插入图片描述
这下子数据是有了,但是这个数据不太对,字段和值混乱了。
难道scan的时候必须按照表的字段顺序来?
试试看
继续调整scan代码

	var user User
	for row.Next() {
		err = row.Scan(&user.ID, &user.Name, &user.Birth)
		if err != nil {
			log.Println(err.Error())
		}
	}

在这里插入图片描述
这次结果正常了,终于符合我们的预期了。

还是对使用scan.next有些疑惑,又翻了翻文档,发现有一个QueryRow方法,我们调整一下代码,看看是不是符合:

	row := db.QueryRow("select * from user where id=?", id)

	var user User
	err = row.Scan(&user.ID, &user.Name, &user.Birth)
	if err != nil {
		log.Panic(err.Error())
	}

运行之后发现是符合预期的。

QueryRow是查询单行数据,获取到的数据是一条,可能直接scan进行操作;
而Query获取到的是数据集,数据集中可能包含0条或多条数据,所以使用Query方法查询出来的数据,就得使用next方法去迭代出来。
那么就好理解了,获取全量数据的场景“select * from user"就必须使用Query来处理了,获取单挑记录可以走QueryRow。

原生SQL-更新操作

接下来就是更新了。
先完善一下更新操作的代码,完善之后是这样:

func (u *UserController) UpdateUserInfo(c *gin.Context) {
	id := c.PostForm("id")
	name := c.PostForm("name")
	birth := c.PostForm("birth")

	db, err := sql.Open("mysql", "root:@tcp(127.0.0.1:3306)/t_gin?charset=utf8&parseTime=true")
	if err != nil {
		log.Println(err.Error())
		c.JSON(http.StatusOK, gin.H{
			"code": 1,
			"data": false,
		})
	}

	result, err := db.Exec("update user set name=?,birth=? where id=?", name, birth, id)
	if err != nil {
		log.Println(err.Error())
	}
	
	c.JSON(http.StatusOK, gin.H{
		"code":0,
		"data":result.RowsAffected(),
	})
}

启动一下进行测试,发现报了个错误

$ go run main.go
# t_gin/controller
controller/user.go:110:30: multiple-value result.RowsAffected() in single-value context

看提示是RowsAffected()返回两个值,不匹配,我们调整一下

	eff, _ := result.RowsAffected()
	c.JSON(http.StatusOK, gin.H{
		"code": 0,
		"data": eff,
	})

然后再运行测试看一看
在这里插入图片描述
好的,看返回是正常的,那我们看看mysql中的数据是不是确实被更新了

mysql> select * from user;
+----+------+----------+
| id | name | birth    |
+----+------+----------+
|  1 | lucy | 2002-5-1 |
+----+------+----------+
1 row in set (0.00 sec)

mysql>

确实被更新了,数据没问题。
这样,更新操作也测试通过了。
至此,插入、获取、更新这3个最基本的数据库操作,我们在gin框架中使用go的原生sql操作都完成了实现。

今天就到这里。

  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值