Examples with TriggerMetadata org.apache.cassandra.schema.TriggerMetadata used on opensource projects

Example 1 with TriggerMetadata

use of org.apache.cassandra.schema.TriggerMetadata in project cassandra by apache.

the class TriggerExecutor method executeInternal.

/**
     * Switch class loader before using the triggers for the column family, if
     * not loaded them with the custom class loader.
     */
private List<Mutation> executeInternal(PartitionUpdate update) {
    Triggers triggers = update.metadata().triggers;
    if (triggers.isEmpty())
        return null;
    List<Mutation> tmutations = Lists.newLinkedList();
    Thread.currentThread().setContextClassLoader(customClassLoader);
    try {
        for (TriggerMetadata td : triggers) {
            ITrigger trigger = cachedTriggers.get(td.classOption);
            if (trigger == null) {
                trigger = loadTriggerInstance(td.classOption);
                cachedTriggers.put(td.classOption, trigger);
            }
            Collection<Mutation> temp = trigger.augment(update);
            if (temp != null)
                tmutations.addAll(temp);
        }
        return tmutations;
    } catch (CassandraException ex) {
        throw ex;
    } catch (Exception ex) {
        throw new RuntimeException(String.format("Exception while executing trigger on table with ID: %s", update.metadata().id), ex);
    } finally {
        Thread.currentThread().setContextClassLoader(parent);
    }
}

Example 2 with TriggerMetadata

use of org.apache.cassandra.schema.TriggerMetadata in project cassandra by apache.

the class TriggersSchemaTest method removeTriggerFromCf.

@Test
public void removeTriggerFromCf() throws Exception {
    TriggerMetadata td = TriggerMetadata.create(triggerName, triggerClass);
    TableMetadata tm = CreateTableStatement.parse(String.format("CREATE TABLE %s (k int PRIMARY KEY, v int)", cfName), ksName).triggers(Triggers.of(td)).build();
    KeyspaceMetadata ksm = KeyspaceMetadata.create(ksName, KeyspaceParams.simple(1), Tables.of(tm));
    MigrationManager.announceNewKeyspace(ksm);
    TableMetadata tm1 = Schema.instance.getTableMetadata(ksName, cfName);
    TableMetadata tm2 = tm1.unbuild().triggers(tm1.triggers.without(triggerName)).build();
    MigrationManager.announceTableUpdate(tm2);
    TableMetadata tm3 = Schema.instance.getTableMetadata(ksName, cfName);
    assertTrue(tm3.triggers.isEmpty());
}

Example 3 with TriggerMetadata

use of org.apache.cassandra.schema.TriggerMetadata in project cassandra by apache.

the class TriggersSchemaTest method newKsContainsCfWithTrigger.

@Test
public void newKsContainsCfWithTrigger() throws Exception {
    TriggerMetadata td = TriggerMetadata.create(triggerName, triggerClass);
    TableMetadata tm = CreateTableStatement.parse(String.format("CREATE TABLE %s (k int PRIMARY KEY, v int)", cfName), ksName).triggers(Triggers.of(td)).build();
    KeyspaceMetadata ksm = KeyspaceMetadata.create(ksName, KeyspaceParams.simple(1), Tables.of(tm));
    MigrationManager.announceNewKeyspace(ksm);
    TableMetadata tm2 = Schema.instance.getTableMetadata(ksName, cfName);
    assertFalse(tm2.triggers.isEmpty());
    assertEquals(1, tm2.triggers.size());
    assertEquals(td, tm2.triggers.get(triggerName).get());
}

Example 4 with TriggerMetadata

use of org.apache.cassandra.schema.TriggerMetadata in project cassandra by apache.

the class TriggersSchemaTest method addTriggerToCf.

@Test
public void addTriggerToCf() throws Exception {
    TableMetadata tm1 = CreateTableStatement.parse(String.format("CREATE TABLE %s (k int PRIMARY KEY, v int)", cfName), ksName).build();
    KeyspaceMetadata ksm = KeyspaceMetadata.create(ksName, KeyspaceParams.simple(1), Tables.of(tm1));
    MigrationManager.announceNewKeyspace(ksm);
    TriggerMetadata td = TriggerMetadata.create(triggerName, triggerClass);
    TableMetadata tm2 = Schema.instance.getTableMetadata(ksName, cfName).unbuild().triggers(Triggers.of(td)).build();
    MigrationManager.announceTableUpdate(tm2);
    TableMetadata tm3 = Schema.instance.getTableMetadata(ksName, cfName);
    assertFalse(tm3.triggers.isEmpty());
    assertEquals(1, tm3.triggers.size());
    assertEquals(td, tm3.triggers.get(triggerName).get());
}