Golang连接MySQL数据库之CRUD

我们这次来学习使用Golang来连接MySQL数据库，并使用Golang实现数据库的CRUD操作。

搭建环境

首先我们创建一个Golang的项目，并配置项目GOPATH，这一步可以参考我的博客Golang环境安装&IDEA开发Golang。

因为我们使用的是MySQL数据库，所以我们需要获取Golang的MySQL数据库驱动。
我们在项目GOPATH的目录下，执行go get命令来获取MySQL的驱动，命令执行成功之后，会直接从网上下载MySQL驱动的包到你的GOPATH目录下的src目录下。

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

Golang的项目环境搭建完成之后，我们还需要创建一张数据库表。

CREATE TABLE `tb_user` (
  `id` int(11) NOT NULL AUTO_INCREMENT,
  `name` varchar(11) DEFAULT NULL,
  `password` varchar(11) DEFAULT NULL,
  PRIMARY KEY (`id`),
  KEY `nameindex` (`name`(10))
) ENGINE=InnoDB DEFAULT CHARSET=utf8;

项目结构

编写结构体

为了能够便于封装从数据库中获取的数据，我们创建一个Golang结构，来封装我们的数据。
我们创建一个bean包，然后在该包下创建一个user.go文件。

user.go

package bean

type User struct {
	id int
	name string
	password string
}

func (user *User) GetId() int {
	return user.id
}

func (user *User) SetId(id int) {
	user.id = id
}

func (user *User) GetName() string {
	return user.name
}

func (user *User) SetName(name string) {
	user.name = name
}

func (user *User) GetPassword() string {
	return user.password
}

func (user *User) SetPassword(password string) {
	user.password = password
}

编写数据库连接工具

由于每次连接数据库都需要获取数据库连接，所以我们直接将数据库的连接操作封装为一个方法，这样就可以不用每次都执行获取连接的步骤了。
我们创建一个util包，在包下创建一个initdb.go文件。

initdb.go

package util

import (
	"database/sql"
	_ "github.com/go-sql-driver/mysql"
	"strings"
)

//我们先将数据库配置信息定义成为常量
const (
	userName = "root"
	password = "admin"
	ip       = "127.0.0.1"
	port     = "3306"
	dbName   = "db_database08"
)

//初始化数据库连接，返回数据库连接的指针引用
func InitDB() *sql.DB {
	//Golang数据连接："用户名:密码@tcp(IP:端口号)/数据库名?charset=utf8"
	path := strings.Join([]string{userName, ":", password, "@tcp(", ip, ":", port, ")/", dbName, "?charset=utf8"}, "")
	//打开数据库,前者是驱动名，所以要导入： _ "github.com/go-sql-driver/mysql"
	db, err := sql.Open("mysql", path)
	if err != nil {
		//如果打开数据库错误，直接panic
		panic(err)
	}
	//设置数据库最大连接数
	db.SetConnMaxLifetime(10)
	//设置上数据库最大闲置连接数
	db.SetMaxIdleConns(5)
	//验证连接
	if err := db.Ping(); err != nil {
		panic(err)
	}
	//将数据库连接的指针引用返回
	return db
}

CRUD操作

Insert操作

我们创建一个insert包，在该包下创建一个insert.go文件

package main

import (
	"fmt"
	"mysql/util"
)

func main() {
	//使用工具获取数据库连接
	db := util.InitDB()
	//开启事务
	tx, err := db.Begin()
	if err != nil {
		//事务开启失败，直接panic
		panic(err)
	}
	//准备SQL语句
	sql := "insert into tb_user (`name`, `password`) values (?, ?)"
	//对SQL语句进行预处理
	stmt, err := db.Prepare(sql)
	if err != nil {
		panic(err)
	}
	result, err := stmt.Exec("阿部多瑞","123")
	if err != nil {
		//SQL执行失败，直接panic
		panic(err)
	}
	//提交事务
	tx.Commit()
	//返回插入记录的id
	fmt.Println(result.LastInsertId())
}

Select操作

我们创建一个select包，在该包下创建一个select.go文件

package main

import (
	"fmt"
	"mysql/bean"
	"mysql/util"
)

func main() {
	//使用工具获取数据库连接
	db := util.InitDB()
	//准备SQL语句
	sql := "select * from tb_user"
	//对SQL语句进行预处理
	stmt, err := db.Prepare(sql)
	if err != nil {
		panic(err)
	}
	rows, err := stmt.Query()
	if err != nil {
		//SQL执行失败，直接panic
		panic(err)
	}
	var users []bean.User
	for rows.Next() {
		var id int
		var name, password string
		err := rows.Scan(&id, &name, &password)
		if err != nil {
			//读取结果集失败
			panic(err)
		}
		var user bean.User
		user.SetId(id)
		user.SetName(name)
		user.SetPassword(password)
		users = append(users, user)
	}
	fmt.Println(users)
}

Update操作

我们创建一个update包，在该包下创建一个update.go文件

package main

import (
	"mysql/util"
)

func main() {
	//使用工具获取数据库连接
	db := util.InitDB()
	//开启事务
	tx, err := db.Begin()
	if err != nil {
		//事务开启失败，直接panic
		panic(err)
	}
	//准备SQL语句
	sql := "update tb_user set `password` = ? where `id` = ?"
	//对SQL语句进行预处理
	stmt, err := db.Prepare(sql)
	if err != nil {
		panic(err)
	}
	_, err = stmt.Exec("789", 1)
	if err != nil {
		//SQL执行失败，直接panic
		panic(err)
	}
	//提交事务
	tx.Commit()
}

Delete操作

我们创建一个delete包，在该包下创建一个delete.go文件

package main

import (
	"mysql/util"
)

func main() {
	//使用工具获取数据库连接
	db := util.InitDB()
	//开启事务
	tx, err := db.Begin()
	if err != nil {
		//事务开启失败，直接panic
		panic(err)
	}
	//准备SQL语句
	sql := "delete from tb_user where `id` = ?"
	//对SQL语句进行预处理
	stmt, err := db.Prepare(sql)
	if err != nil {
		panic(err)
	}
	_, err = stmt.Exec(1)
	if err != nil {
		//SQL执行失败，直接panic
		panic(err)
	}
	//提交事务
	tx.Commit()
}