add setnx avoid get more than one

Sianao · Dec 1, 2024 · bcd3d42 · bcd3d42
1 parent ee9c15b
commit bcd3d42
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git 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
@@ -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
@@ -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,