From edc1c9e3ed1449f69ea00299e6f888ff962b194d Mon Sep 17 00:00:00 2001
From: thijsheijden <hi@thijsheijden.nl>
Date: Mon, 19 Apr 2021 20:16:22 +0200
Subject: [PATCH] Updated keyvaluedriver and its mock
---
internal/drivers/keyvaluedriver/keyvaluedriver.go | 9 ++++-----
.../drivers/keyvaluedriver/mock/mockkeyvaluedriver.go | 10 +++++-----
2 files changed, 9 insertions(+), 10 deletions(-)
diff --git a/internal/drivers/keyvaluedriver/keyvaluedriver.go b/internal/drivers/keyvaluedriver/keyvaluedriver.go
index 3624612..15c3595 100644
--- a/internal/drivers/keyvaluedriver/keyvaluedriver.go
+++ b/internal/drivers/keyvaluedriver/keyvaluedriver.go
@@ -35,13 +35,12 @@ func (d *KeyValueDriver) Start() {
}
// Get retrieves the value from the redis store that belongs to the given key
-func (d *KeyValueDriver) Get(key *string) *string {
- value := d.client.Get(context.Background(), *key).Val()
- return &value
+func (d *KeyValueDriver) Get(key *string) string {
+ return d.client.Get(context.Background(), *key).Val()
}
// Set sets the key value pair in the redis store
-func (d *KeyValueDriver) Set(key *string, value interface{}) error {
- status := d.client.Set(context.Background(), *key, value, 0)
+func (d *KeyValueDriver) Set(key *string, value *string) error {
+ status := d.client.Set(context.Background(), *key, *value, 0)
return status.Err()
}
diff --git a/internal/drivers/keyvaluedriver/mock/mockkeyvaluedriver.go b/internal/drivers/keyvaluedriver/mock/mockkeyvaluedriver.go
index 52f38c4..981b7cd 100644
--- a/internal/drivers/keyvaluedriver/mock/mockkeyvaluedriver.go
+++ b/internal/drivers/keyvaluedriver/mock/mockkeyvaluedriver.go
@@ -2,23 +2,23 @@ package mockkeyvaluedriver
// A KeyValueStore implements methods to set key-value data (mock)
type KeyValueStore struct {
- data map[string]interface{}
+ data map[string]string
}
// CreateKeyValueStore creates a key value store driver (mock)
func CreateKeyValueStore() *KeyValueStore {
return &KeyValueStore{
- data: make(map[string]interface{}),
+ data: make(map[string]string),
}
}
// Set sets a key to a value in the key value store. Expects a non-pointer as value. (mock)
-func (kvs *KeyValueStore) Set(key *string, value interface{}) error {
- kvs.data[*key] = value
+func (kvs *KeyValueStore) Set(key *string, value *string) error {
+ kvs.data[*key] = *value
return nil
}
// Get gets the value for the supplied key from the key value store (mock)
-func (kvs *KeyValueStore) Get(key *string) interface{} {
+func (kvs *KeyValueStore) Get(key *string) string {
return kvs.data[*key]
}
--
GitLab