forked from sqlancer/sqlancer
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathMongoDBConnection.java
More file actions
35 lines (26 loc) · 890 Bytes
/
MongoDBConnection.java
File metadata and controls
35 lines (26 loc) · 890 Bytes
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
package sqlancer.mongodb;
import org.bson.BsonDocument;
import org.bson.BsonString;
import com.mongodb.client.MongoClient;
import com.mongodb.client.MongoDatabase;
import sqlancer.SQLancerDBConnection;
public class MongoDBConnection implements SQLancerDBConnection {
private final MongoClient client;
private final MongoDatabase database;
public MongoDBConnection(MongoClient client, MongoDatabase database) {
this.client = client;
this.database = database;
}
@Override
public String getDatabaseVersion() throws Exception {
return client.getDatabase("dbname").runCommand(new BsonDocument("buildinfo", new BsonString(""))).get("version")
.toString();
}
@Override
public void close() throws Exception {
client.close();
}
public MongoDatabase getDatabase() {
return database;
}
}