From cc09287d752ec6d9c4d4876ab41c4e5859b90f78 Mon Sep 17 00:00:00 2001
From: Luke Shumaker <lukeshu@lukeshu.com>
Date: Wed, 6 Jul 2022 03:05:07 -0600
Subject: Move subvol.go to btrfs/io3_fs.go

---
 pkg/util/lru.go | 73 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 73 insertions(+)
 create mode 100644 pkg/util/lru.go

(limited to 'pkg/util/lru.go')

diff --git a/pkg/util/lru.go b/pkg/util/lru.go
new file mode 100644
index 0000000..2b62e69
--- /dev/null
+++ b/pkg/util/lru.go
@@ -0,0 +1,73 @@
+package util
+
+import (
+	"sync"
+
+	lru "github.com/hashicorp/golang-lru"
+)
+
+type LRUCache[K comparable, V any] struct {
+	initOnce sync.Once
+	inner    *lru.ARCCache
+}
+
+func (c *LRUCache[K, V]) init() {
+	c.initOnce.Do(func() {
+		c.inner, _ = lru.NewARC(128)
+	})
+}
+
+func (c *LRUCache[K, V]) Add(key K, value V) {
+	c.init()
+	c.inner.Add(key, value)
+}
+func (c *LRUCache[K, V]) Contains(key K) bool {
+	c.init()
+	return c.inner.Contains(key)
+}
+func (c *LRUCache[K, V]) Get(key K) (value V, ok bool) {
+	c.init()
+	_value, ok := c.inner.Get(key)
+	if ok {
+		value = _value.(V)
+	}
+	return value, ok
+}
+func (c *LRUCache[K, V]) Keys() []K {
+	c.init()
+	untyped := c.inner.Keys()
+	typed := make([]K, len(untyped))
+	for i := range untyped {
+		typed[i] = untyped[i].(K)
+	}
+	return typed
+}
+func (c *LRUCache[K, V]) Len() int {
+	c.init()
+	return c.inner.Len()
+}
+func (c *LRUCache[K, V]) Peek(key K) (value V, ok bool) {
+	c.init()
+	_value, ok := c.inner.Peek(key)
+	if ok {
+		value = _value.(V)
+	}
+	return value, ok
+}
+func (c *LRUCache[K, V]) Purge() {
+	c.init()
+	c.inner.Purge()
+}
+func (c *LRUCache[K, V]) Remove(key K) {
+	c.init()
+	c.inner.Remove(key)
+}
+
+func (c *LRUCache[K, V]) GetOrElse(key K, fn func() V) V {
+	var value V
+	var ok bool
+	for value, ok = c.Get(key); !ok; value, ok = c.Get(key) {
+		c.Add(key, fn())
+	}
+	return value
+}
-- 
cgit v1.2.3-2-g168b