Golang如何优雅连接MySQL数据库?

 2022-10-27    404  

Go原生就支持连接数据库,所以在使用 Golang 开发时,当需要数据库交互时,即可使用database/sql包。

在Go中访问DB需用sql.DB接口:可创建语句(statement)和事务(transaction),执行查询,获取结果。

使用DB时,除database/sql包,还需引入想使用的特定DB驱动。官方不提供实现,需要先下载三方实现,点击这里查看各种各样的实现版本。

通常DB选型MySQL,所以选型驱动为:github.com/go-sql-driver/mysql,需引入包:

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

包名前的"_"

import 下划线(如:import _ github/demo)的作用:当导入一个包时,该包下的文件里所有init()函数都会被执行,然而,有些时候我们并不需要把整个包都导入进来,仅仅是是希望它执行init()函数而已。这个时候就可以使用 import _ 引用该包。

上面的MySQL驱动中引入的就是MySQL包中各个init()方法,你无法通过包名来调用包中的其他函数。导入时,驱动的初始化函数会调用sql.Register将自己注册在database/sql包的全局变量sql.drivers中,以便以后通过sql.Open访问。

案例用数据表

初始化数据库连接

sql.Open()中的数据库连接串格式为:"用户名:密码@tcp(IP:端口)/数据库?charset=utf8"。DB的类型为:*sql.DB,有DB后即可执行CRUD。

Go将数据库操作分为两类:Query与Exec

  • Query表示查询,它会从数据库获取查询结果(一系列行,可能为空)。
  • Exec表示执行语句,它不会返回行。

常见数据库操作模式:

  • QueryRow只返回一行的查询,作为Query的一个常见特例。
  • Prepare准备一个需要多次使用的语句,供后续执行用。

查询操作

varuserUser
rows,e:=DB.Query("select*fromuserwhereidin(1,2,3)")
ife==nil{
errors.New("queryincurerror")
}
forrows.Next(){
e:=rows.Scan(user.sex,user.phone,user.name,user.id,
user.age)
ife!=nil{
fmt.Println(json.Marshal(user))
}
}
rows.Close()
//单行查询操作
DB.QueryRow("select*fromuserwhereid=1").Scan(user.age,
user.id,user.name,user.phone,user.sex)

执行流程

  1. 使用db.Query()来发送查询到数据库,获取结果集Rows,并检查错误。
  2. 使用rows.Next()作为循环条件,迭代读取结果集。
  3. 使用rows.Scan从结果集中获取一行结果。
  4. 使用rows.Err()在退出迭代后检查错误。
  5. 使用rows.Close()关闭结果集,释放连接。

    增删改和Exec

    通常不会约束你查询必须用Query,只是Query会返回结果集,而Exec不会返回。所以如果你执行的是增删改操作一般用Exec会好一些。Exec返回的结果是Result,Result接口允许获取执行结果的元数据:

    typeResultinterface{
    //用于返回自增ID,并不是所有的关系型数据库都有这个功能。
    LastInsertId()(int64,error)
    //返回受影响的行数。
    RowsAffected()(int64,error)
    }
    

    准备查询

    如果你现在想使用占位符的功能,where 的条件想以参数的形式传入,Go提供了db.Prepare语句来帮你绑定。准备查询的结果是一个准备好的语句(prepared statement),语句中可以包含执行时所需参数的占位符(即绑定值)。准备查询比拼字符串的方式好很多,它可以转义参数,避免SQL注入。同时,准备查询对于一些数据库也省去了解析和生成执行计划的开销,有利于性能。

    占位符

    PostgreSQL使用$N作为占位符,N是一个从1开始递增的整数,代表参数的位置,方便参数的重复使用。MySQL使用?作为占位符,SQLite两种占位符都可以,而Oracle则使用:param1的形式。

    MySQLPostgreSQLOracle
    =====================
    WHEREcol=?WHEREcol=$1WHEREcol=:col
    VALUES(?,?,?)VALUES($1,$2,$3)VALUES(:val1,:val2,:val3)
    stmt,e:=DB.Prepare("select*fromuserwhereid=?")
    query,e:=stmt.Query(1)
    query.Scan()
    

    事务的使用

    通过db.Begin()来开启一个事务,Begin方法会返回一个事务对象Tx。在结果变量Tx上调用Commit()或者Rollback()方法会提交或回滚变更,并关闭事务。在底层,Tx会从连接池中获得一个连接并在事务过程中保持对它的独占。事务对象Tx上的方法与数据库对象sql.DB的方法一一对应,例如Query,Exec等。事务对象也可以准备(prepare)查询,由事务创建的准备语句会显式绑定到创建它的事务。

    //开启事务
    tx,err:=DB.Begin()
    iferr!=nil{
    fmt.Println("txfail")
    }
    //准备sql语句
    stmt,err:=tx.Prepare("DELETEFROMuserWHEREid=?")
    iferr!=nil{
    fmt.Println("Preparefail")
    returnfalse
    }
    //设置参数以及执行sql语句
    res,err:=stmt.Exec(user.id)
    iferr!=nil{
    fmt.Println("Execfail")
    returnfalse
    }
    //提交事务
    tx.Commit()
    
    我们来一个完整的sql操作:packagemain
    
    import(
    "database/sql"
    "encoding/json"
    "fmt"
    _"github.com/go-sql-driver/mysql"
    "github.com/pkg/errors"
    "strings"
    )
    
    //数据库配置
    const(
    userName="root"
    password="123456"
    ip="127.0.0.1"
    port="3306"
    dbName="test"
    )
    
    //Db数据库连接池
    varDB*sql.DB
    
    typeUserstruct{
    idint64
    namestring
    ageint8
    sexint8
    phonestring
    }
    
    //注意方法名大写,就是public
    funcInitDB(){
    //构建连接:"用户名:密码@tcp(IP:端口)/数据库?charset=utf8"
    path:=strings.Join([]string{userName,":",password,"@tcp(",ip,":",port,")/",dbName,"?charset=utf8"},"")
    //打开数据库,前者是驱动名,所以要导入:_"github.com/go-sql-driver/mysql"
    DB,_=sql.Open("mysql",path)
    //设置数据库最大连接数
    DB.SetConnMaxLifetime(100)
    //设置上数据库最大闲置连接数
    DB.SetMaxIdleConns(10)
    //验证连接
    iferr:=DB.Ping();err!=nil{
    fmt.Println("opendatabasefail")
    return
    }
    fmt.Println("connnectsuccess")
    }
    
    //查询操作
    funcQuery(){
    varuserUser
    rows,e:=DB.Query("select*fromuserwhereidin(1,2,3)")
    ife==nil{
    errors.New("queryincurerror")
    }
    forrows.Next(){
    e:=rows.Scan(user.sex,user.phone,user.name,user.id,user.age)
    ife!=nil{
    fmt.Println(json.Marshal(user))
    }
    }
    rows.Close()
    DB.QueryRow("select*fromuserwhereid=1").Scan(user.age,user.id,user.name,user.phone,user.sex)
    
    stmt,e:=DB.Prepare("select*fromuserwhereid=?")
    query,e:=stmt.Query(1)
    query.Scan()
    }
    
    funcDeleteUser(userUser)bool{
    //开启事务
    tx,err:=DB.Begin()
    iferr!=nil{
    fmt.Println("txfail")
    }
    //准备sql语句
    stmt,err:=tx.Prepare("DELETEFROMuserWHEREid=?")
    iferr!=nil{
    fmt.Println("Preparefail")
    returnfalse
    }
    //设置参数以及执行sql语句
    res,err:=stmt.Exec(user.id)
    iferr!=nil{
    fmt.Println("Execfail")
    returnfalse
    }
    //提交事务
    tx.Commit()
    //获得上一个insert的id
    fmt.Println(res.LastInsertId())
    returntrue
    }
    
    funcInsertUser(userUser)bool{
    //开启事务
    tx,err:=DB.Begin()
    iferr!=nil{
    fmt.Println("txfail")
    returnfalse
    }
    //准备sql语句
    stmt,err:=tx.Prepare("INSERTINTOuser(`name`,`phone`)VALUES(?,?)")
    iferr!=nil{
    fmt.Println("Preparefail")
    returnfalse
    }
    //将参数传递到sql语句中并且执行
    res,err:=stmt.Exec(user.name,user.phone)
    iferr!=nil{
    fmt.Println("Execfail")
    returnfalse
    }
    //将事务提交
    tx.Commit()
    //获得上一个插入自增的id
    fmt.Println(res.LastInsertId())
    returntrue
    }
    
    funcmain(){
    InitDB()
    Query()
    deferDB.Close()
    }
    

    参考

    https://www.cnblogs.com/rickiyang/p/11074180.html

  •  标签:  
  • MySQL
  •  

原文链接:https://77isp.com/post/10221.html

=========================================

https://77isp.com/ 为 “云服务器技术网” 唯一官方服务平台,请勿相信其他任何渠道。