From e42b8137a66862613a6def934d9e70ea349f15d3 Mon Sep 17 00:00:00 2001 From: soulteary Date: Wed, 15 Jun 2022 09:29:56 +0800 Subject: [PATCH] chore: avoid hardcode cache directories --- server/internal.go | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/server/internal.go b/server/internal.go index 334e7d5..bded480 100644 --- a/server/internal.go +++ b/server/internal.go @@ -42,6 +42,7 @@ func getInternalResType(url string) int { } const LABEL_NO_VALID_VALUE = "N/A" +const CACHE_META_DIR = "./.aptcache/header/v1" func renderInternalUrls(url string, rw *http.ResponseWriter) { types := getInternalResType(url) @@ -56,8 +57,8 @@ func renderInternalUrls(url string, rw *http.ResponseWriter) { } filesNumberLabel := LABEL_NO_VALID_VALUE - if _, err := os.Stat("./.aptcache/header/v1"); !os.IsNotExist(err) { - files, err := ioutil.ReadDir("./.aptcache/header/v1") + if _, err := os.Stat(CACHE_META_DIR); !os.IsNotExist(err) { + files, err := ioutil.ReadDir(CACHE_META_DIR) if err == nil { filesNumberLabel = strconv.Itoa(len(files)) }