Add AlwaysUseCachedResponse option func

Author: bep

httpcache.go

@@ -101,6 +101,10 @@ type Transport struct {
 	// An empty string signals that this request should not be cached.
 	CacheKey func(req *http.Request) string
 
+	// AlwaysUseCachedResponse is an optional func that when it returns true
+	// a successful response from the cache will be returned without connecting to the server.
+	AlwaysUseCachedResponse func(req *http.Request, key string) bool
+
 	// Around is an optional func.
 	// If set, the Transport will call Around at the start of RoundTrip
 	// and defer the returned func until the end of RoundTrip.
@@ -141,6 +145,9 @@ func (t *Transport) RoundTrip(req *http.Request) (resp *http.Response, err error
 	var cachedResp *http.Response
 	if cacheable {
 		cachedResp, err = t.cachedResponse(req)
+		if err == nil && cachedResp != nil && t.AlwaysUseCachedResponse != nil && t.AlwaysUseCachedResponse(req, cacheKey) {
+			return cachedResp, nil
+		}
 	} else {
 		// Need to invalidate an existing value
 		t.Cache.Delete(cacheKey)

httpcache_test.go

@@ -248,6 +248,27 @@ func TestEnableETagPair(t *testing.T) {
 	}
 }
 
+func TestAlwaysUseCachedResponse(t *testing.T) {
+	resetTest()
+	c := qt.New(t)
+	s.transport.AlwaysUseCachedResponse = func(req *http.Request, key string) bool {
+		return req.Header.Get("Hello") == "world2"
+	}
+
+	{
+		s, _ := doMethod(t, "GET", "/helloheaderasbody", map[string]string{"Hello": "world1"})
+		c.Assert(s, qt.Equals, "world1")
+	}
+	{
+		s, _ := doMethod(t, "GET", "/helloheaderasbody", map[string]string{"Hello": "world2"})
+		c.Assert(s, qt.Equals, "world1")
+	}
+	{
+		s, _ := doMethod(t, "GET", "/helloheaderasbody", map[string]string{"Hello": "world3"})
+		c.Assert(s, qt.Equals, "world3")
+	}
+}
+
 func TestAround(t *testing.T) {
 	resetTest()
 	c := qt.New(t)