2. mysql预处理

什么是预处理？

普通SQL语句执行过程：

1. 客户端对SQL语句进行占位符替换得到完整的SQL语句。

2. 客户端发送完整SQL语句到MySQL服务端

3. MySQL服务端执行完整的SQL语句并将结果返回给客户端。

预处理执行过程：

1. 把SQL语句分成两部分，命令部分与数据部分。

2. 先把命令部分发送给MySQL服务端，MySQL服务端进行SQL预处理。

3. 然后把数据部分发送给MySQL服务端，MySQL服务端对SQL语句进行占位符替换。

4. MySQL服务端执行完整的SQL语句并将结果返回给客户端。

为什么要预处理？

1. 优化MySQL服务器重复执行SQL的方法，可以提升服务器性能，提前让服务器编译，一次编译多次执行，节省后续编译的成本。

2. 避免SQL注入问题。

Go实现MySQL预处理

database/sql中使用下面的Prepare方法来实现预处理操作。

func (db *DB) Prepare(query string) (*Stmt, error)

Prepare方法会先将sql语句发送给MySQL服务端，返回一个准备好的状态用于之后的查询和命令。返回值可以同时执行多个查询和命令。

查询操作的预处理示例代码如下：

// PrepareQueryDemo 预处理查询示例
func PrepareQueryDemo() {
	sqlStr := "select id, name, age from user where id > ?"
	stmt, err := sqlClient.Db.Prepare(sqlStr)
	if err != nil {
		fmt.Printf("prepare failed, err:%v\n", err)
		return
	}
	defer stmt.Close()
	rows, err := stmt.Query(0)
	if err != nil {
		fmt.Printf("query failed, err:%v\n", err)
		return
	}
	defer rows.Close()
	// 循环读取结果集中的数据
	for rows.Next() {
		var u user
		err := rows.Scan(&u.id, &u.name, &u.age)
		if err != nil {
			fmt.Printf("scan failed, err:%v\n", err)
			return
		}
		fmt.Printf("id:%d name:%s age:%d\n", u.id, u.name, u.age)
	}
}

插入、更新和删除操作的预处理十分类似，这里以插入操作的预处理为例：

// PrepareInsertDemo 预处理插入示例
func PrepareInsertDemo(name string, age int) {
	sqlStr := "insert into user(name, age) values (?,?)"
	stmt, err := sqlClient.Db.Prepare(sqlStr)
	if err != nil {
		fmt.Printf("prepare failed, err:%v\n", err)
		return
	}
	defer stmt.Close()
	_, err = stmt.Exec(name, age)
	if err != nil {
		fmt.Printf("insert failed, err:%v\n", err)
		return
	}
	fmt.Println("insert success.")
}

示例

operation.go

package operation

import (
	"fmt"
	sqlClient "main/utils"
)

// PrepareQueryDemo 预处理查询示例
func PrepareQueryDemo() {
	sqlStr := "select id, name, age from user where id > ?"
	stmt, err := sqlClient.Db.Prepare(sqlStr)
	if err != nil {
		fmt.Printf("prepare failed, err:%v\n", err)
		return
	}
	defer stmt.Close()
	rows, err := stmt.Query(0)
	if err != nil {
		fmt.Printf("query failed, err:%v\n", err)
		return
	}
	defer rows.Close()
	// 循环读取结果集中的数据
	for rows.Next() {
		var u user
		err := rows.Scan(&u.id, &u.name, &u.age)
		if err != nil {
			fmt.Printf("scan failed, err:%v\n", err)
			return
		}
		fmt.Printf("id:%d name:%s age:%d\n", u.id, u.name, u.age)
	}
}

// PrepareInsertDemo 预处理插入示例
func PrepareInsertDemo(name string, age int) {
	sqlStr := "insert into user(name, age) values (?,?)"
	stmt, err := sqlClient.Db.Prepare(sqlStr)
	if err != nil {
		fmt.Printf("prepare failed, err:%v\n", err)
		return
	}
	defer stmt.Close()
	_, err = stmt.Exec(name, age)
	if err != nil {
		fmt.Printf("insert failed, err:%v\n", err)
		return
	}
	fmt.Println("insert success.")
}

main.go

package main

import (
	"fmt"
	basicCURD "main/operation"
	sqlClient "main/utils"
)

func main() {

	// 一、连接mysql
	if err := sqlClient.InitDB(); err != nil { // 调用输出化数据库的函数
		fmt.Printf("init db failed,err:%v\n", err)
		return
	}
	db := sqlClient.Db
	defer db.Close()
	fmt.Println("连接成功")

	// 二、基本增删改查
	//// 查询单条语句
	basicCURD.QueryRowDemo()
	//// 查询多态语句
	//basicCURD.QueryMultiRowDemo()
	//// 插入数据
	//basicCURD.InsertRowDemo("高起盛", 19)
	//// 更新数据
	//basicCURD.UpdateRowDemo(999, 1)
	//// 删除数据
	//basicCURD.DeleteRowDemo(7)

	// 三、预处理增删改查
	basicCURD.PrepareQueryDemo()
	basicCURD.InsertRowDemo("唐小虎", 19)
}

/*
连接成功

报错: init db failed,err:sql: unknown driver "mysql" (forgotten import?) 为安装mysql
*/