mongodb工具类

mongodb的一个基本工具类，提供的了基本的数据库操作，具体细节请阅读代码

MongoBaseDao.java


----------


import com.google.common.collect.Lists;
import com.mongodb.*;
import com.thinkgem.jeesite.common.config.Global;
import cz.spring.embedmongo.EmbeddedMongoFactoryBean;
import org.springframework.beans.factory.annotation.Autowired;

import java.util.List;
import java.util.Map;

/**
 * Created by lenovo on 2016/5/12.
 */
public class MongoBaseDao{

    @Autowired
    private EmbeddedMongoFactoryBean embeddedMongoFactoryBean;

    private static String dbName = Global.setProperties("spider.properties").getProperty("mongo.spider");

    /**
     * 保存表数据
     *
     * @param tableName
     * @param map
     * @return
     */
    public WriteResult saveAndUpdate(String tableName, Map map) {
        DBCollection table = MongoUtils.getTable(dbName, tableName);
        WriteResult save = table.save(new BasicDBObject(map));
        return save;
    }

    /**
     * 批量插入
     *
     * @param tableName
     * @param list
     * @return
     */
    public WriteResult insertBatch(String tableName, List<Map> list) {
        DBCollection table = MongoUtils.getTable(dbName, tableName);
        List<DBObject> DBList = Lists.newArrayList();
        for (Map map : list) {
            DBList.add(new BasicDBObject(map));
        }
        WriteResult save = table.insert(DBList);
        return save;
    }

    /**
     * 根据条件查找
     *
     * @param tableName
     * @param map
     * @return
     */
    public DBCursor find(String tableName, Map map) {
        DBCollection table = MongoUtils.getTable(dbName, tableName);
        DBCursor dbObjects = table.find(new BasicDBObject(map));
        return dbObjects;
    }

    /**
     * 查找全部
     *
     * @param tableName
     * @return
     */
    public DBCursor findAll(String tableName) {
        DBCollection table = MongoUtils.getTable(dbName, tableName);
        DBCursor dbObjects = table.find(new BasicDBObject());
        return dbObjects;
    }

    public DBObject findOne(String tableName, Map map) {
        DBCollection table = MongoUtils.getTable(dbName, tableName);
        DBObject one = table.findOne(new BasicDBObject(map));
        return one;
    }
}

MongoUtils .java
----------


import com.mongodb.DB;
import com.mongodb.DBCollection;
import com.mongodb.DBObject;
import com.mongodb.MongoClient;
import cz.spring.embedmongo.EmbeddedMongoFactoryBean;
import org.springframework.beans.factory.annotation.Autowired;

import java.io.IOException;
import java.util.Set;

/**
 * mongodb 工具类
 * Created by lenovo on 2016/5/12.
 */
public class MongoUtils {

    private static MongoClient mongoClient;
    private static DB db;

    @Autowired
    private static EmbeddedMongoFactoryBean embeddedMongoFactoryBean;

    /**
     * 获取mongodb client
     *
     * @return
     */
    public static MongoClient getMongoClient() {
        try {
            mongoClient = embeddedMongoFactoryBean.getObject();
        } catch (IOException e) {
            e.printStackTrace();
        }
        return mongoClient;
    }


    /**
     * 关闭mongodbClient
     */
    public static void closeDbClient() {
        getMongoClient().close();
    }

    /**
     * 获取数据库
     *
     * @param dbName
     * @return
     */
    public static DB getDB(String dbName) {
        db = getMongoClient().getDB(dbName);
        return db;
    }

    /**
     * 删除数据库
     *
     * @param dbName
     */
    public static void dropDataBase(String dbName) {
        db = getDB(dbName);
        db.dropDatabase();
    }

    /**
     * 创建数据库表
     *
     * @param dbName
     * @param tableName
     * @param options
     * @return
     */
    public static DBCollection createTable(String dbName, String tableName, DBObject options) {
        db = getDB(dbName);
        DBCollection collection = db.createCollection(tableName, options);
        return collection;
    }

    /**
     * 获取数据库表
     *
     * @param dbName
     * @param tableName
     * @return
     */
    public static DBCollection getTable(String dbName, String tableName) {
        db = getDB(dbName);
        DBCollection collection = db.getCollection(tableName);
        return collection;
    }

    /**
     * 获取数据库下所有的表名
     *
     * @param dbName
     * @param tableName
     * @return
     */
    public static Set getTables(String dbName, String tableName) {
        db = getDB(dbName);
        Set<String> collections = db.getCollectionNames();
        return collections;
    }
}