com.hangum.tadpole.mongodb.core.utils.MongoDBTableColumn.java Source code

Introduction

Here is the source code for com.hangum.tadpole.mongodb.core.utils.MongoDBTableColumn.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.utils;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Set;

import org.apache.commons.lang.StringUtils;
import org.apache.log4j.Logger;

import com.hangum.tadpole.engine.query.dao.mongodb.CollectionFieldDAO;
import com.mongodb.BasicDBObject;
import com.mongodb.DBObject;

/**
 * mongodb? collection(table)?  .
 * 
 * @author hangum
 *
 */
public class MongoDBTableColumn {
    static Logger logger = Logger.getLogger(MongoDBTableColumn.class);

    /**
     * TableView?   ? ?. 
     *
     * @param dbObject
     * @param mapColumns
     * @return
     */
    public static Map<Integer, String> getTabelColumnView(DBObject dbObject, Map<Integer, String> mapColumns) {
        if (dbObject == null)
            return mapColumns;

        int i = mapColumns.size();
        for (String name : dbObject.keySet()) {
            if (!mapColumns.containsValue(name)) {
                mapColumns.put(i, name);
                i++;
            }
        }

        return mapColumns;
    }

    /**
     * TableViewer?   ?? .
     * 
     * @return
     */
    public static Map<Integer, String> getTabelColumnView(DBObject dbObject) {
        Map<Integer, String> map = new HashMap<Integer, String>();

        if (dbObject == null)
            return map;

        int i = 0;
        Set<String> names = dbObject.keySet();
        for (String name : names) {
            map.put(i, name);
            i++;
        }

        return map;
    }

    /**
     * mongodb table column 
     * 
     * @param indexInfo
     * @param dbObject
     * @return
     */
    public static List<CollectionFieldDAO> tableColumnInfo(List<DBObject> indexInfo, DBObject dbObject) {
        Map<String, Boolean> mapIndex = new HashMap<String, Boolean>();

        // key list parsing
        for (DBObject indexObject : indexInfo) {
            String realKey = StringUtils.substringBetween(indexObject.get("key").toString(), "\"", "\""); //$NON-NLS-1$ //$NON-NLS-2$ //$NON-NLS-3$
            mapIndex.put(realKey, true);
        }

        // column info
        List<CollectionFieldDAO> retColumns = new ArrayList<CollectionFieldDAO>();
        try {
            // ? ? ? ..
            if (dbObject == null)
                return retColumns;

            Set<String> names = dbObject.keySet();
            for (String name : names) {
                CollectionFieldDAO column = new CollectionFieldDAO(name,
                        dbObject.get(name) != null ? dbObject.get(name).getClass().getSimpleName() : "Unknow", //$NON-NLS-1$
                        mapIndex.get(name) != null ? "YES" : "NO"); //$NON-NLS-1$ //$NON-NLS-2$
                // ???  ? ?
                if (dbObject.get(name) instanceof BasicDBObject) {
                    makeTableColumn(column, (BasicDBObject) dbObject.get(name));
                }

                retColumns.add(column);
            }
        } catch (Exception e) {
            logger.error("get MongoDB table column info", e); //$NON-NLS-1$
        }

        return retColumns;
    }

    /**
     * sub column  .
     * 
     * @param column
     * @param dbObject
     */
    private static void makeTableColumn(CollectionFieldDAO column, BasicDBObject dbObject) {
        Set<String> names = dbObject.keySet();

        List<CollectionFieldDAO> listChildField = new ArrayList<CollectionFieldDAO>();
        for (String name : names) {
            CollectionFieldDAO columnSub = new CollectionFieldDAO(name, //$NON-NLS-1$
                    dbObject.get(name) != null ? dbObject.get(name).getClass().getSimpleName() : "Unknow", //$NON-NLS-1$
                    "NO"); //$NON-NLS-1$

            if (dbObject.get(name) instanceof BasicDBObject) {
                makeTableColumn(columnSub, (BasicDBObject) dbObject.get(name));
            }

            listChildField.add(columnSub);
        }

        column.setChildren(listChildField);
    }

}