运维开发网

将gorm集成到go-zero的实现方法

运维开发网 https://www.qedev.com 2022-06-27 19:14 出处:网络
go-zero提供的代码生成器里面,没有提供orm框架操作,但是提供了遍历的缓存操作,所以可以利用gorm当作一个sql语句的生成器,把生成后的sql语句放到go-zero生成的模板中去执行,对gor

go-zero提供的代码生成器里面,没有提供orm框架操作,但是提供了遍历的缓存操作,所以可以利用gorm当作一个sql语句的生成器,把生成后的sql语句放到go-zero生成的模板中去执行,对gor

在go-zero提供的代码生成器中,没有提供orm框架操作,但是提供了遍历缓存操作。但是gorm框架没有好的缓存插件。虽然有gcache,但是不支持Gorm版。

所以我要把这两个结合起来。gorm官方文档中提到了一个接口,可以获取生成的sql语句。

所以gorm可以作为sql语句的生成器,生成的sql语句可以放入go-zero生成的模板中执行。

gorm中的Sql生成器

stmt := DB.Session(amp;Session{DryRun: true}).First(amp;user, 1).Statementstmt.SQL.String() //=gt; SELECT * FROM `users` WHERE `id` = $1 ORDER BY `id`stmt.Vars //=gt; []interface{}{1}

集成到go-zero的效果如下
调用Generalsql,用gorm生成sql,然后执行。

func (m *defaultJojUserModel) FindOne(id int64) (*JojUser, error) {jojUsercenterJojUserIdKey := fmt.Sprintf("%s%v", cacheJojUsercenterJojUserIdPrefix, id)var resp JojUsererr := m.QueryRow(amp;resp, jojUsercenterJojUserIdKey, func(conn sqlx.SqlConn, v interface{}) error {query, values := m.GeneralSQL(func(tx *gorm.DB) *gorm.DB {return tx.First(amp;resp, id)})return conn.QueryRow(v, query, values...)})switch err {case nil:return amp;resp, nilcase sqlc.ErrNotFound:return nil, ErrNotFounddefault:return nil, err}}func (m *defaultJojUserModel) GeneralSQL(queryFn func(tx *gorm.DB) *gorm.DB) (string, []interface{}) {tx := queryFn(m.gormDB.Session(amp;gorm.Session{DryRun: true}))stmt := tx.Statementreturn 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"` // 父评论IdChildComment []ArticleComment `json:"childComment"` //以该评论未父节点的所有评论CreateTime string `json:"createTime"` // 回复时间}func main() {begin := time.Now()for i := 0; i lt; 100000; i++ {GeneralSQL(func(tx *gorm.DB) *gorm.DB {return tx.Find(amp;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(amp;gorm.Session{DryRun: true}))stmt := tx.Statementreturn stmt.SQL.String(), stmt.Vars

关于将gorm集成到go-zero的方法的这篇文章到此为止。关于将gorm集成到go-zero的更多信息

0

精彩评论

暂无评论...
验证码 换一张
取 消