/** {@inheritDoc} */ @Override public String getQueryKeyspace() { return st.getQueryKeyspace(); }
String keyspace = ps.getQueryKeyspace() == null ? "" : ps.getQueryKeyspace(); perKeyspace.put(keyspace, ps.getQueryString());
String prepareKeyspace = toPrepare.getQueryKeyspace(); if (prepareKeyspace != null && (currentKeyspace == null || !currentKeyspace.equals(prepareKeyspace))) { String.format( "Statement was prepared on keyspace %s, can't execute it on %s (%s)", toPrepare.getQueryKeyspace(), connection.keyspace(), toPrepare.getQueryString()));
String keyspace = ps.getQueryKeyspace() == null ? "" : ps.getQueryKeyspace(); perKeyspace.put(keyspace, ps.getQueryString());
String keyspace = ps.getQueryKeyspace() == null ? "" : ps.getQueryKeyspace(); perKeyspace.put(keyspace, ps.getQueryString());
if (statement instanceof BoundStatement) { PreparedStatement pstmt = ((BoundStatement) statement).preparedStatement(); queryKeyspace = pstmt.getQueryKeyspace(); queryString = pstmt.getQueryString();
String keyspace = ps.getQueryKeyspace() == null ? "" : ps.getQueryKeyspace(); perKeyspace.put(keyspace, ps.getQueryString());
String keyspace = ps.getQueryKeyspace() == null ? "" : ps.getQueryKeyspace(); perKeyspace.put(keyspace, ps.getQueryString());
@Test(groups = "short") public void should_get_keyspace_from_session_if_not_set() { Session session = cluster().connect(keyspace2); try { PreparedStatement prepared = session.prepare("select * from users"); assertThat(prepared.getQueryKeyspace()).isEqualTo(keyspace2Internal); // no keyspace is set. BoundStatement bs = prepared.bind(); assertThat(bs.getKeyspace()) .isEqualTo( keyspace2Internal); // keyspace should be set from prepared statement which was set // from session. assertThat(session.execute(bs).one().getString("name")).isEqualTo("test2"); } finally { session.close(); } }
@Test(groups = "short") public void should_have_no_keyspace_if_no_metadata_and_keyspace_not_set() { Session session = cluster().connect(); try { PreparedStatement prepared = session.prepare("select * from " + keyspace2 + ".users"); assertThat(prepared.getQueryKeyspace()).isNull(); // no keyspace is set. BoundStatement bs = prepared.bind(); assertThat(bs.getKeyspace()) .isNull(); // no session, keyspace not set, no metadata so should be null. assertThat(session.execute(bs).one().getString("name")).isEqualTo("test2"); } finally { session.close(); } }
@Test(groups = "short") public void should_get_keyspace_from_metadata_if_not_set() { Session session = cluster().connect(); try { PreparedStatement prepared = session.prepare("select * from " + keyspace2 + ".users where id = ?"); assertThat(prepared.getQueryKeyspace()).isNull(); // no keyspace is set. BoundStatement bs = prepared.bind(2); assertThat(bs.getKeyspace()) .isEqualTo( keyspace2Internal); // keyspace should be set from prepared metadata, we expect it to // be the internal representation. assertThat(session.execute(bs).one().getString("name")).isEqualTo("test2"); } finally { session.close(); } }
private void queryWithKeyspaceOnStatement(Cluster cluster, String sessionKeyspace) { Session session; if (sessionKeyspace != null) { session = cluster.connect(sessionKeyspace); } else { session = cluster.connect(); } try { SimpleStatement statement = new SimpleStatement("SELECT name FROM users WHERE id = 2 and id2 = 3") .setKeyspace(keyspace2Internal); // Keyspace should be propagated to prepared statement. PreparedStatement prepared = session.prepare(statement); assertThat(prepared.getQueryKeyspace()).isEqualTo(keyspace2Internal); BoundStatement bound = prepared.bind(); Row row = session.execute(bound).one(); assertThat(row).isNotNull(); assertThat(row.getString("name")).isEqualTo("test2"); } finally { session.close(); if (cluster != cluster()) { cluster.close(); } } } }
String prepareKeyspace = toPrepare.getQueryKeyspace(); if (prepareKeyspace != null && (currentKeyspace == null || !currentKeyspace.equals(prepareKeyspace))) { toPrepare.getQueryKeyspace(), connection.keyspace(), toPrepare.getQueryString()));
String prepareKeyspace = toPrepare.getQueryKeyspace(); if (prepareKeyspace != null && (currentKeyspace == null || !currentKeyspace.equals(prepareKeyspace))) { toPrepare.getQueryKeyspace(), connection.keyspace(), toPrepare.getQueryString()));
String prepareKeyspace = toPrepare.getQueryKeyspace(); if (prepareKeyspace != null && (currentKeyspace == null || !currentKeyspace.equals(prepareKeyspace))) { toPrepare.getQueryKeyspace(), connection.keyspace(), toPrepare.getQueryString()));
toPrepare.getQueryString(), connection.address); String currentKeyspace = connection.keyspace(); String prepareKeyspace = toPrepare.getQueryKeyspace();
assertThat(prepared.getQueryKeyspace()).isEqualTo(keyspace);