-
Notifications
You must be signed in to change notification settings - Fork 14
/
JenaDriver.java
111 lines (95 loc) · 4.36 KB
/
JenaDriver.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
package cz.cvut.kbss.ontodriver.jena;
import cz.cvut.kbss.ontodriver.Closeable;
import cz.cvut.kbss.ontodriver.OntologyStorageProperties;
import cz.cvut.kbss.ontodriver.config.ConfigParam;
import cz.cvut.kbss.ontodriver.config.Configuration;
import cz.cvut.kbss.ontodriver.config.ConfigurationParameter;
import cz.cvut.kbss.ontodriver.exception.OntoDriverException;
import cz.cvut.kbss.ontodriver.jena.config.Constants;
import cz.cvut.kbss.ontodriver.jena.config.JenaConfigParam;
import cz.cvut.kbss.ontodriver.jena.config.JenaOntoDriverProperties;
import cz.cvut.kbss.ontodriver.jena.connector.*;
import cz.cvut.kbss.ontodriver.jena.exception.JenaDriverException;
import cz.cvut.kbss.ontodriver.jena.util.ConnectionListener;
import java.util.*;
class JenaDriver implements Closeable, ConnectionListener {
private static final List<ConfigurationParameter> CONFIGS = Arrays
.asList(ConfigParam.AUTO_COMMIT, ConfigParam.ONTOLOGY_LANGUAGE, ConfigParam.REASONER_FACTORY_CLASS,
JenaConfigParam.ISOLATION_STRATEGY, JenaConfigParam.STORAGE_TYPE,
JenaConfigParam.TREAT_DEFAULT_GRAPH_AS_UNION);
private volatile boolean open;
private final Configuration configuration;
private final ConnectorFactory connectorFactory;
private final Set<JenaConnection> openConnections;
private boolean autoCommit;
JenaDriver(OntologyStorageProperties storageProperties, Map<String, String> properties) {
assert properties != null;
this.configuration = new Configuration(storageProperties);
CONFIGS.stream().filter(c -> properties.containsKey(c.toString()))
.forEach(c -> configuration.setProperty(c, properties.get(c.toString())));
this.connectorFactory = buildConnectorFactory(properties);
this.openConnections = Collections.synchronizedSet(new HashSet<>());
this.autoCommit = configuration.isSet(ConfigParam.AUTO_COMMIT) ? configuration.is(ConfigParam.AUTO_COMMIT) :
Constants.DEFAULT_AUTO_COMMIT;
this.open = true;
}
private ConnectorFactory buildConnectorFactory(Map<String, String> properties) {
final String isolationStrategy = configuration
.getProperty(JenaConfigParam.ISOLATION_STRATEGY, Constants.DEFAULT_ISOLATION_STRATEGY);
if (configuration.isSet(ConfigParam.REASONER_FACTORY_CLASS)) {
// Once reasoner factory is set, this takes precedence, because only this factory is able to provide
// proper reasoning support
return new InferenceConnectorFactory(configuration, properties);
}
switch (isolationStrategy) {
case JenaOntoDriverProperties.READ_COMMITTED:
return new ReadCommittedConnectorFactory(configuration);
case JenaOntoDriverProperties.SNAPSHOT:
return new SnapshotConnectorFactory(configuration);
default:
throw new IllegalArgumentException("Unsupported transaction isolation strategy " + isolationStrategy);
}
}
JenaConnection acquireConnection() {
ensureOpen();
final StorageConnector connector = connectorFactory.createConnector();
final JenaAdapter adapter = new JenaAdapter(connector, connectorFactory.createInferredConnector(connector));
final JenaConnection connection = new JenaConnection(adapter);
connection.registerListener(this);
connection.setAutoCommit(autoCommit);
openConnections.add(connection);
return connection;
}
private void ensureOpen() {
if (!open) {
throw new IllegalStateException("Driver is closed.");
}
}
@Override
public void connectionClosed(JenaConnection connection) {
openConnections.remove(connection);
}
synchronized void reloadStorage() throws JenaDriverException {
ensureOpen();
try {
connectorFactory.reloadStorage();
} catch (IllegalStateException e) {
throw new JenaDriverException(e);
}
}
@Override
public synchronized void close() throws OntoDriverException {
if (!open) {
return;
}
for (JenaConnection connection : openConnections) {
connection.close();
}
connectorFactory.close();
this.open = false;
}
@Override
public boolean isOpen() {
return open;
}
}