Go操作mysql

下载mysql驱动

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

可能报错：

1.设置好GOPATH，将GOBIN添加到系统环境变量中

2.把Git添加到系统环境变量，要设置的PATH变量值为：c:Program FilesGitin

 

package main

import (

    "database/sql"

    _ "github.com/go-sql-driver/mysql"   //下载的mysql驱动

    "fmt"

)

var db *sql.DB   //连接池

func initDB() (err error) {

    dsn := "root:wyz@tcp(127.0.0.1:3306)/wang"

    //导入了未使用，为什么能通过"mysql"字符串找到"github.com/go-sql-driver/mysql"？

    //init()函数里，向"database/sql"注册了一个"mysql"驱动

    db, err = sql.Open("mysql", dsn)  

    if err != nil{

        return

    }

    err = db.Ping()

    if err != nil {

        return

    }

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

    db.SetMaxOpenConns(100)

    //设置最大空闲连接数（没人用的时候会关闭一些连接）

    db.SetMaxIdleConns(50)

    return

}

type user struct {

    id int64

    name string

    age int

}

// 查询单行记录

func queryRow(id int) *user {

    

    var u1 user

    //1.写查询条件查单条记录

    sqlStr := `select id, name, age from user where id=?`

    //2.执行并拿到结果，返回一行或者nil

    //必须对rowObj对象调用Scan方法，因为该方法会归还数据库连接

    db.QueryRow(sqlStr, id).Scan(&u1.id, &u1.name, &u1.age)

    //返回结果

    return &u1

}

// 查询多行记录

func queryRows(id int) {

    sqlStr := "select id, name, age from user where id > ?"

    rows, err := db.Query(sqlStr, id)

    if err != nil {

        fmt.Println("查询失败：", err)

        return

    }

    // 非常重要：关闭rows释放持有的数据库链接

    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 ", err)

            return

        }

        fmt.Printf("id:%d name:%s age:%d ", u.id, u.name, u.age)

    }

}

// 插入数据

func insertRow() {

    sqlStr := "insert into user(name, age) values (?,?)"

    ret, err := db.Exec(sqlStr, "带头大哥", 38)  //ret用于获取插入那一行的ID

    if err != nil {

        fmt.Printf("插入数据失败:%v ", err)

        return

    }

    lastID, err := ret.LastInsertId()  // 新插入数据的id

    if err != nil {

        fmt.Printf("获取ID失败:%v ", err)

        return

    }

    fmt.Printf("插入数据成功ID：%d. ", lastID)

}

// 更新数据

func updateRow() {

    sqlStr := "update user set age=? where id = ?"

    ret, err := db.Exec(sqlStr, 39, 4) // ret用于获取受影响的行数

    if err != nil {

        fmt.Printf("更新失败:%v ", err)

        return

    }

    n, err := ret.RowsAffected() // 操作影响的行数

    if err != nil {

        fmt.Printf("获取受影响行数失败:%v ", err)

        return

    }

    fmt.Printf("更新成功，影响行数:%d ", n)

}

// 删除数据

func deleteRow() {

    sqlStr := "delete from user where id = ?"

    ret, err := db.Exec(sqlStr, 3)   // ret用于获取受影响的行数

    if err != nil {

        fmt.Printf("删除失败:%v ", err)

        return

    }

    n, err := ret.RowsAffected() // 操作影响的行数

    if err != nil {

        fmt.Printf("获取受影响行数失败:%v ", err)

        return

    }

    fmt.Printf("删除成功，影响行数:%d ", n)

}

func main() {

    err := initDB()

    if err != nil {

        fmt.Println("连接数据库成功")

    }

    u1 := queryRow(1)

    fmt.Println(u1)

    queryRows(1)

}

数据库	占位符语法
MySQL	?
PostgreSQL	$1, $2等
SQLite	? 和$1
Oracle	:name

什么是预处理？

普通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注入问题。

// 预处理查询示例
func prepareQueryDemo() {
	sqlStr := "select id, name, age from user where id > ?"
	stmt, err := db.Prepare(sqlStr)
	if err != nil {
		fmt.Printf("prepare failed, err:%v
", err)
		return
	}
	defer stmt.Close()
	rows, err := stmt.Query(1)  //查询id>1的数据
	if err != nil {
		fmt.Printf("query failed, err:%v
", 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
", err)
			return
		}
		fmt.Printf("id:%d name:%s age:%d
", u.id, u.name, u.age)
	}
}

// 预处理插入示例
func prepareInsertDemo() {
	sqlStr := "insert into user(name, age) values (?,?)"
	stmt, err := db.Prepare(sqlStr)
	if err != nil {
		fmt.Printf("prepare failed, err:%v
", err)
		return
	}
	defer stmt.Close()
	_, err = stmt.Exec("小王子", 18)
	if err != nil {
		fmt.Printf("insert failed, err:%v
", err)
		return
	}
	_, err = stmt.Exec("沙河娜扎", 18)
	if err != nil {
		fmt.Printf("insert failed, err:%v
", err)
		return
	}
	fmt.Println("insert success.")
}

Go实现MySQL事务

（ACID原子性、一致性、隔离性、持久性）

开始事务

func (db *DB) Begin() (*Tx, error)

提交事务

func (tx *Tx) Commit() error

回滚事务

func (tx *Tx) Rollback() error

//  给id为1的age-2   给id为2的age+2 模拟转账操作

func transactionDemo() {

    // 1. 开启事务

    tx, err := db.Begin()

    if err != nil {

        fmt.Printf("开启事务失败:%v ", err)

        return

    }

    // 执行多个SQL操作

    sqlStr1 := `update user set age=age-2 where id=1`

    sqlStr2 := `update xxx set age=age+2 where id=2`

    // 执行SQL1

    _, err = tx.Exec(sqlStr1)

    if err != nil {

        // 要回滚

        tx.Rollback()

        fmt.Println("执行SQL1出错啦，要回滚！")

        return

    }

    // 执行SQL2

    _, err = tx.Exec(sqlStr2)

    if err != nil {

        // 要回滚

        tx.Rollback()

        fmt.Println("执行SQL2出错啦，要回滚！")

        return

    }

    // 上面两步SQL都执行成功，就提交本次事务

    err = tx.Commit()

    if err != nil {

        // 要回滚

        tx.Rollback()

        fmt.Println("提交出错啦，要回滚！")

        return

    }

    fmt.Println("事务执行成功！")

}

123