package com.mongodb.selector;
import com.mongodb.connection.ClusterDescription;
import com.mongodb.connection.ServerDescription;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import static com.mongodb.assertions.Assertions.notNull;
public final class CompositeServerSelector implements ServerSelector {
private final List<ServerSelector> serverSelectors;
public CompositeServerSelector(final List<? extends ServerSelector> serverSelectors) {
notNull("serverSelectors", serverSelectors);
if (serverSelectors.isEmpty()) {
throw new IllegalArgumentException("Server selectors can not be an empty list");
}
ArrayList<ServerSelector> mergedServerSelectors = new ArrayList<ServerSelector>();
for (ServerSelector cur : serverSelectors) {
if (cur == null) {
throw new IllegalArgumentException("Can not have a null server selector in the list of composed selectors");
}
if (cur instanceof CompositeServerSelector) {
mergedServerSelectors.addAll(((CompositeServerSelector) cur).serverSelectors);
} else {
mergedServerSelectors.add(cur);
}
}
this.serverSelectors = Collections.unmodifiableList(mergedServerSelectors);
}
public List<ServerSelector> getServerSelectors() {
return serverSelectors;
}
@Override
public List<ServerDescription> select(final ClusterDescription clusterDescription) {
ClusterDescription curClusterDescription = clusterDescription;
List<ServerDescription> choices = null;
for (ServerSelector cur : serverSelectors) {
choices = cur.select(curClusterDescription);
curClusterDescription = new ClusterDescription(clusterDescription.getConnectionMode(), clusterDescription.getType(), choices,
clusterDescription.getClusterSettings(),
clusterDescription.getServerSettings());
}
return choices;
}
@Override
public boolean equals(final Object o) {
if (this == o) {
return true;
}
if (o == null || getClass() != o.getClass()) {
return false;
}
CompositeServerSelector that = (CompositeServerSelector) o;
if (serverSelectors.size() != that.serverSelectors.size()) {
return false;
}
return serverSelectors.equals(that.serverSelectors);
}
@Override
public int hashCode() {
return serverSelectors != null ? serverSelectors.hashCode() : 0;
}
@Override
public String toString() {
return "{"
+ "serverSelectors=" + serverSelectors
+ '}';
}
}