设为首页收藏本站
开启辅助访问

在路上

 找回密码
 立即注册
在路上 › 站点首页 ›学习 › 查看内容

JAVA单例MongoDB工具类

2016-12-20 13:15| 发布者: zhangjf| 查看: 501| 评论: 0

摘要: JAVA驱动版本： !-- MongoDB驱动 -- dependency groupIdorg.mongodb/groupId artifactIdmongo-java-driver/artifactId version3.0.2/version /dependency 复 ...
JAVA驱动版本：
  1. <!-- MongoDB驱动 -->
  2. <dependency>
  3. <groupId>org.mongodb</groupId>
  4. <artifactId>mongo-java-driver</artifactId>
  5. <version>3.0.2</version>
  6. </dependency>
复制代码

工具类代码如下：
  1. package utils;
  3. import java.util.ArrayList;
  4. import java.util.List;
  6. import org.apache.commons.configuration.CompositeConfiguration;
  7. import org.apache.commons.configuration.ConfigurationException;
  8. import org.apache.commons.configuration.PropertiesConfiguration;
  9. import org.bson.Document;
  10. import org.bson.conversions.Bson;
  11. import org.bson.types.ObjectId;
  13. import com.mongodb.BasicDBObject;
  14. import com.mongodb.MongoClient;
  15. import com.mongodb.MongoClientOptions;
  16. import com.mongodb.MongoClientOptions.Builder;
  17. import com.mongodb.WriteConcern;
  18. import com.mongodb.client.MongoCollection;
  19. import com.mongodb.client.MongoCursor;
  20. import com.mongodb.client.MongoDatabase;
  21. import com.mongodb.client.MongoIterable;
  22. import com.mongodb.client.model.Filters;
  23. import com.mongodb.client.result.DeleteResult;
  25. /**
  26. * MongoDB工具类 Mongo实例代表了一个数据库连接池，即使在多线程的环境中，一个Mongo实例对我们来说已经足够了<br>
  27. * 注意Mongo已经实现了连接池，并且是线程安全的。 <br>
  28. * 设计为单例模式， 因 MongoDB的Java驱动是线程安全的，对于一般的应用，只要一个Mongo实例即可，<br>
  29. * Mongo有个内置的连接池（默认为10个） 对于有大量写和读的环境中，为了确保在一个Session中使用同一个DB时，<br>
  30. * DB和DBCollection是绝对线程安全的<br>
  31. *
  32. * @author zhoulingfei
  33. * @date 2015-5-29 上午11:49:49
  34. * @version 0.0.0
  35. * @Copyright (c)1997-2015 NavInfo Co.Ltd. All Rights Reserved.
  36. */
  37. public enum MongoDBUtil {
  39. /**
  40. * 定义一个枚举的元素，它代表此类的一个实例
  41. */
  42. instance;
  44. private MongoClient mongoClient;
  46. static {
  47. System.out.println("===============MongoDBUtil初始化========================");
  48. CompositeConfiguration config = new CompositeConfiguration();
  49. try {
  50. config.addConfiguration(new PropertiesConfiguration("mongodb.properties"));
  51. } catch (ConfigurationException e) {
  52. e.printStackTrace();
  53. }
  54. // 从配置文件中获取属性值
  55. String ip = config.getString("host");
  56. int port = config.getInt("port");
  57. instance.mongoClient = new MongoClient(ip, port);
  59. // or, to connect to a replica set, with auto-discovery of the primary, supply a seed list of members
  60. // List<ServerAddress> listHost = Arrays.asList(new ServerAddress("localhost", 27017),new ServerAddress("localhost", 27018));
  61. // instance.mongoClient = new MongoClient(listHost);
  63. // 大部分用户使用mongodb都在安全内网下，但如果将mongodb设为安全验证模式，就需要在客户端提供用户名和密码：
  64. // boolean auth = db.authenticate(myUserName, myPassword);
  65. Builder options = new MongoClientOptions.Builder();
  66. // options.autoConnectRetry(true);// 自动重连true
  67. // options.maxAutoConnectRetryTime(10); // the maximum auto connect retry time
  68. options.connectionsPerHost(300);// 连接池设置为300个连接,默认为100
  69. options.connectTimeout(15000);// 连接超时，推荐>3000毫秒
  70. options.maxWaitTime(5000); //
  71. options.socketTimeout(0);// 套接字超时时间，0无限制
  72. options.threadsAllowedToBlockForConnectionMultiplier(5000);// 线程队列数，如果连接线程排满了队列就会抛出“Out of semaphores to get db”错误。
  73. options.writeConcern(WriteConcern.SAFE);//
  74. options.build();
  75. }
  77. // ------------------------------------共用方法---------------------------------------------------
  78. /**
  79. * 获取DB实例 - 指定DB
  80. *
  81. * @param dbName
  82. * @return
  83. */
  84. public MongoDatabase getDB(String dbName) {
  85. if (dbName != null && !"".equals(dbName)) {
  86. MongoDatabase database = mongoClient.getDatabase(dbName);
  87. return database;
  88. }
  89. return null;
  90. }
  92. /**
  93. * 获取collection对象 - 指定Collection
  94. *
  95. * @param collName
  96. * @return
  97. */
  98. public MongoCollection<Document> getCollection(String dbName, String collName) {
  99. if (null == collName || "".equals(collName)) {
  100. return null;
  101. }
  102. if (null == dbName || "".equals(dbName)) {
  103. return null;
  104. }
  105. MongoCollection<Document> collection = mongoClient.getDatabase(dbName).getCollection(collName);
  106. return collection;
  107. }
  109. /**
  110. * 查询DB下的所有表名
  111. */
  112. public List<String> getAllCollections(String dbName) {
  113. MongoIterable<String> colls = getDB(dbName).listCollectionNames();
  114. List<String> _list = new ArrayList<String>();
  115. for (String s : colls) {
  116. _list.add(s);
  117. }
  118. return _list;
  119. }
  121. /**
  122. * 获取所有数据库名称列表
  123. *
  124. * @return
  125. */
  126. public MongoIterable<String> getAllDBNames() {
  127. MongoIterable<String> s = mongoClient.listDatabaseNames();
  128. return s;
  129. }
  131. /**
  132. * 删除一个数据库
  133. */
  134. public void dropDB(String dbName) {
  135. getDB(dbName).drop();
  136. }
  138. /**
  139. * 查找对象 - 根据主键_id
  140. *
  141. * @param collection
  142. * @param id
  143. * @return
  144. */
  145. public Document findById(MongoCollection<Document> coll, String id) {
  146. ObjectId _idobj = null;
  147. try {
  148. _idobj = new ObjectId(id);
  149. } catch (Exception e) {
  150. return null;
  151. }
  152. Document myDoc = coll.find(Filters.eq("_id", _idobj)).first();
  153. return myDoc;
  154. }
  156. /** 统计数 */
  157. public int getCount(MongoCollection<Document> coll) {
  158. int count = (int) coll.count();
  159. return count;
  160. }
  162. /** 条件查询 */
  163. public MongoCursor<Document> find(MongoCollection<Document> coll, Bson filter) {
  164. return coll.find(filter).iterator();
  165. }
  167. /** 分页查询 */
  168. public MongoCursor<Document> findByPage(MongoCollection<Document> coll, Bson filter, int pageNo, int pageSize) {
  169. Bson orderBy = new BasicDBObject("_id", 1);
  170. return coll.find(filter).sort(orderBy).skip((pageNo - 1) * pageSize).limit(pageSize).iterator();
  171. }
  173. /**
  174. * 通过ID删除
  175. *
  176. * @param coll
  177. * @param id
  178. * @return
  179. */
  180. public int deleteById(MongoCollection<Document> coll, String id) {
  181. int count = 0;
  182. ObjectId _id = null;
  183. try {
  184. _id = new ObjectId(id);
  185. } catch (Exception e) {
  186. return 0;
  187. }
  188. Bson filter = Filters.eq("_id", _id);
  189. DeleteResult deleteResult = coll.deleteOne(filter);
  190. count = (int) deleteResult.getDeletedCount();
  191. return count;
  192. }
  194. /**
  195. * FIXME
  196. *
  197. * @param coll
  198. * @param id
  199. * @param newdoc
  200. * @return
  201. */
  202. public Document updateById(MongoCollection<Document> coll, String id, Document newdoc) {
  203. ObjectId _idobj = null;
  204. try {
  205. _idobj = new ObjectId(id);
  206. } catch (Exception e) {
  207. return null;
  208. }
  209. Bson filter = Filters.eq("_id", _idobj);
  210. // coll.replaceOne(filter, newdoc); // 完全替代
  211. coll.updateOne(filter, new Document("$set", newdoc));
  212. return newdoc;
  213. }
  215. public void dropCollection(String dbName, String collName) {
  216. getDB(dbName).getCollection(collName).drop();
  217. }
  219. /**
  220. * 关闭Mongodb
  221. */
  222. public void close() {
  223. if (mongoClient != null) {
  224. mongoClient.close();
  225. mongoClient = null;
  226. }
  227. }
  229. /**
  230. * 测试入口
  231. *
  232. * @param args
  233. */
  234. public static void main(String[] args) {
  236. String dbName = "GC_MAP_DISPLAY_DB";
  237. String collName = "COMMUNITY_BJ";
  238. MongoCollection<Document> coll = MongoDBUtil.instance.getCollection(dbName, collName);
  240. // 插入多条
  241. // for (int i = 1; i <= 4; i++) {
  242. // Document doc = new Document();
  243. // doc.put("name", "zhoulf");
  244. // doc.put("school", "NEFU" + i);
  245. // Document interests = new Document();
  246. // interests.put("game", "game" + i);
  247. // interests.put("ball", "ball" + i);
  248. // doc.put("interests", interests);
  249. // coll.insertOne(doc);
  250. // }
  252. // // 根据ID查询
  253. // String id = "556925f34711371df0ddfd4b";
  254. // Document doc = MongoDBUtil2.instance.findById(coll, id);
  255. // System.out.println(doc);
  257. // 查询多个
  258. // MongoCursor<Document> cursor1 = coll.find(Filters.eq("name", "zhoulf")).iterator();
  259. // while (cursor1.hasNext()) {
  260. // org.bson.Document _doc = (Document) cursor1.next();
  261. // System.out.println(_doc.toString());
  262. // }
  263. // cursor1.close();
  265. // 查询多个
  266. // MongoCursor<Person> cursor2 = coll.find(Person.class).iterator();
  268. // 删除数据库
  269. // MongoDBUtil2.instance.dropDB("testdb");
  271. // 删除表
  272. // MongoDBUtil2.instance.dropCollection(dbName, collName);
  274. // 修改数据
  275. // String id = "556949504711371c60601b5a";
  276. // Document newdoc = new Document();
  277. // newdoc.put("name", "时候");
  278. // MongoDBUtil.instance.updateById(coll, id, newdoc);
  280. // 统计表
  281. // System.out.println(MongoDBUtil.instance.getCount(coll));
  283. // 查询所有
  284. Bson filter = Filters.eq("count", 0);
  285. MongoDBUtil.instance.find(coll, filter);
  287. }
  289. }
复制代码

收藏 邀请
上一篇：用SSLSocketFactory 连接https的地址下一篇：如何用70行Java代码实现深度神经网络算法

最新评论

相关分类

小黑屋|在路上 ( 蜀ICP备15035742号-1 )  

;

GMT+8, 2025-7-8 10:57

Copyright 2015-2025 djqfx

返回顶部