Merge pull request #29 from vincentbernat/feature/cache-miss

feat: add WithOnCacheMiss option

cache.go

@@ -88,6 +88,7 @@ func cache(
 			if err != persist.ErrCacheMiss {
 				cfg.logger.Errorf("get cache error: %s, cache key: %s", err, cacheKey)
 			}
+			cfg.missCacheCallback(c)
 		}
 
 		// cache miss, then call the backend

cache_test.go

@@ -2,17 +2,18 @@ package cache
 
 import (
 	"fmt"
-	"github.com/stretchr/testify/require"
 	"math/rand"
 	"net/http"
 	"net/http/httptest"
 	"sync"
+	"sync/atomic"
 	"testing"
 	"time"
 
 	"github.com/chenyahui/gin-cache/persist"
 	"github.com/gin-gonic/gin"
 	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
 )
 
 func init() {
@@ -53,6 +54,26 @@ func TestCacheByRequestPath(t *testing.T) {
 	assert.Equal(t, w1.Code, w2.Code)
 }
 
+func TestCacheHitMissCallback(t *testing.T) {
+	var cacheHitCount, cacheMissCount int32
+	memoryStore := persist.NewMemoryStore(1 * time.Minute)
+	cachePathMiddleware := CacheByRequestPath(memoryStore, 3*time.Second,
+		WithOnHitCache(func(c *gin.Context) {
+			atomic.AddInt32(&cacheHitCount, 1)
+		}),
+		WithOnMissCache(func(c *gin.Context) {
+			atomic.AddInt32(&cacheMissCount, 1)
+		}),
+	)
+
+	mockHttpRequest(cachePathMiddleware, "/cache?uid=u1", true)
+	mockHttpRequest(cachePathMiddleware, "/cache?uid=u2", true)
+	mockHttpRequest(cachePathMiddleware, "/cache?uid=u3", true)
+
+	assert.Equal(t, cacheHitCount, int32(2))
+	assert.Equal(t, cacheMissCount, int32(1))
+}
+
 func TestCacheDuration(t *testing.T) {
 	memoryStore := persist.NewMemoryStore(1 * time.Minute)
 	cacheURIMiddleware := CacheByRequestURI(memoryStore, 3*time.Second)

examples/options/main.go

@@ -2,26 +2,30 @@ package main
 
 import (
 	"fmt"
+	"sync/atomic"
+	"time"
+
 	cache "github.com/chenyahui/gin-cache"
 	"github.com/chenyahui/gin-cache/persist"
 	"github.com/gin-gonic/gin"
-	"sync/atomic"
-	"time"
 )
 
 func main() {
 	app := gin.New()
 
 	memoryStore := persist.NewMemoryStore(1 * time.Minute)
 
-	var cacheHitCount int32
+	var cacheHitCount, cacheMissCount int32
 	app.GET("/hello",
 		cache.CacheByRequestURI(
 			memoryStore,
 			2*time.Second,
 			cache.WithOnHitCache(func(c *gin.Context) {
 				atomic.AddInt32(&cacheHitCount, 1)
 			}),
+			cache.WithOnMissCache(func(c *gin.Context) {
+				atomic.AddInt32(&cacheMissCount, 1)
+			}),
 		),
 		func(c *gin.Context) {
 			c.String(200, "hello world")
@@ -31,6 +35,9 @@ func main() {
 	app.GET("/get_hit_count", func(c *gin.Context) {
 		c.String(200, fmt.Sprintf("total hit count: %d", cacheHitCount))
 	})
+	app.GET("/get_miss_count", func(c *gin.Context) {
+		c.String(200, fmt.Sprintf("total miss count: %d", cacheMissCount))
+	})
 
 	if err := app.Run(":8080"); err != nil {
 		panic(err)

option.go

@@ -12,7 +12,8 @@ type Config struct {
 
 	getCacheStrategyByRequest GetCacheStrategyByRequest
 
-	hitCacheCallback OnHitCacheCallback
+	hitCacheCallback  OnHitCacheCallback
+	missCacheCallback OnMissCacheCallback
 
 	beforeReplyWithCacheCallback BeforeReplyWithCacheCallback
 
@@ -28,6 +29,7 @@ func newConfigByOpts(opts ...Option) *Config {
 	cfg := &Config{
 		logger:                       Discard{},
 		hitCacheCallback:             defaultHitCacheCallback,
+		missCacheCallback:            defaultMissCacheCallback,
 		beforeReplyWithCacheCallback: defaultBeforeReplyWithCacheCallback,
 		shareSingleFlightCallback:    defaultShareSingleFlightCallback,
 	}
@@ -87,6 +89,20 @@ func WithOnHitCache(cb OnHitCacheCallback) Option {
 	}
 }
 
+// OnMissCacheCallback define the callback when use cache
+type OnMissCacheCallback func(c *gin.Context)
+
+var defaultMissCacheCallback = func(c *gin.Context) {}
+
+// WithOnMissCache will be called when cache miss.
+func WithOnMissCache(cb OnMissCacheCallback) Option {
+	return func(c *Config) {
+		if cb != nil {
+			c.missCacheCallback = cb
+		}
+	}
+}
+
 type BeforeReplyWithCacheCallback func(c *gin.Context, cache *ResponseCache)
 
 var defaultBeforeReplyWithCacheCallback = func(c *gin.Context, cache *ResponseCache) {}