add redis store

pkg/store/redis_store.go

@@ -0,0 +1,151 @@
+package store
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"os"
+	"strings"
+	"time"
+
+	"github.com/redis/go-redis/v9"
+)
+
+type RedisStore struct {
+	prefix         string
+	repoName       string
+	redisClient    RedisClient
+	stackDelimiter *string
+}
+
+type RedisStoreOptions struct {
+	Prefix         *string `mapstructure:"prefix"`
+	StackDelimiter *string `mapstructure:"stack_delimiter"`
+	URL            *string `mapstructure:"url"`
+}
+
+// RedisClient interface allows us to mock the Redis Client in test with only the methods we are using in the
+// RedisStore.
+type RedisClient interface {
+	Get(ctx context.Context, key string) *redis.StringCmd
+	Set(ctx context.Context, key string, value interface{}, expiration time.Duration) *redis.StatusCmd
+}
+
+// Ensure RedisStore implements the store.Store interface.
+var _ Store = (*RedisStore)(nil)
+
+func getRedisOptions(options *RedisStoreOptions) (*redis.Options, error) {
+	if options.URL != nil {
+		opts, err := redis.ParseURL(*options.URL)
+		if err != nil {
+			return &redis.Options{}, fmt.Errorf("failed to parse redis url: %v", err)
+		}
+
+		return opts, nil
+	}
+
+	if os.Getenv("ATMOS_REDIS_URL") != "" {
+		return redis.ParseURL(os.Getenv("ATMOS_REDIS_URL"))
+	}
+
+	return &redis.Options{}, fmt.Errorf("either url must be set in options or REDIS_URL environment variable must be set")
+}
+
+func NewRedisStore(options RedisStoreOptions) (Store, error) {
+	prefix := ""
+	if options.Prefix != nil {
+		prefix = *options.Prefix
+	}
+
+	stackDelimiter := "/"
+	if options.StackDelimiter != nil {
+		stackDelimiter = *options.StackDelimiter
+	}
+
+	opts, err := getRedisOptions(&options)
+	if err != nil {
+		return nil, fmt.Errorf("failed to parse redis url: %v", err)
+	}
+
+	redisClient := redis.NewClient(opts)
+
+	return &RedisStore{
+		prefix:         prefix,
+		redisClient:    redisClient,
+		stackDelimiter: &stackDelimiter,
+	}, nil
+}
+
+func (s *RedisStore) getKey(stack string, component string, key string) (string, error) {
+	if s.stackDelimiter == nil {
+		return "", fmt.Errorf("stack delimiter is not set")
+	}
+
+	prefixParts := []string{s.repoName, s.prefix}
+	prefix := strings.Join(prefixParts, "/")
+
+	return getKey(prefix, *s.stackDelimiter, stack, component, key, "/")
+}
+
+func (s *RedisStore) Get(stack string, component string, key string) (interface{}, error) {
+	if stack == "" {
+		return nil, fmt.Errorf("stack cannot be empty")
+	}
+
+	if component == "" {
+		return nil, fmt.Errorf("component cannot be empty")
+	}
+
+	if key == "" {
+		return nil, fmt.Errorf("key cannot be empty")
+	}
+
+	paramName, err := s.getKey(stack, component, key)
+	if err != nil {
+		return nil, fmt.Errorf("failed to get key: %v", err)
+	}
+
+	ctx := context.Background()
+	jsonData, err := s.redisClient.Get(ctx, paramName).Result()
+	if err != nil {
+		return nil, fmt.Errorf("failed to get key: %v", err)
+	}
+
+	var result interface{}
+	err = json.Unmarshal([]byte(jsonData), &result)
+	if err != nil {
+		return nil, fmt.Errorf("failed to unmarshal file: %v", err)
+	}
+
+	return result, nil
+}
+
+func (s *RedisStore) Set(stack string, component string, key string, value interface{}) error {
+	if stack == "" {
+		return fmt.Errorf("stack cannot be empty")
+	}
+
+	if component == "" {
+		return fmt.Errorf("component cannot be empty")
+	}
+
+	if key == "" {
+		return fmt.Errorf("key cannot be empty")
+	}
+
+	// Construct the full parameter name using getKey
+	paramName, err := s.getKey(stack, component, key)
+	if err != nil {
+		return fmt.Errorf("failed to get key: %v", err)
+	}
+
+	jsonData, err := json.Marshal(value)
+	if err != nil {
+		return fmt.Errorf("failed to marshal value: %v", err)
+	}
+
+	ctx := context.Background()
+	err = s.redisClient.Set(ctx, paramName, jsonData, 0).Err()
+
+	return err
+}