org.apache.jackrabbit.oak.plugins.document.mongo.replica.NodeCollectionProvider Maven / Gradle / Ivy
/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You 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 org.apache.jackrabbit.oak.plugins.document.mongo.replica;
import static com.google.common.collect.ImmutableSet.copyOf;
import static com.google.common.collect.Sets.difference;
import java.net.UnknownHostException;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.Map.Entry;
import java.util.concurrent.ConcurrentHashMap;
import com.mongodb.MongoCredential;
import com.mongodb.ServerAddress;
import org.apache.jackrabbit.oak.plugins.document.Collection;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import com.mongodb.DB;
import com.mongodb.DBCollection;
import com.mongodb.MongoClient;
import com.mongodb.MongoClientException;
import com.mongodb.MongoClientURI;
/**
* This class connects to Mongo instances and returns the NODES collection.
* Connections are cached.
*/
public class NodeCollectionProvider {
private static final Logger LOG = LoggerFactory.getLogger(NodeCollectionProvider.class);
private final Map collections = new ConcurrentHashMap();
private final String originalMongoUri;
private final String dbName;
public NodeCollectionProvider(String originalMongoUri, String dbName) {
this.originalMongoUri = originalMongoUri;
this.dbName = dbName;
}
public void retain(Set hostNames) {
close(copyOf(difference(collections.keySet(), hostNames)));
}
public void close() {
close(collections.keySet());
}
private void close(Set hostNames) {
Iterator> it = collections.entrySet().iterator();
while (it.hasNext()) {
Entry entry = it.next();
if (hostNames.contains(entry.getKey())) {
try {
entry.getValue().getDB().getMongo().close();
it.remove();
} catch (MongoClientException e) {
LOG.error("Can't close Mongo client", e);
}
}
}
}
@SuppressWarnings("deprecation")
public DBCollection get(String hostname) throws UnknownHostException {
if (collections.containsKey(hostname)) {
return collections.get(hostname);
}
MongoClient client;
if (originalMongoUri == null) {
MongoClientURI uri = new MongoClientURI("mongodb://" + hostname);
client = new MongoClient(uri);
} else {
client = prepareClientForHostname(hostname);
}
DB db = client.getDB(dbName);
db.getMongo().slaveOk();
DBCollection collection = db.getCollection(Collection.NODES.toString());
collections.put(hostname, collection);
return collection;
}
private MongoClient prepareClientForHostname(String hostname) throws UnknownHostException {
ServerAddress address;
if (hostname.contains(":")) {
String[] hostSplit = hostname.split(":");
if (hostSplit.length != 2) {
throw new IllegalArgumentException("Not a valid hostname: " + hostname);
}
address = new ServerAddress(hostSplit[0], Integer.parseInt(hostSplit[1]));
} else {
address = new ServerAddress(hostname);
}
MongoClientURI originalUri = new MongoClientURI(originalMongoUri);
List credentialList = new ArrayList(1);
if (originalUri.getCredentials() != null) {
credentialList.add(originalUri.getCredentials());
}
return new MongoClient(address, credentialList, originalUri.getOptions());
}
}