对于服务端编程来说少不了和数据库打交道，今天我们就来看一下使用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数据库的内容，感谢阅读。