This commit is contained in:
Yann Hamon 2025-05-11 00:37:03 +02:00
parent c954a22d7d
commit 24266601d8
6 changed files with 43 additions and 22 deletions

4
pkg/cache/cache.go vendored
View file

@ -1,6 +1,6 @@
package cache
type Cache interface {
Get(key string) ([]byte, error)
Set(key string, schema []byte) error
Get(key string) (any, error)
Set(key string, schema any) error
}

View file

@ -10,18 +10,18 @@ import (
// - This cache caches the parsed Schemas
type inMemory struct {
sync.RWMutex
schemas map[string][]byte
schemas map[string]any
}
// New creates a new cache for downloaded schemas
func NewInMemoryCache() Cache {
return &inMemory{
schemas: make(map[string][]byte),
schemas: make(map[string]any),
}
}
// Get retrieves the JSON schema given a resource signature
func (c *inMemory) Get(key string) ([]byte, error) {
func (c *inMemory) Get(key string) (any, error) {
c.RLock()
defer c.RUnlock()
schema, ok := c.schemas[key]
@ -34,7 +34,7 @@ func (c *inMemory) Get(key string) ([]byte, error) {
}
// Set adds a JSON schema to the schema cache
func (c *inMemory) Set(key string, schema []byte) error {
func (c *inMemory) Set(key string, schema any) error {
c.Lock()
defer c.Unlock()
c.schemas[key] = schema

6
pkg/cache/ondisk.go vendored
View file

@ -27,7 +27,7 @@ func cachePath(folder, key string) string {
}
// Get retrieves the JSON schema given a resource signature
func (c *onDisk) Get(key string) ([]byte, error) {
func (c *onDisk) Get(key string) (any, error) {
c.RLock()
defer c.RUnlock()
@ -41,12 +41,12 @@ func (c *onDisk) Get(key string) ([]byte, error) {
}
// Set adds a JSON schema to the schema cache
func (c *onDisk) Set(key string, schema []byte) error {
func (c *onDisk) Set(key string, schema any) error {
c.Lock()
defer c.Unlock()
if _, err := os.Stat(cachePath(c.folder, key)); os.IsNotExist(err) {
return os.WriteFile(cachePath(c.folder, key), schema, 0644)
return os.WriteFile(cachePath(c.folder, key), schema.([]byte), 0644)
}
return nil
}