package com.mongodb.text;

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

import org.bson.types.ObjectId;

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

public class MongoDb {
	//1.建立一個Mongo的資料庫連線物件
	static Mongo connection = null;
	//2.建立相關資料庫的連線
	static DB db = null;
	public MongoDb(String dbName) throws UnknownHostException, MongoException{
		connection = new Mongo("127.0.0.1:27017");
		db = connection.getDB(dbName);
	}
	public static void main(String[] args) throws UnknownHostException, MongoException {
		//例項化
		MongoDb mongoDb = new MongoDb("foobar");
	    /**
	     * 1.建立一個名字叫javadb的資料庫
	     */
//		mongoDb.createCollection("javadb");
		/**
		 * 2.為集合javadb新增一條資料
		 */
//		DBObject dbs = new BasicDBObject();
//		dbs.put("name", "uspcat.com");
//		dbs.put("age", 2);
//		List<String>  books = new ArrayList<String>();
//		books.add("EXTJS");
//		books.add("MONGODB");
//		dbs.put("books", books);
//		mongoDb.insert(dbs, "javadb");
		/**
		 * 3.批量插入資料
		 */
//		List<DBObject> dbObjects = new ArrayList<DBObject>();
//		DBObject jim = new BasicDBObject("name","jim");
//		DBObject lisi = new BasicDBObject("name","lisi");
//		dbObjects.add(jim);
//		dbObjects.add(lisi);
//		mongoDb.insertBatch(dbObjects, "javadb");
		/**
		 * 4.根據ID刪除資料
		 */
//		mongoDb.deleteById("502870dab9c368bf5b151a04", "javadb");
		/**
		 * 5.根據條件刪除資料
		 */
//		DBObject lisi = new BasicDBObject();
//		lisi.put("name", "lisi");
//		int count = mongoDb.deleteByDbs(lisi, "javadb");
//		System.out.println("刪除資料的條數是: "+count);
		/**
		 * 6.更新操作,為集合增加email屬性
		 */
//		DBObject update = new BasicDBObject();
//		update.put("$set", 
//				new BasicDBObject("eamil","
 
[email protected]"));
//		mongoDb.update(new BasicDBObject(),
//				update,false,true,"javadb");
		/**
		 * 7.查詢出persons集合中的name和age
		 */
//		DBObject keys = new BasicDBObject();
//		keys.put("_id", false);
//		keys.put("name", true);
//		keys.put("age", true);
//		DBCursor cursor = mongoDb.find(null, keys, "persons");
//	    while (cursor.hasNext()) {
//		    DBObject object = cursor.next();
//		    System.out.println(object.get("name"));
//	    }
	    /**
	     * 7.查詢出年齡大於26歲並且英語成績小於80分
	     */
//		DBObject ref = new BasicDBObject();
//		ref.put("age", new BasicDBObject("$gte",26));
//		ref.put("e", new BasicDBObject("$lte",80));
//		DBCursor cursor = mongoDb.find(ref, null, "persons");
//	    while (cursor.hasNext()) {
//		    DBObject object = cursor.next();
//		    System.out.print(object.get("name")+"-->");
//		    System.out.print(object.get("age")+"-->");
//		    System.out.println(object.get("e"));
//	    }
		/**
		 * 8.分頁例子
		 */
		DBCursor cursor = mongoDb.find(null, null, 0, 3, "persons");
	    while (cursor.hasNext()) {
		    DBObject object = cursor.next();
		    System.out.print(object.get("name")+"-->");
		    System.out.print(object.get("age")+"-->");
		    System.out.println(object.get("e"));
	    }		
		//關閉連線物件
		connection.close();
	}
	/**
	 * 穿件一個數據庫集合
	 * @param collName 集合名稱
	 * @param db  資料庫例項
	 */
	public void createCollection(String collName){
		DBObject dbs = new BasicDBObject();
		db.createCollection("javadb", dbs);
	}
	/**
	 * 為相應的集合新增資料
	 * @param dbs
	 * @param collName
	 */
	public void insert(DBObject dbs,String collName){
		//1.得到集合
		DBCollection coll = db.getCollection(collName);
		//2.插入操作
		coll.insert(dbs);
	}
	/**
	 * 為集合批量插入資料
	 * @param dbses
	 * @param collName
	 */
	public void insertBatch(List<DBObject> dbses,String collName){
		//1.得到集合
		DBCollection coll = db.getCollection(collName);
		//2.插入操作
		coll.insert(dbses);
	}
	/**
	 * 根據id刪除資料
	 * @param id
	 * @param collName
	 * @return 返回影響的資料條數
	 */
	public int deleteById(String id,String collName){
		//1.得到集合
		DBCollection coll = db.getCollection(collName);
		DBObject dbs = new BasicDBObject("_id", new ObjectId(id));
		int count = coll.remove(dbs).getN();
		return count;
	}
	/**
	 * 根據條件刪除資料
	 * @param id
	 * @param collName
	 * @return 返回影響的資料條數
	 */	
	public int deleteByDbs(DBObject dbs,String collName){
		//1.得到集合
		DBCollection coll = db.getCollection(collName);
		int count = coll.remove(dbs).getN();
		return count;
	}
	/**
	 * 更新資料
	 * @param find 查詢器
	 * @param update 更新器
	 * @param upsert 更新或插入
	 * @param multi 是否批量更新
	 * @param collName 集合名稱
	 * @return 返回影響的資料條數
	 */
	public int update(DBObject find,
						DBObject update,
						boolean upsert,
						boolean multi,
						String collName){
		//1.得到集合
		DBCollection coll = db.getCollection(collName);
		int count = coll.update(find, update, upsert, multi).getN();
		return count;
	}
	/**
	 * 查詢器(分頁)
	 * @param ref
	 * @param keys
	 * @param start
	 * @param limit
	 * @return
	 */
	public DBCursor find(DBObject ref, 
			DBObject keys,
			int start,
			int limit,
			String collName){
		DBCursor cur = find(ref, keys, collName);
		return cur.limit(limit).skip(start);
	}
	/**
	 * 查詢器(不分頁)
	 * @param ref
	 * @param keys
	 * @param start
	 * @param limit
	 * @param collName
	 * @return
	 */
	public DBCursor find(DBObject ref,
			DBObject keys,
			String collName){
		//1.得到集合
		DBCollection coll = db.getCollection(collName);
		DBCursor cur = coll.find(ref, keys);
		return cur;
	}	
}