Reuse the cache for /tree

1 files changed, 31 insertions, 0 deletions

diff --git a/cache.go b/cache.go
new file mode 100644
index 0000000..e745884
--- /dev/null
+++ b/cache.go
@@ -0,0 +1,31 @@
+// SPDX-License-Identifier: AGPL-3.0-only
+// SPDX-FileContributor: Runxi Yu <https://runxiyu.org>
+
+package main
+
+import (
+	"html/template"
+
+	"github.com/dgraph-io/ristretto/v2"
+	"go.lindenii.runxiyu.org/lindenii-common/clog"
+)
+
+type treeReadmeCacheEntry struct {
+	DisplayTree    []displayTreeEntry
+	ReadmeFilename string
+	ReadmeRendered template.HTML
+}
+
+var treeReadmeCache *ristretto.Cache[[]byte, treeReadmeCacheEntry]
+
+func init() {
+	var err error
+	treeReadmeCache, err = ristretto.NewCache(&ristretto.Config[[]byte, treeReadmeCacheEntry]{
+		NumCounters: 1e4,
+		MaxCost:     1 << 30,
+		BufferItems: 64,
+	})
+	if err != nil {
+		clog.Fatal(1, "Error initializing indexPageCache: "+err.Error())
+	}
+}