com.aconex.scrutineer.jdbc.JdbcIdAndVersionStream Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of scrutineer Show documentation

Compares 2 streams of Ids and Versions to prove that a secondary source is in sync with a primary source of truth. Designed to verify an elasticsearch cluster contains correct information from a JDBC source.

There is a newer version: 7.9.3

Show newest version

package com.aconex.scrutineer.jdbc;

import java.sql.Connection;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.Arrays;
import java.util.Iterator;

import org.slf4j.Logger;

import com.aconex.scrutineer.IdAndVersion;
import com.aconex.scrutineer.IdAndVersionFactory;
import com.aconex.scrutineer.IdAndVersionStream;
import com.aconex.scrutineer.LogUtils;
import com.google.common.base.Predicates;
import com.google.common.collect.Iterables;

public class JdbcIdAndVersionStream implements IdAndVersionStream {

    private static final Logger LOG = LogUtils.loggerForThisClass();

    private final Connection connection;
    private final String sql;
	private final IdAndVersionFactory factory;

    private Statement statement;
    private ResultSet resultSet;
    private Iterator iterator;

    public JdbcIdAndVersionStream(Connection connection, String sql, IdAndVersionFactory factory) {
        this.connection = connection;
        this.sql = sql;
        this.factory = factory;
    }

    @Override
    public void open() {
        long begin = System.currentTimeMillis();
        this.iterator = createIterator(factory);
        LogUtils.info(LOG, "Executed JDBC query in %dms", (System.currentTimeMillis() - begin));
    }

    @Override
    public Iterator iterator() {
        return iterator;
    }

    private Iterator createIterator(IdAndVersionFactory factory) {
        try {
            statement = connection.createStatement();
            resultSet = statement.executeQuery(sql);
            return new IdAndVersionResultSetIterator(resultSet, factory);
        } catch (SQLException e) {
            throw new RuntimeException(e);
        }
    }

    @Override
    public void close() {
        this.iterator = null;
        throwExceptionIfAnyCloseFails(closeResultSet(), closeStatement());
    }

    private void throwExceptionIfAnyCloseFails(SQLException... sqlExceptions) {
        if (!Iterables.all(Arrays.asList(sqlExceptions), Predicates.