summaryrefslogtreecommitdiff
path: root/pkg/cache
diff options
context:
space:
mode:
authorIshita Sequeira <46771830+ishitasequeira@users.noreply.github.com>2024-12-19 19:51:50 +0530
committerGitHub <noreply@github.com>2024-12-19 09:21:50 -0500
commit736f12f55a73d17dbedc040b4f559a2cd33bbff4 (patch)
treed7e31aba857641a20da7bffa5d9c29e3c5c230e4 /pkg/cache
parent03203db49b047f03a8e99b26ef8554942c2513d6 (diff)
Update references of cache, env, health to use modules from registry-scanner (#972)
Signed-off-by: Ishita Sequeira <ishiseq29@gmail.com>
Diffstat (limited to 'pkg/cache')
-rw-r--r--pkg/cache/cache.go20
-rw-r--r--pkg/cache/memcache.go74
-rw-r--r--pkg/cache/memcache_test.go70
3 files changed, 0 insertions, 164 deletions
diff --git a/pkg/cache/cache.go b/pkg/cache/cache.go
deleted file mode 100644
index e59400c..0000000
--- a/pkg/cache/cache.go
+++ /dev/null
@@ -1,20 +0,0 @@
-package cache
-
-import (
- "github.com/argoproj-labs/argocd-image-updater/registry-scanner/pkg/tag"
-)
-
-type ImageTagCache interface {
- HasTag(imageName string, imageTag string) bool
- GetTag(imageName string, imageTag string) (*tag.ImageTag, error)
- SetTag(imageName string, imgTag *tag.ImageTag)
- ClearCache()
- NumEntries() int
-}
-
-// KnownImage represents a known image and the applications using it, without
-// any version/tag information.
-type KnownImage struct {
- ImageName string
- Applications []string
-}
diff --git a/pkg/cache/memcache.go b/pkg/cache/memcache.go
deleted file mode 100644
index 03b4080..0000000
--- a/pkg/cache/memcache.go
+++ /dev/null
@@ -1,74 +0,0 @@
-package cache
-
-import (
- "fmt"
-
- "github.com/argoproj-labs/argocd-image-updater/registry-scanner/pkg/tag"
-
- memcache "github.com/patrickmn/go-cache"
-)
-
-type MemCache struct {
- cache *memcache.Cache
-}
-
-// NewMemCache returns a new instance of MemCache
-func NewMemCache() ImageTagCache {
- mc := MemCache{}
- c := memcache.New(0, 0)
- mc.cache = c
- return &mc
-}
-
-// HasTag returns true if cache has entry for given tag, false if not
-func (mc *MemCache) HasTag(imageName string, tagName string) bool {
- tag, err := mc.GetTag(imageName, tagName)
- if err != nil || tag == nil {
- return false
- } else {
- return true
- }
-}
-
-// SetTag sets a tag entry into the cache
-func (mc *MemCache) SetTag(imageName string, imgTag *tag.ImageTag) {
- mc.cache.Set(tagCacheKey(imageName, imgTag.TagName), *imgTag, -1)
-}
-
-// GetTag gets a tag entry from the cache
-func (mc *MemCache) GetTag(imageName string, tagName string) (*tag.ImageTag, error) {
- var imgTag tag.ImageTag
- e, ok := mc.cache.Get(tagCacheKey(imageName, tagName))
- if !ok {
- return nil, nil
- }
- imgTag, ok = e.(tag.ImageTag)
- if !ok {
- return nil, fmt.Errorf("")
- }
- return &imgTag, nil
-}
-
-func (mc *MemCache) SetImage(imageName, application string) {
- mc.cache.Set(imageCacheKey(imageName), application, -1)
-}
-
-// ClearCache clears the cache
-func (mc *MemCache) ClearCache() {
- for k := range mc.cache.Items() {
- mc.cache.Delete(k)
- }
-}
-
-// NumEntries returns the number of entries in the cache
-func (mc *MemCache) NumEntries() int {
- return mc.cache.ItemCount()
-}
-
-func tagCacheKey(imageName, imageTag string) string {
- return fmt.Sprintf("tags:%s:%s", imageName, imageTag)
-}
-
-func imageCacheKey(imageName string) string {
- return fmt.Sprintf("image:%s", imageName)
-}
diff --git a/pkg/cache/memcache_test.go b/pkg/cache/memcache_test.go
deleted file mode 100644
index ff8bc5f..0000000
--- a/pkg/cache/memcache_test.go
+++ /dev/null
@@ -1,70 +0,0 @@
-package cache
-
-import (
- "testing"
- "time"
-
- memcache "github.com/patrickmn/go-cache"
- "github.com/stretchr/testify/assert"
- "github.com/stretchr/testify/require"
-
- "github.com/argoproj-labs/argocd-image-updater/registry-scanner/pkg/tag"
-)
-
-func Test_MemCache(t *testing.T) {
- imageName := "foo/bar"
- imageTag := "v1.0.0"
- t.Run("Cache hit", func(t *testing.T) {
- mc := NewMemCache()
- newTag := tag.NewImageTag(imageTag, time.Unix(0, 0), "")
- mc.SetTag(imageName, newTag)
- cachedTag, err := mc.GetTag(imageName, imageTag)
- require.NoError(t, err)
- require.NotNil(t, cachedTag)
- assert.Equal(t, imageTag, cachedTag.TagName)
- assert.True(t, mc.HasTag(imageName, imageTag))
- assert.Equal(t, 1, mc.NumEntries())
- })
-
- t.Run("Cache miss", func(t *testing.T) {
- mc := NewMemCache()
- newTag := tag.NewImageTag(imageTag, time.Unix(0, 0), "")
- mc.SetTag(imageName, newTag)
- assert.Equal(t, 1, mc.NumEntries())
- cachedTag, err := mc.GetTag(imageName, "v1.0.1")
- require.NoError(t, err)
- require.Nil(t, cachedTag)
- assert.False(t, mc.HasTag(imageName, "v1.0.1"))
- })
-
- t.Run("Cache clear", func(t *testing.T) {
- mc := NewMemCache()
- newTag := tag.NewImageTag(imageTag, time.Unix(0, 0), "")
- mc.SetTag(imageName, newTag)
- cachedTag, err := mc.GetTag(imageName, imageTag)
- require.NoError(t, err)
- require.NotNil(t, cachedTag)
- assert.Equal(t, imageTag, cachedTag.TagName)
- assert.True(t, mc.HasTag(imageName, imageTag))
- assert.Equal(t, 1, mc.NumEntries())
- mc.ClearCache()
- assert.Equal(t, 0, mc.NumEntries())
- cachedTag, err = mc.GetTag(imageName, imageTag)
- require.NoError(t, err)
- require.Nil(t, cachedTag)
- })
- t.Run("Image Cache Key", func(t *testing.T) {
- mc := MemCache{
- cache: memcache.New(0, 0),
- }
- application := "application1"
- key := imageCacheKey(imageName)
- mc.SetImage(imageName, application)
- app, b := mc.cache.Get(key)
- assert.True(t, b)
- assert.Equal(t, application, app)
- assert.Equal(t, 1, mc.NumEntries())
- mc.ClearCache()
- assert.Equal(t, 0, mc.NumEntries())
- })
-}