go数据库操作

go数据库操作

下载数据库驱动

比如要使用MYSQL，就要下载MYSQL驱动：go get github.com/go-sql-driver/mysql

连接数据库

连接数据的DSN格式为: 
username:password@protocol(address)/dbname?param=value

db, err := sql.Open("mysql", "root:root@tcp(127.0.0.1:3306)/test?charset=utf8")
    if err != nil {
        fmt.Println("failed to open database:", err.Error())
        return
    }
    defer db.Close()　　

返回的DB对象,实际封装了一个数据库连接池,对于goroutine是线程安全的，可以放心使用。这个数据库连接池由"database/sql"包负责自动创建和回收。连接池的大小可以由SetMaxIdleConns指定。 
需要注意的是，创建DB对象成功，并不代表已经成功的连接了数据库，数据库连接只有在真正需要的时候才会被创建。　　

关闭数据库

defer db.Close()

CRUD 
DB中执行SQL通过Exec和Query方法，查询操作是通过Query完成，它会返回一个sql.Rows的结果集，包含一个游标用来遍历查询结果；Exec方法返回的是sql.Result对象，用于检测操作结果，及被影响记录数。

查询

rows, err := db.Query("SELECT * FROM user")
if err != nil {
    fmt.Println("fetech data failed:", err.Error())
    return
}
defer rows.Close()
for rows.Next() {
    var uid int
    var name, password string
    rows.Scan(&uid, &name, &password)
    fmt.Println("uid:", uid, "name:", name, "password:", password)　　

新增

result, err := db.Exec("INSERT INTO user(name,password) VALUES('tom', 'tom')")
if err != nil {
    fmt.Println("insert data failed:", err.Error())
    return
}
id, err := result.LastInsertId()
if err != nil {
    fmt.Println("fetch last insert id failed:", err.Error())
    return
}
fmt.Println("insert new record", id)

　修改

result, err = db.Exec("UPDATE user SET password=? WHERE name=?", "tom_new_password", "tom")
if err != nil {
    fmt.Println("update data failed:", err.Error())
    return
}
num, err := result.RowsAffected()
if err != nil {
    fmt.Println("fetch row affected failed:", err.Error())
    return
}
fmt.Println("update recors number", num)

　删除

result, err = db.Exec("DELETE FROM user WHERE name=?", "tom")
if err != nil {
    fmt.Println("delete data failed:", err.Error())
    return
}
num, err = result.RowsAffected()
if err != nil {
    fmt.Println("fetch row affected failed:", err.Error())
    return
}
fmt.Println("delete record number", num)　　

事务控制

sql.Tx用来支持事务处理

tx, err := db.Begin()
result, err = tx.Exec("DELETE FROM order WHERE uid=? ", 2)
if err != nil {
    fmt.Println("delete data failed:", err.Error())
    return
}
num, err = result.RowsAffected()
if err != nil {
    fmt.Println("fetch row affected failed:", err.Error())
    return
}
fmt.Println("delete record number", num)
result, err = tx.Exec("DELETE FROM user WHERE uid=? ", 2)
if err != nil {
    fmt.Println("delete data failed:", err.Error())
    return
}
num, err = result.RowsAffected()
if err != nil {
    fmt.Println("fetch row affected failed:", err.Error())
    return
}
fmt.Println("delete record number", num)
// 根据条件回滚或者提交
// tx.Rollback()
tx.Commit()　　

预备表达式

sql.Stmt支持预备表达式，可以用来优化SQL查询提高性能，减少SQL注入的风险, DB.Prepare()和Tx.Prepare()都提供了对于预备表达式的支持。

stmt, err := db.Prepare("DELETE FROM order WHERE oid=?")
if err != nil {
    fmt.Println("fetch row affected failed:", err.Error())
    return
}
result, err = stmt.Exec(1)
if err != nil {
    fmt.Println("delete data failed:", err.Error())
    return
}
num, err = result.RowsAffected()
if err != nil {
    fmt.Println("fetch row affected failed:", err.Error())
    return
}
fmt.Println("delete record number", num)

注意点

每次db.Query()操作后，都建议调用rows.Close()