DataObjectReferenceDAO.java

package de.dlr.shepard.neo4Core.dao;

import java.util.Collections;
import java.util.List;
import java.util.stream.StreamSupport;

import de.dlr.shepard.neo4Core.entities.DataObjectReference;
import de.dlr.shepard.util.Constants;
import de.dlr.shepard.util.CypherQueryHelper;

public class DataObjectReferenceDAO extends VersionableEntityDAO<DataObjectReference> {

	/**
	 * Searches the database for references.
	 *
	 * @param dataObjectId identifies the dataObject
	 * @return a List of references
	 */
	public List<DataObjectReference> findByDataObjectNeo4jId(long dataObjectId) {
		String query = String.format("MATCH (d:DataObject)-[hr:has_reference]->%s WHERE ID(d)=%d ",
				CypherQueryHelper.getObjectPart("r", "DataObjectReference", false), dataObjectId)
				+ CypherQueryHelper.getReturnPart("r");

		var queryResult = findByQuery(query, Collections.emptyMap());

		List<DataObjectReference> result = StreamSupport.stream(queryResult.spliterator(), false)
				.filter(r -> r.getDataObject() != null).filter(r -> r.getDataObject().getId().equals(dataObjectId))
				.toList();

		return result;
	}

	public List<DataObjectReference> findByDataObjectShepardId(long dataObjectShepardId) {
		String query = String.format(
				"MATCH (d:DataObject)-[hr:has_reference]->%s WHERE d." + Constants.SHEPARD_ID + "=%d ",
				CypherQueryHelper.getObjectPart("r", "DataObjectReference", false), dataObjectShepardId)
				+ CypherQueryHelper.getReturnPart("r");
		var queryResult = findByQuery(query, Collections.emptyMap());
		List<DataObjectReference> result = StreamSupport.stream(queryResult.spliterator(), false)
				.filter(r -> r.getDataObject() != null)
				.filter(r -> r.getDataObject().getShepardId().equals(dataObjectShepardId)).toList();

		return result;
	}

	@Override
	public Class<DataObjectReference> getEntityType() {
		return DataObjectReference.class;
	}

}