/* * Copyright 2019 Martynas Jusevičius <[email protected]>. * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ package com.atomgraph.server.io; import com.atomgraph.core.io.DatasetProvider; import com.atomgraph.processor.util.Validator; import com.atomgraph.server.exception.ConstraintViolationException; import java.io.IOException; import java.io.InputStream; import java.lang.annotation.Annotation; import java.lang.reflect.Type; import java.util.Iterator; import java.util.List; import javax.ws.rs.core.Context; import javax.ws.rs.core.MediaType; import javax.ws.rs.core.MultivaluedMap; import javax.ws.rs.ext.ContextResolver; import javax.ws.rs.ext.Providers; import org.apache.jena.ontology.Ontology; import org.apache.jena.query.Dataset; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import org.spinrdf.constraints.ConstraintViolation; /** * Dataset provider that validates read triples in each graph against SPIN constraints in an ontology. * * @author Martynas Jusevičius {@literal <[email protected]>} */ public class ValidatingDatasetProvider extends DatasetProvider { private static final Logger log = LoggerFactory.getLogger(ValidatingDatasetProvider.class); @Context private Providers providers; @Override public Dataset readFrom(Class<Dataset> type, Type genericType, Annotation[] annotations, MediaType mediaType, MultivaluedMap<String, String> httpHeaders, InputStream entityStream) throws IOException { return process(super.readFrom(type, genericType, annotations, mediaType, httpHeaders, entityStream)); } public Dataset process(Dataset dataset) { return validate(dataset); } public Dataset validate(Dataset dataset) { Validator validator = new Validator(getOntology().getOntModel()); List<ConstraintViolation> cvs = validator.validate(dataset.getDefaultModel()); if (!cvs.isEmpty()) { if (log.isDebugEnabled()) log.debug("SPIN constraint violations: {}", cvs); throw new ConstraintViolationException(cvs, dataset.getDefaultModel()); } Iterator<String> it = dataset.listNames(); while (it.hasNext()) { String graphURI = it.next(); cvs = validator.validate(dataset.getNamedModel(graphURI)); if (!cvs.isEmpty()) { if (log.isDebugEnabled()) log.debug("SPIN constraint violations: {}", cvs); throw new ConstraintViolationException(cvs, dataset.getNamedModel(graphURI), graphURI); } } return dataset; } public Ontology getOntology() { ContextResolver<Ontology> cr = getProviders().getContextResolver(Ontology.class, null); return cr.getContext(Ontology.class); } public Providers getProviders() { return providers; } }