GoLang—MySQL数据库操作-蒲公英云

Go标准库中没有数据库驱动，如果实现数据库连接与操作，参见获取第三方的数据库驱动。本文以MySQL为例，我们使用go-sql-driver实现数据库的连接和操作，首先在CMD窗口下安装驱动go-sql-driver，指令如下。

go get github.com/go-sql-driver/mysql

数据库驱动安装成功后，在GoLand（ide）中使用go-sql-driver实现数据库的连接和操作，实现方式首先连接MySQL的某个数据库，生成数据库对象db，然后使用数据库对象db对数据表实现怎删改查操作。

数据查询

go-sql-driver提供2种数据查询方式：Query和QueryRow。Query是查询多行数据，也可以查询单行数据，适用范围较广；QueryRow只能查询单行数据，该对象能调用的方法较少，局限性较大。不管Query或QueryRow，两者只查询数据表的数据，如果想将数据表的字段与数据一一对应，需要我们编写函数方法实现，废话不多说，直接看代码。

package main
import (
    "database/sql"
    "encoding/json"
    "fmt"
    _ "github.com/go-sql-driver/mysql"
)
func GetAll(rows *sql.Rows) []map[string]string {
    // 判断传入的查询结果是否为nil
    if rows == nil {
        return nil
    }
    // 定义返回值的数据类型（定义为数组，元素类型为映射，映射的数据为字符串类型）
    var result []map[string]string
    // 获取查询结果rows的列名（即字段名）
    cols, _ := rows.Columns()
    // 定义数组byte
    rawResult := make([][]byte, len(cols))
    // 定义数组dest
    dest := make([]interface{}, len(cols))
    // 将byte数组的内存地址对应分配给dest
    for i, _ := range rawResult {
        dest[i] = &rawResult[i]
    }
    // 遍历每行数据
    for rows.Next()  {
        // 将每行的数据分别存入数组dest
        // 由于数组dest和数组byte的内存地址是相同的
        // 因此两个数组的数据也是相同，只不过两个数组的每个元素的数据格式不同
        rows.Scan(dest...)
        // 根据列的个数定义相应的映射对象sresult
        sresult := make(map[string]string, len(cols))
        // 将当前数据写入映射对象sresult
        for i, raw := range rawResult {
            if raw == nil {
                sresult[cols[i]] = ""
            } else {
                sresult[cols[i]] = string(raw)
            }
        }
        // 将数据写入数组result
        result = append(result, sresult)
    }
    // 将数组result作为返回值
    return result
}
func main(){
    // 连接MySQL数据库
    db, _ := sql.Open("mysql", "root:1234@tcp(127.0.0.1:3306)/music_db?charset=utf8")
    // 查询方式一，先使用Prepare再使用Query或QueryRow
    p , _ := db.Prepare("select * from index_song")
    rows, _ := p.Query()
    // 查询方式二，直接使用Query或QueryRow
    //rows, _ := db.Query("select * from index_song")
    s2 := GetAll(rows)
    for _, v := range s2{
        // 将映射对象转化JSON数据
        mjson, _ :=json.Marshal(v)
        mString :=string(mjson)
        fmt.Println(mString)
    }
    // 查询方式三，查询单条数据
    // 适合单条数据的某些字段，如果查询全部字段，建议使用Query
    row := db.QueryRow("select name,album from index_song where id=1")
    var name,album string
    row.Scan(&name,&album)
    fmt.Println(name,album)
}

上述代码中，涉及了两个Go的基础语法，下划线（_）和三个点（…），本文不再详细讲述语法知识，读者可以点击相关链接查看。
函数方法GetAll是将Query查询对象rows作为函数参数，函数返回值result为数组类型，数组元素的数据类型为映射对象。函数首先从查询对象rows获取数据表的字段名，生成对象cols，将对象cols的字段名作为返回值result的映射对象的key，而数据表的数据作为映射对象的value。具体的实现过程不过多讲述，详细可看代码注释。

数据增改删

数据的增改删操作主要由Exec方法实现，实现方法较为简单，直接看到代码即可理解

package main
import (
    "database/sql"
    "fmt"
    _ "github.com/go-sql-driver/mysql"
)
func main(){
    // 连接MySQL数据库
    db, _ := sql.Open("mysql", "root:1234@tcp(127.0.0.1:3306)/music_db?charset=utf8")
    // 新增数据
    ret, _ := db.Exec("insert into index_label(id,name) values(15,'测试测试')")
    // LastInsertId是获取数据新增后的ID
    insID, _ := ret.LastInsertId()
    fmt.Println(insID)
    // 修改数据
    ret2, _ := db.Exec("update index_label set name= 'GOGO' where id = ?", 15)
    // RowsAffected是获取已被修改的数据总数
    aff_nums, _ := ret2.RowsAffected()
    fmt.Println(aff_nums)
    // 删除数据
    ret3, _ := db.Exec("delete from index_label where id = ?",15)
    delNums, _ := ret3.RowsAffected()
    fmt.Println(delNums)
}

事务

事务由Begin函数方法实现，只需在数据库连接对象调用Begin函数方法即可，然后由Begin函数方法生成的对象执行数据操作即可，可以根据执行结果判断当前事务是否执行或回滚，详细代码如下。

package main
import (
    "database/sql"
    "fmt"
    _ "github.com/go-sql-driver/mysql"
)
func main(){
    // 连接MySQL数据库
    db, _ := sql.Open("mysql", "root:1234@tcp(127.0.0.1:3306)/music_db?charset=utf8")
    //开启事务
    tx, _ := db.Begin()
    //id为1的price+1，id为2的price-1
    ret4, _ := tx.Exec("update index_label set name = '测试事务1' where id = ?", 10)
    ret5, _ := tx.Exec("update index_label set name = '测试事务2' where id = ?", 2)
    //如果id不存在,受影响行数则为0
    //接收影响行数，为0则失败
    updNums1, _ := ret4.RowsAffected()
    updNums2, _ := ret5.RowsAffected()
    if updNums1 > 0 && updNums2 > 0 { //只有两条更新同时成功，那么才提交
        tx.Commit() //提交事务
        fmt.Println("Success")
    } else { //否则回滚
        tx.Rollback() //回退事务
        fmt.Println("Fail")
    }
}