D-Net

package eu.dnetlib.enabling.is.jdbc;

import java.io.IOException;

import java.util.Collection;

import java.util.List;

import java.util.Map;

import javax.sql.DataSource;

import org.antlr.stringtemplate.StringTemplate;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

import org.apache.cxf.helpers.IOUtils;

import org.springframework.beans.factory.annotation.Required;

import org.springframework.jdbc.core.JdbcTemplate;

import com.google.common.collect.Lists;

import eu.dnetlib.enabling.is.rmi.InformationServiceException;

import eu.dnetlib.enabling.is.rmi.Operation;

public class DatabaseUtils {

	private JdbcTemplate jdbcTemplate;

	private static final Log log = LogFactory.getLog(DatabaseUtils.class);

	public List<String> searchJson(final String sql) {

		log.debug(sql);

		return jdbcTemplate.query(sql, new JsonRowMapper());

	}

	public List<Map<String, Object>> searchSimple(final String sql) {

		log.debug(sql);

		return jdbcTemplate.queryForList(sql);

	}

	public int update(final String sql) {

		log.debug(sql);

		return jdbcTemplate.update(sql);

	}

	public boolean isRegisteredService(final String address) {

		final String sql = "SELECT service from service_protocols where address = ?";

		final List<String> list = jdbcTemplate.queryForList(sql, String.class, address);

		return list != null && !list.isEmpty();

	}

	public boolean isRegisteredDs(final String code, final String type) {

		final String sql = "SELECT id from datastructures where code = ? and type = ?";

		final List<String> list = jdbcTemplate.queryForList(sql, String.class, code, type);

		return list != null && !list.isEmpty();

	}

	public String findServiceIdByAddress(final String... addrs) {

		return findServiceIdByAddress(Lists.newArrayList(addrs));

	}

	public String findServiceIdByAddress(final Collection<String> addresses) {

		final String sql = "SELECT service from service_protocols where address = ?";

		for (String addr : addresses) {

			final List<String> list = jdbcTemplate.queryForList(sql, String.class, addr);

			if (list != null && !list.isEmpty()) { return list.get(0); }

		}

		return null;

	}

	public String findDsIdByCode(final String code, final String type) {

		final String sql = "SELECT id from datastructures where code = ? and type = ?";

		final List<String> list = jdbcTemplate.queryForList(sql, String.class, code, type);

		return list == null || list.isEmpty() ? null : list.get(0);

	}

	public void registerNotificationTrigger(final Operation operation, final String table) throws InformationServiceException {

		try {

			final StringTemplate st = new StringTemplate();

			switch (operation) {

			case INSERT:

				st.setTemplate(IOUtils.toString(getClass().getResourceAsStream("/eu/dnetlib/enabling/is/templates/trigger_insert.sql.st")));

				break;

			case DELETE:

				st.setTemplate(IOUtils.toString(getClass().getResourceAsStream("/eu/dnetlib/enabling/is/templates/trigger_delete.sql.st")));

				break;

			case UPDATE:

				st.setTemplate(IOUtils.toString(getClass().getResourceAsStream("/eu/dnetlib/enabling/is/templates/trigger_update.sql.st")));

				break;

			default:

				throw new InformationServiceException("Unknown operation: " + operation);

			}

			st.setAttribute("table", table);

			update(st.toString());

		} catch (IOException e) {

			throw new InformationServiceException("Error registering trigger on table " + table);

		}

	}

	public DataSource getDataSource() {

		return jdbcTemplate.getDataSource();

	}

	public JdbcTemplate getJdbcTemplate() {

		return jdbcTemplate;

	}

	@Required

	public void setJdbcTemplate(final JdbcTemplate jdbcTemplate) {

		this.jdbcTemplate = jdbcTemplate;

	}

}