ApacheBeam和Python。尝试对ApacheKafka进行SASL_SSLOAUTHBEARER验证发生错误。
创始人
2024-09-05 11:30:12
0次

一种可能的解决方法是在 Apache Beam 中使用 KafkaPython 库,并配置正确的安全协议和认证机制。下面是一个代码示例:

import apache_beam as beam
from apache_beam.transforms import window
from kafka import KafkaProducer
from kafka import KafkaConsumer
from kafka.errors import KafkaError

class KafkaCustomSink(beam.PTransform):
    """A custom Apache Beam sink for producing messages to Apache Kafka"""

    def __init__(self, bootstrap_servers, topic, security_protocol='SASL_SSL', sasl_mechanism='OAUTHBEARER', sasl_oauth_token_provider=pks_utils.oauth2_provider_from_environment()):
        self.bootstrap_servers = bootstrap_servers
        self.topic = topic
        self.security_protocol = security_protocol
        self.sasl_mechanism = sasl_mechanism
        self.sasl_oauth_token_provider = sasl_oauth_token_provider

    # In this transform, the elements of the PCollection are simply
    # sent as messages to the Kafka topic.
    def expand(self, pcoll):
        producer = KafkaProducer(
            bootstrap_servers=self.bootstrap_servers,
            security_protocol=self.security_protocol,
            sasl_mechanism=self.sasl_mechanism,
            sasl_oauth_token_provider=self.sasl_oauth_token_provider)

        def send_to_kafka(element):
            future = producer.send(self.topic, element)
            try:
                record_metadata = future.get(timeout=10)
            except KafkaError as ex:
                # Decide what to do if produce request failed...
                print('Error producing to topic {0}: {1}'.format(self.topic, str(ex)))
            except:
                # Decide what to do if produce request failed...
                print('Error producing to topic {0}: unexpected error'.format(self.topic))
            else:
                # Print metadata of the produced message for reference...
                print('Message produced with metadata: {0}'.format(str(record_metadata)))

        pcoll | 'Kafka send' >> beam.Map(send_to_kafka)

        return pcoll

# Example usage of the KafkaCustomSink custom sink...
with beam.Pipeline() as pipeline:
    data = ['{"value": "ABCD"}', '{"value": "EFGH"}', '{"value": "IJKL"}']
    (pipeline
     | "Create data" >> beam.Create(data)
     | "Write to Kafka" >> KafkaCustomSink(bootstrap_servers='kafka-brokers', topic='test-topic')
     )

在代码示例中,我们使用 KafkaCustomSink 类作为 Apache Beam 的自定义 sink,将 PCollections 中的消息发送到 Apache Kafka。在初始化 KafkaProducer 时,我们设置

相关内容

热门资讯

每日必看教程开挂攻略!指尖四川... 每日必看教程开挂攻略!指尖四川麻将多功能开挂攻略,指尖四川麻将切实真的是有开挂攻略1、指尖四川麻将辅...
实测必看作弊器技巧!梦黄麻将多... 实测必看作弊器技巧!梦黄麻将多功能作弊器技巧,梦黄麻将果然存在有作弊器技巧1、梦黄麻将免费脚本咨询教...
研究成果辅助器教程!小白大作战... 研究成果辅助器教程!小白大作战多功能辅助器教程,小白大作战确实真的有辅助器教程所有人都在同一条线上,...
分享认知辅助软件!小闲巴渝棋牌... 分享认知辅助软件!小闲巴渝棋牌多功能辅助软件,小闲巴渝棋牌竟然是有辅助软件1、不需要AI权限,帮助你...
分享认知透视器app!温岭逸趣... 分享认知透视器app!温岭逸趣麻将多功能透视器app,温岭逸趣麻将真是是有透视器app1)温岭逸趣麻...
专业讨论作弊器插件!欢乐划水麻... 专业讨论作弊器插件!欢乐划水麻将多功能作弊器插件,欢乐划水麻将果然真的是有作弊器插件1、用户打开应用...
最新技巧外挂攻略!逍遥皖北棋牌... 最新技巧外挂攻略!逍遥皖北棋牌多功能外挂攻略,逍遥皖北棋牌都是存在有外挂攻略亲,关键说明,逍遥皖北棋...
今日公布辅助app!乐通多功能... 今日公布辅助app!乐通多功能辅助app,乐通一直有辅助app1、乐通破解器简单,乐通机器人操作,乐...
揭秘一下辅助攻略!青橙竞技多功... 揭秘一下辅助攻略!青橙竞技多功能辅助攻略,青橙竞技一贯真的是有辅助攻略1、玩家可以在青橙竞技线上大神...
推荐一款透视挂技巧!无锡麻将多... 推荐一款透视挂技巧!无锡麻将多功能透视挂技巧,无锡麻将原来有透视挂技巧1、起透看视 无锡麻将辅助软件...