diff --git a/cache/redis.go b/cache/redis.go index c79d291..850c23d 100644 --- a/cache/redis.go +++ b/cache/redis.go @@ -8,6 +8,7 @@ import ( "net/http" "os" "path/filepath" + "time" "github.com/redis/go-redis/v9" ) @@ -74,6 +75,10 @@ func (c *Redis) Upgrade(key string) { if c.Nil() { return } + res, _ := c.db.SetNX(context.Background(), key, "set", time.Minute).Result() + if !res { + return + } resp, err := http.Get("https://" + key) if err != nil { log.Println("get err ", err) diff --git a/router/default.go b/router/default.go index a726efd..2f4da30 100644 --- a/router/default.go +++ b/router/default.go @@ -55,7 +55,6 @@ func NewRouter(c *cache.Redis) *mux.Router { path := fmt.Sprintf("github.com/%s/%s/releases/download/%s/%s", userBaisc[0], userBaisc[1], vars["version"], vars["file"]) c.Incr(path) if c.Exists(path) { - fmt.Println("this way ") r.URL.Path = path http.FileServer(http.Dir("./cache")).ServeHTTP(w, r) v, ok := r.Header[http.CanonicalHeaderKey("X-Real-IP")] diff --git a/service/log.go b/service/log.go index 42f9889..1b4b295 100644 --- a/service/log.go +++ b/service/log.go @@ -128,7 +128,7 @@ func DefaultLogFormatter(param LogFormatterParams) { resetColor = param.ResetColor() } - fmt.Printf("[Proxy] %s | %d %s %s| %13v | %15s |%s %-7s %s %#v %s\n", + fmt.Printf("[Proxy]|%s %d %s %s| %13v | %15s |%s %-7s %s %#v %s\n", statusColor, param.StatusCode, resetColor, time.Now().Format("01-02 15:04:05"), param.ContentLength, param.ClientIP,