Added keyspace to CasquatchTrace

tmobile · Aug 19, 2019 · f8ffd19 · f8ffd19
1 parent 443abc9
commit f8ffd19
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/casquatch-driver/src/main/java/com/tmobile/opensource/casquatch/CasquatchDao.java b/casquatch-driver/src/main/java/com/tmobile/opensource/casquatch/CasquatchDao.java
@@ -69,11 +69,14 @@ public CasquatchTrace(ResultSet resultSet) {
                     this.profile = resultSet.getExecutionInfo().getStatement().getExecutionProfileName();
                     if(resultSet.getExecutionInfo().getStatement().getConsistencyLevel() !=null ) {
                         this.consistencyLevel = resultSet.getExecutionInfo().getStatement().getConsistencyLevel().name();
-                        this.keyspace = resultSet.getExecutionInfo().getStatement().getKeyspace().toString();
+                        if(resultSet.getExecutionInfo().getStatement().getKeyspace()!=null) {
+                            this.keyspace = resultSet.getExecutionInfo().getStatement().getKeyspace().toString();
+                        }
                     }
                 }
             }
         }
+
         public String toString() {
             try {
                 return new ObjectMapper().writeValueAsString(this);