java下一个简单的mongodb的helper类

类java下一个简单的mongodb的helper类，参考了下面地址的写法，做了一些修改，调整了全局变量定义，修改了init方法,增加了save方法。
http://blog.csdn.net/phinecos/article/details/8444515

点击(此处)折叠或打开

1. package cls;
   
2. 
   
3. import java.net.UnknownHostException;
   
4. import java.util.Map;
   
5. import java.util.concurrent.ConcurrentHashMap;
   
6. 
   
7. import org.slf4j.LoggerFactory;
   
8. import org.slf4j.Logger;
   
9. 
   
10. import com.mongodb.DB;
    
11. import com.mongodb.DBCollection;
    
12. import com.mongodb.DBCursor;
    
13. import com.mongodb.DBObject;
    
14. import com.mongodb.Mongo;
15. import com.mongodb.WriteResult;
    
16. 
    
17. public class MongoDbHelper {
    
18.     protected static Logger logger = LoggerFactory.getLogger("MongoDbHelper");
    
19.     private Mongo myMongo = null;
    
20.     private DB myDb = null;
    
21.     private static final String serverAddr="127.0.0.1";
    
22.     private static final int port=27017;
    
23.     private static final String myDbName="test";
    
24.     private static final String userName="admin";
    
25.     private static final String passWord="123456";
    
26.     private static Map<String, DBCollection> dbCollectionMap = new ConcurrentHashMap<String, DBCollection>();
    
27.     
    
28.     public MongoDbHelper(){
    
29.         this.init();     
    
30.     }
    
31. 
    
32.      private void init() {
    
33.      if (this.myMongo == null) {
    
34.      try {
    
35.      this.myMongo = new Mongo( MongoDbHelper.serverAddr , MongoDbHelper.port);
    
36.      if (null != this.myMongo) {
    
37.      this.myDb = this.myMongo.getDB(MongoDbHelper.myDbName);
    
38.      //如果需要用户认证则先切换到admin数据库进行用户认证后，再切回用户数据库；
    
39.      if(userName!=""){
    
40.          this.myDb=this.myMongo.getDB("admin");
    
41.          this.myDb.authenticate(userName, passWord.toCharArray());
    
42.          this.myDb=this.myMongo.getDB(MongoDbHelper.myDbName);
    
43.      }
    
44.      }
    
45.      } catch (UnknownHostException e) {
    
46.      logger.error("连接mongoDb失败, 服务器地址: " + MongoDbHelper.serverAddr + ", 端口: " + MongoDbHelper.port);
    
47.      throw new RuntimeException(e);
    
48.      }
    
49.      }
    
50.      }
    
51.      private DBCollection getDBCollection(String collectionName) {
    
52.      DBCollection collection = null;
    
53.      if (dbCollectionMap.containsKey(collectionName)) {
    
54.      collection = dbCollectionMap.get(collectionName);
    
55.      } else {
    
56.      collection = this.myDb.getCollection(collectionName);
    
57.      if (null != collection) {
    
58.      dbCollectionMap.put(collectionName, collection);
    
59.      }
    
60.      }
    
61.      return collection;
    
62.      }
    
63.     
    
64.      public boolean isDocumentExsit(String collectionName, DBObject query) {
    
65.      boolean result = false;
    
66.      DBCursor dbCursor = null;
    
67.      DBCollection collection = this.getDBCollection(collectionName);
    
68.      if (null != collection) {
    
69.      dbCursor = collection.find(query);
    
70.      if (null != dbCursor && dbCursor.hasNext()) {
    
71.      result = true;
    
72.      }
    
73.      }
    
74.      return result;
    
75.      }
    
76.     
    
77.      public DBCursor selectDocument(String collectionName, DBObject query) {
    
78.          DBCursor result = null;
    
79.      DBCursor dbCursor = null;
    
80.      DBCollection collection = this.getDBCollection(collectionName);
    
81.      if (null != collection) {
    
82.      dbCursor = collection.find(query);
    
83.      if (null != dbCursor && dbCursor.hasNext()) {
    
84.      result = dbCursor;
    
85.      }
    
86.      }
    
87.      return result;
    
88.      }
    
89.     
    
90.      public void insertDocument(String collectionName, DBObject newDocument) {
    
91.      DBCollection collection = this.getDBCollection(collectionName);
    
92.      if (null != collection) {
    
93.      if (!this.isDocumentExsit(collectionName, newDocument)) {//insert only doc not exist
    
94.      collection.insert(newDocument);
    
95.      }
    
96.      }
    
97.      }
    
98.     
    
99.      public boolean updateDocument(String collectionName, DBObject query, DBObject updatedDocument) {
    
100.      boolean result = false;
     
101.      WriteResult writeResult = null;
     
102.      DBCollection collection = this.getDBCollection(collectionName);
     
103.      if (null != collection) {
     
104.      writeResult = collection.update(query, updatedDocument);
     
105.      if (null != writeResult) {
     
106.      if (writeResult.getN() > 0) {
     
107.      result = true;
     
108.      }
     
109.      }
     
110.      }
     
111.      return result;
     
112.      }
     
113.         //以下代码为测试mongodb的save功能
     
114.         /*
     
115.         DBObject upObject =new BasicDBObject("_id", new ObjectId("52fdbe36e97fee8601000000"));
     
116.         upObject.put("enterpriseName","cccc");
     
117.         myMongoDb.saveDoc("testColl", upObject);*/
     
118.         //============================================
     
119.      public boolean saveDoc(String collectionName,DBObject saveObject){
     
120.          boolean result=false;
     
121.          WriteResult writeResult = null;
     
122.      DBCollection collection = this.getDBCollection(collectionName);
     
123.      if (null != collection) {
     
124.      writeResult = collection.save(saveObject);
     
125.      if (null != writeResult) {
     
126.      if (writeResult.getN() > 0) {
     
127.      result = true;
     
128.      }
     
129.      }
     
130.      }
     
131.      return result;
     
132.      }
     
133.     
     
134.      public boolean deleteDocument(String collectionName, DBObject query) {
     
135.      boolean result = false;
     
136.      WriteResult writeResult = null;
     
137.      DBCollection collection = this.getDBCollection(collectionName);
     
138.      if (null != collection) {
     
139.      writeResult = collection.remove(query);
     
140.      if (null != writeResult) {
     
141.      if (writeResult.getN() > 0) {
     
142.      result = true;
     
143.      }
     
144.      }
     
145.      }
     
146.      return result;
     
147.      }
     
148.     
     
149.     
     
150. }