golang数据库操作初体验

https://golang.org/s/sqldrivers

安装

直接执行go get,然后会下载到你的$GOPATH中，如果用的go mod也一样，只不过下载的路径不一样。

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

导入驱动

database/sql这个包是要导入的，然后导入go-sql-driver,包前面的 “_"表示执行包的init函数，函数init里面直接将自已注册到database/sql包中，然后就能用这个包里面的方法访问数据库了。

import (
    "database/sql"
    _ "github.com/go-sql-driver/mysql"   
)
​
func init() {
    sql.Register("mysql", &MySQLDriver{})
}

连接数据库

type dbObj struct {
    db *sql.DB
}
​
func (d *dbObj) Open() *sql.DB{
    var err error
    d.db, err = sql.Open("mysql", "lc:111111@/test")
    if err != nil {
        panic(err)
    }
    return d.db
}

sql.Opendb.ping()

sql.Open

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

它接受两个参数:

[username[:password]@][protocol[(address)]]/dbname[?param1=value1&...¶mN=valueN]

示例

来点基本的CURD操作，还是挺简单的。

package db
​
import (
    "database/sql"
    "fmt"
    _ "github.com/go-sql-driver/mysql"
    "log"
)
​
type  userInfo struct {
    id int
    orgcode string
    name string
    version int
}
​
type dbObj struct {
    db *sql.DB
}
​
func (d *dbObj) Open() *sql.DB{
    var err error
    d.db, err = sql.Open("mysql", "lc:111111@/test")
    if err != nil {
        panic(err)
    }
    return d.db
}
func (d *dbObj) Close(){
    d.Close()
}
​
​
func SelectAll() {
    dbc :=&dbObj{}
    db := dbc.Open()
    defer db.Close()
​
    stmt ,_ :=db.Prepare("SELECT orgcode,`name` FROM  userinfo WHERE id > ?")
    rows, _ :=stmt.Query(0) //query为多行
    defer rows.Close()
    user :=&userInfo{}
​
    for rows.Next() {
        err :=rows.Scan(&user.orgcode,&user.name)
        if err != nil {
            log.Fatal(err)
        }
        fmt.Println(user.orgcode , ":", user.name)
    }
}
​
func Select() {
    dbc :=&dbObj{}
    db := dbc.Open()
    defer db.Close()
​
    stmt ,_ :=db.Prepare("SELECT orgcode,`name` FROM  userinfo WHERE ID= ?")
    rows :=stmt.QueryRow(1008) //QueryRow为单行
​
    user :=&userInfo{}
​
​
    err :=rows.Scan(&user.orgcode,&user.name)
    if err != nil {
        log.Fatal(err)
    }
    fmt.Println(user.orgcode , ":", user.name)
​
}
​
func Insert() {
    dbc :=&dbObj{}
    db := dbc.Open()
    defer db.Close()
​
    result, err :=db.Exec("INSERT  userinfo (orgcode,imei,`name`) VALUE(?,?,?)","cccc",1009,"cccc")
    if err != nil {
        log.Fatal(err)
    }
    rowsaffected,err :=  result.RowsAffected()
    if err != nil {
        fmt.Printf("获取受影响行数失败,err:%v",err)
        return
    }
    fmt.Println("受影响行数:" ,rowsaffected )
}
​
func Delete() {
    dbc :=&dbObj{}
    db := dbc.Open()
    defer db.Close()
​
    result, err :=db.Exec("DELETE FROM userinfo WHERE id=?",1009)
    if err != nil {
        log.Fatal(err)
    }
    rowsaffected,err :=  result.RowsAffected()
    if err != nil {
        fmt.Printf("获取受影响行数失败,err:%v",err)
        return
    }
    fmt.Println("受影响行数:" ,rowsaffected )
}
​
func Update() {
    dbc :=&dbObj{}
    db := dbc.Open()
    defer db.Close()
​
    result, err :=db.Exec("UPDATE userinfo SET `name`= ? WHERE id=?", "lcbbb",1008)
    if err != nil {
        log.Fatal(err)
    }
​
    rowsaffected,err :=  result.RowsAffected()
    if err != nil {
        fmt.Printf("获取受影响行数失败,err:%v",err)
        return
    }
    fmt.Println("受影响行数:" ,rowsaffected )
}
​
​
​
func Transaction() {
    dbc :=&dbObj{}
    db := dbc.Open()
    defer db.Close()
​
    tx ,_:=db.Begin()
    tx.Exec("UPDATE userinfo SET `name`= ? WHERE id=?", "lcaaa",1007)
    result, err :=tx.Exec("UPDATE userinfo SET `name`= ? WHERE id=?", "lcbbb",1008)
    if err != nil {
        log.Fatal(err)
    }
    tx.Commit() //提交事务
    rowsaffected,err :=  result.RowsAffected()
    if err != nil {
        fmt.Printf("获取受影响行数失败,err:%v",err)
        return
    }
    fmt.Println("受影响行数:" ,rowsaffected )
}
​
​
func ConcurrenceUpdate() {
    dbc :=&dbObj{}
    db := dbc.Open()
    defer db.Close()
​
    getone :=func(db *sql.DB,id int) *userInfo{
​
​
        stmt ,_ :=db.Prepare("SELECT orgcode,`name`,version FROM  userinfo WHERE ID= ?")
        rows :=stmt.QueryRow(id) //
​
        user :=&userInfo{}
​
​
        err :=rows.Scan(&user.orgcode,&user.name, &user.version)
        if err != nil {
            log.Fatal(err)
        }
​
        return user
​
    }
​
    udateone :=func(db *sql.DB,name string,id int,version int){
        result, err :=db.Exec("UPDATE userinfo SET `name`= ?, version=version+1 WHERE id=? AND version=?", name,id,version)
        if err != nil {
            log.Fatal(err)
        }
​
        rowsaffected,err :=  result.RowsAffected()
        if err != nil {
            fmt.Printf("并发更新获取受影响行数失败,err:%v",err)
            return
        }
        fmt.Println("并发更新受影响行数:" ,rowsaffected )
    }
​
    num :=10
    for i:=0; i<num ;i++{
        go func(){
            u :=getone(db,1008)
            fmt.Printf("获取数据:%v\r\n",u)
            udateone(db,"lc并发更新测试", 1008,u.version)
        }()
    }
​
    select{}
}

以上只是一些基本的数据库操作，像连接池等还没有做，这个等后续看了再去写。