aws_msk_iam_sasl_signer_kafka_python_producer_example.py

1+import boto3
2+from kafka import KafkaProducer
3+from aws_msk_iam_sasl_signer import MSKAuthTokenProvider
4+
5+class MSKTokenProvider():
6+    def __call__(self):
7+        region = boto3.session.Session().region_name
8+        token, _ = MSKAuthTokenProvider.generate_auth_token(region)
9+        return token
10+
11+tp = MSKTokenProvider()
12+
13+producer = KafkaProducer(
14+    bootstrap_servers='your_broker_address',
15+    security_protocol='SASL_SSL',
16+    sasl_mechanism='OAUTHBEARER',
17+    sasl_oauth_token_provider=tp,
18+)
19+
20+producer.send('mskiamexample', b'test message')