fix: set header from cache instead of add (#3)

chenyahui · Sep 14, 2021 · 36a138f · 36a138f
1 parent 3889a49
commit 36a138f
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 6 deletions.
diff --git a/cache.go b/cache.go
@@ -166,9 +166,10 @@ func replyWithCache(
 	respCache *responseCache,
 ) {
 	c.Writer.WriteHeader(respCache.Status)
+
 	for key, values := range respCache.Header {
 		for _, val := range values {
-			c.Writer.Header().Add(key, val)
+			c.Writer.Header().Set(key, val)
 		}
 	}
 

diff --git a/go.mod b/go.mod
@@ -8,5 +8,4 @@ require (
 	github.com/go-redis/redis/v8 v8.10.0
 	github.com/stretchr/testify v1.7.0
 	golang.org/x/sync v0.0.0-20210220032951-036812b2e83c
-	gopkg.in/yaml.v3 v3.0.0-20210107192922-496545a6307b // indirect
 )
diff --git a/go.sum b/go.sum
@@ -156,6 +156,5 @@ gopkg.in/yaml.v2 v2.2.4/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
 gopkg.in/yaml.v2 v2.2.8/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
 gopkg.in/yaml.v2 v2.3.0 h1:clyUAQHOM3G0M3f5vQj7LuJrETvjVot3Z5el9nffUtU=
 gopkg.in/yaml.v2 v2.3.0/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
+gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c h1:dUUwHk2QECo/6vqA44rthZ8ie2QXMNeKRTHCNY2nXvo=
 gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
-gopkg.in/yaml.v3 v3.0.0-20210107192922-496545a6307b h1:h8qDotaEPuJATrMmW04NCwg7v22aHH28wwpauUhK9Oo=
-gopkg.in/yaml.v3 v3.0.0-20210107192922-496545a6307b/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
diff --git a/option.go b/option.go
@@ -33,15 +33,15 @@ func WithCacheStrategyByRequest(getGetCacheStrategyByRequest GetCacheStrategyByR
 
 type OnHitCacheCallback func(c *gin.Context)
 
-func WithOnHitCache(cb OnHitCacheCallback) Option{
+func WithOnHitCache(cb OnHitCacheCallback) Option {
 	return func(c *Config) {
 		if cb != nil {
 			c.hitCacheCallback = cb
 		}
 	}
 }
 
-var defaultHitCacheCallback = func(c *gin.Context){}
+var defaultHitCacheCallback = func(c *gin.Context) {}
 
 type Logger interface {
 	Errorf(string, ...interface{})