mock_redis.go

package redis

import (
	"fmt"
	"sync"
)

type MockClient struct {
	lock  sync.RWMutex
	cache map[string]string
}

func NewMockClient() *MockClient {
	return &MockClient{
		lock:  sync.RWMutex{},
		cache: make(map[string]string),
	}
}

func (m *MockClient) Get(key string) (v string, err error) {
	m.lock.RLock()
	val, ok := m.cache[key]
	m.lock.RUnlock()

	if ok == true {
		return val, nil
	}
	return "", NewNotFoundError(fmt.Sprintf("key %s not found", key))
}
func (m *MockClient) Set(key, value string, expiresSeconds uint) error {
	m.lock.Lock()
	m.cache[key] = value
	m.lock.Unlock()

	return nil
}
func (m *MockClient) Delete(keys ...string) error {
	m.lock.Lock()
	for _, key := range keys {
		delete(m.cache, key)
	}
	m.lock.Unlock()
	return nil
}
func (m *MockClient) Close() error { return nil }