golang——database sql包学习


1、database/sql包

sql包提供了保证SQL或类SQL数据库的泛用接口。

使用sql包时必须注入(至少)一个数据库驱动。

(1)获取mysql driver:go get -v github.com/go-sql-driver/mysql

(2)代码示例:

package main

import (
    "database/sql"
    "fmt"
    "log"
    "time"

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

// 检查错误
func checkErr(err error) {
    if err != nil {
        log.Fatalln(err)
    }
}

// 事务错误
func checkTxErr(err error, tx *sql.Tx) {
    if err != nil {
        log.Println(err)
        err = tx.Rollback()
        checkErr(err)
    }
}

// 数据库实体
type User struct {
    ID      int
    Name    string
    Age     int
    Sex     int
    AddDate time.Time
}

func main() {
    //1、获取数据库连接
    db, err := sql.Open("mysql", "root:123456@tcp(127.0.0.1:3306)/test?parseTime=true")
    checkErr(err)
    defer db.Close()
    fmt.Println("数据库连接成功")
    //2、判断连接是否有效
    err = db.Ping()
    checkErr(err)
    fmt.Println("数据库连接有效")
    //3、创建表
    sql := `
        CREATE TABLE IF NOT EXISTS users(
            id INT NOT NULL AUTO_INCREMENT,
            name VARCHAR(100) NOT NULL,
            age INT NOT NULL,
            sex TINYINT,
            add_date DATETIME,
            PRIMARY KEY(id) 
        )
    `
    _, err = db.Exec(sql)
    checkErr(err)

    //4、添加数据
    // sql = "INSERT INTO users (name,age,sex,add_date) VALUES (?,?,?,?)"
    // res, err := db.Exec(sql, "张三", 18, 1, time.Now())
    // checkErr(err)
    // fmt.Println(res.LastInsertId())

    //5、查询数据
    sql = "SELECT id,name,age,sex,add_date FROM users"
    rows, err := db.Query(sql)
    checkErr(err)
    defer rows.Close()
    user := User{}
    for rows.Next() {
        err = rows.Scan(&user.ID, &user.Name, &user.Age, &user.Sex, &user.AddDate)
        checkErr(err)
        fmt.Println(user, user.AddDate.Format("2006/01/02 15:04:05"))
    }
    err = rows.Err()
    checkErr(err)

    //6、查询一行
    sql = "SELECT id,name,age,sex,add_date FROM users"
    row := db.QueryRow(sql)
    err = row.Scan(&user.ID, &user.Name, &user.Age, &user.Sex, &user.AddDate)
    checkErr(err)
    fmt.Println(user)

    //7、命令
    sql = "UPDATE users SET name=? WHERE id=?;"
    stmt, err := db.Prepare(sql)
    checkErr(err)
    defer stmt.Close()
    result, err := stmt.Exec("李四", 1)
    checkErr(err)
    fmt.Println(result.RowsAffected())

    //8、查询
    sql = "SELECT id,name,age,sex,add_date FROM users WHERE id=?"
    stmt2, err := db.Prepare(sql)
    checkErr(err)
    defer stmt2.Close()
    row = stmt2.QueryRow(1)
    err = row.Scan(&user.ID, &user.Name, &user.Age, &user.Sex, &user.AddDate)
    checkErr(err)
    fmt.Println(user)

    //9、事务
    tx, err := db.Begin()
    checkErr(err)
    _, err = tx.Exec("UPDATE users SET age=? WHERE id=?", 20, 1)
    checkTxErr(err, tx)
    _, err = tx.Exec("UPDATE users SET sex=? WHERE id=?", 1, 1)
    checkTxErr(err, tx)
    err = tx.Commit()
    checkTxErr(err, tx)

    //10、查询一行
    sql = "SELECT id,name,age,sex,add_date FROM users"
    row = db.QueryRow(sql)
    err = row.Scan(&user.ID, &user.Name, &user.Age, &user.Sex, &user.AddDate)
    checkErr(err)
    fmt.Println(user)

}

2、数据库

2.1、type DB struct{}

DB是一个数据库句柄,代表一个具有零到多个底层连接的连接池。

它可以安全的被多个go程同时使用。

连接池的大小可以用SetMaxIdleConns方法控制。

2.2、常用方法

(1)func Open(driverName, dataSourceName string) (*DB, error)

打开数据库,返回数据库句柄,DB可以安全的被多个go程同时使用,并会维护自身的闲置连接池。

Open函数只需调用一次,很少需要关闭DB。

(2)func (db *DB) Driver() driver.Driver

返回数据库下层驱动。

(3)func (db *DB) Ping() error

检查与数据库的连接是否仍有效,如果需要会创建连接。

(4)func (db *DB) Close() error

关闭数据库,释放任何打开的资源。

一般不会关闭DB,因为DB句柄通常被多个go程共享,并长期活跃。

(5)func (db *DB) SetMaxOpenConns(n int)

设置与数据库建立连接的最大数目。

如果n大于0且小于最大闲置连接数,会将最大闲置连接数减小到匹配最大开启连接数的限制。

如果n <= 0,不会限制最大开启连接数,默认为0(无限制)。

(6)func (db *DB) SetMaxIdleConns(n int)

设置连接池中的最大闲置连接数。

如果n大于最大开启连接数,则新的最大闲置连接数会减小到匹配最大开启连接数的限制。

如果n <= 0,不会保留闲置连接。

(7)func (db *DB) Exec(query string, args ...interface{}) (Result, error)

执行一次命令(包括查询、删除、更新、插入等),不返回任何执行结果。

参数args表示query中的占位参数。

(8)func (db DB) Query(query string, args ...interface{}) (Rows, error)

执行一次查询,返回多行结果(即Rows),一般用于执行select命令。

(9)func (db DB) QueryRow(query string, args ...interface{}) Row

执行一次查询,并期望返回最多一行结果(即Row)。

总是返回非nil的值,直到返回值的Scan方法被调用时,才会返回被延迟的错误。

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

创建一个准备好的状态用于之后的查询和命令。

返回值可以同时执行多个查询和命令。

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

开始一个事务。

隔离水平由数据库驱动决定。

3、数据表

3.1、type Rows{}

Rows是查询的结果。

它的游标指向结果集的第零行,使用Next方法来遍历各行结果。

3.2、常用方法

(1)func (rs *Rows) Columns() ([]string, error)

返回列名。

如果Rows已经关闭会返回错误。

(2)func (rs *Rows) Scan(dest ...interface{}) error

Scan将当前行各列结果填充进dest指定的各个值中。

如果某个参数的类型为*[]byte,Scan会保存对应数据的拷贝,该拷贝为调用者所有,可以安全的,修改或无限期的保存。

如果参数类型为*RawBytes可以避免拷贝;参见RawBytes的文档获取其使用的约束。

如果某个参数的类型为*interface{},Scan会不做转换的拷贝底层驱动提供的值。

如果值的类型为[]byte,会进行数据的拷贝,调用者可以安全使用该值。

(3)func (rs *Rows) Next() bool

Next准备用于Scan方法的下一行结果。

如果成功会返回真,如果没有下一行或者出现错误会返回假。

每一次调用Scan方法,甚至包括第一次调用该方法,都必须在前面先调用Next方法。

(4)func (rs *Rows) Close() error

关闭Rows,阻止对其更多的列举。

如果Next方法返回假,Rows会自动关闭。

检查Err方法结果的条件。

Close方法是幂等的(多次调用无效的成功),不影响Err方法的结果。

(5)func (rs *Rows) Err() error

Err返回可能的、在迭代时出现的错误。

Err需在显式或隐式调用Close方法后调用。

4、数据行

4.1、type Row{}

QueryRow方法返回Row,代表单行查询结果。

4.2、常用方法

(1)func (r *Row) Scan(dest ...interface{}) error

Scan将该行查询结果各列分别保存进dest参数指定的值中。

如果该查询匹配多行,Scan会使用第一行结果并丢弃其余各行。

如果没有匹配查询的行,Scan会返回ErrNoRows。

5、SQL命令状态

5.1、type Stmt struct{}

Stmt是准备好的状态。

Stmt可以安全的被多个go程同时使用。

5.2、常用方法

(1)func (s *Stmt) Exec(args ...interface{}) (Result, error)

使用提供的参数执行准备好的命令状态,返回Result类型的该状态执行结果的总结。

(2)func (s Stmt) Query(args ...interface{}) (Rows, error)

使用提供的参数执行准备好的查询状态,返回Rows类型查询结果。

(3)func (s Stmt) QueryRow(args ...interface{}) Row

使用提供的参数执行准备好的查询状态。

如果在执行时遇到了错误,该错误会被延迟,直到返回值的Scan方法被调用时才释放。

返回值总是非nil的。

如果没有查询到结果,*Row类型返回值的Scan方法会返回ErrNoRows;否则,Scan方法会扫描结果第一行并丢弃其余行。

(4)func (s *Stmt) Close() error

关闭状态。

6、事务

6.1、type Tx struct{}

Tx代表一个进行中的数据库事务。

一次事务必须以对Commit或Rollback的调用结束。

调用Commit或Rollback后,所有对事务的操作都会失败并返回错误值ErrTxDone。

6.2、常用方法

(1)func (tx *Tx) Exec(query string, args ...interface{}) (Result, error)

执行命令,但不返回结果。例如执行insert和update。

(2)func (tx Tx) Query(query string, args ...interface{}) (Rows, error)

执行查询并返回零到多行结果(Rows),一般执行select命令。

(3)func (tx Tx) QueryRow(query string, args ...interface{}) Row

执行查询并期望返回最多一行结果(Row)。

总是返回非nil的结果,查询失败的错误会延迟到在调用该结果的Scan方法时释放。

(4)func (tx Tx) Prepare(query string) (Stmt, error)

准备一个专用于该事务的状态。

返回的该事务专属状态操作在Tx递交会回滚后不能再使用。

(5)func (tx Tx) Stmt(stmt Stmt) *Stmt

使用已存在的状态生成一个该事务特定的状态。

(6)func (tx *Tx) Commit() error

提交事务。

(7)func (tx *Tx) Rollback() error

回滚事务。

golang——database sql包学习介绍到这里,更多go学习请参考编程字典go教程 和问答部分,谢谢大家对编程字典的支持。


原文链接:https://www.cnblogs.com/dzhy/p/11065300.html