使用Spring Kafka自定义Apache Kafka头的转换

// 导入必要的包
import org.springframework.kafka.core.KafkaTemplate;
import org.springframework.kafka.support.KafkaHeaders;
import org.springframework.messaging.Message;
import org.springframework.messaging.support.MessageBuilder;
import org.springframework.stereotype.Component;

@Component
public class KafkaProducer {

    private final KafkaTemplate<String, String> kafkaTemplate;

    public KafkaProducer(KafkaTemplate<String, String> kafkaTemplate) {
        this.kafkaTemplate = kafkaTemplate;
    }

    public void sendMessageWithCustomHeader(String message, String customHeaderKey, String customHeaderValue) {
        // 创建一个带有自定义头的消息
        Message<String> kafkaMessage = MessageBuilder
                .withPayload(message)
                .setHeader(KafkaHeaders.TOPIC, "topic-name")
                .setHeader(customHeaderKey, customHeaderValue)
                .build();

        // 发送消息
        kafkaTemplate.send(kafkaMessage);
    }
}