no.nav.common.kafka.spring.DatabaseUtils Maven / Gradle / Ivy

Go to download
package no.nav.common.kafka.spring;

import lombok.SneakyThrows;
import no.nav.common.kafka.consumer.feilhandtering.StoredConsumerRecord;
import no.nav.common.kafka.producer.feilhandtering.StoredProducerRecord;
import org.apache.kafka.common.TopicPartition;
import org.springframework.jdbc.core.JdbcTemplate;

import java.sql.ResultSet;
import java.sql.Timestamp;
import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors;

import static java.lang.String.format;
import static no.nav.common.kafka.util.DatabaseConstants.*;

public class DatabaseUtils {

    @SneakyThrows
    public static List fetchProducerRecords(ResultSet resultSet) {
        try (resultSet) {
            List records = new ArrayList<>();

            while (resultSet.next()) {
                long id = resultSet.getInt(ID);
                String topic = resultSet.getString(TOPIC);
                String headersJson = resultSet.getString(HEADERS_JSON);
                byte[] key = resultSet.getBytes(KEY);
                byte[] value = resultSet.getBytes(VALUE);

                records.add(new StoredProducerRecord(id, topic, key, value, headersJson));
            }

            return records;
        }
    }

    @SneakyThrows
    public static List fetchConsumerRecords(ResultSet resultSet) {
        try (resultSet) {
            List records = new ArrayList<>();

            while (resultSet.next()) {
                long id = resultSet.getInt(ID);
                String topic = resultSet.getString(TOPIC);
                int partition = resultSet.getInt(PARTITION);
                long offset = resultSet.getLong(RECORD_OFFSET);
                byte[] key = resultSet.getBytes(KEY);
                byte[] value = resultSet.getBytes(VALUE);
                String headersJson = resultSet.getString(HEADERS_JSON);
                int retries = resultSet.getInt(RETRIES);
                Timestamp lastRetry = resultSet.getTimestamp(LAST_RETRY);
                long timestamp = resultSet.getLong(RECORD_TIMESTAMP);

                records.add(new StoredConsumerRecord(id, topic, partition, offset, key, value, headersJson, retries, lastRetry, timestamp));
            }

            return records;
        }
    }

    @SneakyThrows
    public static List fetchTopicPartitions(ResultSet resultSet) {
        List topicPartitions = new ArrayList<>();

        while (resultSet.next()) {
            String topic = resultSet.getString(TOPIC);
            int partition = resultSet.getInt(PARTITION);

            topicPartitions.add(new TopicPartition(topic, partition));
        }

        return topicPartitions;
    }

    @SneakyThrows
    public static long incrementAndGetPostgresSequence(JdbcTemplate jdbcTemplate, String sequenceName) {
        String sql = format("SELECT nextval('%s')", sequenceName);
        return fetchSequence(jdbcTemplate, sql);
    }

    @SneakyThrows
    public static long incrementAndGetOracleSequence(JdbcTemplate jdbcTemplate, String sequenceName) {
        String sql = format("SELECT %s.NEXTVAL FROM dual", sequenceName);
        return fetchSequence(jdbcTemplate, sql);
    }

    @SneakyThrows
    private static long fetchSequence(JdbcTemplate jdbcTemplate, String sql) {
        return jdbcTemplate.queryForObject(sql, (rs, rn) -> rs.getLong(1));
    }

    public static String inClause(int number) {
        StringBuilder sb = new StringBuilder();
        sb.append("IN (");
        for (int i = 0; i < number; i++ ) {
            sb.append("?");
            if (i < number - 1) {
                sb.append(",");
            }
        }
        sb.append(")");
        return sb.toString();
    }

    public static String toPostgresArray(List values) {
        return "{" + values.stream().map(Object::toString).collect(Collectors.joining(",")) + "}";
    }
}