com.mageddo.tobby.converter.ProducedRecordConverter Maven / Gradle / Ivy

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.converter;

import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.Base64;
import java.util.UUID;

import com.mageddo.tobby.ProducedRecord;
import com.mageddo.tobby.ProducerRecord;
import com.mageddo.tobby.UncheckedSQLException;

public class ProducedRecordConverter {

  private ProducedRecordConverter() {
  }

  public static ProducedRecord map(ResultSet rs) {
    try {
      return ProducedRecord.builder()
          .key(getBytesFromBase64(rs, "TXT_KEY"))
          .value(getBytesFromBase64(rs, "TXT_VALUE"))
          .topic(rs.getString("NAM_TOPIC"))
          .createdAt(rs.getTimestamp("DAT_CREATED")
              .toLocalDateTime())
          .id(UUID.fromString(rs.getString("IDT_TTO_RECORD")))
          .partition(getInteger(rs))
          .headers(HeadersConverter.decodeFromBase64(rs.getString("TXT_HEADERS")))
          .build();
    } catch (SQLException e) {
      throw new UncheckedSQLException(e);
    }
  }

  private static Integer getInteger(ResultSet rs) throws SQLException {
    final int v = rs.getInt("NUM_PARTITION");
    return rs.wasNull() ? null : v;
  }

  private static byte[] getBytesFromBase64(ResultSet rs, String name) throws SQLException {
    final String base64 = rs.getString(name);
    return Base64
        .getDecoder()
        .decode(base64);
  }

  public static ProducedRecord from(UUID id, ProducerRecord record) {
    return ProducedRecord.builder()
        .id(id)
        .topic(record.getTopic())
        .partition(record.getPartition())
        .key(record.getKey())
        .value(record.getValue())
        .headers(record.getHeaders())
        .build()
        ;
  }
}