com.hangum.tadpole.mongodb.core.test.ReplicaSetConnnection.java Source code

Introduction

Here is the source code for com.hangum.tadpole.mongodb.core.test.ReplicaSetConnnection.java

Source

/*******************************************************************************
 * Copyright (c) 2013 hangum.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the GNU Lesser Public License v2.1
 * which accompanies this distribution, and is available at
 * http://www.gnu.org/licenses/old-licenses/gpl-2.0.html
 * 
 * Contributors:
 *     hangum - initial API and implementation
 ******************************************************************************/
package com.hangum.tadpole.mongodb.core.test;

import java.net.UnknownHostException;
import java.util.ArrayList;
import java.util.List;
import java.util.Set;

import org.apache.commons.lang.StringUtils;

import com.hangum.tadpole.mongodb.core.utils.MongoDBTableColumn;
import com.hangum.tadpole.sql.dao.mongodb.CollectionFieldDAO;
import com.mongodb.DB;
import com.mongodb.DBCollection;
import com.mongodb.DBObject;
import com.mongodb.Mongo;
import com.mongodb.MongoException;
import com.mongodb.ServerAddress;

public class ReplicaSetConnnection {
    public static String serverurl = "localhost";
    public static int port = 4000;

    public static String SERVER_URL = "localhost:4001,localhost:4002";

    /**
     * db  ? 
     * 
     * @param args
     */
    public static void main(String[] args) {
        try {
            ReplicaSetConnnection testMongoCls = new ReplicaSetConnnection();

            Mongo mongo = testMongoCls.connection(serverurl, port);
            System.out.println("");
            if (mongo != null) {
                DB testDB = mongo.getDB("test");
                // Set<String> collectionNames =
                // testMongoCls.getCollectionNames(testDB);
                // for (String collection : collectionNames) {
                testMongoCls.getCollectionInfo(testDB, "store");
                // }
            }
            //      mongo.close();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    /**
     * mongo db .
     * 
     * @return
     */
    public Mongo connection(String uri, int port) throws Exception {
        List<ServerAddress> listServerList = new ArrayList<ServerAddress>();

        listServerList.add(new ServerAddress(serverurl, port));

        String[] urls = StringUtils.split(SERVER_URL, ",");
        for (String ipPort : urls) {
            String[] strIpPort = StringUtils.split(ipPort, ":");

            listServerList.add(new ServerAddress(strIpPort[0], Integer.parseInt(strIpPort[1])));
        }

        Mongo m = null;
        //      try {
        m = new Mongo(listServerList);//uri, port);
        //         List<String> listDB = m.getDatabaseNames();
        //         for (String dbName : listDB) {
        //            System.out.println(dbName);
        //         }

        // authentication(optional)
        // boolean auth = db.authenticate(myUserName, myPassword);

        //      } catch (UnknownHostException e) {
        //         e.printStackTrace();
        //      } catch (MongoException e) {
        //         e.printStackTrace();
        //      }

        return m;
    }

    /**
     * collection list
     * 
     * @param db
     * @return
     */
    public Set<String> getCollectionNames(DB db) {
        Set<String> colls = db.getCollectionNames();

        for (String s : colls) {
            System.out.println(s);
        }

        return colls;
    }

    /**
     * collection info
     * 
     * @param db
     * @param collection
     * @return
     */
    public String getCollectionInfo(DB db, String collection) {
        DBCollection coll = db.getCollection(collection);

        System.out.println("#################################################");
        System.out.println("[collection name]" + coll.getName());
        System.out.println("[count]" + coll.getCount());

        // index list
        List<DBObject> listIndex = coll.getIndexInfo();
        List<CollectionFieldDAO> columnInfo = MongoDBTableColumn.tableColumnInfo(listIndex,
                db.getCollection(collection).findOne());
        for (CollectionFieldDAO collectionFieldDAO : columnInfo) {
            System.out.println("[field]" + collectionFieldDAO.getField());

            if (!collectionFieldDAO.getChildren().isEmpty()) {
                List<CollectionFieldDAO> childColl = collectionFieldDAO.getChildren();
                for (CollectionFieldDAO collectionFieldDAO2 : childColl) {
                    System.out.println("\t [child field]" + collectionFieldDAO2.getField());

                    if (!collectionFieldDAO2.getChildren().isEmpty()) {
                        List<CollectionFieldDAO> childColl2 = collectionFieldDAO2.getChildren();
                        for (CollectionFieldDAO collectionFieldDAO3 : childColl2) {
                            System.out.println("\t\t [child field]" + collectionFieldDAO3.getField());
                        }

                    }
                }

            }
        }

        //      //
        //      StringBuffer sbJson = new StringBuffer();
        //
        //      DBCursor cursor = coll.find();
        //      while (cursor.hasNext()) {
        //         DBObject dbObj = cursor.next();
        //         String jsonData = dbObj.toString();
        //         System.out.println("[data] \t" + jsonData);
        //
        //         sbJson.append(jsonData);
        //      }
        //
        //      System.out
        //            .println("#####[fully text]#########################################################");

        // System.out.println("\t\t ******[detail data][start]*******************************************");
        // ObjectMapper om = new ObjectMapper();
        // try {
        // Map<String, Object> mapObj = om.readValue(sbJson.toString(),
        // new TypeReference<Map<String, Object>>() {});
        // System.out.println("[json to object]" + mapObj);
        //
        // Set<String> keys = mapObj.keySet();
        // for (String key : keys) System.out.print(key + "\t:\t");
        // System.out.println();
        //
        //
        // } catch (JsonParseException e) {
        // e.printStackTrace();
        // } catch (JsonMappingException e) {
        // e.printStackTrace();
        // } catch (IOException e) {
        // e.printStackTrace();
        // }

        System.out.println("\t\t ******[detail data][end]*******************************************");

        return "";
    }
}