reintroduce infohash cache

nbdy · Nov 20, 2023 · 4ec9796 · 4ec9796
1 parent 2370cea
commit 4ec9796
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/cache/cache.go b/cache/cache.go
@@ -12,7 +12,7 @@ import (
 
 var InfoHashCache = mapset.NewSet[[20]byte]()
 
-func PopulateInfoHashCacheFromDatabase(database clover.DB) {
+func PopulateInfoHashCacheFromDatabase(database *clover.DB) {
 	all, _ := database.FindAll(query.NewQuery(db.TorrentTable))
 	for _, d := range all {
 		ih := d.Get("InfoHash").(string)

diff --git a/main.go b/main.go
@@ -80,6 +80,8 @@ func main() {
 	cfg := config.ParseArguments()
 	database := db.OpenDatabase(cfg)
 
+	cache.PopulateInfoHashCacheFromDatabase(database)
+
 	db.AddToBlacklist(database, ReadFileLines(cfg.NameBlacklist), "0")
 	db.AddToBlacklist(database, ReadFileLines(cfg.FileBlacklist), "1")