RouterNode.java

/**
 * 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;
    }
}