Examples with GetRecordsByIdHandler io.arlas.server.ogc.csw.operation.getrecordbyid.GetRecordsByIdHandler used on opensource projects

Example 1 with GetRecordsByIdHandler

use of io.arlas.server.ogc.csw.operation.getrecordbyid.GetRecordsByIdHandler 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);
    }
}