D-Net

package eu.dnetlib.goldoa.service;

import eu.dnetlib.goldoa.domain.Budget;

import eu.dnetlib.goldoa.domain.Organization;

import eu.dnetlib.goldoa.domain.OrganizationManagerException;

import eu.dnetlib.goldoa.domain.Vocabulary;

import org.springframework.dao.EmptyResultDataAccessException;

import org.springframework.jdbc.core.BatchPreparedStatementSetter;

import org.springframework.jdbc.core.JdbcTemplate;

import org.springframework.jdbc.core.PreparedStatementSetter;

import org.springframework.jdbc.core.RowMapper;

import org.springframework.jdbc.datasource.DataSourceUtils;

import org.springframework.transaction.annotation.Transactional;

import javax.sql.DataSource;

import java.sql.Connection;

import java.sql.PreparedStatement;

import java.sql.ResultSet;

import java.sql.SQLException;

import java.sql.Types;

import java.util.ArrayList;

import java.util.Arrays;

import java.util.List;

import java.util.UUID;

/**

 * Created by antleb on 3/3/15.

 */

@Transactional

public class OrganizationManagerImpl implements OrganizationManager {

	private DataSource dataSource;

	private final String SEARCH_ORGANIZATIONS = "select id, name from organisation where lower(name) like lower(?)";

	private final String GET_ORGANIZATION = "select * from organisation o left join organization_budget ob on ob.organization=o.id where o.id = ?";

	private final String INSERT_ORGANIZATION = "insert into organization (id, name, shortname, source) values (?, ?, ?, 'user')";

	private final String INSERT_ORGANIZATION_BUDGET = "insert into organization_budget (organization, budget) values (?, ?)";

	@Override

	public List<Vocabulary> search(String term) {

		return new JdbcTemplate(dataSource).query(SEARCH_ORGANIZATIONS, new Object[]{"%" + term + "%"}, new int[]{Types.VARCHAR}, new RowMapper<Vocabulary>() {

			@Override

			public Vocabulary mapRow(ResultSet rs, int rowNum) throws SQLException {

				return new Vocabulary(rs.getString("id"), rs.getString("name"));

			}

		});

	}

	@Override

	public Organization getById(String organizationId) throws OrganizationManagerException {

		try {

			return new JdbcTemplate(dataSource).queryForObject(GET_ORGANIZATION, new Object[]{organizationId}, new int[]{Types.VARCHAR}, new RowMapper<Organization>() {

				@Override

				public Organization mapRow(ResultSet rs, int rowNum) throws SQLException {

					Organization org = new Organization();

					org.setId(rs.getString("id"));

					org.setName(rs.getString("name"));

					org.setShortName(rs.getString("shortname"));

					org.setBudgets(new ArrayList<String>());

					org.setBudgets(Arrays.asList((String[]) rs.getArray("budgets").getArray()));

					return org;

				}

			});

		} catch (EmptyResultDataAccessException e) {

			throw new OrganizationManagerException(OrganizationManagerException.ErrorCause.NOT_EXISTS);

		}

	}

	@Override

	public String saveOrganization(final Organization organization) throws OrganizationManagerException {

		if (organization.getId() == null)

			organization.setId(organization.getShortName());

		try {

			getById(organization.getId());

			throw new OrganizationManagerException(OrganizationManagerException.ErrorCause.ALREADY_EXISTS);

		} catch (OrganizationManagerException e) {

			if (e.getErrorCause() == OrganizationManagerException.ErrorCause.ALREADY_EXISTS)

				throw e;

		}

		new JdbcTemplate(dataSource).update(INSERT_ORGANIZATION, new PreparedStatementSetter() {

			@Override

			public void setValues(PreparedStatement ps) throws SQLException {

				ps.setString(1, organization.getId());

				ps.setString(2, organization.getName());

				ps.setString(3, organization.getShortName());

			}

		});

		new JdbcTemplate(dataSource).batchUpdate(INSERT_ORGANIZATION_BUDGET, new BatchPreparedStatementSetter() {

			@Override

			public void setValues(PreparedStatement ps, int i) throws SQLException {

				ps.setString(1, organization.getId());

				ps.setString(2, organization.getBudgets().get(i));

			}

			@Override

			public int getBatchSize() {

				return organization.getBudgets()!=null?organization.getBudgets().size():0;

			}

		});

		return organization.getId();

	}

	private Connection getConnection() {

		return DataSourceUtils.getConnection(dataSource);

	}

	public DataSource getDataSource() {

		return dataSource;

	}

	public void setDataSource(DataSource dataSource) {

		this.dataSource = dataSource;

	}

}