private static void mockGetScriptForEndpoint(Rows<String, String> response, String filter) { /* mock data so that the getScriptForEndpoint at the end of the method will work */ Calendar now = Calendar.getInstance(); Row<String, String> row0 = mockRow(); ColumnList<String> columnList0 = mockColumnList(row0); mockColumn(columnList0, "filter_id", filter); mockColumn(columnList0, "revision", 1L); mockColumn(columnList0, "active", false); mockColumn(columnList0, "creation_date", now.getTime()); mockColumn(columnList0, "filter_code", "System.out.println(\"hello world\")".getBytes()); // what we put here doesn't matter mockColumn(columnList0, "filter_name", "name"); mockColumn(columnList0, "filter_type", "INBOUND"); mockColumn(columnList0, "canary", false); when(response.getRowByIndex(0)).thenReturn(row0); when(response.isEmpty()).thenReturn(true, false); // true on the first request, false thereafter when(response.size()).thenReturn(1); } }
private static void mockGetScriptForEndpoint(Rows<String, String> response, String filter) { /* mock data so that the getScriptForEndpoint at the end of the method will work */ Calendar now = Calendar.getInstance(); Row<String, String> row0 = mockRow(); ColumnList<String> columnList0 = mockColumnList(row0); mockColumn(columnList0, "filter_id", filter); mockColumn(columnList0, "revision", 1L); mockColumn(columnList0, "active", false); mockColumn(columnList0, "creation_date", now.getTime()); mockColumn(columnList0, "filter_code", "System.out.println(\"hello world\")".getBytes()); // what we put here doesn't matter mockColumn(columnList0, "filter_name", "name"); mockColumn(columnList0, "filter_type", "type"); mockColumn(columnList0, "canary", false); when(response.getRowByIndex(0)).thenReturn(row0); when(response.isEmpty()).thenReturn(true, false); // true on the first request, false thereafter when(response.size()).thenReturn(1); } }
when(response.getRowByIndex(0)).thenReturn(row0);
when(response.getRowByIndex(0)).thenReturn(row0);
@Test public void testGetAllKeysRoot() throws ConnectionException { LOG.info("Starting testGetAllKeysRoot..."); List<String> keys = new ArrayList<String>(); for (char key = 'A'; key <= 'Z'; key++) { String keyName = Character.toString(key); keys.add(keyName); } OperationResult<Rows<String, String>> result = keyspace .prepareQuery(CF_STANDARD1) .getKeySlice(keys.toArray(new String[keys.size()])) .execute(); Assert.assertEquals(26, result.getResult().size()); Row<String, String> row; row = result.getResult().getRow("A"); Assert.assertEquals("A", row.getKey()); row = result.getResult().getRow("B"); Assert.assertEquals("B", row.getKey()); row = result.getResult().getRow("NonExistent"); Assert.assertNull(row); row = result.getResult().getRowByIndex(10); Assert.assertEquals("K", row.getKey()); LOG.info("... testGetAllKeysRoot"); }
when(response.getRowByIndex(0)).thenReturn(row0);
when(response.getRowByIndex(0)).thenReturn(row0);
when(response.getRowByIndex(0)).thenReturn(row0);
when(response.getRowByIndex(0)).thenReturn(row0); when(response.getRowByIndex(1)).thenReturn(row1);
when(response.getRowByIndex(0)).thenReturn(row0); when(response.getRowByIndex(1)).thenReturn(row1);
public String findKey(String cluster, String instanceId, String dc) { try { final String selectClause = String.format( "SELECT * FROM %s WHERE %s = '%s' and %s = '%s' and %s = '%s' ", CF_NAME_INSTANCES, CN_CLUSTER, cluster, CN_INSTANCEID, instanceId, CN_LOCATION, dc); logger.info(selectClause); final ColumnFamily<String, String> CF_INSTANCES_NEW = ColumnFamily.newColumnFamily(KS_NAME, StringSerializer.get(), StringSerializer.get()); OperationResult<CqlResult<String, String>> result = bootKeyspace.prepareQuery(CF_INSTANCES_NEW) .withCql(selectClause).execute(); if (result == null || result.getResult().getRows().size() == 0) return null; Row<String, String> row = result.getResult().getRows().getRowByIndex(0); return row.getKey(); } catch (Exception e) { logger.warn("Caught an Unknown Exception during find a row matching cluster[" + cluster + "], id[" + instanceId + "], and region[" + dc + "] ... -> " + e.getMessage()); throw new RuntimeException(e); } }
public String findKey(String app, String id, String location, String datacenter) { try { final String selectClause = String.format( "SELECT * FROM %s USING CONSISTENCY LOCAL_QUORUM WHERE %s = '%s' and %s = '%s' and %s = '%s' and %s = '%s' ", "tokens", CN_APPID, app, CN_ID, id, CN_LOCATION, location, CN_DC, datacenter); logger.info(selectClause); final ColumnFamily<String, String> CF_INSTANCES_NEW = ColumnFamily.newColumnFamily(KS_NAME, StringSerializer.get(), StringSerializer.get()); OperationResult<CqlResult<String, String>> result = bootKeyspace.prepareQuery(CF_INSTANCES_NEW) .withCql(selectClause).execute(); if (result == null || result.getResult().getRows().size() == 0) return null; Row<String, String> row = result.getResult().getRows().getRowByIndex(0); return row.getKey(); } catch (Exception e) { logger.warn("Caught an Unknown Exception during find a row matching cluster[" + app + "], id[" + id + "], and region[" + datacenter + "] ... -> " + e.getMessage()); throw new RuntimeException(e); } }
when(response.getRowByIndex(0)).thenReturn(row0);
when(response.getRowByIndex(0)).thenReturn(row0);
when(response.getRowByIndex(0)).thenReturn(row0); when(response.getRowByIndex(1)).thenReturn(row1); when(response2.getRowByIndex(0)).thenReturn(response2row0); when(response3.getRowByIndex(0)).thenReturn(response3row0);
when(response.getRowByIndex(0)).thenReturn(row0); when(response.getRowByIndex(1)).thenReturn(row1); when(response2.getRowByIndex(0)).thenReturn(response2row0); when(response3.getRowByIndex(0)).thenReturn(response3row0);
when(response.getRowByIndex(0)).thenReturn(row0);
when(response.getRowByIndex(0)).thenReturn(row0);
when(response.getRowByIndex(0)).thenReturn(row0);
Row<K, C> lastRow = rows.getRowByIndex(rows.size() - 1); String lastToken = partitioner.getTokenForKey(lastRow.getRawKey()); checkpointManager.trackCheckpoint(startToken, currentToken); if (!lastToken.equals(partitioner.getTokenForKey(rows.getRowByIndex(i).getRawKey()))) { break;