Added key range and length, random read/write

Signed-off-by: zviedris <mzviedris@gmail.com>

Scripts/redis_get_set.py

@@ -12,6 +12,8 @@
 import time
 from locust import User, events, TaskSet, task, constant
 import redis
+import string
+import random
 import gevent.monkey
 gevent.monkey.patch_all()
 
@@ -22,6 +24,8 @@ def load_config(filepath):
         configs = json.load(property_file)
     return configs
 
+def randStr(chars = string.ascii_uppercase + string.digits, N=10):
+    return ''.join(random.choice(chars) for _ in range(N))
 
 filename = "redis.json"
 
@@ -65,15 +69,16 @@ def write(self, key, value, command='SET'):
                 request_type=command, name=key, response_time=total_time, exception=e)
         else:
             total_time = int((time.time() - start_time) * 1000)
-            length = 1
+            length = len(value)
             events.request_success.fire(
                 request_type=command, name=key, response_time=total_time, response_length=length)
         return result
 
 
 class RedisLocust(User):
     wait_time = constant(0.1)
-    key_range = 500
+    key_range = configs["key_range"]
+    key_length = configs["key_length"]
 
     def __init__(self, *args, **kwargs):
         super(RedisLocust, self).__init__(*args, **kwargs)
@@ -83,19 +88,21 @@ def __init__(self, *args, **kwargs):
 
     @task(2)
     def get_time(self):
-        for i in range(self.key_range):
+        #for i in range(self.key_range):
+            i = randint(1, self.key_range-1)
             self.key = 'key'+str(i)
             self.client.query(self.key)
 
     @task(1)
     def write(self):
-        for i in range(self.key_range):
+        #for i in range(self.key_range):
+            i = randint(1, self.key_range-1)
             self.key = 'key'+str(i)
-            self.value = 'value'+str(i)
+            self.value = randStr(N=self.key_length)
             self.client.write(self.key, self.value)
 
-    @task(1)
-    def get_key(self):
-        var = str(randint(1, self.key_range-1))
-        self.key = 'key'+var
-        self.value = 'value'+var
+#    @task(1)
+#    def get_key(self):
+#        var = str(randint(1, self.key_range-1))
+#        self.key = 'key'+var
+#        self.value = 'value'+var

Scripts/redis_orig.json

@@ -1,5 +1,7 @@
 {
     "redis_host": "$REDIS_HOST",
     "redis_port": "$REDIS_PORT",
-    "redis_password": "$REDIS_PW"
+    "redis_password": "$REDIS_PW",
+    "key_length": 5000,
+    "key_range": 1000
 }