fix: use protobuf instead of gob

darkweak · Aug 14, 2024 · 9301a77 · 9301a77
1 parent d01c621
commit 9301a77
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 12 deletions.
diff --git a/pkg/api/souin.go b/pkg/api/souin.go
@@ -85,10 +85,12 @@ func (s *SouinAPI) BulkDelete(key string, purge bool) {
 					mapping.Mapping[k] = v
 				}
 
-				buf := bytes.NewBuffer([]byte{})
-				if gob.NewEncoder(buf).Encode(mapping) == nil {
-					_ = current.Set(core.MappingKeyPrefix+key, buf.Bytes(), 0)
+				v, e := proto.Marshal(&mapping)
+				if e != nil {
+					fmt.Println("Impossible to re-encode the mapping", core.MappingKeyPrefix+key)
+					current.Delete(core.MappingKeyPrefix + key)
 				}
+				_ = current.Set(core.MappingKeyPrefix+key, v, storageToInfiniteTTLMap[current.Name()])
 			}
 		}
 

diff --git a/plugins/caddy/httpcache_test.go b/plugins/caddy/httpcache_test.go
@@ -1034,15 +1034,7 @@ func TestExpires(t *testing.T) {
 		}
 	}
 
-	cacheChecker(caddyTester, "/expires-only", "Hello, expires-only!", int(expiresValue.Sub(time.Now()).Seconds())-1)
+	cacheChecker(caddyTester, "/expires-only", "Hello, expires-only!", int(time.Until(expiresValue)-1))
 	cacheChecker(caddyTester, "/expires-with-max-age", "Hello, expires-with-max-age!", 59)
 	cacheChecker(caddyTester, "/expires-with-s-maxage", "Hello, expires-with-s-maxage!", 4)
 }
-
-type testCancelHandler struct{}
-
-func (t *testCancelHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
-	fmt.Println("Request")
-	time.Sleep(time.Second)
-	_, _ = w.Write([]byte("Hello, cancel-handler!"))
-}