gorm怎么整合进go-zero

本篇内容主要讲解"gorm怎么整合进go-zero"，感兴趣的朋友不妨来看看。本文介绍的方法操作简单快捷，实用性强。下面就让小编来带大家学习"gorm怎么整合进go-zero"吧!

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).Statementstmt.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"有了更深的了解，不妨来实际操作一番吧！这里是网站，更多相关内容可以进入相关频道进行查询，关注我们，继续学习！