gorm整合进go-zero的实现方法
短信预约 -IT技能 免费直播动态提醒
go-zero提供的代码生成器里面，没有提供orm框架操作，但是提供了遍历的缓存操作。但是gorm框架的话，没有比较好的缓存插件，虽然有一个gcache,但不支持gorm2.0版本。
所以我打算把这两个结合起来。在gorm官方文档中提到了一个接口，可以获取到生成的sql语句。
所以可以利用gorm当作一个sql语句的生成器，把生成后的sql语句放到go-zero生成的模板中去执行。
gorm中的sql生成器
stmt := DB.Session(&Session{DryRun: true}).First(&user, 1).Statement
stmt.SQL.String() //=> SELECT * FROM `users` WHERE `id` = $1 ORDER BY `id`
stmt.Vars //=> []interface{}{1}整合到go-zero中的效果如下
调用GeneralSQL，使用gorm生成sql后，去执行
func (m *defaultJojUserModel) FindOne(id int64) (*JojUser, error) {
jojUsercenterJojUserIdKey := fmt.Sprintf("%s%v", cacheJojUsercenterJojUserIdPrefix, id)
var resp JojUser
err := m.QueryRow(&resp, jojUsercenterJojUserIdKey, func(conn sqlx.SqlConn, v interface{}) error {
query, values := m.GeneralSQL(func(tx *gorm.DB) *gorm.DB {
return tx.First(&resp, id)
})
return conn.QueryRow(v, query, values...)
})
switch err {
case nil:
return &resp, nil
case sqlc.ErrNotFound:
return nil, ErrNotFound
default:
return nil, err
}
}
func (m *defaultJojUserModel) GeneralSQL(queryFn func(tx *gorm.DB) *gorm.DB) (string, []interface{}) {
tx := queryFn(m.gormDB.Session(&gorm.Session{DryRun: true}))
stmt := tx.Statement
return stmt.SQL.String(), stmt.Vars
}
关于效率的问题，做了一个简单的测试，使用gorm生成sql的话，生成1e5次耗时304.1878ms
type ArticleComment struct {
Id int `json:"id"`
Content string `json:"content"`
ReplyComment *ArticleComment `json:"replyToComment"` // 该评论回复的评论
ParentCommentId int `json:"parentCommentId"` // 父评论Id
ChildComment []ArticleComment `json:"childComment"` //以该评论未父节点的所有评论
CreateTime string `json:"createTime"` // 回复时间
}
func main() {
begin := time.Now()
for i := 0; i < 100000; i++ {
GeneralSQL(func(tx *gorm.DB) *gorm.DB {
return tx.Find(&Article{}, []int{1, 2})
})
}
end := time.Now()
fmt.Println(end.Sub(begin))
func GeneralSQL(queryFn func(tx *gorm.DB) *gorm.DB) (string, []interface{}) {
tx := queryFn(db.Session(&gorm.Session{DryRun: true}))
stmt := tx.Statement
return stmt.SQL.String(), stmt.Vars到此这篇关于gorm整合进go-zero的方法的文章就介绍到这了,更多相关gorm整合进go-zero内容请搜索编程网以前的文章或继续浏览下面的相关文章希望大家以后多多支持编程网！
免责声明：
① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的，并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据，供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。
② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341
