org.molgenis.data.mysql.MySqlEntityFactory 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 static java.util.Objects.requireNonNull;

import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.List;

import org.molgenis.data.AttributeMetaData;
import org.molgenis.data.DataConverter;
import org.molgenis.data.DataService;
import org.molgenis.data.Entity;
import org.molgenis.data.EntityManager;
import org.molgenis.data.EntityMetaData;
import org.molgenis.data.Fetch;
import org.molgenis.data.support.DefaultEntity;
import org.molgenis.fieldtypes.IntField;
import org.molgenis.fieldtypes.MrefField;
import org.molgenis.fieldtypes.XrefField;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.jdbc.core.RowMapper;
import org.springframework.stereotype.Component;

@Component
public class MySqlEntityFactory
{
	private static final Logger LOG = LoggerFactory.getLogger(MySqlEntityFactory.class);

	private final EntityManager entityManager;
	private final DataService dataService;

	@Autowired
	public MySqlEntityFactory(EntityManager entityManager, DataService dataService)
	{
		this.entityManager = requireNonNull(entityManager);
		this.dataService = requireNonNull(dataService);
	}

	public RowMapper createRowMapper(EntityMetaData entityMeta, Fetch fetch, JdbcTemplate jdbcTemplate,
			String tableName)
	{
		return new EntityMapper(entityMeta, fetch, jdbcTemplate, tableName);
	}

	private class EntityMapper implements RowMapper
	{
		private static final int GROUP_CONCAT_MAX_LEN = 1024;
		private final EntityMetaData entityMetaData;
		private final Fetch fetch;
		private final JdbcTemplate jdbcTemplate;
		private final String tableName;

		private EntityMapper(EntityMetaData entityMetaData, Fetch fetch, JdbcTemplate jdbcTemplate, String tableName)
		{
			this.entityMetaData = requireNonNull(entityMetaData);
			this.fetch = fetch; // can be null
			this.jdbcTemplate = requireNonNull(jdbcTemplate);
			this.tableName = requireNonNull(tableName);
		}

		@Override
		public Entity mapRow(ResultSet resultSet, int i) throws SQLException
		{
			Entity e = new DefaultEntity(entityMetaData, dataService);

			// TODO performance, iterate over fetch if available
			for (AttributeMetaData att : entityMetaData.getAtomicAttributes())
			{
				if (fetch == null || fetch.hasField(att.getName()))
				{
					if (att.getExpression() != null)
					{
						continue;
					}

					if (att.getDataType() instanceof MrefField)
					{
						EntityMetaData refEntityMeta = att.getRefEntity();
						String mrefIds = resultSet.getString(att.getName());
						if (mrefIds != null)
						{
							Iterable mrefEntities;
							if (refEntityMeta.getIdAttribute().getDataType() instanceof IntField)
							{
								List mrefIntegerIds;
								if (mrefIds.length() >= GROUP_CONCAT_MAX_LEN)
								{
									// this list is just as long as it's allowed to be so it probably got truncated.
									// Retrieve the IDs explicitly in a separate query.
									String mrefSelectSql = getMrefSelectSql(e, att);
									if (LOG.isDebugEnabled())
									{
										LOG.debug("Fetching MySQL [{}] data for SQL [{}]", refEntityMeta.getName(),
												mrefSelectSql);
									}
									mrefIntegerIds = jdbcTemplate.queryForList(mrefSelectSql, Integer.class);
								}
								else
								{
									mrefIntegerIds = DataConverter.toIntList(mrefIds);
								}

								// convert ids to (lazy) entities
								mrefEntities = entityManager.getReferences(refEntityMeta, mrefIntegerIds);

							}
							else
							{
								List