Commit 3c4799b2 authored by Noel Alonso's avatar Noel Alonso
Browse files

Añade métodos para manejar datos georeferenciados

parent 4c44e284

src/main/java/es/redmic/elasticsearchlib/common/repository/IRWBaseESRepository.java

0 → 100644
+17 −0
Original line number Diff line number Diff line 
package es.redmic.elasticsearchlib.common.repository;



import org.elasticsearch.common.xcontent.XContentBuilder;



import es.redmic.models.es.common.dto.EventApplicationResult;

import es.redmic.models.es.common.model.BaseES;



public interface IRWBaseESRepository<TModel extends BaseES<?>> {



	EventApplicationResult save(TModel modelToIndex);



	EventApplicationResult update(TModel modelToIndex);



	EventApplicationResult update(String id, XContentBuilder doc);



	EventApplicationResult delete(String id);

}

src/main/java/es/redmic/elasticsearchlib/geodata/repository/GeoItemsProcessingFunction.java

0 → 100644
+46 −0
Original line number Diff line number Diff line 
package es.redmic.elasticsearchlib.geodata.repository;



import java.util.ArrayList;

import java.util.List;



import org.elasticsearch.search.SearchHit;

import org.elasticsearch.search.SearchHitField;



import com.fasterxml.jackson.databind.ObjectMapper;



import es.redmic.elasticsearchlib.common.utils.IProcessItemFunction;

import es.redmic.models.es.geojson.common.model.Feature;



public class GeoItemsProcessingFunction<TModel extends Feature<?, ?>> implements IProcessItemFunction<TModel> {



	List<TModel> items = new ArrayList<TModel>();

	Class<TModel> typeOfTModel;

	protected ObjectMapper objectMapper;



	public GeoItemsProcessingFunction(Class<TModel> typeOfTModel, ObjectMapper objectMapper) {



		this.typeOfTModel = typeOfTModel;

		this.objectMapper = objectMapper;

	}



	@Override

	public void process(SearchHit hit) {



		TModel item = mapper(hit);

		items.add(item);

	}



	private TModel mapper(SearchHit hit) {



		TModel item = objectMapper.convertValue(hit.getSourceAsMap(), this.typeOfTModel);

		SearchHitField parent = hit.getFields().get("_parent");

		item.set_parentId(parent.getValue().toString());



		return item;

	}



	@Override

	public List<?> getResults() {

		return items;

	}

}
 
 No newline at end of file

src/main/java/es/redmic/elasticsearchlib/geodata/repository/RGeoDataESRepository.java

0 → 100644
+108 −0
Original line number Diff line number Diff line 
package es.redmic.elasticsearchlib.geodata.repository;



import java.util.List;



import org.elasticsearch.action.search.SearchResponse;

import org.elasticsearch.index.query.BoolQueryBuilder;

import org.elasticsearch.index.query.QueryBuilder;



import com.fasterxml.jackson.databind.JavaType;

import com.vividsolutions.jts.geom.Geometry;



import es.redmic.elasticsearchlib.common.query.DataQueryUtils;

import es.redmic.elasticsearchlib.common.repository.RBaseESRepository;

import es.redmic.exception.data.ItemNotFoundException;

import es.redmic.models.es.common.query.dto.MgetDTO;

import es.redmic.models.es.common.query.dto.SimpleQueryDTO;

import es.redmic.models.es.geojson.common.model.Feature;

import es.redmic.models.es.geojson.common.model.GeoHitWrapper;

import es.redmic.models.es.geojson.common.model.GeoHitsWrapper;

import es.redmic.models.es.geojson.common.model.GeoSearchWrapper;

import es.redmic.models.es.geojson.properties.model.GeoDataProperties;



public abstract class RGeoDataESRepository<TModel extends Feature<?, ?>, TQueryDTO extends SimpleQueryDTO>

		extends RBaseESRepository<TModel, TQueryDTO> {



	public RGeoDataESRepository(String[] index, String[] type) {

		super(index, type);

	}



	/*

	 * Sobrescribe método base para añadir query de control de accesso a datos

	 */



	public GeoHitWrapper<?, ?> findById(String id) {



		BoolQueryBuilder query = DataQueryUtils.getItemsQuery(id);



		GeoSearchWrapper<?, ?> result = findBy(query);



		if (result.getHits() == null || result.getHits().getHits() == null || result.getHits().getHits().size() != 1)

			throw new ItemNotFoundException("id", id);



		return result.getHits().getHits().get(0);

	}



	public GeoHitsWrapper<?, ?> mget(MgetDTO dto) {



		List<String> ids = dto.getIds();



		GeoSearchWrapper<?, ?> result = findBy(DataQueryUtils.getItemsQuery(ids), dto.getFields());



		if (result.getHits() == null || result.getHits().getHits() == null)

			throw new ItemNotFoundException("ids", dto.getIds().toString());



		if (result.getHits().getHits().size() != ids.size()) {



			for (GeoHitWrapper<?, ?> hit : result.getHits().getHits()) {

				ids.remove(hit.get_id());

			}



			throw new ItemNotFoundException("ids", ids.toString());

		}



		return result.getHits();

	}



	protected GeoSearchWrapper<?, ?> findBy(QueryBuilder queryBuilder) {



		return findBy(queryBuilder, null);

	}



	protected GeoSearchWrapper<?, ?> findBy(QueryBuilder queryBuilder, List<String> returnFields) {



		return searchResponseToWrapper(searchRequest(queryBuilder, returnFields),

				getSourceType(GeoSearchWrapper.class));

	}



	public GeoSearchWrapper<?, ?> find(TQueryDTO queryDTO) {



		SearchResponse result = super.searchRequest(queryDTO);



		return searchResponseToWrapper(result, getSourceType(GeoSearchWrapper.class));

	}



	@Override

	protected List<?> scrollQueryReturnItems(QueryBuilder builder) {



		return scrollQueryReturnItems(builder, new GeoItemsProcessingFunction<TModel>(typeOfTModel, objectMapper));

	}



	public SimpleQueryDTO createSimpleQueryDTOFromTextQueryParams(Integer from, Integer size) {



		SimpleQueryDTO queryDTO = new SimpleQueryDTO();



		if (from != null)

			queryDTO.setFrom(from);

		if (size != null)

			queryDTO.setSize(size);



		return queryDTO;

	}



	@Override

	protected JavaType getSourceType(Class<?> wrapperClass) {

		return objectMapper.getTypeFactory().constructParametricType(wrapperClass, GeoDataProperties.class,

				Geometry.class);

	}

}

src/main/java/es/redmic/elasticsearchlib/geodata/repository/RWGeoDataESRepository.java

0 → 100644
+83 −0
Original line number Diff line number Diff line 
package es.redmic.elasticsearchlib.geodata.repository;



import org.elasticsearch.common.xcontent.XContentBuilder;

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



import es.redmic.elasticsearchlib.common.repository.IRWBaseESRepository;

import es.redmic.elasticsearchlib.common.utils.ElasticPersistenceUtils;

import es.redmic.models.es.common.dto.EventApplicationResult;

import es.redmic.models.es.common.query.dto.SimpleQueryDTO;

import es.redmic.models.es.geojson.common.model.Feature;



public abstract class RWGeoDataESRepository<TModel extends Feature<?, ?>, TQueryDTO extends SimpleQueryDTO>

		extends RGeoDataESRepository<TModel, TQueryDTO> implements IRWBaseESRepository<TModel> {



	@Autowired

	ElasticPersistenceUtils<TModel> elasticPersistenceUtils;



	public RWGeoDataESRepository(String[] index, String[] type) {

		super(index, type);

	}



	@Override

	public EventApplicationResult save(TModel modelToIndex) {



		EventApplicationResult checkInsert = checkInsertConstraintsFulfilled(modelToIndex);



		if (!checkInsert.isSuccess()) {

			return checkInsert;

		}



		return elasticPersistenceUtils.save(getIndex()[0], getType()[0], modelToIndex, modelToIndex.getUuid());

	}



	@Override

	public EventApplicationResult update(TModel modelToIndex) {



		EventApplicationResult checkUpdate = checkUpdateConstraintsFulfilled(modelToIndex);



		if (!checkUpdate.isSuccess()) {

			return checkUpdate;

		}



		return elasticPersistenceUtils.update(getIndex()[0], getType()[0], modelToIndex,

				modelToIndex.getId().toString());

	}



	@Override

	public EventApplicationResult update(String id, XContentBuilder doc) {



		return elasticPersistenceUtils.update(getIndex()[0], getType()[0], id, doc);

	}



	@Override

	public EventApplicationResult delete(String id) {



		EventApplicationResult checkDelete = checkDeleteConstraintsFulfilled(id);



		if (!checkDelete.isSuccess()) {

			return checkDelete;

		}



		return elasticPersistenceUtils.delete(getIndex()[0], getType()[0], id);

	}



	/*

	 * Función que comprueba que las restricciones necesarias para añadir el item se

	 * cumplen. Por ejemplo que no exista el identificador, que no haya códigos

	 * repetidos. En caso de no cumplirse devuelve la excepción etc

	 */

	protected abstract EventApplicationResult checkInsertConstraintsFulfilled(TModel modelToIndex);



	/*

	 * Función que comprueba que las restricciones necesarias para editar el item se

	 * cumplen. Por ejemplo que no haya códigos repetidos, etc

	 */

	protected abstract EventApplicationResult checkUpdateConstraintsFulfilled(TModel modelToIndex);



	/*

	 * Función que comprueba que las restricciones necesarias para borrar el item se

	 * cumplen. Por ejemplo que no esté referenciado en otros servicios.

	 */

	protected abstract EventApplicationResult checkDeleteConstraintsFulfilled(String modelToIndex);

}
 
 No newline at end of file