package io.trygvis.spring;

import io.trygvis.queue.JdbcQueueService;
import io.trygvis.queue.Queue;
import io.trygvis.queue.QueueExecutor;
import io.trygvis.queue.QueueService;
import io.trygvis.queue.QueueSystem;
import org.springframework.dao.DataAccessException;
import org.springframework.jdbc.core.ConnectionCallback;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.transaction.annotation.Transactional;

import java.sql.Connection;
import java.sql.SQLException;
import java.util.Date;
import java.util.List;

public class SpringQueueService implements QueueService {

    public final JdbcTemplate jdbcTemplate;

    public JdbcQueueService queueService;

    public SpringQueueService(QueueSystem queueSystem, JdbcTemplate jdbcTemplate) {
        this.jdbcTemplate = jdbcTemplate;
        this.queueService = queueSystem.createQueueService();
    }

    @Transactional
    public QueueExecutor getQueue(final String name, final int interval, final boolean autoCreate) throws SQLException {
        return jdbcTemplate.execute(new ConnectionCallback<QueueExecutor>() {
            @Override
            public QueueExecutor doInConnection(Connection c) throws SQLException, DataAccessException {
                return queueService.lookupQueue(c, name, interval, autoCreate);
            }
        });
    }

    @Transactional
    public void schedule(final Queue queue, final Date scheduled, final List<String> arguments) throws SQLException {
        jdbcTemplate.execute(new ConnectionCallback<Object>() {
            @Override
            public Object doInConnection(Connection c) throws SQLException, DataAccessException {
                queueService.getQueue(queue.name).schedule(c, scheduled, arguments);
                return null;
            }
        });
    }
}