package io.trygvis.queue;

import org.springframework.beans.factory.annotation.*;
import org.springframework.jdbc.core.*;
import org.springframework.stereotype.*;

import java.sql.*;
import java.util.Date;
import java.util.*;

import static java.util.Arrays.*;

@Component
public class TaskDao {

    @Autowired
    private JdbcTemplate jdbcTemplate;

    public long insert(String queue, Date scheduled, String arguments) {
        jdbcTemplate.update("INSERT INTO task(id, run_count, queue, scheduled, arguments) " +
                "VALUES(nextval('task_seq'), 0, ?, ?, ?)", queue, scheduled, arguments);
        return jdbcTemplate.queryForObject("SELECT currval('task_seq')", Long.class);
    }

    public Task findById(long id) {
        return jdbcTemplate.queryForObject("SELECT " + TaskRowMapper.fields + " FROM task WHERE id=?",
                new TaskRowMapper(), id);
    }

    public List<Task> findByNameAndCompletedIsNull(String name) {
        return jdbcTemplate.query("SELECT " + TaskRowMapper.fields + " FROM task WHERE queue=? AND completed IS NULL",
                new TaskRowMapper(), name);
    }

    public void update(Task task) {
        jdbcTemplate.update("UPDATE task SET scheduled=?, last_run=?, run_count=?, completed=? WHERE id=?",
                task.scheduled, task.lastRun, task.runCount, task.completed, task.id);
    }

    private class TaskRowMapper implements RowMapper<Task> {
        public static final String fields = "id, queue, scheduled, last_run, run_count, completed, arguments";

        public Task mapRow(ResultSet rs, int rowNum) throws SQLException {
            return new Task(
                    rs.getLong(1),
                    rs.getString(2),
                    rs.getTimestamp(3),
                    rs.getTimestamp(4),
                    rs.getInt(5),
                    rs.getTimestamp(6),
                    asList(rs.getString(7).split(" ")));
        }
    }
}