org.molgenis.data.mysql.AsyncJdbcTemplate Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of molgenis-data-mysql Show documentation

The newest version!

package org.molgenis.data.mysql;

import java.util.concurrent.ExecutionException;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.Future;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.TimeoutException;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.dao.DataAccessException;
import org.springframework.dao.QueryTimeoutException;
import org.springframework.jdbc.core.JdbcTemplate;

/**
 * JdbcTemplate implementation that calls {@link #execute(String)} on a different thread.
 */
public class AsyncJdbcTemplate
{
	private final static Logger LOG = LoggerFactory.getLogger(AsyncJdbcTemplate.class);

	private final ExecutorService executors = Executors.newCachedThreadPool();
	private JdbcTemplate jdbcTemplate;

	public AsyncJdbcTemplate(JdbcTemplate jdbcTemplate)
	{
		this.jdbcTemplate = jdbcTemplate;
	}

	public void execute(final String sql) throws DataAccessException
	{
		LOG.debug("executing:" + sql + "...");
		Future result = executors.submit(new Runnable()
		{
			@Override
			public void run()
			{
				jdbcTemplate.execute(sql);
				LOG.debug("Executed " + sql);
			}
		});

		try
		{
			result.get(30, TimeUnit.SECONDS);
		}
		catch (InterruptedException | TimeoutException e)
		{
			LOG.warn("Interrupted awaiting SQL statement: " + sql, e);
			throw new QueryTimeoutException("Interrupted awaiting SQL statement: " + sql, e);
		}
		catch (ExecutionException e)
		{
			LOG.error("Error executing SQL statement", e.getCause());
			throw (DataAccessException) e.getCause();
		}
	}

}