Package: LocalModel$Containment

LocalModel$Containment

name	instruction	branch	complexity	line	method
static {...}	M: 0 C: 34 100%	M: 0 C: 0 100%	M: 0 C: 1 100%	M: 0 C: 2 100%	M: 0 C: 1 100%

Coverage

1: /**
2: * Copyright (C) 2020 Czech Technical University in Prague
3: * <p>
4: * This program is free software: you can redistribute it and/or modify it under the terms of the GNU General Public
5: * License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later
6: * version.
7: * <p>
8: * This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied
9: * warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
10: * details. You should have received a copy of the GNU General Public License along with this program. If not, see
11: * <http://www.gnu.org/licenses/>.
12: */
13: package cz.cvut.kbss.ontodriver.jena.connector;
14:
15: import org.apache.jena.query.Dataset;
16: import org.apache.jena.query.DatasetFactory;
17: import org.apache.jena.rdf.model.*;
18:
19: import java.util.*;
20:
21: /**
22: * Tracks transactional changes.
23: */
24: class LocalModel {
25:
26: private final Dataset added;
27: private final Dataset removed;
28:
29: /**
30: * Whether default graph should be treated as union of all graphs.
31: */
32: private final boolean defaultAsUnion;
33:
34: enum Containment {
35: ADDED, REMOVED, UNKNOWN
36: }
37:
38: LocalModel(boolean defaultAsUnion) {
39: this.added = DatasetFactory.create();
40: this.removed = DatasetFactory.create();
41: this.defaultAsUnion = defaultAsUnion;
42: }
43:
44: Collection<Statement> enhanceStatements(Collection<Statement> statements, Resource subject, Property property,
45: RDFNode value, Collection<String> contexts) {
46: if (contexts.isEmpty()) {
47: return enhanceStatements(statements, subject, property, value, addedDefault(), removedDefault());
48: } else {
49: Collection<Statement> enhanced = statements;
50: for (String ctx : contexts) {
51: enhanced = enhanceStatements(statements, subject, property, value, added.getNamedModel(ctx),
52: removed.getNamedModel(ctx));
53: }
54: return enhanced;
55: }
56: }
57:
58: private Model addedDefault() {
59: return defaultAsUnion ? added.getUnionModel().union(added.getDefaultModel()) : added.getDefaultModel();
60: }
61:
62: private Model removedDefault() {
63: return defaultAsUnion ? removed.getUnionModel().union(removed.getDefaultModel()) : removed.getDefaultModel();
64: }
65:
66: private static Collection<Statement> enhanceStatements(Collection<Statement> toEnhance, Resource subject,
67: Property property, RDFNode value, Model addedModel,
68: Model removedModel) {
69: final Set<Statement> statements = new HashSet<>(toEnhance);
70: statements.addAll(addedModel.listStatements(subject, property, value).toList());
71: statements.removeAll(removedModel.listStatements(subject, property, value).toList());
72: return statements;
73: }
74:
75: Containment contains(Resource subject, Property property, RDFNode value, Collection<String> contexts) {
76: if (contexts.isEmpty()) {
77: if (removedDefault().contains(subject, property, value)) {
78: return Containment.REMOVED;
79: } else {
80: return addedDefault().contains(subject, property, value) ? Containment.ADDED :
81: Containment.UNKNOWN;
82: }
83: } else {
84: Containment result = Containment.UNKNOWN;
85: for (String c : contexts) {
86: if (removed.getNamedModel(c).contains(subject, property, value)) {
87: return Containment.REMOVED;
88: } else if (added.getNamedModel(c).contains(subject, property, value)) {
89: result = Containment.ADDED;
90: }
91: }
92: return result;
93: }
94: }
95:
96: void addStatements(List<Statement> statements, String context) {
97: registerStatements(statements, context, added, removed);
98: }
99:
100: private void registerStatements(List<Statement> statements, String context, Dataset addTo, Dataset removeFrom) {
101: if (context != null) {
102: addTo.getNamedModel(context).add(statements);
103: removeFrom.getNamedModel(context).remove(statements);
104: } else {
105: addTo.getDefaultModel().add(statements);
106: removeFrom.getDefaultModel().remove(statements);
107: if (defaultAsUnion) {
108: removeFrom.listNames().forEachRemaining(n -> removeFrom.getNamedModel(n).remove(statements));
109: }
110: }
111: }
112:
113: void removeStatements(List<Statement> statements, String context) {
114: registerStatements(statements, context, removed, added);
115: }
116:
117: Dataset getAdded() {
118: return added;
119: }
120:
121: Dataset getRemoved() {
122: return removed;
123: }
124:
125: List<String> getContexts() {
126: final Iterator<String> it = added.listNames();
127: final List<String> contexts = new ArrayList<>();
128: it.forEachRemaining(contexts::add);
129: return contexts;
130: }
131: }