import java.net.UnknownHostException;

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;

/**
 * mongoDB CURD test
 *
 * author：svennee
 */
public class testMongoDB {
        public static void main(String[] args) {
                try {
                        /**
                         * Connection MongoDB
                         */
                        /*
                         * MongoObject have a constructor(String, int)
                         *
                         * String = host：127.0.0.1(default value) or 192.168.1.155
                         * int = port：27017(default value)
                         */
                        Mongo mongo = new Mongo("192.168.1.161", 27017);
                        DB db = mongo.getDB("n_db_test");
                        DBCollection city = db.getCollection("city");

/**
                         * Insert data
                         */
                        BasicDBObject needAddObject = new BasicDBObject();
                        needAddObject.put("cid", "c001");
                        needAddObject.put("city_name", "武汉");
                        needAddObject.put("country", "湖北");
                        city.insert(needAddObject);
                        System.out.println("Insert data success~");

/**
                         * Read data
                         */
                        BasicDBObject queryCondition = new BasicDBObject();
                        queryCondition.put("city_name", "武汉");
                        DBCursor queryCursor = city.find(queryCondition);
                        DBCursor queryAllCursor = city.find();
                        while (queryCursor.hasNext()) {
                                System.out.println("Read condition data："+queryCursor.hashCode() + " ReadValue：" + queryCursor.next());
                        }
                        while (queryAllCursor.hasNext()) {
                                System.out.println("Read All data："+queryAllCursor.hashCode() + " ReadValue：" + queryAllCursor.next());
                        }
                        System.out.println("Read data success~");
                        
                        /**
                         * Modify data
                         */
                        // One method(这种强大些！)
                        BasicDBObject needModifyData = new BasicDBObject();
                        needModifyData.put("cid", "c001");
                        DBCursor modifyCursor = city.find(needModifyData);
                        if (modifyCursor.hasNext()) {
                                DBObject modifyObject = modifyCursor.next();
                                modifyObject.put("cid", "c002");
                                city.save(modifyObject);
                        }
                        
                        // Two method
                        BasicDBObject old_val = new BasicDBObject(),new_val = new BasicDBObject();
                        old_val.put("city_name", "武汉");
                        new_val.put("city_name", "武汉苏州");
                        city.update(old_val, new_val);
                        System.out.println("Modfiy data success~");

/**
                         * Delete data
                         */
                        BasicDBObject needDelData = new BasicDBObject();
                        needDelData.put("city_name", "武汉苏州");
                        DBCursor delCursor = city.find(needDelData);
                        while (delCursor.hasNext()) {
                                DBObject delObject = delCursor.next();
                                city.remove(delObject);
                        }
                        System.out.println("Delete data success~");
                        
                } catch (UnknownHostException e) {
                        System.err.println("UnknownHostException");
                        e.printStackTrace();
                } catch (MongoException e) {
                        System.err.println("MongoException");
                        e.printStackTrace();
                }

}

}