package com.x.mongodb;

import java.net.UnknownHostException;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.bson.types.ObjectId;

import com.mongodb.BasicDBObject;
import com.mongodb.DB;
import com.mongodb.DBCollection;
import com.mongodb.DBObject;
import com.mongodb.Mongo;
import com.mongodb.MongoException;


public final class MongoDBUtil {

	private static final String HOST = "127.0.0.1";

	private static final String dbName = "test";

	private static Mongo mongo;

	private static DB db;

	static {
		try {
			mongo = new Mongo(HOST);
			db = mongo.getDB(dbName);
			// db.authenticate(username, passwd)
		} catch (UnknownHostException e) {
			e.printStackTrace();
		} catch (MongoException e) {
			e.printStackTrace();
		}
	}

	private MongoDBUtil() {
	}

	/**
	 * 添加操作
	 * <br>------------------------------<br>
	 * @param map
	 * @param collectionName
	 */
	public static void add(Map<String, Object> map, String collectionName) {
		DBObject dbObject = new BasicDBObject(map);
		getCollection(collectionName).insert(dbObject);
	}
	
	/**
	 * 添加操作
	 * <br>------------------------------<br>
	 * @param list
	 * @param collectionName
	 */
	public static void add(List<Map<String, Object>> list, String collectionName) {
		for (Map<String, Object> map : list) {
			add(map, collectionName);
		}
	}
	
	/**
	 * 删除操作
	 * <br>------------------------------<br>
	 * @param map
	 * @param collectionName
	 */
	public static void delete(Map<String, Object> map, String collectionName) {
		DBObject dbObject = new BasicDBObject(map);
		getCollection(collectionName).remove(dbObject);
	}
	
	/**
	 * 删除操作,根据主键
	 * <br>------------------------------<br>
	 * @param id			
	 * @param collectionName
	 */
	public static void delete(String id, String collectionName) {
		Map<String, Object> map = new HashMap<String, Object>();
		map.put("_id", new ObjectId(id));
		delete(map, collectionName);
	}
	
	/**
	 * 删除全部
	 * <br>------------------------------<br>
	 * @param collectionName
	 */
	public static void deleteAll(String collectionName) {
		getCollection(collectionName).drop();
	}
	
	/**
	 * 修改操作</br>
	 * 会用一个新文档替换现有文档,文档key结构会发生改变</br>
	 * 比如原文档{"_id":"123","name":"zhangsan","age":12}当根据_id修改age 
	 * value为{"age":12}新建的文档name值会没有,结构发生了改变
	 * <br>------------------------------<br>
	 * @param whereMap		
	 * @param valueMap		
	 * @param collectionName
	 */
	public static void update(Map<String, Object> whereMap, Map<String, Object> valueMap, String collectionName) {
		executeUpdate(collectionName, whereMap, valueMap, new UpdateCallback(){
			public DBObject doCallback(DBObject valueDBObject) {
				return valueDBObject;
			}
		});
	} 
	
	/**
	 * 修改操作,使用$set修改器</br>
	 * 用来指定一个键值,如果键不存在,则自动创建,会更新原来文档, 不会生成新的, 结构不会发生改变
	 * <br>------------------------------<br>
	 * @param whereMap		
	 * @param valueMap		
	 * @param collectionName
	 */
	public static void updateSet(Map<String, Object> whereMap, Map<String, Object> valueMap, String collectionName) {
		executeUpdate(collectionName, whereMap, valueMap, new UpdateCallback(){
			public DBObject doCallback(DBObject valueDBObject) {
				return new BasicDBObject("$set", valueDBObject);
			}
		});
	} 
	
	/**
	 * 修改操作,使用$inc修改器</br>
	 * 修改器键的值必须为数字</br>
	 * 如果键存在增加或减少键的值, 如果不存在创建键
	 * <br>------------------------------<br>
	 * @param whereMap		
	 * @param valueMap		
	 * @param collectionName
	 */
	public static void updateInc(Map<String, Object> whereMap, Map<String, Integer> valueMap, String collectionName) {
		executeUpdate(collectionName, whereMap, valueMap, new UpdateCallback(){
			public DBObject doCallback(DBObject valueDBObject) {