org.zalando.nakadiproducer.eventlog.impl.EventLogRepositoryImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of nakadi-producer-spring-boot-starter Show documentation
Spring Boot Auto Configuration for Nakadi event producer
There is a newer version: 30.0.0-RC1
Show newest version
package org.zalando.nakadiproducer.eventlog.impl;

import java.sql.Timestamp;
import java.time.Instant;
import java.util.Collection;
import java.util.HashMap;
import java.util.Map;

import org.springframework.dao.EmptyResultDataAccessException;
import org.springframework.jdbc.core.BeanPropertyRowMapper;
import org.springframework.jdbc.core.namedparam.MapSqlParameterSource;
import org.springframework.jdbc.core.namedparam.NamedParameterJdbcTemplate;
import org.springframework.jdbc.support.GeneratedKeyHolder;

public class EventLogRepositoryImpl implements EventLogRepository {
    private NamedParameterJdbcTemplate jdbcTemplate;

    public EventLogRepositoryImpl(NamedParameterJdbcTemplate jdbcTemplate) {
        this.jdbcTemplate = jdbcTemplate;
    }

    @Override
    public Collection findByLockedByAndLockedUntilGreaterThan(String lockedBy, Instant lockedUntil) {
        Map namedParameterMap = new HashMap<>();
        namedParameterMap.put("lockedBy", lockedBy);
        namedParameterMap.put("lockedUntil", toSqlTimestamp(lockedUntil));
        return jdbcTemplate.query(
            "SELECT * FROM nakadi_events.event_log where locked_by = :lockedBy and locked_until > :lockedUntil",
            namedParameterMap,
            new BeanPropertyRowMapper<>(EventLog.class)
        );
    }

    @Override
    public void lockSomeMessages(String lockId, Instant now, Instant lockExpires) {
        Map namedParameterMap = new HashMap<>();
        namedParameterMap.put("lockId", lockId);
        namedParameterMap.put("now", toSqlTimestamp(now));
        namedParameterMap.put("lockExpires", toSqlTimestamp(lockExpires));
        jdbcTemplate.update(
            "UPDATE nakadi_events.event_log SET locked_by = :lockId, locked_until = :lockExpires where locked_until is null or locked_until < :now",
            namedParameterMap
        );
    }

    @Override
    public void delete(EventLog eventLog) {
        Map namedParameterMap = new HashMap<>();
        namedParameterMap.put("id", eventLog.getId());
        jdbcTemplate.update(
            "DELETE FROM nakadi_events.event_log where id = :id",
            namedParameterMap
        );
    }

    @Override
    public void persist(EventLog eventLog) {
        Timestamp now = toSqlTimestamp(Instant.now());
        MapSqlParameterSource namedParameterMap = new MapSqlParameterSource();
        namedParameterMap.addValue("eventType", eventLog.getEventType());
        namedParameterMap.addValue("eventBodyData", eventLog.getEventBodyData());
        namedParameterMap.addValue("flowId", eventLog.getFlowId());
        namedParameterMap.addValue("created", now);
        namedParameterMap.addValue("lastModified", now);
        namedParameterMap.addValue("lockedBy", eventLog.getLockedBy());
        namedParameterMap.addValue("lockedUntil", eventLog.getLockedUntil());
        GeneratedKeyHolder generatedKeyHolder = new GeneratedKeyHolder();
        jdbcTemplate.update(
            "INSERT INTO " +
                "    nakadi_events.event_log " +
                "    (event_type, event_body_data, flow_id, created, last_modified, locked_by, locked_until) " +
                "VALUES " +
                "    (:eventType, :eventBodyData, :flowId, :created, :lastModified, :lockedBy, :lockedUntil)",
            namedParameterMap,
            generatedKeyHolder
        );

        eventLog.setId((Integer) generatedKeyHolder.getKeys().get("id"));
    }

    private Timestamp toSqlTimestamp(Instant now) {
        if (now == null) {
            return null;
        }
        return Timestamp.from(now);
    }

    @Override
    public void deleteAll() {
        jdbcTemplate.update("DELETE from nakadi_events.event_log", new HashMap<>());
    }

    @Override
    public EventLog findOne(Integer id) {
        Map namedParameterMap = new HashMap<>();
        namedParameterMap.put("id", id);
        try {
            return jdbcTemplate.queryForObject(
                "SELECT * FROM nakadi_events.event_log where id = :id",
                namedParameterMap,
                new BeanPropertyRowMapper<>(EventLog.class)
            );
        } catch (EmptyResultDataAccessException ignored) {
            return null;
        }
    }
}