Experimental commit for A-B testing with Clover Fraser release

and on the SDC application

Change-Id: I6e1bd84a6d674a2c4c4484722b20415f5402a59c
Signed-off-by: Stephen Wong <stephen.kf.wong@gmail.com>

1 files changed, 12 insertions, 2 deletions

diff --git a/clover/servicemesh/route_rules.py b/clover/servicemesh/route_rules.py
index cc2ee0c..935940e 100644
--- a/clover/servicemesh/route_rules.py
+++ b/clover/servicemesh/route_rules.py
@@ -12,6 +12,8 @@ import subprocess
 import sys
 import yaml
 
+from clover.orchestration.kube_client import KubeClient
+
 #istioctl='$HOME/istio-0.6.0/bin/istioctl'
 # The assumption is that istioctl is already in the user's path
 ISTIOCTL='istioctl'
@@ -85,14 +87,22 @@ def parse_route_rules(routerules):
 def _derive_key_from_test_id(test_id):
     return 'route-rules-' + str(test_id)
 
+def _get_redis_ip():
+    k8s_client = KubeClient()
+    redis_pod = k8s_client.find_pod_by_name('redis')
+    redis_ip = redis_pod.get('pod_ip')
+    return redis_ip
+
 def set_route_rules(test_id):
-    r = redis.StrictRedis(host='localhost', port=6379, db=0)
+    redis_ip = _get_redis_ip()
+    r = redis.StrictRedis(host=redis_ip, port=6379, db=0)
     key = _derive_key_from_test_id(test_id)
     rr = get_route_rules()
     r.set(key, rr)
 
 def fetch_route_rules(test_id):
-    r = redis.StrictRedis(host='localhost', port=6379, db=0)
+    redis_ip = _get_redis_ip()
+    r = redis.StrictRedis(host=redis_ip, port=6379, db=0)
     key = _derive_key_from_test_id(test_id)
     rr = r.get(key)
     return yaml.load(rr)