implemented updated AstyanaxContext methods

sanjeevgour · Nov 21, 2015 · 305e534 · 305e534
1 parent 0a5b3fa
commit 305e534
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 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
@@ -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
@@ -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) {