com.mageddo.tobby.producer.kafka.converter.HeadersConverter Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of tobby-transactional-outbox Show documentation

null

There is a newer version: 2.1.6-alpha

Show newest version

package com.mageddo.tobby.producer.kafka.converter;

import java.util.ArrayList;
import java.util.List;

import org.apache.kafka.clients.producer.ProducerRecord;
import org.apache.kafka.common.header.Header;
import org.apache.kafka.common.header.Headers;
import org.apache.kafka.common.header.internals.RecordHeader;
import org.apache.kafka.common.header.internals.RecordHeaders;

import lombok.AccessLevel;
import lombok.NoArgsConstructor;

import static com.mageddo.tobby.Headers.isTobbyHeader;

@NoArgsConstructor(access = AccessLevel.PRIVATE)
public class HeadersConverter {

  public static Headers toKafkaHeaders(com.mageddo.tobby.Headers headers) {
    if (headers == null) {
      return null;
    }
    final RecordHeaders kafkaHeaders = new RecordHeaders();
    for (com.mageddo.tobby.Header header : headers.asList()) {
      kafkaHeaders.add(toKafkaHeader(header));
    }
    return kafkaHeaders;
  }

  private static Header toKafkaHeader(com.mageddo.tobby.Header header) {
    return new RecordHeader(header.getKey(), header.getValue());
  }

  public static  com.mageddo.tobby.Headers fromKafkaHeaders(ProducerRecord record) {
    final List headers = new ArrayList<>();
    for (Header header : record.headers()) {
      if(isTobbyHeader(header.key())){
        continue;
      }
      headers.add(fromKafkaHeader(header));
    }
    return new com.mageddo.tobby.Headers(headers);
  }

  public static com.mageddo.tobby.Header fromKafkaHeader(Header header) {
    return new com.mageddo.tobby.Header(header.key(), header.value());
  }
}