小白学Go语言(golang):操作MySQL数据库

对于服务端编程来说少不了和数据库打交道，今天我们就来看一下使用Go语言如何操作数据库，我们使用常用的MySQL数据库来做说明。

在Go语言中要操作数据库需要使用标准库提供的database/sql包以及具体的数据库驱动包，本文使用github.com/go-sql-driver/mysql作为MySQL数据库的驱动包。

首先看一下所使用的数据库和表，本文使用test数据库和student表，数据库中表数据如下：

首先我们来看看如何连接数据库，函数代码如下：

func myConnect(myConfig *mysql.Config) error {
    var err error
    db, err = sql.Open("mysql", myConfig.FormatDSN()) //注册MySQL数据库驱动
    if err != nil {
        return err //返回错误信息
    }
    pingErr := db.Ping() //测试是否可以连接成功
    if pingErr != nil {
        return pingErr //返回错误信息
    }
    return nil //无错误返回
}

代码说明详见注释，注意函数中db是在外部定义的全局变量。

连接数据库成功以后下面来看如何从数据库表查询数据，函数代码如下：

func studentById(id int64) (Student, error) {
    var student Student //声明student变量用于存储查询到的记录

    row := db.QueryRow("SELECT * FROM student WHERE id = ?", id) //根据id查询一行数据
    if err := row.Scan(&student.Id, &student.Name, &student.Class); err != nil { //存储行记录到student并判断错误
        if err == sql.ErrNoRows { //未查询到数据
            return student, fmt.Errorf("studentById %d: no such student.", id) //返回错误信息
        }
        return student, fmt.Errorf("studentById %d: %v", id, err) //返回错误信息
    }

    return student, nil //返回记录信息
}

代码说明详见注释，注意函数中Student类型是外部定义的一个结构体类型。查询语句中通过绑定id参数的方式来避免sql注入的风险。上面if语句先初始化一个局部变量err，再对err做判断。

下面再来看看如何向数据库中插入一条记录，函数代码如下：

func insertStudent(student *Student) (int64, error) {
    result, err := db.Exec("INSERT INTO student (name, class) VALUES (?, ?)", student.Name, student.Class) //执行插入语句
    if err != nil {
        return 0, fmt.Errorf("insertStudent: %v", err) //返回错误信息
    }
    id, err := result.LastInsertId()
    if err != nil {
        return 0, fmt.Errorf("insertStudent: %v", err) //返回错误信息
    }

    return id, nil //返回插入的记录id
}

连接、查询、插入数据库的可执行的完整代码如下：

package main

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

var db *sql.DB
type Student struct {
    Id      int64
    Name    string
    Class   string
}

func main() {
    err := myConnect(&mysql.Config{ //连接数据库
        User:   "root",
        Passwd: "12345678",
        Net:    "tcp",
        Addr:   "mysql:3306", //数据库服务器地址
        DBName: "test", //数据库名字
    })
    if err != nil {
        log.Fatal(err) //打印错误日志并终止
    }
    fmt.Println("Mysql connected!")

    student, err := studentById(1) //查询id为1的记录
    if err != nil {
        log.Fatal(err) //打印错误日志并终止
    }
    fmt.Printf("Student found: %+v\n", student)

    id, err := insertStudent(&Student{ //插入新纪录
        Name:   "Lily",
        Class:  "Class Three",
    })
    if err != nil {
        log.Fatal(err) //打印错误日志并终止
    }
    fmt.Printf("Id of added student: %v\n", id)
}

func myConnect(myConfig *mysql.Config) error {
    var err error
    db, err = sql.Open("mysql", myConfig.FormatDSN()) //注册MySQL数据库驱动
    if err != nil {
        return err //返回错误信息
    }
    pingErr := db.Ping() //测试是否可以连接成功
    if pingErr != nil {
        return pingErr //返回错误信息
    }
    return nil //无错误返回
}

func studentById(id int64) (Student, error) {
    var student Student //声明student变量用于存储查询到的记录

    row := db.QueryRow("SELECT * FROM student WHERE id = ?", id) //根据id查询一行数据
    if err := row.Scan(&student.Id, &student.Name, &student.Class); err != nil { //存储行记录到student并判断错误
        if err == sql.ErrNoRows { //未查询到数据
            return student, fmt.Errorf("studentById %d: no such student.", id) //返回错误信息
        }
        return student, fmt.Errorf("studentById %d: %v", id, err) //返回错误信息
    }

    return student, nil //返回记录信息
}

func insertStudent(student *Student) (int64, error) {
    result, err := db.Exec("INSERT INTO student (name, class) VALUES (?, ?)", student.Name, student.Class) //执行插入语句
    if err != nil {
        return 0, fmt.Errorf("insertStudent: %v", err) //返回错误信息
    }
    id, err := result.LastInsertId()
    if err != nil {
        return 0, fmt.Errorf("insertStudent: %v", err) //返回错误信息
    }

    return id, nil //返回插入的记录id
}

执行以上代码输出如下：

数据库中数据如下：

以上就是Go语言操作MySQL数据库的内容，感谢阅读。