add CacheDiscovery

smallnest · Oct 31, 2023 · b73eba8 · b73eba8
1 parent 39c51c0
commit b73eba8
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/client/discovery.go b/client/discovery.go
@@ -3,6 +3,7 @@ package client
 import (
 	"encoding/json"
 	"os"
+	"path/filepath"
 	"sync"
 )
 
@@ -39,6 +40,15 @@ func CacheDiscovery(threshold int, cachedFile string, discovery ServiceDiscovery
 		cachedFile = ".cache/discovery.json"
 	}
 
+	cachedFileDir := filepath.Dir(cachedFile)
+
+	if _, err := os.Stat(cachedFileDir); os.IsNotExist(err) {
+		// 目录不存在，创建目录
+		if err := os.MkdirAll(cachedFileDir, os.ModePerm); err != nil {
+			panic(err)
+		}
+	}
+
 	return &cachedServiceDiscovery{
 		threshold:        threshold,
 		cachedFile:       cachedFile,