update: add singleflight into query

asjdf · Feb 24, 2023 · 43552a6 · 43552a6
1 parent c22e20a
commit 43552a6
Show file tree

Hide file tree

Showing 6 changed files with 356 additions and 250 deletions.
diff --git a/cache/afterQuery.go b/cache/afterQuery.go
diff --git a/cache/beforeQuery.go b/cache/beforeQuery.go
diff --git a/cache/cache.go b/cache/cache.go
@@ -61,12 +61,7 @@ func (c *Gorm2Cache) Initialize(db *gorm.DB) (err error) {
 		return err
 	}
 
-	err = db.Callback().Query().Before("gorm:query").Register("gorm:cache:before_query", BeforeQuery(c))
-	if err != nil {
-		return err
-	}
-
-	err = db.Callback().Query().After("gorm:query").Register("gorm:cache:after_query", AfterQuery(c))
+	err = newQueryHandler(c).Bind(db)
 	if err != nil {
 		return err
 	}