@Override public void beforeMigration(PartitionMigrationEvent event) { if (isPrimaryReplicaMigrationEvent(event)) { ownerMigrationsStarted.incrementAndGet(); } migrationAwareService.beforeMigration(event); }
@Override public void commitMigration(PartitionMigrationEvent event) { try { migrationAwareService.commitMigration(event); } finally { if (isPrimaryReplicaMigrationEvent(event)) { int completed = ownerMigrationsCompleted.incrementAndGet(); assert completed <= ownerMigrationsStarted.get(); } } }
@Override public void rollbackMigration(PartitionMigrationEvent event) { try { migrationAwareService.rollbackMigration(event); } finally { if (isPrimaryReplicaMigrationEvent(event)) { int completed = ownerMigrationsCompleted.incrementAndGet(); assert completed <= ownerMigrationsStarted.get(); } } }
@Override public void beforeMigration(PartitionMigrationEvent event) { if (isPrimaryReplicaMigrationEvent(event)) { ownerMigrationsStarted.incrementAndGet(); } migrationAwareService.beforeMigration(event); }
@Override public void commitMigration(PartitionMigrationEvent event) { try { migrationAwareService.commitMigration(event); } finally { if (isPrimaryReplicaMigrationEvent(event)) { int completed = ownerMigrationsCompleted.incrementAndGet(); assert completed <= ownerMigrationsStarted.get(); } } }
@Override public void rollbackMigration(PartitionMigrationEvent event) { try { migrationAwareService.rollbackMigration(event); } finally { if (isPrimaryReplicaMigrationEvent(event)) { int completed = ownerMigrationsCompleted.incrementAndGet(); assert completed <= ownerMigrationsStarted.get(); } } }