Go使用MySQL数据库

Posted 2021-04-04 码道仕

简介

Go官方并不提供数据库驱动，而只是为开发数据库驱动定义了一些标准接口。我们操作mysql数据库得需要使用第三方维护的MySQL驱动，推荐使用github.com/go-sql-driver/mysql，主要理由如下：

• 维护的比较好
• 完全支持database/sql接口
• 支持keepalive，保持长连接

有了驱动，其实Go使用MySQL和其他语言差不多，只是具体的方法调用和API组织结构不太一样。

准备工作

use golang;

CREATE TABLE `user` (
    `id` int(11) NOT NULL AUTO_INCREMENT,
    `username` varchar(100) NOT NULL DEFAULT '' COMMENT '用户名',
    `email` varchar(50) NOT NULL DEFAULT '' COMMENT '用户邮箱',
    `ctime` int(11) NOT NULL COMMENT '创建时间',
    `utime` int(11) NOT NULL COMMENT '更新时间',
    PRIMARY KEY (`id`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8;

先通过ssh登录服务器（因为安全考虑，不开放MySQL的root用户的远程登录权限），用root用户本地登录mysql新建一个用于测试的可远程登录的用户：

mysql> GRANT ALL PRIVILEGES ON golang.* to 'golang'@'%' IDENTIFIED BY '123456';
mysql> GRANT ALL PRIVILEGES ON golang.* to 'golang'@'localhost' IDENTIFIED BY '123456';
mysql> FLUSH PRIVILEGES;

接下来就可以通过golang用户登录新建的数据库进行操作了。

操作MySQL

先get数据库驱动，命令行运行go get github.com/go-sql-driver/mysql会自动将驱动下载到$GOPATH/src中。

package main

import (
    "database/sql"
    "fmt"
    "time"

    _ "github.com/go-sql-driver/mysql"
)

func checkError(e error) {
    if e != nil {
        panic(e)
    }
}

func main() {
    // 打开MySQL连接
    db, err := sql.Open("mysql", "golang:123456@tcp(127.0.0.1:3306)/golang?charset=utf8")
    checkError(err)
    fmt.Println("connected")
    // ?为占位符，防止SQL注入
    stmt, err := db.Prepare("insert into user(username, gender, email, ctime, utime) values(?, ?, ?, ?, ?)")
    checkError(err)
    timestamp := time.Now().Unix()
    // 向占位符传参，Exec是一个不定参数函数，传入的参数与Prepare阶段设置的占位符相等
    res, err := stmt.Exec("zhangsan", 1, "[e-mail]", timestamp, timestamp)
    checkError(err)
    // 获取生成的数据id
    id, err := res.LastInsertId()
    checkError(err)
    fmt.Println(id)

    stmt, err = db.Prepare("update user set email=? where id=?")
    checkError(err)
    res, err = stmt.Exec("[e-mail]", id)
    checkError(err)
    aff, err := res.RowsAffected()
    checkError(err)
    fmt.Println("affected", aff)

    rows, err := db.Query("select id, username, email from user")
    for rows.Next() {
        var id int
        var username string
        var email string
        err = rows.Scan(&id, &username, &email)
        checkError(err)
        fmt.Println(id, username, email)
    }

    stmt, _ = db.Prepare("delete from user where id>?")
    res, _ = stmt.Exec(1)
    aff, _ = res.RowsAffected()
    fmt.Println("affected", aff)
}

整个数据库操作的流程大概如下：

• sql.Open()函数用来打开一个注册过的数据库驱动；
• db.Prepare()函数用来返回准备要执行的sql操作，然后返回准备完毕的执行状态；
• stmt.Exec()函数用来执行stmt准备好的SQL语句；
• db.Query()函数用来直接执行Sql返回Rows结果。

数据库驱动的注册在数据库驱动包导入时执行，注册的逻辑写在包的init函数中。导入驱动包只需要import _ "driver/package"，只需要执行其init()，而不需要调用包中导出的方法，调用的是Go语言规定的sql接口。