Add data setter to cache package

Author: nmorsman

cache/cache.go

@@ -46,6 +46,13 @@ type GetOpts[T any] struct {
 	Generator func() (T, error)
 }
 
+type SetOpts[T any] struct {
+	Key   string
+	TTL   int64
+	Grace int64
+	Data  T
+}
+
 var DefaultOpts = &Opts{
 	DefaultTTL:   time.Minute,
 	DefaultGrace: 0,
@@ -265,3 +272,45 @@ func (c *Cache[T]) Get(key string, generator func() (T, error)) (T, error) {
 		Generator: generator,
 	})
 }
+
+//
+// Cache setter with opts
+//
+
+func (c *Cache[T]) SetWithOpts(opts *SetOpts[T]) {
+	getOpts := &GetOpts[T]{
+		Key:   opts.Key,
+		TTL:   opts.TTL,
+		Grace: opts.Grace,
+		Generator: func() (T, error) {
+			return opts.Data, nil
+		},
+	}
+
+	c.mu.RLock()
+	item, exists := c.items[opts.Key]
+	c.mu.RUnlock()
+
+	// Update data if container exists, otherwise create
+	if exists {
+		c.updateCacheItem(getOpts)
+	} else {
+		item = c.createCacheItem(getOpts)
+	}
+
+	// Wait for data to be generated
+	<-item.ready.signal
+}
+
+//
+// Cache setter with default opts
+//
+
+func (c *Cache[T]) Set(key string, data T) {
+	c.SetWithOpts(&SetOpts[T]{
+		Key:   key,
+		Data:  data,
+		TTL:   c.defaultTTL,
+		Grace: c.defaultGrace,
+	})
+}

cache/cache_test.go

@@ -131,6 +131,61 @@ func TestCacheGrace(t *testing.T) {
 	assert.Equal(t, d1, d2)
 }
 
+func TestCacheSet(t *testing.T) {
+	cache := New[int64]()
+	cache.Set("test", 42)
+
+	data, err := cache.Get("test", func() (int64, error) {
+		return 123, nil
+	})
+
+	assert.NoError(t, err)
+	assert.Equal(t, int64(42), data)
+}
+
+func TestCacheSetUpdate(t *testing.T) {
+	cache := New[int64]()
+	generator := func() (int64, error) {
+		return cache.Get("test", func() (int64, error) {
+			return 123, nil
+		})
+	}
+
+	d1, e1 := generator()
+	assert.NoError(t, e1)
+	assert.Equal(t, int64(123), d1)
+
+	cache.Set("test", 42)
+
+	d2, e2 := generator()
+	assert.NoError(t, e2)
+	assert.Equal(t, int64(42), d2)
+}
+
+func TestCacheSetWithOpts(t *testing.T) {
+	cache := New[int64]()
+	generator := func() (int64, error) {
+		return cache.Get("test", func() (int64, error) {
+			return 123, nil
+		})
+	}
+
+	d1, e1 := generator()
+	assert.NoError(t, e1)
+	assert.Equal(t, int64(123), d1)
+
+	cache.SetWithOpts(&SetOpts[int64]{
+		Key:   "test",
+		Data:  42,
+		TTL:   time.Minute.Nanoseconds(),
+		Grace: time.Minute.Nanoseconds(),
+	})
+
+	d2, e2 := generator()
+	assert.NoError(t, e2)
+	assert.Equal(t, int64(42), d2)
+}
+
 func TestCachePurgeExpired(t *testing.T) {
 	cache := NewWithOpts[int64](&Opts{
 		DefaultTTL: 0,