com.mongodb.ClusterDescription Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of mongo-java-driver Show documentation
Show all versions of mongo-java-driver Show documentation
The MongoDB Java Driver uber-artifact, containing mongodb-driver, mongodb-driver-core, and bson
/*
* Copyright (c) 2008-2014 MongoDB, Inc.
*
* Licensed 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 com.mongodb;
import org.bson.util.annotations.Immutable;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.List;
import java.util.Set;
import java.util.TreeSet;
import static java.lang.String.format;
import static org.bson.util.Assertions.notNull;
/**
* Immutable snapshot state of a cluster.
*/
@Immutable
class ClusterDescription {
private final ClusterConnectionMode connectionMode;
private final ClusterType type;
private final Set all;
public ClusterDescription(final ClusterConnectionMode connectionMode, final ClusterType type,
final List serverDescriptions) {
notNull("all", serverDescriptions);
this.connectionMode = notNull("connectionMode", connectionMode);
this.type = notNull("type", type);
Set serverDescriptionSet = new TreeSet(new Comparator() {
public int compare(final ServerDescription o1, final ServerDescription o2) {
return o1.getAddress().toString().compareTo(o2.getAddress().toString());
}
});
serverDescriptionSet.addAll(serverDescriptions);
this.all = Collections.unmodifiableSet(serverDescriptionSet);
}
/**
* Return whether the cluster is compatible with the driver.
*
* @return true if the cluster is compatible with the driver.
*/
public boolean isCompatibleWithDriver() {
for (final ServerDescription cur : all) {
if (!cur.isCompatibleWithDriver()) {
return false;
}
}
return true;
}
public ClusterConnectionMode getConnectionMode() {
return connectionMode;
}
public ClusterType getType() {
return type;
}
/**
* Returns the Set of all server descriptions in this cluster, sorted by the String value of the ServerAddress of each one.
*
* @return the set of server descriptions
*/
public Set getAll() {
return all;
}
public ServerDescription getByServerAddress(final ServerAddress serverAddress) {
for (ServerDescription cur : getAll()) {
if (cur.isOk() && cur.getAddress().equals(serverAddress)) {
return cur;
}
}
return null;
}
/**
* While it may seem counter-intuitive that a MongoDb cluster can have more than one primary, it can in the case where the client's view
* of the cluster is a set of mongos servers, any of which can serve as the primary.
*
* @return a list of servers that can act as primaries\
*/
public List getPrimaries() {
return getServersByPredicate(new Predicate() {
public boolean apply(final ServerDescription serverDescription) {
return serverDescription.isPrimary();
}
});
}
public List getSecondaries() {
return getServersByPredicate(new Predicate() {
public boolean apply(final ServerDescription serverDescription) {
return serverDescription.isSecondary();
}
});
}
public List getSecondaries(final TagSet tags) {
return getServersByPredicate(new Predicate() {
public boolean apply(final ServerDescription serverDescription) {
return serverDescription.isSecondary() && serverDescription.hasTags(tags);
}
});
}
public List getAny() {
return getServersByPredicate(new Predicate() {
public boolean apply(final ServerDescription serverDescription) {
return serverDescription.isOk();
}
});
}
public List getAnyPrimaryOrSecondary() {
return getServersByPredicate(new Predicate() {
public boolean apply(final ServerDescription serverDescription) {
return serverDescription.isPrimary() || serverDescription.isSecondary();
}
});
}
public List getAnyPrimaryOrSecondary(final TagSet tags) {
return getServersByPredicate(new Predicate() {
public boolean apply(final ServerDescription serverDescription) {
return (serverDescription.isPrimary() || serverDescription.isSecondary()) && serverDescription.hasTags(tags);
}
});
}
@Override
public boolean equals(final Object o) {
if (this == o) {
return true;
}
if (o == null || getClass() != o.getClass()) {
return false;
}
final ClusterDescription that = (ClusterDescription) o;
if (!all.equals(that.all)) {
return false;
}
if (connectionMode != that.connectionMode) {
return false;
}
return true;
}
@Override
public int hashCode() {
int result = all.hashCode();
result = 31 * result + connectionMode.hashCode();
return result;
}
@Override
public String toString() {
return "ClusterDescription{"
+ "type=" + getType()
+ ", connectionMode=" + connectionMode
+ ", all=" + all
+ '}';
}
public String getShortDescription() {
StringBuilder serverDescriptions = new StringBuilder();
String delimiter = "";
for (ServerDescription cur : all) {
serverDescriptions.append(delimiter).append(cur.getShortDescription());
delimiter = ", ";
}
return format("{type=%s, servers=[%s]", type, serverDescriptions);
}
private interface Predicate {
boolean apply(ServerDescription serverDescription);
}
private List getServersByPredicate(final Predicate predicate) {
final List membersByTag = new ArrayList();
for (final ServerDescription cur : all) {
if (predicate.apply(cur)) {
membersByTag.add(cur);
}
}
return membersByTag;
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy