Merge pull request #37 from elecbug/elecbug/master

feat: update cache clear part

Author: elecbug

graph/algorithm/cache.go

@@ -11,21 +11,32 @@ var cachedAllShortestPaths = make(map[string]graph.Paths)
 var cachedAllShortestPathLengths = make(map[string]graph.PathLength)
 var cacheMu sync.RWMutex
 
-// CacheClear clears the cached shortest paths and their lengths.
-func CacheClear() {
+// ClearCache clears the cached shortest paths and their lengths.
+func ClearCache() {
 	cacheMu.Lock()
 	defer cacheMu.Unlock()
 	cachedAllShortestPaths = make(map[string]graph.Paths)
 	cachedAllShortestPathLengths = make(map[string]graph.PathLength)
 }
 
-// AutoCacheClear starts a goroutine that clears the cache at regular intervals defined by tick.
-func AutoCacheClear(tick time.Duration) {
+// ClearCacheForGraph clears the cache for a specific graph.
+func ClearCacheForGraph(g *graph.Graph) {
+	cacheMu.Lock()
+	defer cacheMu.Unlock()
+
+	gh := g.Hash()
+
+	delete(cachedAllShortestPaths, gh)
+	delete(cachedAllShortestPathLengths, gh)
+}
+
+// AutoClearCache starts a goroutine that clears the cache at regular intervals defined by tick.
+func AutoClearCache(tick time.Duration) {
 	go func() {
 		for {
 			time.Sleep(tick)
 
-			CacheClear()
+			ClearCache()
 		}
 	}()
 }