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

知识点掌握了，还需要不断练习才能熟练运用。下面golang学习网给大家带来一个Golang开发实战，手把手教大家学习《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 
到这里，我们也就讲完了《gorm整合进go-zero的实现方法》的内容了。个人认为，基础知识的学习和巩固，是为了更好的将其运用到项目中，欢迎关注golang学习网公众号，带你了解更多关于golang的知识点！