From 305e53486afa76d0ecbb7f4f81dc86f67c8459c2 Mon Sep 17 00:00:00 2001
From: Tom Breitkopf <thomas.j.breitkopf@gmail.com>
Date: Sat, 21 Nov 2015 15:55:09 -0500
Subject: [PATCH] implemented updated AstyanaxContext methods

---
 .../com/netflix/zuul/dependency/cassandra/CassandraHelper.java  | 2 +-
 .../com/netflix/zuul/scriptManager/ZuulFilterDAOCassandra.java  | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/zuul-netflix/src/main/java/com/netflix/zuul/dependency/cassandra/CassandraHelper.java b/zuul-netflix/src/main/java/com/netflix/zuul/dependency/cassandra/CassandraHelper.java
index d2ee147786..1c5f81eaf6 100644
--- a/zuul-netflix/src/main/java/com/netflix/zuul/dependency/cassandra/CassandraHelper.java
+++ b/zuul-netflix/src/main/java/com/netflix/zuul/dependency/cassandra/CassandraHelper.java
@@ -87,7 +87,7 @@ public Keyspace getZuulCassKeyspace() throws Exception {
                     .buildKeyspace(ThriftFamilyFactory.getInstance());
 
             context.start();
-            zuulCassKeyspace = context.getEntity();
+            zuulCassKeyspace = context.getClient();
             return zuulCassKeyspace;
         } catch (Exception e) {
             LOG.error("Exception occurred when initializing Cassandra keyspace: " + e);
diff --git a/zuul-netflix/src/main/java/com/netflix/zuul/scriptManager/ZuulFilterDAOCassandra.java b/zuul-netflix/src/main/java/com/netflix/zuul/scriptManager/ZuulFilterDAOCassandra.java
index caef2de973..3edff9cbbf 100644
--- a/zuul-netflix/src/main/java/com/netflix/zuul/scriptManager/ZuulFilterDAOCassandra.java
+++ b/zuul-netflix/src/main/java/com/netflix/zuul/scriptManager/ZuulFilterDAOCassandra.java
@@ -502,7 +502,7 @@ private static class CassandraGatewayProd implements CassandraGateway {
 
 
         public CassandraGatewayProd(AstyanaxContext<Keyspace> context) {
-            keyspace = context.getEntity();
+            keyspace = context.getClient();
         }
 
         public CassandraGatewayProd(Keyspace keyspace) {