Are you incurring important cross Availability Zone visitors prices when operating an Apache Kafka consumer in containerized environments on Amazon Elastic Kubernetes Service (Amazon EKS) that devour information from Amazon Managed Streaming for Apache Kafka (Amazon MSK) matters?
If you happen to’re not accustomed to Apache Kafka’s rack consciousness characteristic, we strongly advocate beginning with the weblog put up on the way to Cut back community visitors prices of your Amazon MSK shoppers with rack consciousness for an in-depth rationalization of the characteristic and the way Amazon MSK helps it.
Though the answer described in that put up makes use of an Amazon Elastic Compute Cloud (Amazon EC2) occasion deployed in a single Availability Zone to devour messages from an Amazon MSK subject, fashionable cloud-native architectures demand extra dynamic and scalable approaches. Amazon EKS has emerged as a number one platform for deploying and managing distributed functions. The dynamic nature of Kubernetes introduces distinctive implementation challenges in comparison with static consumer deployments. On this put up, we stroll you thru an answer for implementing rack consciousness in shopper functions which are dynamically deployed throughout a number of Availability Zones utilizing Amazon EKS.
Right here’s a fast recap of some key Apache Kafka terminology from the referenced weblog. An Apache Kafka consumer shopper will register to learn towards a subject. A subject is the logical information construction that Apache Kafka organizes information into. A subject is segmented right into a single or many partitions. Partitions are the unit of parallelism in Apache Kafka. Amazon MSK supplies excessive availability by replicating every partition of a subject throughout brokers in numerous Availability Zones. As a result of there are replicas of every partition that reside throughout the completely different brokers that make up your MSK cluster, Amazon MSK additionally tracks whether or not a reproduction partition is in sync with the newest information for that partition. This implies there’s one partition that Amazon MSK acknowledges as containing probably the most up-to-date information, and this is named the chief partition. The gathering of replicated partitions is named in-sync replicas. This record of in-sync replicas is used internally when the cluster must elect a brand new chief partition if the present chief had been to turn into unavailable.
When shopper functions learn from a subject, the Apache Kafka protocol facilitates a community alternate to find out which dealer presently has the chief partition that the patron must learn from. Which means the patron might be instructed to learn from a dealer in a unique Availability Zone than itself, resulting in cross-zone visitors cost in your AWS account. To assist optimize this value, Amazon MSK helps the rack consciousness characteristic, utilizing which shoppers can ask an Amazon MSK cluster to supply a reproduction partition to learn from, throughout the identical Availability Zone because the consumer, even when it isn’t the present chief partition. The cluster accomplishes this by checking for an in-sync duplicate on a dealer throughout the identical Availability Zone as the patron.
The problem with Kafka shoppers on Amazon EKS
In Amazon EKS, the underlying items of computes are EC2 cases which are abstracted as Kubernetes nodes. The nodes are organized into node teams for ease of administration, scaling, and grouping of functions on sure EC2 occasion sorts. As a greatest observe for resilience, the nodes in a node group are unfold throughout a number of Availability Zones. Amazon EKS makes use of the underlying Amazon EC2 metadata concerning the Availability Zone that it’s positioned in, and it injects that data into the node’s metadata throughout node configuration. Particularly, the Availability Zone (AZ ID) is injected into the node metadata.
When an software is deployed in a Kubernetes Pod on Amazon EKS, it goes by means of a means of binding to a node that meets the pod’s necessities. As proven within the following diagram, whenever you deploy consumer functions on Amazon EKS, the pod for the applying could be sure to a node with out there capability in any Availability Zone. Additionally, the pod doesn’t mechanically inherit the Availability Zone data from the node that it’s sure to, a bit of knowledge vital for rack consciousness. The next structure diagram illustrates Kafka shoppers operating on Amazon EKS with out rack consciousness.

To set the consumer configuration for rack consciousness, the pod must know what Availability Zone it’s positioned in, dynamically, as it’s sure to a node. Throughout its lifecycle, the identical pod could be evicted from the node it was sure to beforehand and moved to a node in a unique Availability Zone, if the matching standards allow that. Making the pod conscious of its Availability Zone dynamically units the rack consciousness parameter consumer.rack in the course of the initialization of the applying container that’s encapsulated within the pod.
After rack consciousness is enabled on the MSK cluster, what occurs if the dealer in the identical Availability Zone because the consumer (hosted on Amazon EKS or elsewhere) turns into unavailable? The Apache Kafka protocol is designed to help a distributed information storage system. Assuming clients comply with the most effective observe of implementing a replication issue > 1, Apache Kafka can dynamically reroute the patron consumer to the following out there in-sync duplicate on a unique dealer. This resilience stays constant even after implementing nearest duplicate fetching, or rack consciousness. Enabling rack consciousness optimizes the networking alternate to desire a partition throughout the identical Availability Zone, but it surely doesn’t compromise the patron’s means to function if the closest duplicate is unavailable.
On this put up, we stroll you thru an instance of the way to use the Kubernetes metadata label, topology.k8s.aws/zone-id, assigned to every node by Amazon EKS, and use an open supply coverage engine, Kyverno, to deploy a coverage that mutates the pods which are within the binding state to dynamically inject the node’s AZ ID into the pod’s metadata as an annotation, as depicted within the following diagram. This annotation, in flip, is utilized by the container to create an surroundings variable that’s assigned the pod’s annotated AZ ID data. The surroundings variable is then used within the container postStart lifecycle hook to generate the Kafka consumer configuration file with rack consciousness setting. The next structure diagram illustrates Kafka shoppers operating on Amazon EKS with rack consciousness.

Answer Walkthrough
Stipulations
For this walkthrough, we use AWS CloudShell to run the scripts which are offered inline as you progress. For a clean expertise, earlier than getting began, be sure that to have kubectl and eksctl put in and configured within the AWS CloudShell surroundings, following the set up directions for Linux (amd64). Helm can also be required to be set up on AWS CloudShell, utilizing the directions for Linux.
Additionally, test if the envsubst software is put in in your CloudShell surroundings by invoking:
If the software isn’t put in, you possibly can set up it utilizing the command:
We additionally assume you have already got an MSK cluster deployed in an Amazon Digital Personal Cloud (VPC) in three Availability Zones with the title MSK-AZ-Conscious. On this walkthrough, we use AWS Id and Entry Administration (IAM) authentication for consumer entry management to the MSK cluster. If you happen to’re utilizing a cluster in your account with a unique title, exchange the cases of MSK-AZ-Conscious within the directions.
We comply with the identical MSK cluster configuration talked about within the Rack Consciousness weblog talked about beforehand, with some modifications. (Make sure you’ve set duplicate.selector.class = org.apache.kafka.frequent.duplicate.RackAwareReplicaSelector for the explanations mentioned there). In our configuration, we add one line: num.partitions = 6. Though not obligatory, this ensures that matters which are mechanically created may have a number of partitions to help clearer demonstrations in subsequent sections.
Lastly, we use the Amazon MSK Knowledge Generator with the next configuration:
Working the MSK Knowledge Generator with this configuration will mechanically create a six-partition subject named MSK-AZ-Conscious-Matter on our cluster for us, and it’ll push information to that subject. To comply with together with the walkthrough, we advocate and assume that you just deploy the MSK Knowledge Generator to create the subject and populate it with simulated information.
Create the EKS cluster
Step one is to put in an EKS cluster in the identical Amazon VPC subnets because the MSK cluster. You’ll be able to modify the title of the MSK cluster by altering that surroundings variable MSK_CLUSTER_NAME in case your cluster is created with a unique title than advised. You may also change the Amazon EKS cluster title by altering EKS_CLUSTER_NAME.
The surroundings variables that we outline listed here are used all through the walkthrough.
The final step is to replace the kubeconfig with an entry for the EKS cluster:
Subsequent, you must create an IAM coverage, MSK-AZ-Conscious-Coverage, to permit entry from the Amazon EKS pods to the MSK cluster. Be aware right here that we’re utilizing MSK-AZ-Conscious because the cluster title.
Create a file, msk-az-aware-policy.json, with the IAM coverage template:
To create the IAM coverage, use the next command. It first replaces the placeholders within the coverage file with values from related surroundings variables, after which creates the IAM coverage:
Configure EKS Pod Id
Amazon EKS Pod Id gives a simplified expertise for acquiring IAM permissions for pods on Amazon EKS. This requires putting in an add-on Amazon EKS Pod Id Agent to the EKS cluster:
Affirm that the add-on has been put in and its standing is ACTIVE and that the standing of all of the pods related to the add-on is Working.
After you’ve put in the add-on, you must create a pod id affiliation between a Kubernetes service account and the IAM coverage created earlier:
Set up Kyverno
Kyverno is an open supply coverage engine for Kubernetes that permits for validation, mutation, and era of Kubernetes sources utilizing insurance policies written in YAML, thus simplifying the enforcement of safety and compliance necessities. You must set up Kyverno to dynamically inject metadata into the Amazon EKS pods as they enter the binding state to tell them of Availability Zone ID.
In AWS CloudShell, create a file named kyverno-values.yaml. This file defines the Kubernetes RBAC permissions for Kyverno’s Admission Controller to learn Amazon EKS node metadata as a result of the default Kyverno (v. 1.13 onwards) settings don’t permit this:
After this file is created, you possibly can set up Kyverno utilizing helm and offering the values file created within the earlier step:
Beginning with Kyverno v 1.13, the Admission Controller is configured to disregard the AdmissionReview requests for pods in binding state. This must be modified by modifying the Kyverno ConfigMap:
The kubectl edit command makes use of the default editor configured in your surroundings (in our case Linux VIM).
This may open the ConfigMap in a textual content editor.
As highlighted within the following screenshot, [Pod/binding,*,*] ought to be faraway from the resourceFilters discipline for the Kyverno Admission Controller to course of AdmissionReview requests for pods in binding state.

If Linux VIM is your default editor, you possibly can delete the entry utilizing VIM command 18x, which means delete (or lower) 18 characters from the present cursor place. Save the modified configuration utilizing the VIM command :wq, which means write (or save) the file and give up.
After deleting, the resourceFilters discipline ought to look just like the next screenshot.

When you have a unique editor configured in your surroundings, comply with the suitable steps to realize an identical final result.
Configure Kyverno coverage
You must configure the coverage that can make the pods rack conscious. This coverage is customized from the advised method within the Kyverno weblog put up, Assigning Node Metadata to Pods. Create a brand new file with the title kyverno-inject-node-az-id.yaml:
It instructs Kyverno to observe for pods in binding state. After Kyverno receives the AdmissionReview request for a pod, it units the variable node to the title of the node to which the pod is being sure. It additionally units one other variable node_az_id to the Availability Zone ID by calling the Kubernetes API /api/v1/nodes/node to get the node metadata label topology.k8s.aws/zone-id. Lastly, it defines a mutate rule to inject the obtained AZ ID into the pod’s metadata as an annotation node_az_id.
After you’ve created the file, apply the coverage utilizing the next command:
Deploy a pod with out rack consciousness
Now let’s visualize the issue assertion. To do that, connect with one of many EKS pods and test the way it interacts with the MSK cluster whenever you run a Kafka shopper from the pod.
First, get the bootstrap string of the MSK cluster. Search for the Amazon Useful resource Names (ARNs) of the MSK cluster:
Utilizing the cluster ARN, you may get the bootstrap string with the next command:
Create a brand new file named kafka-no-az.yaml:
This pod manifest doesn’t make use of the Availability Zone ID injected into the metadata annotation and therefore doesn’t add consumer.rack to the consumer.properties configuration.
Deploy the pods utilizing the next command:
Run the next command to substantiate that the pods have been deployed and are within the Working state:
Choose a pod id from the output of the earlier command, and connect with it utilizing:
Run the Kafka shopper:
This command will dump all of the ensuing logs into the file, non-rack-aware-consumer.log. There’s a variety of data in these logs, and we encourage you to open them and take a deeper look. Subsequent, study the EKS pod in motion. To do that, run the next command to tail the file to view fetch request outcomes to the MSK cluster. You’ll discover a handful of significant logs to assessment as the patron entry varied partitions of the Kafka subject:
Observe your log output, which ought to look just like the next:
You’ve now related to a selected pod within the EKS cluster and run a Kafka shopper to learn from the MSK subject with out rack consciousness. Keep in mind that this pod is operating inside a single Availability Zone.
Reviewing the log output, you discover rack: values as use1-az2, use1-az4, and use1-az6 because the pod makes calls to completely different partitions of the subject. These rack values characterize the Availability Zone IDs that our brokers are operating inside. Which means our EKS pod is creating networking connections to brokers throughout three completely different Availability Zones, which might be accruing networking prices in our account.
Additionally discover that you haven’t any strategy to test which node, and subsequently Availability Zone, this EKS pod is operating in. You’ll be able to observe within the logs that it’s calling to MSK brokers in numerous Availability Zones, however there isn’t a strategy to know which dealer is in the identical Availability Zone because the EKS pod you’ve related to. Delete the deployment whenever you’re carried out:
Deploy a pod with rack consciousness
Now that you’ve got skilled the patron habits with out rack consciousness, you must inject the Availability Zone ID to make your pods rack-aware.
Create a brand new file named kafka-az-aware.yaml:
As you possibly can observe, the pod manifest defines an surroundings variable NODE_AZ_ID, assigning it the worth from the pod’s personal metadata annotation node_az_id that was injected by Kyverno. The manifest then makes use of the pod’s postStart lifecycle script so as to add consumer.rack into the consumer.properties configuration, setting it equal to the worth within the surroundings variable NODE_AZ_ID.
Deploy the pods utilizing the next command:
Run the next command to substantiate that the pods have been deployed and are within the Working state:
Confirm that Availability Zone Ids have been injected into the pods
Your output ought to look just like:
Or:

Choose a pod id from the output of the get pods command and shell-in to it.
The output of the get $pod command matches the order of outcomes from the get pods command. This matching will assist you perceive what Availability Zone your pod is operating in so you possibly can evaluate it to log outputs later.
After you’ve related to your pod, run the Kafka shopper:
Much like earlier than, this command will dump all of the ensuing logs into the file, rack-aware-consumer.log. You create a brand new file so there’s no overlap between the Kafka shoppers you’ve run. There’s a variety of data in these logs, and we encourage you to open them and take a deeper look. If you wish to see the rack consciousness of your EKS pod in motion, run the next command to tail the file to view fetch request outcomes to the MSK cluster. You’ll be able to observe a handful of significant logs to assessment right here as the patron entry varied partitions of the Kafka subject:
Observe your log output, which ought to look just like the next:
For every log line, now you can observe two rack: values. The primary rack: worth reveals the present chief, the second rack: reveals the rack that’s getting used to fetch messages.
For instance, have a look at MSK-AZ-Conscious-Matter-5. The chief is recognized as rack: use1-az4, however the fetch request is shipped to use1-az6 as indicated by to node b-2.mskazaware.hxrzlh.c6.kafka.us-east-1.amazonaws.com:9098 (id: 2 rack: use1-az6) (org.apache.kafka.shoppers.shopper.internals.AbstractFetch)
You’ll discover one thing comparable in all different log strains. The fetch is all the time to the dealer in use1-az6, which maps to our expectation, given the pod we related to was on this Availability Zone.
Congratulations! You’re consuming from the closest duplicate on Amazon EKS.
Clear Up
Delete the deployment when completed:
To delete the EKS Pod Id affiliation:
To delete the IAM coverage:
To delete the EKS cluster:
If you happen to adopted together with this put up utilizing the Amazon MSK Knowledge Generator, you should definitely delete your deployment so it’s now not making an attempt to generate and ship information after you delete the remainder of your sources.
Clear up will depend upon which deployment possibility you used. To learn extra concerning the deployment choices and the sources created for the Amazon MSK Knowledge Generator, consult with Getting Began within the GitHub repository.
Creating an MSK cluster was a prerequisite of this put up, and in the event you’d like to wash up the MSK cluster as effectively, you should use the next command:
aws kafka delete-cluster --cluster-arn "${MSK_CLUSTER_ARN}"
There is no such thing as a extra value to utilizing AWS CloudShell, however in the event you’d prefer to delete your shell, consult with the Delete a shell session house listing within the AWS CloudShell Person Information.
Conclusion
Apache Kafka nearest duplicate fetching, or rack consciousness, is a strategic cost-optimization approach. By implementing it for Amazon MSK shoppers on Amazon EKS, you possibly can considerably scale back cross-zone visitors prices whereas sustaining strong, distributed streaming architectures. Open supply instruments similar to Kyverno can simplify complicated configuration challenges and drive significant financial savings.The answer we’ve demonstrated supplies a robust, repeatable method to dynamically injecting Availability Zone data into Kubernetes pods, optimize Kafka shopper routing, and reduce scale back switch prices.
Extra sources
To be taught extra about rack consciousness with Amazon MSK, consult with Cut back community visitors prices of your Amazon MSK shoppers with rack consciousness.
Concerning the authors
Austin Groeneveld is a Streaming Specialist Options Architect at Amazon Internet Companies (AWS), primarily based within the San Francisco Bay Space. On this function, Austin is keen about serving to clients speed up insights from their information utilizing the AWS platform. He’s significantly fascinated by the rising function that information streaming performs in driving innovation within the information analytics house. Exterior of his work at AWS, Austin enjoys watching and enjoying soccer, touring, and spending high quality time along with his household.
Farooq Ashraf is a Senior Options Architect at AWS, specializing in SaaS, Generative AI, and MLOps. He’s keen about mixing multi-tenant SaaS ideas with Cloud providers to innovate scalable options for the digital enterprise, and has a number of weblog posts, and workshops to his credit score.
