Java RowMutation类代码示例

import org.apache.cassandra.db.RowMutation; //导入依赖的package包/类
public RowMutation mutationForKey(ByteBuffer key, String keyspace, Long timestamp, ThriftClientState clientState, List<ByteBuffer> variables, CFMetaData metadata)
throws InvalidRequestException
{
    RowMutation rm = new RowMutation(keyspace, key);

    QueryProcessor.validateKeyAlias(metadata, keyName);

    if (columns.size() < 1)
    {
        // No columns, delete the row
        rm.delete(columnFamily, (timestamp == null) ? getTimestamp(clientState) : timestamp);
    }
    else
    {
        // Delete specific columns
        for (Term column : columns)
        {
            ByteBuffer columnName = column.getByteBuffer(metadata.comparator, variables);
            validateColumnName(columnName);
            rm.delete(columnFamily, columnName, (timestamp == null) ? getTimestamp(clientState) : timestamp);
        }
    }

    return rm;
}
 

import org.apache.cassandra.db.RowMutation; //导入依赖的package包/类
/**
 * mark all of the column families we're modifying as dirty at this position
 */
private void markDirty(RowMutation rowMutation, ReplayPosition repPos)
{
    for (ColumnFamily columnFamily : rowMutation.getColumnFamilies())
    {
        // check for null cfm in case a cl write goes through after the cf is
        // defined but before a new segment is created.
        CFMetaData cfm = Schema.instance.getCFMetaData(columnFamily.id());
        if (cfm == null)
        {
            logger.error("Attempted to write commit log entry for unrecognized column family: {}", columnFamily.id());
        }
        else
        {
            markCFDirty(cfm.cfId, repPos.position);
        }
    }
}
 

import org.apache.cassandra.db.RowMutation; //导入依赖的package包/类
public void run()
{
    long totalSize = RowMutation.serializer.serializedSize(rowMutation, MessagingService.current_version) + CommitLogSegment.ENTRY_OVERHEAD_SIZE;
    if (totalSize > DatabaseDescriptor.getCommitLogSegmentSize())
    {
        logger.warn("Skipping commitlog append of extremely large mutation ({} bytes)", totalSize);
        return;
    }

    if (!activeSegment.hasCapacityFor(totalSize))
    {
        CommitLogSegment oldSegment = activeSegment;
        activateNextSegment();
        // Now we can run the user defined command just before switching to the new commit log.
        // (Do this here instead of in the recycle call so we can get a head start on the archive.)
        archiver.maybeArchive(oldSegment.getPath(), oldSegment.getName());
    }
    try
    {
        activeSegment.write(rowMutation);
    }
    catch (IOException e)
    {
        throw new FSWriteError(e, activeSegment.getPath());
    }
}
 

import org.apache.cassandra.db.RowMutation; //导入依赖的package包/类
public Collection<RowMutation> execute(Collection<? extends IMutation> updates) throws InvalidRequestException
{
    boolean hasCounters = false;
    Collection<RowMutation> tmutations = null;
    for (IMutation mutation : updates)
    {
        for (ColumnFamily cf : mutation.getColumnFamilies())
        {
            List<RowMutation> intermediate = execute(mutation.key(), cf);
            if (intermediate == null)
                continue;

            validate(intermediate);
            if (tmutations == null)
                tmutations = intermediate;
            else
                tmutations.addAll(intermediate);
        }
        if (mutation instanceof CounterMutation)
            hasCounters = true;
    }
    if (tmutations != null && hasCounters)
        throw new InvalidRequestException("Counter mutations and trigger mutations cannot be applied together atomically.");
    return tmutations;
}
 

import org.apache.cassandra.db.RowMutation; //导入依赖的package包/类
private void testCompaction(String columnFamilyName, int insertsPerTable) throws IOException, ExecutionException, InterruptedException
{
    CompactionManager.instance.disableAutoCompaction();

    Keyspace keyspace = Keyspace.open("Keyspace1");
    ColumnFamilyStore store = keyspace.getColumnFamilyStore(columnFamilyName);

    Set<DecoratedKey> inserted = new HashSet<DecoratedKey>();
    for (int j = 0; j < insertsPerTable; j++) {
        DecoratedKey key = Util.dk(String.valueOf(j));
        RowMutation rm = new RowMutation("Keyspace1", key.key);
        rm.add(columnFamilyName, ByteBufferUtil.bytes("0"), ByteBufferUtil.EMPTY_BYTE_BUFFER, j);
        rm.apply();
        inserted.add(key);
        store.forceBlockingFlush();
        assertEquals(inserted.size(), Util.getRangeSlice(store).size());
    }
    CompactionManager.instance.performMaximal(store);
    assertEquals(1, store.getSSTables().size());
}
 

import org.apache.cassandra.db.RowMutation; //导入依赖的package包/类
private void mutate(List<RowMutation> cmds, org.apache.cassandra.db.ConsistencyLevel clvl) throws BackendException {
    try {
        schedule(DatabaseDescriptor.getRpcTimeout());
        try {
            if (atomicBatch) {
                StorageProxy.mutateAtomically(cmds, clvl);
            } else {
                StorageProxy.mutate(cmds, clvl);
            }
        } catch (RequestExecutionException e) {
            throw new TemporaryBackendException(e);
        } finally {
            release();
        }
    } catch (TimeoutException ex) {
        log.debug("Cassandra TimeoutException", ex);
        throw new TemporaryBackendException(ex);
    }
}
 

import org.apache.cassandra.db.RowMutation; //导入依赖的package包/类
/**
 * mark all of the column families we're modifying as dirty at this position
 */
private void markDirty(RowMutation rowMutation, ReplayPosition repPos)
{
    for (ColumnFamily columnFamily : rowMutation.getColumnFamilies())
    {
        // check for null cfm in case a cl write goes through after the cf is
        // defined but before a new segment is created.
        CFMetaData cfm = Schema.instance.getCFMetaData(columnFamily.id());
        if (cfm == null)
        {
            logger.error("Attempted to write commit log entry for unrecognized column family: " + columnFamily.id());
        }
        else
        {
            markCFDirty(cfm.cfId, repPos.position);
        }
    }
}
 

import org.apache.cassandra.db.RowMutation; //导入依赖的package包/类
public static void trace(final ByteBuffer sessionIdBytes, final String message, final int elapsed)
{
    final ByteBuffer eventId = ByteBuffer.wrap(UUIDGen.getTimeUUIDBytes());
    final String threadName = Thread.currentThread().getName();

    StageManager.getStage(Stage.TRACING).execute(new WrappedRunnable()
    {
        public void runMayThrow() throws Exception
        {
            CFMetaData cfMeta = CFMetaData.TraceEventsCf;
            ColumnFamily cf = ColumnFamily.create(cfMeta);
            Tracing.addColumn(cf, Tracing.buildName(cfMeta, eventId, ByteBufferUtil.bytes("source")), FBUtilities.getBroadcastAddress());
            Tracing.addColumn(cf, Tracing.buildName(cfMeta, eventId, ByteBufferUtil.bytes("thread")), threadName);
            if (elapsed >= 0)
                Tracing.addColumn(cf, Tracing.buildName(cfMeta, eventId, ByteBufferUtil.bytes("source_elapsed")), elapsed);
            Tracing.addColumn(cf, Tracing.buildName(cfMeta, eventId, ByteBufferUtil.bytes("activity")), message);
            RowMutation mutation = new RowMutation(Tracing.TRACE_KS, sessionIdBytes);
            mutation.add(cf);
            StorageProxy.mutate(Arrays.asList(mutation), ConsistencyLevel.ANY);
        }
    });
}
 

import org.apache.cassandra.db.RowMutation; //导入依赖的package包/类
private void testCompaction(String columnFamilyName, int insertsPerTable) throws IOException, ExecutionException, InterruptedException
{
    CompactionManager.instance.disableAutoCompaction();

    Table table = Table.open("Keyspace1");
    ColumnFamilyStore store = table.getColumnFamilyStore(columnFamilyName);

    Set<DecoratedKey> inserted = new HashSet<DecoratedKey>();
    for (int j = 0; j < insertsPerTable; j++) {
        DecoratedKey key = Util.dk(String.valueOf(j));
        RowMutation rm = new RowMutation("Keyspace1", key.key);
        rm.add(new QueryPath(columnFamilyName, null, ByteBufferUtil.bytes("0")), ByteBufferUtil.EMPTY_BYTE_BUFFER, j);
        rm.apply();
        inserted.add(key);
        store.forceBlockingFlush();
        assertEquals(inserted.size(), Util.getRangeSlice(store).size());
    }
    CompactionManager.instance.performMaximal(store);
    assertEquals(1, store.getSSTables().size());
}
 

import org.apache.cassandra.db.RowMutation; //导入依赖的package包/类
private static SSTableReader makeSSTable()
{
    Table t = Table.open("Keyspace1");
    for (int i = 0; i < 100; i++)
    {
        RowMutation rm = new RowMutation(t.name, ByteBufferUtil.bytes(Long.toString(System.nanoTime())));
        rm.add(new QueryPath("Standard1", null, ByteBufferUtil.bytes("cola")), ByteBufferUtil.bytes("value"), 0);
        rm.apply();
    }
    try
    {
        t.getColumnFamilyStore("Standard1").forceBlockingFlush();
        return t.getColumnFamilyStore("Standard1").getSSTables().iterator().next();
    }
    catch (Exception any)
    {
        throw new RuntimeException(any);
    }
}
 

import org.apache.cassandra.db.RowMutation; //导入依赖的package包/类
public static void trace(final ByteBuffer sessionIdBytes, final String message, final int elapsed)
{
    final ByteBuffer eventId = ByteBuffer.wrap(UUIDGen.getTimeUUIDBytes());
    final String threadName = Thread.currentThread().getName();

    StageManager.getStage(Stage.TRACING).execute(new WrappedRunnable()
    {
        public void runMayThrow()
        {
            CFMetaData cfMeta = CFMetaData.TraceEventsCf;
            ColumnFamily cf = ColumnFamily.create(cfMeta);
            Tracing.addColumn(cf, Tracing.buildName(cfMeta, eventId, ByteBufferUtil.bytes("source")), FBUtilities.getBroadcastAddress());
            Tracing.addColumn(cf, Tracing.buildName(cfMeta, eventId, ByteBufferUtil.bytes("thread")), threadName);
            if (elapsed >= 0)
                Tracing.addColumn(cf, Tracing.buildName(cfMeta, eventId, ByteBufferUtil.bytes("source_elapsed")), elapsed);
            Tracing.addColumn(cf, Tracing.buildName(cfMeta, eventId, ByteBufferUtil.bytes("activity")), message);
            Tracing.mutateWithCatch(new RowMutation(Tracing.TRACE_KS, sessionIdBytes, cf));
        }
    });
}
 

import org.apache.cassandra.db.RowMutation; //导入依赖的package包/类
/** {@inheritDoc} */
public void mutationForKey(RowMutation mutation, String keyspace, Long timestamp) throws InvalidRequestException
{
    CFMetaData metadata = validateColumnFamily(keyspace, columnFamily);
    QueryProcessor.validateKeyAlias(metadata, keyName);

    AbstractType comparator = metadata.getComparatorFor(null);

    if (columns.size() < 1) // No columns, delete the row
        mutation.delete(new QueryPath(columnFamily), (timestamp == null) ? getTimestamp() : timestamp);
    else    // Delete specific columns
    {
        for (Term column : columns)
        {
            ByteBuffer columnName = column.getByteBuffer(comparator);
            validateColumnName(columnName);
            mutation.delete(new QueryPath(columnFamily, null, columnName), (timestamp == null) ? getTimestamp() : timestamp);
        }
    }
}
 

import org.apache.cassandra.db.RowMutation; //导入依赖的package包/类
private void testCompaction(String columnFamilyName, int insertsPerTable) throws IOException, ExecutionException, InterruptedException
{
    CompactionManager.instance.disableAutoCompaction();

    Table table = Table.open("Keyspace1");
    ColumnFamilyStore store = table.getColumnFamilyStore(columnFamilyName);

    Set<DecoratedKey> inserted = new HashSet<DecoratedKey>();
    for (int j = 0; j < insertsPerTable; j++) {
        DecoratedKey key = Util.dk(String.valueOf(j));
        RowMutation rm = new RowMutation("Keyspace1", key.key);
        rm.add(new QueryPath(columnFamilyName, null, ByteBufferUtil.bytes("0")), ByteBufferUtil.EMPTY_BYTE_BUFFER, j);
        rm.apply();
        inserted.add(key);
        store.forceBlockingFlush();
        assertEquals(inserted.size(), Util.getRangeSlice(store).size());
    }
    CompactionManager.instance.performMajor(store);
    assertEquals(1, store.getSSTables().size());
}
 

import org.apache.cassandra.db.RowMutation; //导入依赖的package包/类
private static SSTable makeSSTable()
{
    Table t = Table.open("Keyspace1");
    for (int i = 0; i < 100; i++)
    {
        RowMutation rm = new RowMutation(t.name, ByteBufferUtil.bytes(Long.toString(System.nanoTime())));
        rm.add(new QueryPath("Standard1", null, ByteBufferUtil.bytes("cola")), ByteBufferUtil.bytes("value"), 0);
        try
        {
            rm.apply();
        }
        catch (IOException ex) 
        {
            throw new RuntimeException(ex);
        }
    }
    try
    {
        t.getColumnFamilyStore("Standard1").forceBlockingFlush();
        return t.getColumnFamilyStore("Standard1").getSSTables().iterator().next();
    }
    catch (Exception any)
    {
        throw new RuntimeException(any);
    }
}
 

import org.apache.cassandra.db.RowMutation; //导入依赖的package包/类
@Test
public void testOneRow() throws IOException, ExecutionException, InterruptedException, NoSuchAlgorithmException
{
    CompactionManager.instance.disableAutoCompaction();

    Table table = Table.open("Keyspace1");
    ColumnFamilyStore cfs = table.getColumnFamilyStore("Standard1");

    ByteBuffer key = ByteBufferUtil.bytes("k");
    RowMutation rm = new RowMutation("Keyspace1", key);
    rm.add(new QueryPath("Standard1", null, ByteBufferUtil.bytes("c")), ByteBufferUtil.EMPTY_BYTE_BUFFER, 0);
    rm.apply();
    cfs.forceBlockingFlush();

    assertBytes(cfs, Integer.MAX_VALUE);
    assertDigest(cfs, Integer.MAX_VALUE);
}