StructuredDataReferenceDAO.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.StructuredDataReference;
import de.dlr.shepard.util.CypherQueryHelper;
import de.dlr.shepard.util.TraversalRules;

public class StructuredDataReferenceDAO extends GenericDAO<StructuredDataReference> {

	/**
	 * Searches the database for references.
	 *
	 * @param dataObjectId identifies the dataObject
	 * @return a List of references
	 */
	public List<StructuredDataReference> findByDataObject(long dataObjectId) {
		String query = String.format("MATCH (d:DataObject)-[hr:has_reference]->%s WHERE ID(d)=%d ",
				CypherQueryHelper.getObjectPart("r", "StructuredDataReference", false), dataObjectId)
				+ CypherQueryHelper.getReturnPart("r");
		var queryResult = findByQuery(query, Collections.emptyMap());
		List<StructuredDataReference> result = StreamSupport.stream(queryResult.spliterator(), false)
				.filter(r -> r.getDataObject() != null).filter(r -> r.getDataObject().getId().equals(dataObjectId))
				.toList();
		return result;
	}

	public List<StructuredDataReference> findReachableReferences(TraversalRules traversalRule, long collectionId,
			long startId, String userName) {
		String query = getSearchForReachableReferencesQuery(traversalRule, collectionId, startId, userName);
		var queryResult = findByQuery(query, Collections.emptyMap());
		List<StructuredDataReference> ret = StreamSupport.stream(queryResult.spliterator(), false).toList();
		return ret;
	}

	public List<StructuredDataReference> findReachableReferences(long collectionId, long startId, String userName) {
		String query = getSearchForReachableReferencesQuery(collectionId, startId, userName);
		var queryResult = findByQuery(query, Collections.emptyMap());
		List<StructuredDataReference> ret = StreamSupport.stream(queryResult.spliterator(), false).toList();
		return ret;
	}

	public List<StructuredDataReference> findReachableReferences(long collectionId, String userName) {
		String query = getSearchForReachableReferencesQuery(collectionId, userName);
		var queryResult = findByQuery(query, Collections.emptyMap());
		List<StructuredDataReference> ret = StreamSupport.stream(queryResult.spliterator(), false).toList();
		return ret;
	}

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