代码拉取完成，页面将自动刷新
推荐使用kratos工具快速生成项目，如我们生成一个叫kratos-demo的项目。目录结构如下：
├── CHANGELOG.md
├── OWNERS
├── README.md
├── api
│ ├── api.bm.go
│ ├── api.pb.go
│ ├── api.proto
│ └── client.go
├── cmd
│ ├── cmd
│ └── main.go
├── configs
│ ├── application.toml
│ ├── db.toml
│ ├── grpc.toml
│ ├── http.toml
│ ├── memcache.toml
│ └── redis.toml
├── go.mod
├── go.sum
├── internal
│ ├── dao
│ │ ├── dao.bts.go
│ │ ├── dao.go
│ │ ├── db.go
│ │ ├── mc.cache.go
│ │ ├── mc.go
│ │ └── redis.go
│ ├── di
│ │ ├── app.go
│ │ ├── wire.go
│ │ └── wire_gen.go
│ ├── model
│ │ └── model.go
│ ├── server
│ │ ├── grpc
│ │ │ └── server.go
│ │ └── http
│ │ └── server.go
│ └── service
│ └── service.go
└── test
└── docker-compose.yaml
创建项目成功后，进入项目中的configs目录，mysql.toml，我们可以看到：
[demo]
addr = "127.0.0.1:3306"
dsn = "{user}:{password}@tcp(127.0.0.1:3306)/{database}?timeout=1s&readTimeout=1s&writeTimeout=1s&parseTime=true&loc=Local&charset=utf8mb4,utf8"
readDSN = ["{user}:{password}@tcp(127.0.0.2:3306)/{database}?timeout=1s&readTimeout=1s&writeTimeout=1s&parseTime=true&loc=Local&charset=utf8mb4,utf8","{user}:{password}@tcp(127.0.0.3:3306)/{database}?timeout=1s&readTimeout=1s&writeTimeout=1s&parseTime=true&loc=Local&charset=utf8,utf8mb4"]
active = 20
idle = 10
idleTimeout ="4h"
queryTimeout = "200ms"
execTimeout = "300ms"
tranTimeout = "400ms"
在该配置文件中我们可以配置mysql的读和写的dsn、连接地址addr、连接池的闲置连接数idle、最大连接数active以及各类超时。
如果配置了readDSN，在进行读操作的时候会优先使用readDSN的连接。
进入项目的internal/dao目录，打开db.go，其中：
var cfg struct {
Client *sql.Config
}
checkErr(paladin.Get("db.toml").UnmarshalTOML(&dc))
使用paladin配置管理工具将上文中的db.toml中的配置解析为我们需要使用db的相关配置。
// Dao dao.
type Dao struct {
db *sql.DB
}
在dao的主结构提中定义了mysql的连接池对象。
d = &dao{
db: sql.NewMySQL(dc.Demo),
}
使用kratos/pkg/database/sql包的NewMySQL方法进行连接池对象的初始化，需要传入上文解析的配置。
// Ping ping the resource.
func (d *dao) Ping(ctx context.Context) (err error) {
return d.db.Ping(ctx)
}
生成的dao层模板中自带了mysql相关的ping方法，用于为负载均衡服务的健康监测提供依据，详见blademaster。
// Close close the resource.
func (d *dao) Close() {
d.db.Close()
}
在关闭dao层时，通过调用mysql连接池对象的Close方法，我们可以关闭该连接池，从而释放相关资源。
// GetDemo 用户角色
func (d *dao) GetDemo(c context.Context, did int64) (demo int8, err error) {
err = d.db.QueryRow(c, _getDemoSQL, did).Scan(&demo)
if err != nil && err != sql.ErrNoRows {
log.Error("d.GetDemo.Query error(%v)", err)
return
}
return demo, nil
}
db.QueryRow方法用于返回最多一条记录的查询，在QueryRow方法后使用Scan方法即可将mysql的返回值转换为Golang的数据类型。
当mysql查询不到对应数据时，会返回sql.ErrNoRows，如果不需处理，可以参考如上代码忽略此error。
// ResourceLogs ResourceLogs.
func (d *dao) GetDemos(c context.Context, dids []int64) (demos []int8, err error) {
rows, err := d.db.Query(c, _getDemosSQL, dids)
if err != nil {
log.Error("query error(%v)", err)
return
}
defer rows.Close()
for rows.Next() {
var tmpD int8
if err = rows.Scan(&tmpD); err != nil {
log.Error("scan demo log error(%v)", err)
return
}
demos = append(demos, tmpD)
}
return
}
db.Query方法一般用于批量查询的场景，返回*sql.Rows和error信息。 我们可以使用rows.Next()方法获得下一行的返回结果，并且配合使用rows.Scan()方法将该结果转换为Golang的数据类型。当没有下一行时，rows.Next方法将返回false，此时循环结束。
注意，在使用完毕rows对象后，需要调用rows.Close方法关闭连接，释放相关资源。
// DemoExec exec
func (d *Dao) DemoExec(c context.Context, id int64) (rows int64, err error) {
res, err := d.db.Exec(c, _demoUpdateSQL, id)
if err != nil {
log.Error("db.DemoExec.Exec(%s) error(%v)", _demoUpdateSQL, err)
return
}
return res.RowsAffected()
}
执行UPDATE/DELETE/INSERT语句时，使用db.Exec方法进行语句执行，返回*sql.Result和error信息：
// A Result summarizes an executed SQL command.
type Result interface {
LastInsertId() (int64, error)
RowsAffected() (int64, error)
}
Result接口支持获取影响行数和LastInsertId（一般用于获取Insert语句插入数据库后的主键ID）
kratos/pkg/database/sql包支持事务操作，具体操作示例如下：
开启一个事务：
tx := d.db.Begin()
if err = tx.Error; err != nil {
log.Error("db begin transcation failed, err=%+v", err)
return
}
在事务中执行语句：
res, err := tx.Exec(_demoSQL, did)
if err != nil {
return
}
rows := res.RowsAffected()
提交事务：
if err = tx.Commit().Error; err!=nil{
log.Error("db commit transcation failed, err=%+v", err)
}
回滚事务：
if err = tx.Rollback().Error; err!=nil{
log.Error("db rollback failed, err=%+v", rollbackErr)
}
此处可能存在不合适展示的内容，页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容，可点击提交进行申诉，我们将尽快为您处理。
马建仓 AI 助手
