use of org.apache.kafka.common.metadata.ProducerIdsRecord in project kafka by apache.
the class ProducerIdControlManagerTest method testSnapshotIterator.
@Test
public void testSnapshotIterator() {
ProducerIdsBlock range = null;
for (int i = 0; i < 100; i++) {
range = generateProducerIds(producerIdControlManager, i % 4, 100);
}
Iterator<List<ApiMessageAndVersion>> snapshotIterator = producerIdControlManager.iterator(Long.MAX_VALUE);
assertTrue(snapshotIterator.hasNext());
List<ApiMessageAndVersion> batch = snapshotIterator.next();
assertEquals(1, batch.size(), "Producer IDs record batch should only contain a single record");
assertEquals(range.firstProducerId() + range.size(), ((ProducerIdsRecord) batch.get(0).message()).nextProducerId());
assertFalse(snapshotIterator.hasNext(), "Producer IDs iterator should only contain a single batch");
ProducerIdControlManager newProducerIdManager = new ProducerIdControlManager(clusterControl, snapshotRegistry);
snapshotIterator = producerIdControlManager.iterator(Long.MAX_VALUE);
while (snapshotIterator.hasNext()) {
snapshotIterator.next().forEach(message -> newProducerIdManager.replay((ProducerIdsRecord) message.message()));
}
// Verify that after reloading state from this "snapshot", we don't produce any overlapping IDs
long lastProducerID = range.firstProducerId() + range.size() - 1;
range = generateProducerIds(producerIdControlManager, 1, 100);
assertTrue(range.firstProducerId() > lastProducerID);
}
use of org.apache.kafka.common.metadata.ProducerIdsRecord in project kafka by apache.
the class ProducerIdControlManagerTest method testInitialResult.
@Test
public void testInitialResult() {
ControllerResult<ProducerIdsBlock> result = producerIdControlManager.generateNextProducerId(1, 100);
assertEquals(0, result.response().firstProducerId());
assertEquals(1000, result.response().size());
ProducerIdsRecord record = (ProducerIdsRecord) result.records().get(0).message();
assertEquals(1000, record.nextProducerId());
}
use of org.apache.kafka.common.metadata.ProducerIdsRecord in project kafka by apache.
the class ProducerIdControlManagerTest method testMonotonic.
@Test
public void testMonotonic() {
producerIdControlManager.replay(new ProducerIdsRecord().setBrokerId(1).setBrokerEpoch(100).setNextProducerId(42));
ProducerIdsBlock range = producerIdControlManager.generateNextProducerId(1, 100).response();
assertEquals(42, range.firstProducerId());
// Can't go backwards in Producer IDs
assertThrows(RuntimeException.class, () -> {
producerIdControlManager.replay(new ProducerIdsRecord().setBrokerId(1).setBrokerEpoch(100).setNextProducerId(40));
}, "Producer ID range must only increase");
range = producerIdControlManager.generateNextProducerId(1, 100).response();
assertEquals(42, range.firstProducerId());
// Gaps in the ID range are okay.
producerIdControlManager.replay(new ProducerIdsRecord().setBrokerId(1).setBrokerEpoch(100).setNextProducerId(50));
range = producerIdControlManager.generateNextProducerId(1, 100).response();
assertEquals(50, range.firstProducerId());
}
use of org.apache.kafka.common.metadata.ProducerIdsRecord in project kafka by apache.
the class ProducerIdControlManager method generateNextProducerId.
ControllerResult<ProducerIdsBlock> generateNextProducerId(int brokerId, long brokerEpoch) {
clusterControlManager.checkBrokerEpoch(brokerId, brokerEpoch);
long firstProducerIdInBlock = nextProducerId.get();
if (firstProducerIdInBlock > Long.MAX_VALUE - ProducerIdsBlock.PRODUCER_ID_BLOCK_SIZE) {
throw new UnknownServerException("Exhausted all producerIds as the next block's end producerId " + "has exceeded the int64 type limit");
}
ProducerIdsBlock block = new ProducerIdsBlock(brokerId, firstProducerIdInBlock, ProducerIdsBlock.PRODUCER_ID_BLOCK_SIZE);
long newNextProducerId = block.nextBlockFirstId();
ProducerIdsRecord record = new ProducerIdsRecord().setNextProducerId(newNextProducerId).setBrokerId(brokerId).setBrokerEpoch(brokerEpoch);
return ControllerResult.of(Collections.singletonList(new ApiMessageAndVersion(record, (short) 0)), block);
}
use of org.apache.kafka.common.metadata.ProducerIdsRecord in project kafka by apache.
the class ProducerIdControlManager method iterator.
Iterator<List<ApiMessageAndVersion>> iterator(long epoch) {
List<ApiMessageAndVersion> records = new ArrayList<>(1);
long producerId = nextProducerId.get(epoch);
if (producerId > 0) {
records.add(new ApiMessageAndVersion(new ProducerIdsRecord().setNextProducerId(producerId).setBrokerId(0).setBrokerEpoch(0L), (short) 0));
}
return Collections.singleton(records).iterator();
}
Aggregations