/**
* Copyright (C) 2010 EdgyTech LLC.
*
* 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.edgytech.umongo;
import com.mongodb.BasicDBList;
import com.mongodb.BasicDBObject;
import com.mongodb.CommandResult;
import com.mongodb.DBObject;
import com.mongodb.Mongo;
import com.mongodb.MongoClient;
import com.mongodb.ServerAddress;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import java.util.logging.Level;
import javax.swing.ImageIcon;
import org.xml.sax.SAXException;
/**
*
* @author antoine
*/
public class RouterNode extends BaseTreeNode {
MongoClient mongo;
ServerAddress addr;
BasicDBList shards;
public RouterNode(ServerAddress addr, MongoClient mongo) throws IOException, SAXException {
this.addr = addr;
this.mongo = mongo;
xmlLoad(Resource.getXmlDir(), Resource.File.routerNode, null);
}
@Override
protected void populateChildren() {
CommandResult res = mongo.getDB("admin").command("listShards");
shards = (BasicDBList) res.get("shards");
if (shards == null) {
return;
}
for (Object obj : shards) {
try {
DBObject shard = (DBObject) obj;
String shardName = (String) shard.get("_id");
String hosts = (String) shard.get("host");
String repl = null;
int slash = hosts.indexOf('/');
if (slash >= 0) {
repl = hosts.substring(0, slash);
hosts = hosts.substring(slash + 1);
}
String[] hostList = hosts.split(",");
ArrayList<ServerAddress> addrs = new ArrayList<ServerAddress>();
for (String host : hostList) {
int colon = host.indexOf(':');
if (colon >= 0) {
addrs.add(new ServerAddress(host.substring(0, colon), Integer.parseInt(host.substring(colon + 1))));
} else {
addrs.add(new ServerAddress(host));
}
}
if (repl != null || addrs.size() > 1) {
addChild(new ReplSetNode(repl, addrs, mongo.getMongoClientOptions(), shardName));
} else {
addChild(new ServerNode(addrs.get(0), mongo.getMongoClientOptions(), false, false));
}
} catch (Exception e) {
getLogger().log(Level.WARNING, null, e);
}
}
// add config servers
try {
res = mongo.getDB("admin").command("getCmdLineOpts");
String configStr = (String) ((BasicDBObject) res.get("parsed")).get("configdb");
String[] configsvrs = configStr.split(",");
for (String host : configsvrs) {
int colon = host.indexOf(':');
ServerAddress addr;
if (colon >= 0) {
addr = new ServerAddress(host.substring(0, colon), Integer.parseInt(host.substring(colon + 1)));
} else {
addr = new ServerAddress(host);
}
addChild(new ServerNode(addr, mongo.getMongoClientOptions(), false, true));
}
} catch (Exception e) {
getLogger().log(Level.WARNING, null, e);
}
}
public ServerAddress getAddress() {
return addr;
}
public MongoClient getMongoClient() {
return mongo;
}
@Override
protected void updateNode() {
label = "MongoS: " + mongo.getConnectPoint();
}
@Override
protected void refreshNode() {
}
BasicDBList getShards() {
return shards;
}
String[] getShardNames() {
if (!shards.isEmpty()) {
String[] items = new String[shards.size()];
for (int i = 0; i < shards.size(); ++i) {
DBObject shard = (DBObject) shards.get(i);
items[i] = (shard.get("_id")).toString();
}
return items;
}
return null;
}
}