fix: fix key of singleflight not contains vars

README.md

@@ -6,6 +6,7 @@
 - 即插即用
 - 旁路缓存
 - 穿透防护
+- 击穿防护
 - 多存储介质（内存/redis）
 
 ## 使用说明

cache/query.go

@@ -57,17 +57,17 @@ func (h *queryHandler) BeforeQuery() func(db *gorm.DB) {
 		db.InstanceSet("gorm:cache:vars", db.Statement.Vars)
 
 		if util.ShouldCache(tableName, cache.Config.Tables) {
-			hitted := false
+			hit := false
 			defer func() {
-				if hitted {
+				if hit {
 					cache.IncrHitCount()
 				} else {
 					cache.IncrMissCount()
 				}
 			}()
 
 			// singleFlight Check
-			singleFlightKey := fmt.Sprintf("%s:%s", tableName, sql) // todo: key with vars
+			singleFlightKey := util.GenSingleFlightKey(tableName, sql, db.Statement.Vars...)
 			h.singleFlight.mu.Lock()
 			if h.singleFlight.m == nil {
 				h.singleFlight.m = make(map[string]*call)
@@ -88,7 +88,7 @@ func (h *queryHandler) BeforeQuery() func(db *gorm.DB) {
 					_ = db.AddError(err)
 					return
 				}
-				hitted = true
+				hit = true
 				db.RowsAffected = c.rowsAffected
 				db.Error = multierror.Append(util.SingleFlightHit) // 为保证后续流程不走，必须设一个error
 				if c.err != nil {
@@ -190,13 +190,13 @@ func (h *queryHandler) BeforeQuery() func(db *gorm.DB) {
 
 			if cache.Config.CacheLevel == config.CacheLevelAll || cache.Config.CacheLevel == config.CacheLevelOnlyPrimary {
 				if tryPrimaryCache() {
-					hitted = true
+					hit = true
 					return
 				}
 			}
 			if cache.Config.CacheLevel == config.CacheLevelAll || cache.Config.CacheLevel == config.CacheLevelOnlySearch {
-				if !hitted && trySearchCache() {
-					hitted = true
+				if !hit && trySearchCache() {
+					hit = true
 				}
 			}
 		}

util/key.go

@@ -44,3 +44,17 @@ func GenSearchCacheKey(instanceId string, tableName string, sql string, vars ...
 func GenSearchCachePrefix(instanceId string, tableName string) string {
 	return GormCachePrefix + ":" + instanceId + ":s:" + tableName
 }
+
+func GenSingleFlightKey(tableName string, sql string, vars ...interface{}) string {
+	buf := strings.Builder{}
+	buf.WriteString(sql)
+	for _, v := range vars {
+		pv := reflect.ValueOf(v)
+		if pv.Kind() == reflect.Ptr {
+			buf.WriteString(fmt.Sprintf(":%v", pv.Elem()))
+		} else {
+			buf.WriteString(fmt.Sprintf(":%v", v))
+		}
+	}
+	return fmt.Sprintf("%s:%s", tableName, buf.String())
+}