Go操作mysql

下载mysql驱动

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

可能报错:

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

2.把Git添加到系统环境变量,要设置的PATH变量值为:c:\Program Files\Git\bin

 
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\n", err)
            return
        }
        fmt.Printf("id:%d name:%s age:%d\n", 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\n", err)
        return
    }
    lastID, err := ret.LastInsertId()  // 新插入数据的id
    if err != nil {
        fmt.Printf("获取ID失败:%v\n", err)
        return
    }
    fmt.Printf("插入数据成功ID:%d.\n", lastID)
}

// 更新数据
func updateRow() {
    sqlStr := "update user set age=? where id = ?"
    ret, err := db.Exec(sqlStr, 39, 4) // ret用于获取受影响的行数
    if err != nil {
        fmt.Printf("更新失败:%v\n", err)
        return
    }
    n, err := ret.RowsAffected() // 操作影响的行数
    if err != nil {
        fmt.Printf("获取受影响行数失败:%v\n", err)
        return
    }
    fmt.Printf("更新成功,影响行数:%d\n", n)
}

// 删除数据
func deleteRow() {
    sqlStr := "delete from user where id = ?"
    ret, err := db.Exec(sqlStr, 3)   // ret用于获取受影响的行数
    if err != nil {
        fmt.Printf("删除失败:%v\n", err)
        return
    }
    n, err := ret.RowsAffected() // 操作影响的行数
    if err != nil {
        fmt.Printf("获取受影响行数失败:%v\n", err)
        return
    }
    fmt.Printf("删除成功,影响行数:%d\n", 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\n", err) return } defer stmt.Close() rows, err := stmt.Query(1) //查询id>1的数据 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 prepareInsertDemo() { sqlStr := "insert into user(name, age) values (?,?)" stmt, err := db.Prepare(sqlStr) if err != nil { fmt.Printf("prepare failed, err:%v\n", err) return } defer stmt.Close() _, err = stmt.Exec("小王子", 18) if err != nil { fmt.Printf("insert failed, err:%v\n", err) return } _, err = stmt.Exec("沙河娜扎", 18) if err != nil { fmt.Printf("insert failed, err:%v\n", 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\n", 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

你可能感兴趣的:(Go操作mysql)