private String statementQueryString(final Statement statement) { String query; if (statement instanceof RegularStatement) { query = statementQueryString((RegularStatement) statement); } else if (statement instanceof BoundStatement) { query = statementQueryString((BoundStatement) statement); } else if (statement instanceof BatchStatement) { query = statementQueryString((BatchStatement) statement); } else { query = "unknown;"; } return query; }
private String statementQueryString(final BatchStatement batchStatement) { StringBuffer sb = new StringBuffer(); sb.append("BEGIN BATCH "); for (Statement statement : batchStatement.getStatements()) { sb.append(statementQueryString(statement)); } sb.append(" APPLY BATCH;"); return sb.toString(); }
private Query extractQuery(final long startTime, final long execTime, final Statement statement) { final String queryString = statementQueryString(statement); final Query.StatementType queryType = queryType(queryString); return Query.create(startTime, execTime, host, queryType, statement.getKeyspace(), "", queryString, extractConsistencyLevel(statement)); }