Refresh schema on SCHEMA_CHANGE response

olim7t · olim7t · commit d0141fc062ce · 2017-09-18T12:26:27.000-07:00
diff --git a/core/src/main/java/com/datastax/oss/driver/internal/core/cql/CqlRequestHandler.java b/core/src/main/java/com/datastax/oss/driver/internal/core/cql/CqlRequestHandler.java
@@ -382,7 +382,12 @@ public void onResponse(Frame responseFrame) {
         Message responseMessage = responseFrame.message;
         if (responseMessage instanceof SchemaChange) {
           // TODO schema agreement, and chain setFinalResult to the result
-          setFinalResult((Result) responseMessage, responseFrame, this);
+          SchemaChange schemaChange = (SchemaChange) responseMessage;
+          context
+              .metadataManager()
+              .refreshSchema(schemaChange.keyspace, false, false)
+              .whenComplete(
+                  ((metadata, error) -> setFinalResult(schemaChange, responseFrame, this)));
         } else if (responseMessage instanceof Result) {
           LOG.debug("[{}] Got result, completing", logPrefix);
           setFinalResult((Result) responseMessage, responseFrame, this);
