-
Notifications
You must be signed in to change notification settings - Fork 14
/
RemoteRepositoryWrapper.java
87 lines (73 loc) · 2.74 KB
/
RemoteRepositoryWrapper.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
package cz.cvut.kbss.ontodriver.rdf4j.connector.init;
import cz.cvut.kbss.ontodriver.Wrapper;
import cz.cvut.kbss.ontodriver.config.DriverConfiguration;
import cz.cvut.kbss.ontodriver.exception.OntoDriverException;
import cz.cvut.kbss.ontodriver.rdf4j.exception.Rdf4jDriverException;
import org.apache.http.client.HttpClient;
import org.apache.http.client.utils.HttpClientUtils;
import org.eclipse.rdf4j.http.client.HttpClientSessionManager;
import org.eclipse.rdf4j.http.client.SharedHttpClientSessionManager;
import org.eclipse.rdf4j.model.ValueFactory;
import org.eclipse.rdf4j.repository.Repository;
import org.eclipse.rdf4j.repository.RepositoryConnection;
import org.eclipse.rdf4j.repository.RepositoryException;
import org.eclipse.rdf4j.repository.http.HTTPRepository;
import java.io.File;
/**
* Wrapper for RDF4J {@link HTTPRepository} allowing to set custom {@link HttpClient} for it to use.
*/
public class RemoteRepositoryWrapper implements Repository, Wrapper {
private final HTTPRepository delegate;
private HttpClient httpClient;
public RemoteRepositoryWrapper(HTTPRepository delegate, DriverConfiguration configuration) {
this.delegate = delegate;
this.httpClient = HttpClientFactory.createHttpClient(configuration);
final HttpClientSessionManager sessionManager = delegate.getHttpClientSessionManager();
if (sessionManager instanceof SharedHttpClientSessionManager) {
((SharedHttpClientSessionManager) sessionManager).setHttpClient(httpClient);
}
}
@Override
public void setDataDir(File file) {
delegate.setDataDir(file);
}
@Override
public File getDataDir() {
return delegate.getDataDir();
}
@Override
public void init() throws RepositoryException {
delegate.init();
}
@Override
public boolean isInitialized() {
return delegate.isInitialized();
}
@Override
public void shutDown() throws RepositoryException {
delegate.shutDown();
if (httpClient != null) {
HttpClientUtils.closeQuietly(httpClient);
this.httpClient = null;
}
}
@Override
public boolean isWritable() throws RepositoryException {
return delegate.isWritable();
}
@Override
public RepositoryConnection getConnection() throws RepositoryException {
return delegate.getConnection();
}
@Override
public ValueFactory getValueFactory() {
return delegate.getValueFactory();
}
@Override
public <T> T unwrap(Class<T> cls) throws OntoDriverException {
if (cls.isAssignableFrom(delegate.getClass())) {
return cls.cast(delegate);
}
throw new Rdf4jDriverException("No instance of class " + cls + " found.");
}
}