Add StopAdjust and StopAll

diskcache.go

@@ -91,6 +91,7 @@
 	mu                 sync.Mutex
 	keyMu              *keyrwmutex.KeyRWMutex
 	adjustMu           sync.Mutex
+	adjustStopCh       chan struct{}
 }
 
 // DiskCacheOption is an option for DiskCache.
@@ -240,7 +241,7 @@
 			c.mu.Lock()
 			defer c.mu.Unlock()
 			c.totalBytes += size
 			c.m.Set(key, &cacheItem{
 				path:  path,
 				bytes: size,
 			}, ttlcache.DefaultTTL)
@@ -253,6 +254,12 @@
 	return c, nil
 }
 
+// StopAll stops all the goroutines of the cache.
+func (c *DiskCache) StopAll() {
+	c.StartAutoCleanup()
+	c.StopAdjust()
+}
+
 // StartAutoCleanup starts the goroutine of automatic cache cleanup
 func (c *DiskCache) StartAutoCleanup() {
 	go c.m.Start()
@@ -263,6 +270,11 @@
 	c.m.Stop()
 }
 
+// StopAdjust
+func (c *DiskCache) StopAdjust() {
+	c.adjustStopCh <- struct{}{}
+}
+
 // DeleteExpired deletes expired caches.
 func (c *DiskCache) DeleteExpired() {
 	c.m.DeleteExpired()
@@ -278,7 +290,7 @@
 func (c *DiskCache) StoreWithTTL(key string, req *http.Request, res *http.Response, ttl time.Duration) error {
 	c.keyMu.LockKey(key)
 	defer func() {
 		_ = c.keyMu.UnlockKey(key)
 	}()
 	p := filepath.Join(c.cacheRoot, KeyToPath(key, c.cacheDirLen))
 	dir := filepath.Dir(p)
@@ -332,7 +344,7 @@
 func (c *DiskCache) Load(key string) (*http.Request, *http.Response, error) {
 	c.keyMu.RLockKey(key)
 	defer func() {
 		_ = c.keyMu.RUnlockKey(key)
 	}()
 	i := c.m.Get(key)
 	if i == nil {
@@ -393,6 +405,11 @@
 			return
 		}
 		c.mu.Unlock()
+		select {
+		case <-c.adjustStopCh:
+			return
+		default:
+		}
 	}
 }
 
@@ -400,7 +417,7 @@
 	defer func() {
 		c.d.remove(ci.key)
 	}()
 	_ = os.Remove(ci.path)
 	c.mu.Lock()
 	defer c.mu.Unlock()
 	if c.totalBytes < ci.bytes {