1.连接

Go语言中的database/sql包提供了保证SQL或类SQL数据库的泛用接口,并不提供具体的数据库驱动。
使用database/sql包时必须注入(至少)一个数据库驱动

1.1下载依赖

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

1.2使用MySQL驱动

  1. func Open(driverName, dataSourceName string) (*DB, error)
  1. import "database/sql"
  2. import _ "github.com/go-sql-driver/mysql"
  3. func main() {
  4. dsn :="user:password@tcp(127.0.0.1:3306)/dbname"
  5. db, err := sql.Open("mysql", dsn)
  6. if err != nil {
  7. panic(err)
  8. }
  9. defer db.Close()
  10. }

1.3初始化连接

Open函数可能只是验证其参数格式是否正确,实际上并不创建与数据库的连接。
如果要检查数据源的名称是否真实有效,应该调用Ping方法。

  1. import (
  2. "database/sql"
  3. "fmt"
  4. _ "github.com/go-sql-driver/mysql"
  5. )
  6. // 定义一个初始化数据库的函数
  7. var db *sql.DB
  8. func initDB() (err error) {
  9. dsn :="user:password@tcp(127.0.0.1:3306)/sql_test?charset=utf8mb4&parseTime=True"
  10. db, err = sql.Open("mysql", dsn) //Open函数验证其参数格式是否正确
  11. if err != nil {
  12. return err
  13. }
  14. err = db.Ping() //Ping方法检查数据源的名称是否真实有效
  15. if err != nil {
  16. return err
  17. }
  18. return nil
  19. }
  20. func main() {
  21. err := initDB()
  22. if err != nil {
  23. fmt.Printf("init db failed,err:%v\n",err)
  24. return
  25. }
  26. }

sql.DB是表示连接的数据库对象(结构体实例),它保存了连接数据库相关的所有信息。
它内部维护着一个具有零到多个底层连接的连接池,它可以安全地被多个goroutine同时使用。

1.4SetMaxOpenConns

  1. func (db *DB) SetMaxOpenConns(n int)

SetMaxOpenConns设置与数据库建立连接的最大数目。 如果n大于0且小于最大闲置连接数,会将最大闲置连接数减小到匹配最大开启连接数的限制。 如果n<=0,不会限制最大开启连接数,默认为0(无限制)。

1.5SetMaxIdleConns

  1. func (db *DB) SetMaxIdleConns(n int)

SetMaxIdleConns设置连接池中的最大闲置连接数。 如果n大于最大开启连接数,则新的最大闲置连接数会减小到匹配最大开启连接数的限制。 如果n<=0,不会保留闲置连接。

2.CRUD

2.1建表

  1. CREATE TABLE `user` (
  2. `id` BIGINT(20) NOT NULL AUTO_INCREMENT,
  3. `name` VARCHAR(20) DEFAULT '',
  4. `age` INT(11) DEFAULT '0',
  5. PRIMARY KEY(`id`)
  6. )ENGINE=InnoDB AUTO_INCREMENT=1 DEFAULT CHARSET=utf8mb4;

2.2单行查询

  1. import (
  2. "database/sql"
  3. "fmt"
  4. _ "github.com/go-sql-driver/mysql"
  5. )
  6. var db *sql.DB
  7. type user struct {
  8. id int
  9. age int
  10. name string
  11. }
  12. // 定义一个初始化数据库的函数
  13. func initDB() (err error) {
  14. // DSN:Data Source Name
  15. dsn := "root:123456@tcp(127.0.0.1:3306)/sql_test?charset=utf8mb4&parseTime=True"
  16. // 不会校验账号密码是否正确
  17. // 注意!!!这里不要使用:=,我们是给全局变量赋值,然后在main函数中使用全局变量db
  18. db, err = sql.Open("mysql", dsn)
  19. if err != nil {
  20. return err
  21. }
  22. // 尝试与数据库建立连接(校验dsn是否正确)
  23. err = db.Ping()
  24. if err != nil {
  25. return err
  26. }
  27. return nil
  28. }
  29. func queryRowDemo(){
  30. sqlStr := "select id, name, age from user where id=?"
  31. var u user
  32. //确保QueryRow之后调用Scan方法,否则持有的数据库链接不会被释放
  33. err := db.QueryRow(sqlStr, 1).Scan(&u.id, &u.name, &u.age)
  34. if err != nil {
  35. fmt.Printf("scan failed,err:%v\n",err)
  36. return
  37. }
  38. fmt.Printf("id:%d,name:%s age:%d",u.id,u.name,u.age)
  39. }
  40. func main() {
  41. err := initDB() // 调用输出化数据库的函数
  42. if err != nil {
  43. fmt.Printf("init db failed,err:%v\n", err)
  44. return
  45. }
  46. queryRowDemo()
  47. }

2.3多行查询

多行查询db.Query()执行一次查询,返回多行结果(即Rows)

  1. func queryMultiRowDemo() {
  2. sqlStr :="select id,name,age from user where id > ?"
  3. rows, err := db.Query(sqlStr, 0)
  4. if err != nil {
  5. fmt.Printf("query failed,err:%v\n",err)
  6. return
  7. }
  8. defer rows.Close()
  9. for rows.Next(){
  10. var u user
  11. err := rows.Scan(&u.id, &u.name, &u.age)
  12. if err != nil {
  13. fmt.Printf("scan failed,err:%v\n",err)
  14. return
  15. }
  16. fmt.Printf("id:%d name:%s age:%d\n",u.id,u.name,u.age)
  17. }
  18. }

2.4插入数据

插入、更新和删除操作都使用Exec方法
Exec执行一次命令(包括查询、删除、更新、插入等),返回的Result是对已执行的SQL命令的总结。

func insertRowDemo() {
    sqlStr := "insert into user(name,age) values (?,?)"
    result, err := db.Exec(sqlStr, "cheng", 18)
    if err != nil {
        fmt.Printf("insert failed,err:%v\n",err)
        return
    }
    id, err := result.LastInsertId()
    if err != nil {
        fmt.Printf("get lastinsert id failed,err:%v\n",err)
        return
    }
    fmt.Println(result)
    fmt.Println("id",id)

}

2.5更新数据

func updateRowDemo()  {
    sqlStr := "update user set age = ? where id = ?"
    result, err := db.Exec(sqlStr, 23, 2)
    if err != nil {
        fmt.Printf("update failed,err:%v",err)
        return
    }
    n, err := result.RowsAffected()   // 操作影响的行数
    if err != nil {
        fmt.Printf("get rowsaffected failed,err:%v",err)
        return
    }
    fmt.Println(n)   //影响的行数
}

2.6删除数据

func deleteRowDemo()  {
    sqlStr := "delete from user where id = ?"
    result, err := db.Exec(sqlStr, 3)
    if err != nil {
        fmt.Printf("delete failed,err:%v\n",err)
        return
    }
    n, err := result.RowsAffected()
    if err != nil {
        fmt.Printf("get failed,err:%v",err)
    }
    fmt.Println(n)
}

3.MySQL预处理

3.1什么是预处理?

普通SQL语句执行过程:

  1. 客户端对SQL语句进行占位符替换得到完整的SQL语句。
  2. 客户端发送完整SQL语句到MySQL服务端
  3. MySQL服务端执行完整的SQL语句并将结果返回给客户端。

预处理执行过程:

  1. 把SQL语句分成两部分,命令部分与数据部分。
  2. 先把命令部分发送给MySQL服务端,MySQL服务端进行SQL预处理。
  3. 然后把数据部分发送给MySQL服务端,MySQL服务端对SQL语句进行占位符替换。
  4. MySQL服务端执行完整的SQL语句并将结果返回给客户端。

    3.2为什么要预处理?

  5. 优化MySQL服务器重复执行SQL的方法,可以提升服务器性能。

  6. 提前让服务器编译,一次编译多次执行,节省后续编译的成本。
  7. 避免SQL注入问题。

3.3Go实现MySQL预处理

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

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

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

查询

func prepareMultiRowDemo() {
    sqlStr :="select id,name,age from user where id > ?"
    prepare, err := db.Prepare(sqlStr)
    if err != nil {
        fmt.Printf("prepare failed,err:%v",err)
        return
    }
    defer prepare.Close()
    rows, err := prepare.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)
    }
}

插入

func prepareInsertRowDemo() {
    sqlStr := "insert into user(name,age) values (?,?)"
    prepare, err := db.Prepare(sqlStr)
    if err != nil {
        fmt.Printf("prepare failed, err:%v\n", err)
        return
    }
    defer prepare.Close()
    result, err := prepare.Exec("cheng", 25)
    if err != nil {
        fmt.Printf("insert failed,err:%v\n",err)
        return
    }
    id, err := result.LastInsertId()
    if err != nil {
        fmt.Printf("get lastinsert id failed,err:%v\n",err)
        return
    }
    fmt.Println("id:",id)
    fmt.Println("insert success")

}

3.4SQL注入问题

我们任何时候都不应该自己拼接SQL语句

// sql注入示例
func sqlDemo(id int)  {
    sqlStr := fmt.Sprintf("select id,name,age from user where id='%d'",id)
    fmt.Printf("SQL:%s\n",sqlStr)
    var u user
    err := db.QueryRow(sqlStr).Scan(&u.id, &u.name, &u.age)
    if err != nil {
        fmt.Printf("exec failed,err:%v\n",err)
        return
    }
    fmt.Printf("user:%#v\n",u)
}

补充:不同的数据库中,SQL语句使用的占位符语法不尽相同。

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

4.Go实现MySQL事务

在MySQL中只有使用了Innodb数据库引擎的数据库或表才支持事务。
事务处理可以用来维护数据库的完整性,保证成批的SQL语句要么全部执行,要么全部不执行

4.1事务的ACID

条件 解释
原子性 一个事务(transaction)中的所有操作,要么全部完成,要么全部不完成,不会结束在中间某个环节。事务在执行过程中发生错误,会被回滚(Rollback)到事务开始前的状态,就像这个事务从来没有执行过一样。
一致性 在事务开始之前和事务结束以后,数据库的完整性没有被破坏。这表示写入的资料必须完全符合所有的预设规则,这包含资料的精确度、串联性以及后续数据库可以自发性地完成预定的工作。
隔离性 数据库允许多个并发事务同时对其数据进行读写和修改的能力,隔离性可以防止多个事务并发执行时由于交叉执行而导致数据的不一致。事务隔离分为不同级别,包括读未提交(Read uncommitted)、读提交(read committed)、可重复读(repeatable read)和串行化(Serializable)。
持久性 事务处理结束后,对数据的修改就是永久的,即便系统故障也不会丢失。

4.2事务相关方法

func (db *DB) Begin() (*Tx, error) //开始事务
func (tx *Tx) Commit() error      //提交事务
func (tx *Tx) Rollback() error    //回滚事务

4.3事务示例

该事物操作能够确保两次更新操作要么同时成功要么同时失败,不会存在中间状态。

func transactionDemo() {
    tx, err := db.Begin()  //开启事务
    if err != nil {
        if tx != nil {
            tx.Rollback()  // 回滚
        }
        fmt.Printf("begin failed,err:%v",err)
    }
    sqlStr := "update user set age = 30 where id = ?"
    result, err := tx.Exec(sqlStr, 2)
    if err != nil {
        tx.Rollback()   //回滚
        fmt.Printf("exec failed,err:%v",err)
        return
    }
    n, err := result.RowsAffected()
    if err != nil {
        tx.Rollback()   //回滚
        fmt.Printf("exec rows failed,err:%v",err)
        return
    }
    fmt.Println("影响的行数:",n)
    if n==1{
        tx.Commit()
        fmt.Println("事务提交啦")
    }else {
        tx.Rollback()  //事务回滚
        fmt.Println("事务回滚啦")
    }
    fmt.Println("success")

}