/** Return true if the column family exists */ public boolean cfExists( String keyspace, String cfName ) { KeyspaceDefinition ksDef = cluster.describeKeyspace( keyspace ); if ( ksDef == null ) { return false; } for ( ColumnFamilyDefinition cf : ksDef.getCfDefs() ) { if ( cfName.equals( cf.getName() ) ) { return true; } } return false; }
private boolean doesLockCFExist(KeyspaceDefinition keyspaceDef) { for (ColumnFamilyDefinition cfdef : keyspaceDef.getCfDefs()) { if (cfdef.getName().equals(lockManagerConfigurator.getLockManagerCF())) { return true; } } return false; }
public ThriftKsDef(KeyspaceDefinition keyspaceDefinition) { name = keyspaceDefinition.getName(); strategyClass = keyspaceDefinition.getStrategyClass(); strategyOptions = keyspaceDefinition.getStrategyOptions(); setReplicationFactor(keyspaceDefinition.getReplicationFactor()); cfDefs = keyspaceDefinition.getCfDefs(); }
@Test public void testDescribeKeyspace() throws Exception { KeyspaceDefinition keyspaceDetail = cassandraCluster.describeKeyspace("Keyspace1"); assertNotNull(keyspaceDetail); assertEquals(22, keyspaceDetail.getCfDefs().size()); } }
@Test public void testDescribeKeyspace() throws Exception { ThriftCluster cassandraCluster = new ThriftCluster("Test Cluster", cassandraHostConfigurator, user1Credentials); KeyspaceDefinition keyspaceDetail = cassandraCluster.describeKeyspace("Keyspace1"); assertNotNull(keyspaceDetail); assertEquals(22, keyspaceDetail.getCfDefs().size()); }
@Test public void testInitWithDefaults() { KeyspaceDefinition keyspaceDef = cluster.describeKeyspace(lm.getKeyspace().getKeyspaceName()); assertNotNull(keyspaceDef); assertTrue(verifyCFCreation(keyspaceDef.getCfDefs())); }
@Test public void testEditColumnFamily() throws Exception { BasicColumnFamilyDefinition columnFamilyDefinition = new BasicColumnFamilyDefinition(); columnFamilyDefinition.setKeyspaceName("DynKeyspace3"); columnFamilyDefinition.setName("DynamicCF"); ColumnFamilyDefinition cfDef = new ThriftCfDef(columnFamilyDefinition); KeyspaceDefinition keyspaceDefinition = HFactory.createKeyspaceDefinition("DynKeyspace3", "org.apache.cassandra.locator.SimpleStrategy", 1, Arrays.asList(cfDef)); cassandraCluster.addKeyspace(keyspaceDefinition); KeyspaceDefinition fromCluster = cassandraCluster.describeKeyspace("DynKeyspace3"); cfDef = fromCluster.getCfDefs().get(0); columnFamilyDefinition = new BasicColumnFamilyDefinition(cfDef); BasicColumnDefinition columnDefinition = new BasicColumnDefinition(); columnDefinition.setName(StringSerializer.get().toByteBuffer("birthdate")); columnDefinition.setIndexName("birthdate_idx"); columnDefinition.setIndexType(ColumnIndexType.KEYS); columnDefinition.setValidationClass(ComparatorType.LONGTYPE.getClassName()); columnFamilyDefinition.addColumnDefinition(columnDefinition); columnDefinition = new BasicColumnDefinition(); columnDefinition.setName(StringSerializer.get().toByteBuffer("nonindexed_field")); columnDefinition.setValidationClass(ComparatorType.LONGTYPE.getClassName()); columnFamilyDefinition.addColumnDefinition(columnDefinition); cassandraCluster.updateColumnFamily(new ThriftCfDef(columnFamilyDefinition)); fromCluster = cassandraCluster.describeKeyspace("DynKeyspace3"); assertEquals("birthdate",StringSerializer.get().fromByteBuffer(fromCluster.getCfDefs().get(0).getColumnMetadata().get(0).getName())); assertEquals("birthdate_idx",fromCluster.getCfDefs().get(0).getColumnMetadata().get(0).getIndexName()); assertEquals("nonindexed_field",StringSerializer.get().fromByteBuffer(fromCluster.getCfDefs().get(0).getColumnMetadata().get(1).getName())); }
/** * Returns true if a given keyspace has a column family. * * @param ksdef the keyspace. * @param cfname the column family name. * @return true if the keyspace has the column family, false otherwise. */ public static boolean hasColumnFamily(final KeyspaceDefinition ksdef, final String cfname) { for (final ColumnFamilyDefinition cfdef : ksdef.getCfDefs()) { if (cfdef.getName().equals(cfname)) { return true; } } return false; } }
/** * Returns true if a given keyspace has a column family. * * @param ksdef the keyspace. * @param cfname the column family name. * @return true if the keyspace has the column family, false otherwise. */ public static boolean hasColumnFamily(final KeyspaceDefinition ksdef, final String cfname) { for (final ColumnFamilyDefinition cfdef : ksdef.getCfDefs()) { if (cfdef.getName().equals(cfname)) { return true; } } return false; } }
/** * Private helper to obtain a column family from a keyspace by name * * @param name the column family name * @param keyspaceDefinition the keyspace * @return the column family if found, null otherwise */ protected ColumnFamilyDefinition getColumnFamilyFromKeyspace(String name, KeyspaceDefinition keyspaceDefinition) { ColumnFamilyDefinition definition = null; for (ColumnFamilyDefinition columnFamilyDefinitionEntry : keyspaceDefinition.getCfDefs()) { if (columnFamilyDefinitionEntry.getName().equals(name)) { definition = columnFamilyDefinitionEntry; break; } } return definition; }
private boolean doesLockCFExist(KeyspaceDefinition keyspaceDef) { for (ColumnFamilyDefinition cfdef : keyspaceDef.getCfDefs()) { if (cfdef.getName().equals(lockManagerConfigurator.getLockManagerCF())) { return true; } } return false; }
@Test public void testEditBasicColumnFamilyDefinition() throws Exception { BasicKeyspaceDefinition ksDef = new BasicKeyspaceDefinition(); ksDef.setName("Keyspace2"); ksDef.setReplicationFactor(1); ksDef.setStrategyClass("SimpleStrategy"); cassandraCluster.addKeyspace(ksDef); BasicColumnFamilyDefinition cfDef = new BasicColumnFamilyDefinition(); cfDef.setName("DynCf2"); cfDef.setKeyspaceName("Keyspace2"); cassandraCluster.addColumnFamily(cfDef); KeyspaceDefinition fromCluster = cassandraCluster.describeKeyspace("Keyspace2"); cfDef = new BasicColumnFamilyDefinition(fromCluster.getCfDefs().get(0)); cfDef.setDefaultValidationClass(ComparatorType.LONGTYPE.getClassName()); cassandraCluster.updateColumnFamily(cfDef); String cfid2 = cassandraCluster.dropColumnFamily("Keyspace2", "DynCf2"); assertNotNull(cfid2); } }
/** * Private helper to determine if a column family has already been created in a keyspace * * @param columnFamilyDefinition the column family definition * @param keyspaceDefinition the keyspace definition * @return true if the column family is already present in the keyspace */ protected boolean keyspaceContainsColumnFamily(ColumnFamilyDefinition columnFamilyDefinition, KeyspaceDefinition keyspaceDefinition) { boolean contains = false; for (ColumnFamilyDefinition columnFamilyDefinitionEntry : keyspaceDefinition.getCfDefs()) { if (columnFamilyDefinitionEntry.getName().equals(columnFamilyDefinition.getName())) { contains = true; break; } } return contains; }
KeyspaceDefinition ksdef = cluster.describeKeyspace("myKeyspace"); for (ColumnFamilyDefinition cfdef: ksdef.getCfDefs()) { System.out.println(cfdef.getName()); if (cfdef.getName().equals(cfname)) {
public ThriftKsDef(KeyspaceDefinition keyspaceDefinition) { name = keyspaceDefinition.getName(); strategyClass = keyspaceDefinition.getStrategyClass(); strategyOptions = keyspaceDefinition.getStrategyOptions(); setReplicationFactor(keyspaceDefinition.getReplicationFactor()); cfDefs = keyspaceDefinition.getCfDefs(); }
public ThriftKsDef(KeyspaceDefinition keyspaceDefinition) { name = keyspaceDefinition.getName(); strategyClass = keyspaceDefinition.getStrategyClass(); strategyOptions = keyspaceDefinition.getStrategyOptions(); setReplicationFactor(keyspaceDefinition.getReplicationFactor()); cfDefs = keyspaceDefinition.getCfDefs(); }
private static void checkColumns(List<ColumnFamilyModel> expectedColumnFamilies, Keyspace keyspace, KeyspaceDefinition keyspaceDefinition) { List<ColumnFamilyDefinition> columnFamilyDefinitions = keyspaceDefinition.getCfDefs(); for (ColumnFamilyModel expectedColumnFamilyModel : expectedColumnFamilies) { ColumnFamilyDefinition columnFamily = checkColumnFamilyName(columnFamilyDefinitions, expectedColumnFamilyModel); ColumnType columnType = checkColumnFamilyType(expectedColumnFamilyModel, columnFamily); String expectedColumnFamilyName = expectedColumnFamilyModel.getName(); List<RowModel> expectedRows = expectedColumnFamilyModel.getRows(); checkNumberOfRowsIntoColumnFamily(keyspace, expectedColumnFamilyName, expectedRows.size()); if (ColumnType.STANDARD == columnType) { for (RowModel expectedRowModel : expectedRows) { checkStandardColumns(keyspace, expectedColumnFamilyModel, expectedRowModel); } } else { if (ColumnType.SUPER == columnType) { for (RowModel expectedRowModel : expectedRows) { List<ColumnModel> expectedColumns = expectedRowModel.getColumns(); checkNotStandardColumnsInSuperColumns(expectedRowModel, expectedColumns.size()); checkSuperColumns(keyspace, expectedColumnFamilyModel, expectedRowModel); } } else { throw new IllegalArgumentException("Column type is not STANDARD or SUPER."); } } } }
private static void checkColumns(List<ColumnFamilyModel> expectedColumnFamilies, Keyspace keyspace, KeyspaceDefinition keyspaceDefinition) { List<ColumnFamilyDefinition> columnFamilyDefinitions = keyspaceDefinition.getCfDefs(); for (ColumnFamilyModel expectedColumnFamilyModel : expectedColumnFamilies) { ColumnFamilyDefinition columnFamily = checkColumnFamilyName(columnFamilyDefinitions, expectedColumnFamilyModel); ColumnType columnType = checkColumnFamilyType(expectedColumnFamilyModel, columnFamily); String expectedColumnFamilyName = expectedColumnFamilyModel.getName(); List<RowModel> expectedRows = expectedColumnFamilyModel.getRows(); checkNumberOfRowsIntoColumnFamily(keyspace, expectedColumnFamilyName, expectedRows.size()); if (ColumnType.STANDARD == columnType) { for (RowModel expectedRowModel : expectedRows) { checkStandardColumns(keyspace, expectedColumnFamilyModel, expectedRowModel); } } else { if (ColumnType.SUPER == columnType) { for (RowModel expectedRowModel : expectedRows) { List<ColumnModel> expectedColumns = expectedRowModel.getColumns(); checkNotStandardColumnsInSuperColumns(expectedRowModel, expectedColumns.size()); checkSuperColumns(keyspace, expectedColumnFamilyModel, expectedRowModel); } } else { throw new IllegalArgumentException("Column type is not STANDARD or SUPER."); } } } }
public static void strictAssertEquals(DataSet dataset, Cluster cluster, Keyspace keyspace) { KeyspaceDefinition keyspaceDefinition = getKeyspaceDefinition(cluster, keyspace); checkKeyspaceName(dataset.getKeyspace().getName(), keyspace.getKeyspaceName()); checkColumnsSize(dataset.getColumnFamilies(), keyspaceDefinition.getCfDefs()); checkColumns(dataset.getColumnFamilies(), keyspace, keyspaceDefinition); }
public static void strictAssertEquals(DataSet dataset, Cluster cluster, Keyspace keyspace) { KeyspaceDefinition keyspaceDefinition = getKeyspaceDefinition(cluster, keyspace); checkKeyspaceName(dataset.getKeyspace().getName(), keyspace.getKeyspaceName()); checkColumnsSize(dataset.getColumnFamilies(), keyspaceDefinition.getCfDefs()); checkColumns(dataset.getColumnFamilies(), keyspace, keyspaceDefinition); }