gorm怎么整合進(jìn)go-zero

本篇內(nèi)容主要講解“gorm怎么整合進(jìn)go-zero”，感興趣的朋友不妨來看看。本文介紹的方法操作簡(jiǎn)單快捷，實(shí)用性強(qiáng)。下面就讓小編來帶大家學(xué)習(xí)“gorm怎么整合進(jìn)go-zero”吧!

go-zero提供的代碼生成器里面，沒有提供orm框架操作，但是提供了遍歷的緩存操作。但是gorm框架的話，沒有比較好的緩存插件，雖然有一個(gè)gcache,但不支持gorm2.0版本。

所以我打算把這兩個(gè)結(jié)合起來。在gorm官方文檔中提到了一個(gè)接口，可以獲取到生成的sql語句。

所以可以利用gorm當(dāng)作一個(gè)sql語句的生成器，把生成后的sql語句放到go-zero生成的模板中去執(zhí)行。

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中的效果如下
調(diào)用GeneralSQL，使用gorm生成sql后，去執(zhí)行

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
}

關(guān)于效率的問題，做了一個(gè)簡(jiǎn)單的測(cè)試，使用gorm生成sql的話，生成1e5次耗時(shí)304.1878ms

type ArticleComment struct {
	Id              int              `json:"id"`
	Content         string           `json:"content"`
	ReplyComment    *ArticleComment  `json:"replyToComment"`  // 該評(píng)論回復(fù)的評(píng)論
	ParentCommentId int              `json:"parentCommentId"` // 父評(píng)論Id
	ChildComment    []ArticleComment `json:"childComment"`    //以該評(píng)論未父節(jié)點(diǎn)的所有評(píng)論
	CreateTime      string           `json:"createTime"`      // 回復(fù)時(shí)間
}

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

到此，相信大家對(duì)“gorm怎么整合進(jìn)go-zero”有了更深的了解，不妨來實(shí)際操作一番吧！這里是億速云網(wǎng)站，更多相關(guān)內(nèi)容可以進(jìn)入相關(guān)頻道進(jìn)行查詢，關(guān)注我們，繼續(xù)學(xué)習(xí)！