Partition assignment in Kafka - Mini Project: Build & Apply

Partition assignment

📖 Scenario: You are managing a Kafka topic with multiple partitions. You want to assign partitions to consumers in a group so that each consumer handles specific partitions.

🎯 Goal: Build a simple Kafka consumer group partition assignment simulation using Python dictionaries. You will create a data structure for partitions, configure consumers, assign partitions to consumers, and display the assignments.

📋 What You'll Learn

Create a dictionary with partitions and their IDs

Create a list of consumer IDs

Assign partitions to consumers evenly

Print the final partition assignment

💡 Why This Matters

🌍 Real World

Kafka partition assignment is important to distribute workload evenly among consumers in a group, ensuring efficient data processing.

💼 Career

Understanding partition assignment helps in roles like data engineering, backend development, and system architecture where Kafka is used for streaming data.

1

Create partitions dictionary

Create a dictionary called partitions with these exact entries: 0: 'partition_0', 1: 'partition_1', 2: 'partition_2', 3: 'partition_3', 4: 'partition_4'

Kafka

# Your code here

Hint

Use curly braces to create a dictionary with keys 0 to 4 and values as 'partition_0' to 'partition_4'.

2

Create consumers list

Create a list called consumers with these exact string values: 'consumer_1', 'consumer_2', 'consumer_3'

Kafka

partitions = {0: 'partition_0', 1: 'partition_1', 2: 'partition_2', 3: 'partition_3', 4: 'partition_4'}
# Your code here

Hint

Use square brackets to create a list with the three consumer strings.

3

Assign partitions to consumers

Create a dictionary called assignment that assigns partitions to consumers evenly. Use a for loop with variables partition_id and partition_name to iterate over partitions.items(). Assign each partition to a consumer by cycling through consumers using the modulo operator. Store the result as a list of partition names for each consumer.

Kafka

partitions = {0: 'partition_0', 1: 'partition_1', 2: 'partition_2', 3: 'partition_3', 4: 'partition_4'}
consumers = ['consumer_1', 'consumer_2', 'consumer_3']
# Your code here

Hint

Initialize assignment as a dictionary with empty lists for each consumer. Use modulo to cycle through consumers.

4

Print partition assignment

Write a print statement to display the assignment dictionary.

Kafka

partitions = {0: 'partition_0', 1: 'partition_1', 2: 'partition_2', 3: 'partition_3', 4: 'partition_4'}
consumers = ['consumer_1', 'consumer_2', 'consumer_3']
assignment = {consumer: [] for consumer in consumers}
for partition_id, partition_name in partitions.items():
    consumer = consumers[partition_id % len(consumers)]
    assignment[consumer].append(partition_name)
# Your code here

Hint

Use print(assignment) to show the final dictionary.

Practice

1.

What is the main purpose of partition assignment in Kafka?

easy

A. To encrypt messages before sending

B. To split topic data among consumers to share the workload

C. To compress data for faster transmission

D. To create new topics automatically

Solution

1. Step 1: Understand partition assignment role

   Partition assignment divides topic partitions among consumers to distribute processing.

2. Step 2: Compare options

   Only To split topic data among consumers to share the workload describes splitting data among consumers; others describe unrelated tasks.

3. Final Answer:

   To split topic data among consumers to share the workload -> Option B

4. Quick Check:

   Partition assignment = workload sharing [OK]

Hint: Partition assignment shares topic data among consumers [OK]

Common Mistakes:

• Confusing partition assignment with encryption
• Thinking it compresses data
• Assuming it creates topics

2.

Which of the following is a valid Kafka partition assignment strategy?

easy

A. RoundRobin

B. HashMap

C. QuickSort

D. BinarySearch

Solution

1. Step 1: Recall Kafka partition assignment strategies

   Kafka supports strategies like Range, RoundRobin, and Sticky.

2. Step 2: Identify valid strategy

   RoundRobin is a known Kafka strategy; others are unrelated algorithms.

3. Final Answer:

   RoundRobin -> Option A

4. Quick Check:

   RoundRobin = valid strategy [OK]

Hint: Remember Kafka uses RoundRobin, Range, Sticky strategies [OK]

Common Mistakes:

• Confusing general algorithms with Kafka strategies
• Choosing unrelated computer science terms
• Ignoring Kafka-specific terms

3.

Given a topic with 4 partitions and 2 consumers using the Range strategy, how are partitions assigned?

Partitions: 0, 1, 2, 3
Consumers: C1, C2
Strategy: Range

medium

A. Partitions are assigned randomly

B. C1 gets partitions 0 and 2; C2 gets partitions 1 and 3

C. C1 gets all partitions; C2 gets none

D. C1 gets partitions 0 and 1; C2 gets partitions 2 and 3

Solution

1. Step 1: Understand Range strategy behavior

   Range assigns consecutive partitions to each consumer in order.

2. Step 2: Apply Range to 4 partitions and 2 consumers

   Partitions 0 and 1 go to C1; partitions 2 and 3 go to C2.

3. Final Answer:

   C1 gets partitions 0 and 1; C2 gets partitions 2 and 3 -> Option D

4. Quick Check:

   Range = consecutive partition blocks [OK]

Hint: Range assigns consecutive partitions to consumers [OK]

Common Mistakes:

• Mixing RoundRobin with Range strategy
• Assuming random assignment
• Giving all partitions to one consumer

4.

Identify the error in this partition assignment code snippet using the Sticky strategy:

assignor = StickyAssignor()
assignor.assign(partitions=[0,1,2], consumers=["C1"])

medium

A. Consumers list must be integers

B. Partitions list should be empty

C. StickyAssignor requires at least two consumers for assignment

D. StickyAssignor does not support partition assignment

Solution

1. Step 1: Understand StickyAssignor requirements

   Sticky strategy balances partitions among multiple consumers to minimize movement.

2. Step 2: Check given consumers list

   Only one consumer is provided, so StickyAssignor cannot balance partitions.

3. Final Answer:

   StickyAssignor requires at least two consumers for assignment -> Option C

4. Quick Check:

   Sticky needs multiple consumers [OK]

Hint: Sticky strategy needs multiple consumers to balance partitions [OK]

Common Mistakes:

• Assuming single consumer works with Sticky
• Confusing consumer data types
• Thinking Sticky doesn't assign partitions

5.

You have 5 consumers and a topic with 10 partitions. You want to maintain processing order per partition and balance load evenly. Which partition assignment strategy should you choose?

hard

A. Sticky, because it balances partitions while minimizing reassignment

B. RoundRobin, because it distributes partitions evenly across consumers

C. Range, because it assigns consecutive partitions to consumers

D. Random, because it assigns partitions unpredictably

Solution

1. Step 1: Analyze requirements

   Need to keep processing order per partition and balance load evenly.

2. Step 2: Compare strategies

   Range assigns consecutive partitions but may imbalance load; RoundRobin balances evenly but can cause frequent partition moves; Sticky balances load and minimizes partition reassignment, preserving order.

3. Step 3: Choose best fit

   Sticky strategy best meets both order and load balance requirements.

4. Final Answer:

   Sticky, because it balances partitions while minimizing reassignment -> Option A

5. Quick Check:

   Sticky = balanced load + order preservation [OK]

Hint: Sticky balances load and keeps partition order [OK]

Common Mistakes:

• Choosing Range despite load imbalance
• Picking RoundRobin ignoring partition order
• Selecting random assignment which is unpredictable