【笔记】Go语言操作Mysql数据库

Go语言操作Mysql数据库

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

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

连接数据库

<username>：用户名
<password>：密码
<host>：数据库地址
<port>：数据库端口
<database>：需要连接的数据库名

db, err := sql.Open("mysql", "<username>:<password>@tcp(<host>:<port>)/<database>")

执行SQL

<sql>：SQL语句

_, err := db.Exec("<sql>")

• 通过?作为可变参数的占位符

即便占位符得到的数据是字符串，也不要用成对的单引号将占位符引起来

_, err := db.Exec("SELECT * FROM user WHERE username=?", "name")

其他数据库的占位符

MySQL：?
PostgreSQL：$1
SQLite：?、$1
Oracle：:name

• 如果新增数据成功，可以获取id字段

result, err := db.Exec("INSERT INTO user VALUES(null, username, password)")
id, err := result.LastInsertId()

• 如果修改数据成功，可以获取受影响的行数

result, err := db.Exec("UPDATE user SET key=value")
rows, err := result.RowsAffected()

• 如果删除数据成功，可以获取受影响的行数

result, err := db.Exec("DELETE FROM user WHERE key=value")
rows, err := result.RowsAffected()

执行有数据返回的SQL

查询单条数据

• 在执行Scan()函数后，会自动将数据库连接放回池中
• 根据SELECT后拼接的字段名，按顺序解析结果，要注意使用对应的数据类型

var username string
var password string
row, err := db.QueryRow("SELECT username, password FROM user WHERE id=1")
row.Scan(username, password)

var username string
var password string
row, err := db.QueryRow("SELECT username, password FROM user WHERE id=1").Scan(username, password)

• 也可以将结果直接封装到结构体

查询多条数据

执行查询语句

rows, err := db.Query("SELECT username, password FROM user")
defer rows.Close()

创建结构体

• 创建映射数据表的结构体

type User struct {
    Username string
    Password string
}

遍历查询结果集

• 将结果集封装到结构体中

for rows.Next() {
    user := new(User)
    err := rows.Scan(&user.Username, $user.Password)
}

设置数据库连接池最大连接数

<num>：最大连接数，大于0且小于最大闲置连接数

db.SetMaxOpenConns(<num>)

设置数据库连接池最大闲置连接数

• 将不使用的连接自动关闭

db.SetMaxIdleConns(<num>)

封装数据库获取连接

package tool

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

var db *sql.DB

// 初始化数据库连接
func DatabaseInit() (err error) {
	// 获取配置文件
	cfg := GetConfig()
	// 获取数据库连接
	db, err = sql.Open("mysql", "<username>:<password>@tcp(<ip>:<port>)/<database>")
	if err != nil {
		fmt.Println("数据库初始化时DSN格式有误")
		return err
	}
	// 尝试与数据库建立连接，检测密码是否正确
	err = db.Ping()
	if err != nil {
		fmt.Println("数据库初始化时登陆失败")
		return err
	}
	fmt.Println("连接数据库成功")
	return
}

// 获取数据库连接
func DatabaseGetConnect() *sql.DB {
	if db != nil {
		return db
	}
	err := DatabaseInit()
	if err != nil {
		return nil
	}
	return db
}

Go语言实现Mysql预处理

• 提前让服务器编译SQL，一次编译多次执行，提升性能
• 避免SQL注入问题

// 先编译SQL
stmt, err := db.Perpare("SELECT * FROM user WHERE username=?")
defer stms.Close()
// 再添加参数
rows, err := stmt.Query("user")
defer rows.Close()
for rows.Next() {
    var user User
    err := rows.Scan(&user.username)
}

Go语言实现Mysql事务

// 开启事务
tx, err := db.Begin()
result, err := tx.Exec("<sql>")
if err != nil {
    // 回滚事务
    tx.Rollback()
}
// 提交事务
err = tx.Commit()

哔哩哔哩——80技术
哔哩哔哩——地鼠文档