use of io.arlas.server.core.model.CollectionReferences in project ARLAS-server by gisaia.
the class CSWRESTService method getCollectionReferencesForGetRecords.
private CollectionReferences getCollectionReferencesForGetRecords(String[] elements, String[] excludes, int maxRecords, int startPosition, String[] ids, String q, String constraint, BoundingBox boundingBox) throws IOException, ArlasException {
CollectionReference metacollection = collectionReferenceService.getCollectionReference(getMetacollectionName());
// First we check if there is only "metacollection" that is returned. If this is the case, it means that the queried param is a config param. Thus all collections should be returned
CollectionReferences collectionReferences = ogcDao.getCollectionReferences(elements, null, 2, startPosition - 1, ids, q, constraint, boundingBox);
if (collectionReferences.totalCollectionReferences == 1 && collectionReferences.collectionReferences.size() == 1 && collectionReferences.collectionReferences.get(0).collectionName.equals(getMetacollectionName())) {
return ogcDao.getAllCollectionReferencesExceptOne(elements, excludes, maxRecords, startPosition - 1, metacollection);
} else {
return ogcDao.getCollectionReferencesExceptOne(elements, excludes, maxRecords, startPosition - 1, ids, q, constraint, boundingBox, metacollection);
}
}
use of io.arlas.server.core.model.CollectionReferences in project ARLAS-server by gisaia.
the class CSWRESTService method doKVP.
@Timed
@Path("/csw")
@GET
@Produces({ MediaType.APPLICATION_XML, MediaType.TEXT_XML, ATOM.APPLICATION_ATOM_XML, MIME_TYPE__OPENSEARCH_XML })
@ApiOperation(value = "CSW", produces = MediaType.APPLICATION_XML + "," + MediaType.TEXT_XML + "," + ATOM.APPLICATION_ATOM_XML + "," + MIME_TYPE__OPENSEARCH_XML, notes = "CSW")
@ApiResponses(value = { @ApiResponse(code = 200, message = "Successful operation"), @ApiResponse(code = 500, message = "Arlas Server Error.", response = Error.class) })
public Response doKVP(@ApiParam(name = "version", value = "version", allowMultiple = false, required = true) @QueryParam(value = "version") String version, @ApiParam(name = "acceptversions", value = "acceptversions", allowMultiple = false, required = true) @QueryParam(value = "acceptversions") String acceptVersions, @ApiParam(name = "service", value = "service", allowMultiple = false, required = true) @QueryParam(value = "service") String service, @ApiParam(name = "request", value = "request", allowMultiple = false, required = true) @QueryParam(value = "request") String request, @ApiParam(name = "elementname", value = "elementname", allowMultiple = false, required = true) @QueryParam(value = "elementname") String elementName, @ApiParam(name = "elementsetname", value = "elementsetname", allowMultiple = false, required = true) @QueryParam(value = "elementsetname") String elementSetName, @ApiParam(name = "filter", value = "filter", allowMultiple = false, required = true) @QueryParam(value = "filter") String filter, @ApiParam(name = "constraint", value = "constraint", allowMultiple = false, required = true) @QueryParam(value = "constraint") String constraint, @ApiParam(name = "constraintLanguage", value = "constraintLanguage", allowMultiple = false, required = true) @QueryParam(value = "constraintLanguage") String constraintLanguage, @ApiParam(name = "startposition", value = "startposition", allowMultiple = false, required = false) @QueryParam(value = "startposition") Integer startPosition, @ApiParam(name = "maxrecords", value = "maxrecords", allowMultiple = false, required = false) @QueryParam(value = "maxrecords") Integer maxRecords, @ApiParam(name = "sections", value = "sections", allowMultiple = false, required = false) @QueryParam(value = "sections") String sections, @ApiParam(name = "acceptformats", value = "acceptformats", allowMultiple = false, required = false) @QueryParam(value = "acceptformats") String acceptFormats, @ApiParam(name = "q", value = "q", allowMultiple = false, required = false) @QueryParam(value = "q") String query, @ApiParam(name = "bbox", value = "bbox", allowMultiple = false, required = false) @QueryParam(value = "bbox") String bbox, @ApiParam(name = "outputformat", value = "outputformat", allowMultiple = false, required = false) @QueryParam(value = "outputformat") String outputFormat, @ApiParam(name = "outputschema", value = "outputschema", allowMultiple = false, required = false) @QueryParam(value = "outputschema") String outputSchema, @ApiParam(name = "typenames", value = "typenames", allowMultiple = false, required = false) @QueryParam(value = "typenames") String typeNames, @ApiParam(name = "recordids", value = "recordids", allowMultiple = false, required = false) @QueryParam(value = "recordids") String recordIds, @ApiParam(name = "id", value = "id", allowMultiple = false, required = false) @QueryParam(value = "id") String id, @ApiParam(name = "language", value = "language", allowMultiple = false, required = false) @QueryParam(value = "language") String language, @ApiParam(hidden = true) @HeaderParam(value = "Column-Filter") Optional<String> columnFilter, // --------------------------------------------------------
@ApiParam(name = "pretty", value = Documentation.FORM_PRETTY, allowMultiple = false, defaultValue = "false", required = false) @QueryParam(value = "pretty") Boolean pretty, @Context HttpHeaders headers) throws ArlasException, DatatypeConfigurationException, IOException {
String acceptFormatMediaType = MediaType.APPLICATION_XML;
String outputFormatMediaType = MediaType.APPLICATION_XML;
for (MediaType mediaType : headers.getAcceptableMediaTypes()) {
if (mediaType.getSubtype().contains("opensearchdescription")) {
OpenSearchHandler openSearchHandler = cswHandler.openSearchHandler;
OpenSearchDescription description = openSearchHandler.getOpenSearchDescription(serverBaseUri);
return Response.ok(description).build();
} else if (mediaType.getSubtype().contains("atom")) {
outputFormatMediaType = MediaType.APPLICATION_ATOM_XML;
}
}
if (request == null & version == null & service == null) {
request = "GetCapabilities";
version = CSWConstant.SUPPORTED_CSW_VERSION;
service = CSWConstant.CSW;
}
String[] sectionList;
if (sections == null) {
sectionList = new String[] { "All" };
} else {
sectionList = sections.split(",");
for (String section : sectionList) {
if (!Arrays.asList(CSWConstant.SECTION_NAMES).contains(section)) {
throw new OGCException(OGCExceptionCode.INVALID_PARAMETER_VALUE, "Invalid sections", "sections", Service.CSW);
}
}
}
if (acceptFormats != null) {
if (acceptFormats.equals("text/xml")) {
acceptFormatMediaType = MediaType.TEXT_XML;
} else if (acceptFormats.equals("application/xml")) {
acceptFormatMediaType = MediaType.APPLICATION_XML;
} else {
throw new OGCException(OGCExceptionCode.INVALID_PARAMETER_VALUE, "Invalid acceptFormats", "acceptFormats", Service.CSW);
}
}
if (outputFormat != null) {
if (outputFormat.equals("application/xml")) {
outputFormatMediaType = MediaType.APPLICATION_XML;
} else if (outputFormat.equals("application/atom+xml")) {
outputFormatMediaType = MediaType.APPLICATION_ATOM_XML;
} else {
throw new OGCException(OGCExceptionCode.INVALID_PARAMETER_VALUE, "Invalid outputFormat", "outputFormat", Service.CSW);
}
}
RequestUtils.checkRequestTypeByName(request, CSWConstant.SUPPORTED_CSW_REQUESTYPE, Service.CSW);
CSWRequestType requestType = CSWRequestType.valueOf(request);
CSWCheckParam.checkQuerySyntax(requestType, elementName, elementSetName, acceptVersions, version, service, outputSchema, typeNames, bbox, recordIds, query, id, constraintLanguage);
String[] ids = null;
if (recordIds != null && recordIds.length() > 0) {
ids = recordIds.split(",");
} else if (id != null) {
ids = new String[] { id };
}
BoundingBox boundingBox = null;
if (bbox != null && bbox.length() > 0) {
// west, south, east, north CSW spec
double[] bboxList = GeoFormat.toDoubles(bbox, Service.CSW);
if (!(isBboxLatLonInCorrectRanges(bboxList) && bboxList[3] > bboxList[1]) && bboxList[0] != bboxList[2]) {
throw new OGCException(OGCExceptionCode.INVALID_PARAMETER_VALUE, FluidSearchService.INVALID_BBOX, "bbox", Service.CSW);
}
boundingBox = new BoundingBox(bboxList[3], bboxList[1], bboxList[0], bboxList[2]);
}
startPosition = Optional.ofNullable(startPosition).orElse(1);
maxRecords = Optional.ofNullable(maxRecords).orElse(cswHandler.ogcConfiguration.queryMaxFeature.intValue());
elementSetName = Optional.ofNullable(elementSetName).orElse("summary");
String[] elements = new String[] {};
if (elementName != null) {
elements = new String[elementName.split(",").length];
int i = 0;
for (String element : elementName.split(",")) {
if (element.contains(":")) {
elements[i] = elementName.split(":")[1];
element = elements[i];
} else {
elements[i] = element;
}
if (!Arrays.asList(CSWConstant.DC_FIELDS).contains(element.toLowerCase()))
throw new OGCException(OGCExceptionCode.INVALID_PARAMETER_VALUE, "Invalid elementName", "elementName", Service.CSW);
i++;
}
}
List<CollectionReference> collections;
switch(requestType) {
case GetCapabilities:
GetCapabilitiesHandler getCapabilitiesHandler = cswHandler.getCapabilitiesHandler;
List<String> responseSections = Arrays.asList(sectionList);
String serviceUrl = serverBaseUri + "ogc/csw/?";
getCapabilitiesHandler.setCapabilitiesType(responseSections, serviceUrl, serverBaseUri + "ogc/csw/opensearch");
if (cswHandler.inspireConfiguration.enabled) {
collections = collectionReferenceService.getAllCollectionReferences(columnFilter);
collections.removeIf(collectionReference -> collectionReference.collectionName.equals(getMetacollectionName()));
filterCollectionsByColumnFilter(columnFilter, collections);
if (CollectionUtils.isNotEmpty(collections)) {
getCapabilitiesHandler.addINSPIRECompliantElements(collections, responseSections, serviceUrl, language);
}
}
JAXBElement<CapabilitiesType> getCapabilitiesResponse = getCapabilitiesHandler.getCSWCapabilitiesResponse();
return Response.ok(getCapabilitiesResponse).type(acceptFormatMediaType).build();
case GetRecords:
GetRecordsHandler getRecordsHandler = cswHandler.getRecordsHandler;
CollectionReferences collectionReferences = getCollectionReferencesForGetRecords(elements, null, maxRecords, startPosition, ids, query, constraint, boundingBox);
collections = new ArrayList<>(collectionReferences.collectionReferences);
filterCollectionsByColumnFilter(columnFilter, collections);
long recordsMatched = collectionReferences.totalCollectionReferences;
if (recordIds != null && recordIds.length() > 0) {
if (collections.size() == 0) {
throw new OGCException(OGCExceptionCode.NOT_FOUND, "Document not Found", "id", Service.CSW);
}
}
GetRecordsResponseType getRecordsResponse = getRecordsHandler.getCSWGetRecordsResponse(collections, ElementSetName.valueOf(elementSetName), startPosition - 1, recordsMatched, elements, outputSchema);
return Response.ok(getRecordsResponse).type(outputFormatMediaType).build();
case GetRecordById:
GetRecordsByIdHandler getRecordsByIdHandler = cswHandler.getRecordsByIdHandler;
CollectionReferences recordCollectionReferences = ogcDao.getCollectionReferences(elements, null, maxRecords, startPosition - 1, ids, query, constraint, boundingBox);
collections = new ArrayList<>(recordCollectionReferences.collectionReferences);
ColumnFilterUtil.assertCollectionsAllowed(columnFilter, collections);
if (outputSchema != null && outputSchema.equals(CSWConstant.SUPPORTED_CSW_OUTPUT_SCHEMA[2])) {
GetRecordByIdResponse getRecordByIdResponse = getRecordsByIdHandler.getMDMetadaTypeResponse(collections, ElementSetName.valueOf(elementSetName));
return Response.ok(getRecordByIdResponse).type(outputFormatMediaType).build();
} else {
AbstractRecordType abstractRecordType = getRecordsByIdHandler.getAbstractRecordTypeResponse(collections, ElementSetName.valueOf(elementSetName));
return Response.ok(abstractRecordType).type(outputFormatMediaType).build();
}
default:
throw new OGCException(OGCExceptionCode.INTERNAL_SERVER_ERROR, "Internal error: Unhandled request '" + request + "'.", Service.CSW);
}
}
use of io.arlas.server.core.model.CollectionReferences in project ARLAS-server by gisaia.
the class ElasticOGCCollectionReferenceDao method getCollectionReferences.
private CollectionReferences getCollectionReferences(BoolQueryBuilder boolQueryBuilder, String[] includes, String[] excludes, int size, int from) throws ArlasException {
CollectionReferences collectionReferences = new CollectionReferences();
collectionReferences.collectionReferences = new ArrayList<>();
// Exclude old include_fields for support old collection
if (excludes != null) {
String[] copy = Arrays.copyOf(excludes, excludes.length + 1);
copy[excludes.length] = "include_fields";
excludes = copy;
} else {
excludes = new String[] { "include_fields" };
}
try {
SearchRequest request = new SearchRequest(arlasIndex);
SearchSourceBuilder searchSourceBuilder = new SearchSourceBuilder();
searchSourceBuilder.query(boolQueryBuilder).from(from).size(size).fetchSource(includes, excludes);
request.source(searchSourceBuilder);
SearchResponse response = client.search(request);
collectionReferences.totalCollectionReferences = response.getHits().getTotalHits().value;
for (SearchHit hit : response.getHits().getHits()) {
String source = hit.getSourceAsString();
try {
collectionReferences.collectionReferences.add(new CollectionReference(hit.getId(), mapper.readerFor(CollectionReferenceParameters.class).readValue(source)));
} catch (IOException e) {
throw new InternalServerErrorException("Can not fetch collection", e);
}
}
collectionReferences.nbCollectionReferences = collectionReferences.collectionReferences.size();
} catch (IndexNotFoundException e) {
throw new InternalServerErrorException("Unreachable collections", e);
}
return collectionReferences;
}
Aggregations