private void assertTotalResults(String table, String q, long expected) throws BlurException, TException { BlurQuery bquery = new BlurQuery(); Query query = new Query(); query.setQuery(q); bquery.setQuery(query); BlurResults results = client().query(table, bquery); assertEquals("Should find our row.", expected, results.getTotalResults()); }
@Override public int hashCode() { final int prime = 31; int result = 1; result = prime * result + ((query == null) ? 0 : query.hashCode()); result = prime * result + ((table == null) ? 0 : table.hashCode()); return result; }
@Override public boolean equals(Object obj) { if (this == obj) return true; if (obj == null) return false; if (getClass() != obj.getClass()) return false; QueryCacheKey other = (QueryCacheKey) obj; if (query == null) { if (other.query != null) return false; } else if (!query.equals(other.query)) return false; if (table == null) { if (other.table != null) return false; } else if (!table.equals(other.table)) return false; return true; }
private void runSearch(int start, int fetch) throws SQLException { currentStart = start; currentFetch = fetch; BlurQuery blurQuery = new BlurQuery(); blurQuery.minimumNumberOfResults = Long.MAX_VALUE; blurQuery.maxQueryTime = Long.MAX_VALUE; blurQuery.uuid = UUID.randomUUID().toString(); blurQuery.fetch = fetch; blurQuery.start = start; blurQuery.query = new Query(); blurQuery.query.query = parser.getWhere(); blurQuery.query.rowQuery = false; try { schema = client.schema(tableName); } catch (BlurException e) { e.printStackTrace(); throw new SQLException(e); } catch (TException e) { e.printStackTrace(); throw new SQLException(e); } selector = new Selector(); setupSelector(selector, schema, columnNames); selector.recordOnly = !blurQuery.query.rowQuery; Map<String, Map<String, ColumnDefinition>> columnFamilies = schema.getFamilies(); Map<String, ColumnDefinition> cfSet = columnFamilies.get(columnFamily); columnFamilies.clear(); columnFamilies.put(columnFamily, cfSet); blurResultSetMetaData = new BlurResultSetMetaData(columnNames, columnFamilies); try { results = client.query(tableName, blurQuery); } catch (BlurException e) { e.printStackTrace(); throw new SQLException(e); } catch (TException e) { e.printStackTrace(); throw new SQLException(e); } if (results.totalResults > 0) { size = results.results.size(); } totalResults = results.totalResults; }