gorm整合进go-zero的实现方法

 更新时间:2022年03月17日 11:53:04   作者:jiangxiaoju  
go-zero提供的代码生成器里面,没有提供orm框架操作,但是提供了遍历的缓存操作,所以可以利用gorm当作一个sql语句的生成器,把生成后的sql语句放到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).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内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

相关文章

  • golang与非golang程序探测beyla源码解读

    golang与非golang程序探测beyla源码解读

    这篇文章主要为大家介绍了beyla源码解读之golang与非golang程序的探测实例,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪
    2023-12-12
  • go mod tidy报错解决方法详解

    go mod tidy报错解决方法详解

    这篇文章主要为大家介绍了go mod tidy报错解决方法详解,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪
    2023-12-12
  • Golang使用singleflight解决并发重复请求

    Golang使用singleflight解决并发重复请求

    高并发的场景下,经常会出现并发重复请求资源的情况,singleflight是golang内置的一个包,这个包提供了对重复函数调用的抑制功能,所以下面我们就来看看如何使用它解决并发重复请求吧
    2023-08-08
  • golang如何解决go get命令无响应问题

    golang如何解决go get命令无响应问题

    文章介绍了在Go语言中处理由于官方库被封禁导致依赖下载失败的方法,包括设置代理和直接克隆依赖包到GOPATH/src下
    2024-12-12
  • Go编译32位GNU静态链接库的方法

    Go编译32位GNU静态链接库的方法

    Go链接库系统的难用可谓是人尽皆知,不同Go版本编译出来的不兼容,而且只支持GNU的,不能编译出Windows上的dll和lib。这篇文章给大家介绍Go编译32位GNU静态链接库的方法,感兴趣的朋友一起看看吧
    2020-05-05
  • Golang 并发读写锁的具体实现

    Golang 并发读写锁的具体实现

    Go语言中的sync.RWMutex提供了读写锁机制,允许多个协程并发读取共享资源,但在写操作时保持独占性,本文主要介绍了Golang 并发读写锁的具体实现,感兴趣的可以了解一下
    2025-02-02
  • 解决Go中使用seed得到相同随机数的问题

    解决Go中使用seed得到相同随机数的问题

    这篇文章主要介绍了Go中使用seed得到相同随机数的问题,需要的朋友可以参考下
    2019-10-10
  • Golang中time.After的使用理解与释放问题

    Golang中time.After的使用理解与释放问题

    这篇文章主要给大家介绍了关于Golang中time.After的使用理解与释放问题,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2018-08-08
  • Golang学习笔记(五):函数

    Golang学习笔记(五):函数

    这篇文章主要介绍了Golang学习笔记(五):函数的相关资料,本文讲解了基本语法、多返回值及命名返回参数、参数传递:传值与传指针、参数传递:可变参数、匿名函数、函数作为值、类型等内容,需要的朋友可以参考下
    2015-05-05
  • 使用Go语言实现心跳机制

    使用Go语言实现心跳机制

    心跳最典型的应用场景是是探测服务是否存活,这篇文章主要来和大家介绍一下如何使用Go语言实现一个简单的心跳程序,感兴趣的可以了解下
    2024-01-01

最新评论