Examples with PendingStats org.apache.cassandra.repair.consistent.admin.PendingStats used on opensource projects

Example 1 with PendingStats

use of org.apache.cassandra.repair.consistent.admin.PendingStats in project cassandra by apache.

the class PendingRepairStatTest method pendingRepairStats.

@Test
public void pendingRepairStats() {
    InstrumentedLocalSessions sessions = new InstrumentedLocalSessions();
    sessions.start();
    cfs.disableAutoCompaction();
    SSTableReader sstable1 = createSSTable(0, 10);
    SSTableReader sstable2 = createSSTable(10, 10);
    SSTableReader sstable3 = createSSTable(10, 20);
    LocalSession session1 = createSession();
    sessions.putSessionUnsafe(session1);
    LocalSession session2 = createSession();
    sessions.putSessionUnsafe(session2);
    PendingStats stats;
    stats = sessions.getPendingStats(cfm.id, Collections.singleton(FULL_RANGE));
    Assert.assertEquals(0, stats.total.numSSTables);
    // set all sstables to pending
    mutateRepaired(sstable1, UNREPAIRED_SSTABLE, session1.sessionID);
    mutateRepaired(sstable2, UNREPAIRED_SSTABLE, session2.sessionID);
    mutateRepaired(sstable3, UNREPAIRED_SSTABLE, session2.sessionID);
    stats = sessions.getPendingStats(cfm.id, Collections.singleton(FULL_RANGE));
    Assert.assertEquals(Sets.newHashSet(session1.sessionID, session2.sessionID), stats.total.sessions);
    Assert.assertEquals(3, stats.total.numSSTables);
    Assert.assertEquals(3, stats.pending.numSSTables);
    Assert.assertEquals(0, stats.failed.numSSTables);
    Assert.assertEquals(0, stats.finalized.numSSTables);
    // set the 2 sessions to failed and finalized
    session1.setState(FAILED);
    sessions.save(session1);
    session2.setState(FINALIZED);
    sessions.save(session2);
    stats = sessions.getPendingStats(cfm.id, Collections.singleton(FULL_RANGE));
    Assert.assertEquals(3, stats.total.numSSTables);
    Assert.assertEquals(0, stats.pending.numSSTables);
    Assert.assertEquals(1, stats.failed.numSSTables);
    Assert.assertEquals(2, stats.finalized.numSSTables);
    // remove sstables from pending sets
    mutateRepaired(sstable1, UNREPAIRED_SSTABLE, NO_PENDING_REPAIR);
    mutateRepaired(sstable2, session2.repairedAt, NO_PENDING_REPAIR);
    mutateRepaired(sstable3, session2.repairedAt, NO_PENDING_REPAIR);
    stats = sessions.getPendingStats(cfm.id, Collections.singleton(FULL_RANGE));
    Assert.assertTrue(stats.total.sessions.isEmpty());
}

Example 2 with PendingStats

use of org.apache.cassandra.repair.consistent.admin.PendingStats in project cassandra by apache.

the class LocalSessions method getPendingStats.

public PendingStats getPendingStats(TableId tid, Collection<Range<Token>> ranges) {
    ColumnFamilyStore cfs = Schema.instance.getColumnFamilyStoreInstance(tid);
    Preconditions.checkArgument(cfs != null);
    PendingStat.Builder pending = new PendingStat.Builder();
    PendingStat.Builder finalized = new PendingStat.Builder();
    PendingStat.Builder failed = new PendingStat.Builder();
    Map<UUID, PendingStat> stats = cfs.getPendingRepairStats();
    for (Map.Entry<UUID, PendingStat> entry : stats.entrySet()) {
        UUID sessionID = entry.getKey();
        PendingStat stat = entry.getValue();
        Verify.verify(sessionID.equals(Iterables.getOnlyElement(stat.sessions)));
        LocalSession session = sessions.get(sessionID);
        Verify.verifyNotNull(session);
        if (!Iterables.any(ranges, r -> r.intersects(session.ranges)))
            continue;
        switch(session.getState()) {
            case FINALIZED:
                finalized.addStat(stat);
                break;
            case FAILED:
                failed.addStat(stat);
                break;
            default:
                pending.addStat(stat);
        }
    }
    return new PendingStats(cfs.keyspace.getName(), cfs.name, pending.build(), finalized.build(), failed.build());
}

Example 3 with PendingStats

use of org.apache.cassandra.repair.consistent.admin.PendingStats in project cassandra by apache.

the class ActiveRepairService method getPendingStats.

@Override
public List<CompositeData> getPendingStats(List<String> schemaArgs, String rangeString) {
    List<CompositeData> stats = new ArrayList<>();
    Collection<Range<Token>> userRanges = rangeString != null ? RepairOption.parseRanges(rangeString, DatabaseDescriptor.getPartitioner()) : null;
    for (ColumnFamilyStore cfs : SchemaArgsParser.parse(schemaArgs)) {
        String keyspace = cfs.keyspace.getName();
        Collection<Range<Token>> ranges = userRanges != null ? userRanges : StorageService.instance.getLocalReplicas(keyspace).ranges();
        PendingStats cfStats = consistent.local.getPendingStats(cfs.metadata().id, ranges);
        stats.add(cfStats.toComposite());
    }
    return stats;
}