48. 访问MySql数据库增删改查和连接池及空字段处理 | 厚土Go学习笔记

Posted 2021-04-06 Golang语言社区

和上一节相比，go 语言访问 mysql 数据库可以有更好的写法，今天来讲一下连接池。同时，也演示一下当表字段内容为 NULL 时，go 语言的处理。

首先我们建立一个新的数据库 cofoxdb 和数据表 user

新增管理员

切换tab

设置用户权限

新建数据库 cofoxdb

双击数据库成为当前库，点击图标后写入 SQL 建表脚本

建表 SQL 脚本

drop TABLE if exists `user`;
CREATE TABLE `user` (
  `id` int(11) NOT NULL AUTO_INCREMENT COMMENT '流水号',
  `userName` varchar(45) NOT NULL COMMENT '用户名【不可更改】',
  `password` varchar(255) NOT NULL COMMENT '密码',
  `nickName` varchar(45) NOT NULL COMMENT '昵称',
  `registTime` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '用户注册时间',
  `lastTimeLogin` datetime DEFAULT NULL COMMENT '上次登录时间',
  `newLoginTime` datetime DEFAULT NULL COMMENT '最新登录时间（当前登录时间）',
  `bak` varchar(1000) DEFAULT NULL COMMENT '备注',
  `online` char(1) DEFAULT 'N' COMMENT '当前在线，Y/N
Y：在线
N：不在线',
  `createTime` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '记录创建时间',
  `creator` varchar(45) DEFAULT NULL COMMENT '记录创建人',
  `updateTime` datetime DEFAULT NULL COMMENT '记录修改时间',
  `updator` varchar(45) DEFAULT NULL COMMENT '记录修改人',
  PRIMARY KEY (`id`,`userName`,`nickName`)
) ENGINE=InnoDB AUTO_INCREMENT=3 DEFAULT CHARSET=utf8 COMMENT='All Registered users';

由于在一个应用中会有多处代码需要链接数据库，所以我们准备一个全局变量，供所有需要者调用。同时声明的也有 error 变量。

var db *sql.DB
var err error

需要给 db 实例化，建立一个 init() 函数，这样，在 main() 函数执行前就可以把数据库链接完成初始化了。

func init()  {
    db, err = sql.Open("mysql", "cofox:Q1w2e3r4@tcp(127.0.0.1:3306)/cofoxdb?charset=utf8")
    check(err)

    db.SetMaxOpenConns(2000)
    db.SetMaxIdleConns(1000)
    check(db.Ping())
}

▪ db.SetMaxOpenConns(2000) 是设置这个连接池最大链接数是 2000 个。

▪ db.SetMaxIdleConns(1000) 设置的是连接池内最低保持 1000 个待用链接。这样当有需要访问的程序请求时，就可以从连接池内分配一条已有的链接。提高访问效率。

▪ db.Ping() 是为了让程序和数据库进行真正的链接（sql.Open并没有建立真正的连接关系，只是初始化。）

插入数据

直接使用 db.Prepare ，因为 db 已经初始化了。

res.LastInsertId() 执行后返回最新的 id。如果是批量数据插入的话，这个会返回第一条记录的 id。

func insert()  {
    stmt, err := db.Prepare(`INSERT user (userName, password, nickName) VALUES (?, ?, ?)`)
    check(err)

    res, err := stmt.Exec("cofox_1","123456","冷静的狐狸")
    check(err)

    id, err := res.LastInsertId()
    check(err)

    fmt.Println(id)
    stmt.Close()

}

修改数据

也是直接使用 db

res.RowsAffected() 提交执行，返回修改了的记录数。

func update() {
    stmt, err := db.Prepare("UPDATE user set nickName=?, updateTime=?, updator=?, bak=? WHERE id=?")
    check(err)

    res, err := stmt.Exec("厚土火焰山",time.Now().Format("2006-01-02 15:04:05"),"root","测试更新
go直连数据库", 1)
    check(err)

    num, err := res.RowsAffected()
    check(err)

    fmt.Println(num)
    stmt.Close()
}

删除数据

同上

res.RowsAffected()提交执行，返回删除了的记录数。

func remove() {
    stmt, err := db.Prepare("DELETE FROM cofoxdb WHERE id=?")
    check(err)

    res, err := stmt.Exec(7)
    check(err)

    num, err := res.RowsAffected()
    check(err)

    fmt.Println(num)
    stmt.Close()

}

查询数据

因为表字段较多，很多字段在新增后或许仍然没有写入相应的数据，这些字段如果没有默认值的话，就会是 NULL 值。

NULL 值在 go 语言中是不能写入 string time.Time 的。所以这里我们使用 "database/sql" 提供的 sql.NullString 类型。当然 Null** 类型还有很多 NullInt64、NullFloat64、NullBool

var id int
var userName string
var password string
var nickName string
var registTime string
var lastTimeLogin sql.NullString
var newLoginTime sql.NullString
var bak sql.NullString
var online sql.NullString
var createTime sql.NullString
var creator sql.NullString
var updateTime sql.NullString
var updator sql.NullString

我们用这个类型来处理字段有可能为 NULL 的数据。这样就可以正常读取记录值了。

这些 NullString 的类型结构是这样的

type NullString struct {
    String string
    Valid  bool // Valid is true if String is not NULL
}

都是有两个字段在里面。而 String 字段就是我们最终想要的东西。所以，在输出或使用的时候，我们这样组织代码

lastTimeLogin.String, newLoginTime.String, bak.String, online.String, createTime.String, creator.String, updateTime.String, updator.String

执行 insert() 后，我们再执行 query2()，得到如下结果

id = "3", userName = "cofox_1", password = "123456", nickName = "冷静的狐狸", registTime = "2017-09-07 17:39:02", lastTimeLogin = "", newLoginTime = "", bak = "", online = "N", createTime = "2017-09-07 17:39:02", creator = "", updateTime = "", updator = ""

看完整代码示例

package main

import (
    "database/sql"
    _"github.com/go-sql-driver/mysql"
    "fmt"
    "log"
    "time"
)

var db *sql.DB
var err error

func init()  {
    db, err = sql.Open("mysql", "cofox:Q1w2e3r4@tcp(127.0.0.1:3306)/cofoxdb?charset=utf8")
    check(err)

    db.SetMaxOpenConns(2000)
    db.SetMaxIdleConns(1000)
    check(db.Ping())
}

func main() {
    //query()
    query2()
    //insert()
    //update()
    //remove()
}

//查询数据
func query() {
    rows, err := db.Query("SELECT * FROM user")
    check(err)

    for rows.Next() {
        columns, _ := rows.Columns()

        scanArgs := make([]interface{}, len(columns))
        values := make([]interface{}, len(columns))

        for i := range values {
            scanArgs[i] = &values[i]
        }

        //将数据保存到 record 字典
        err = rows.Scan(scanArgs...)
        record := make(map[string]string)
        for i, col := range values {
            if col != nil {
                record[columns[i]] = string(col.([]byte))
            }
        }
        fmt.Println(record)
    }
    rows.Close()

}

func query2()  {
    rows, err := db.Query("SELECT id, userName, password, nickName, registTime, lastTimeLogin, newLoginTime, bak, online, createTime, creator, updateTime, updator FROM user")
    check(err)

    for rows.Next(){
        var id int
        var userName string
        var password string
        var nickName string
        var registTime string
        var lastTimeLogin sql.NullString
        var newLoginTime sql.NullString
        var bak sql.NullString
        var online sql.NullString
        var createTime sql.NullString
        var creator sql.NullString
        var updateTime sql.NullString
        var updator sql.NullString

        //注意这里的Scan括号中的参数顺序，和 SELECT 的字段顺序要保持一致。
        if err := rows.Scan(&id, &userName, &password, &nickName, &registTime, &lastTimeLogin, &newLoginTime, &bak, &online, &createTime, &creator, &updateTime, &updator); err != nil {
            log.Fatal(err)
        }

        fmt.Printf("id = "%d", userName = "%s", password = "%s", nickName = "%s", registTime = "%s", lastTimeLogin = "%s", newLoginTime = "%s", bak = "%s", online = "%s", createTime = "%s", creator = "%s", updateTime = "%s", updator = "%s"
",id, userName, password, nickName, registTime, lastTimeLogin.String, newLoginTime.String, bak.String, online.String, createTime.String, creator.String, updateTime.String, updator.String)

    }

    if err := rows.Err(); err != nil {
        log.Fatal(err)
    }
    rows.Close()
}

//插入数据
func insert()  {
    stmt, err := db.Prepare(`INSERT user (userName, password, nickName) VALUES (?, ?, ?)`)
    check(err)

    res, err := stmt.Exec("cofox_1","123456","冷静的狐狸")
    check(err)

    id, err := res.LastInsertId()
    check(err)

    fmt.Println(id)
    stmt.Close()

}

//修改数据
func update() {
    stmt, err := db.Prepare("UPDATE user set nickName=?, updateTime=?, updator=?, bak=? WHERE id=?")
    check(err)

    res, err := stmt.Exec("厚土火焰山",time.Now().Format("2006-01-02 15:04:05"),"root","测试更新
go直连数据库", 1)
    check(err)

    num, err := res.RowsAffected()
    check(err)

    fmt.Println(num)
    stmt.Close()
}

//删除数据
func remove() {
    stmt, err := db.Prepare("DELETE FROM cofoxdb WHERE id=?")
    check(err)

    res, err := stmt.Exec(7)
    check(err)

    num, err := res.RowsAffected()
    check(err)

    fmt.Println(num)
    stmt.Close()

}

func check(err error) {
    if err != nil{
        fmt.Println(err)
        panic(err)
    }
}

---

关于作者

↓↓↓「阅读原文」是传送门