Examples with BrokerRegistration org.apache.kafka.metadata.BrokerRegistration used on opensource projects

Example 1 with BrokerRegistration

use of org.apache.kafka.metadata.BrokerRegistration in project kafka by apache.

the class ReplicationControlManager method handleBrokerFenced.

/**
 * Generate the appropriate records to handle a broker being fenced.
 *
 * First, we remove this broker from any non-singleton ISR. Then we generate a
 * FenceBrokerRecord.
 *
 * @param brokerId      The broker id.
 * @param records       The record list to append to.
 */
void handleBrokerFenced(int brokerId, List<ApiMessageAndVersion> records) {
    BrokerRegistration brokerRegistration = clusterControl.brokerRegistrations().get(brokerId);
    if (brokerRegistration == null) {
        throw new RuntimeException("Can't find broker registration for broker " + brokerId);
    }
    generateLeaderAndIsrUpdates("handleBrokerFenced", brokerId, NO_LEADER, records, brokersToIsrs.partitionsWithBrokerInIsr(brokerId));
    records.add(new ApiMessageAndVersion(new FenceBrokerRecord().setId(brokerId).setEpoch(brokerRegistration.epoch()), FENCE_BROKER_RECORD.highestSupportedVersion()));
}

Example 2 with BrokerRegistration

use of org.apache.kafka.metadata.BrokerRegistration in project kafka by apache.

the class ClusterDelta method replay.

public void replay(FenceBrokerRecord record) {
    BrokerRegistration broker = getBrokerOrThrow(record.id(), record.epoch(), "fence");
    changedBrokers.put(record.id(), Optional.of(broker.cloneWithFencing(true)));
}

Example 3 with BrokerRegistration

use of org.apache.kafka.metadata.BrokerRegistration in project kafka by apache.

the class ClusterDelta method replay.

public void replay(UnfenceBrokerRecord record) {
    BrokerRegistration broker = getBrokerOrThrow(record.id(), record.epoch(), "unfence");
    changedBrokers.put(record.id(), Optional.of(broker.cloneWithFencing(false)));
}

Example 4 with BrokerRegistration

use of org.apache.kafka.metadata.BrokerRegistration in project kafka by apache.

the class ClusterControlManagerTest method testUnregister.

@Test
public void testUnregister() throws Exception {
    RegisterBrokerRecord brokerRecord = new RegisterBrokerRecord().setBrokerId(1).setBrokerEpoch(100).setIncarnationId(Uuid.fromString("fPZv1VBsRFmnlRvmGcOW9w")).setRack("arack");
    brokerRecord.endPoints().add(new BrokerEndpoint().setSecurityProtocol(SecurityProtocol.PLAINTEXT.id).setPort((short) 9092).setName("PLAINTEXT").setHost("example.com"));
    SnapshotRegistry snapshotRegistry = new SnapshotRegistry(new LogContext());
    ClusterControlManager clusterControl = new ClusterControlManager(new LogContext(), Uuid.randomUuid().toString(), new MockTime(0, 0, 0), snapshotRegistry, 1000, new StripedReplicaPlacer(new Random()), new MockControllerMetrics());
    clusterControl.activate();
    clusterControl.replay(brokerRecord);
    assertEquals(new BrokerRegistration(1, 100, Uuid.fromString("fPZv1VBsRFmnlRvmGcOW9w"), Collections.singletonMap("PLAINTEXT", new Endpoint("PLAINTEXT", SecurityProtocol.PLAINTEXT, "example.com", 9092)), Collections.emptyMap(), Optional.of("arack"), true), clusterControl.brokerRegistrations().get(1));
    UnregisterBrokerRecord unregisterRecord = new UnregisterBrokerRecord().setBrokerId(1).setBrokerEpoch(100);
    clusterControl.replay(unregisterRecord);
    assertFalse(clusterControl.brokerRegistrations().containsKey(1));
}

Example 5 with BrokerRegistration

use of org.apache.kafka.metadata.BrokerRegistration in project kafka by apache.

the class ReplicationControlManager method unregisterBroker.

public ControllerResult<Void> unregisterBroker(int brokerId) {
    BrokerRegistration registration = clusterControl.brokerRegistrations().get(brokerId);
    if (registration == null) {
        throw new BrokerIdNotRegisteredException("Broker ID " + brokerId + " is not currently registered");
    }
    List<ApiMessageAndVersion> records = new ArrayList<>();
    handleBrokerUnregistered(brokerId, registration.epoch(), records);
    return ControllerResult.of(records, null);
}