MongoDbHelper.cs 32 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864
  1. using MongoDB.Bson;
  2. using MongoDB.Driver;
  3. using MongoDB.Driver.Builders;
  4. using System;
  5. using System.Collections.Generic;
  6. using System.Linq;
  7. using System.Linq.Expressions;
  8. using System.Reflection;
  9. using System.Text;
  10. using System.Threading.Tasks;
  11. namespace Ant.Service.Mongodb
  12. {
  13. /// <summary>
  14. /// Mongo db的数据库帮助类
  15. /// </summary>
  16. public class MongoDbHelper
  17. {
  18. /// <summary>
  19. /// 数据库的实例
  20. /// </summary>
  21. private MongoDatabase _db;
  22. /// <summary>
  23. /// ObjectId的键
  24. /// </summary>
  25. private readonly string OBJECTID_KEY = "_id";
  26. public MongoDbHelper(string host, string timeOut)
  27. {
  28. this._db = new MongoDb(host, timeOut).GetDataBase();
  29. }
  30. public MongoDbHelper(MongoDatabase db)
  31. {
  32. this._db = db;
  33. }
  34. #region 查询实例
  35. //Query.All(“name”, “a”, “b”);//通过多个元素来匹配数组
  36. //Query.And(Query.EQ(“name”, “a”), Query.EQ(“title”, “t”));//同时满足多个条件
  37. //Query.EQ(“name”, “a”);//等于
  38. //Query.Exists(“type”, true);//判断键值是否存在
  39. //Query.GT(“value”, 2);//大于>
  40. //Query.GTE(“value”, 3);//大于等于>=
  41. //Query.In(“name”, “a”, “b”);//包括指定的所有值,可以指定不同类型的条件和值
  42. //Query.LT(“value”, 9);//小于<
  43. //Query.LTE(“value”, 8);//小于等于<=
  44. //Query.Mod(“value”, 3, 1);//将查询值除以第一个给定值,若余数等于第二个给定值则返回该结果
  45. //Query.NE(“name”, “c”);//不等于
  46. //Query.Nor(Array);//不包括数组中的值
  47. //Query.Not(“name”);//元素条件语句
  48. //Query.NotIn(“name”, “a”, 2);//返回与数组中所有条件都不匹配的文档
  49. //Query.Or(Query.EQ(“name”, “a”), Query.EQ(“title”, “t”));//满足其中一个条件
  50. //Query.Size(“name”, 2);//给定键的长度
  51. //Query.Type(“_id”, BsonType.ObjectId);//给定键的类型
  52. //Query.Where(BsonJavaScript);//执行JavaScript
  53. //Query.Matches(“Title”, str);//模糊查询 相当于sql中like – str可包含正则表达式
  54. #endregion
  55. public T GetNextSequence<T>(IMongoQuery query, SortByDocument sortBy, UpdateDocument update, string collectionName, string indexName)
  56. {
  57. if (this._db == null)
  58. {
  59. return default(T);
  60. }
  61. try
  62. {
  63. MongoCollection<T> mc = this._db.GetCollection<T>(collectionName);
  64. query = this.InitQuery(query);
  65. sortBy = this.InitSortBy(sortBy, OBJECTID_KEY);
  66. update = this.InitUpdateDocument(update, indexName);
  67. var ido = mc.FindAndModify(query, sortBy, update, true, false);
  68. return ido.GetModifiedDocumentAs<T>();
  69. }
  70. catch (Exception ex)
  71. {
  72. return default(T);
  73. }
  74. }
  75. #region 插入数据
  76. /// <summary>
  77. /// 将数据插入进数据库
  78. /// </summary>
  79. /// <typeparam name="T">需要插入数据的类型</typeparam>
  80. /// <param name="t">需要插入的具体实体</param>
  81. public bool Insert<T>(T t)
  82. {
  83. //集合名称
  84. string collectionName = typeof(T).Name;
  85. return Insert<T>(t, collectionName);
  86. }
  87. /// <summary>
  88. /// 将数据插入进数据库
  89. /// </summary>
  90. /// <typeparam name="T">需要插入数据库的实体类型</typeparam>
  91. /// <param name="t">需要插入数据库的具体实体</param>
  92. /// <param name="collectionName">指定插入的集合</param>
  93. public bool Insert<T>(T t, string collectionName)
  94. {
  95. if (this._db == null)
  96. {
  97. return false;
  98. }
  99. try
  100. {
  101. MongoCollection<BsonDocument> mc = this._db.GetCollection<BsonDocument>(collectionName);
  102. //将实体转换为bson文档
  103. BsonDocument bd = t.ToBsonDocument();
  104. //进行插入操作
  105. WriteConcernResult result = mc.Insert(bd);
  106. if (!string.IsNullOrEmpty(result.ErrorMessage))
  107. {
  108. return false;
  109. }
  110. return true;
  111. }
  112. catch (Exception ex)
  113. {
  114. return false;
  115. }
  116. }
  117. /// <summary>
  118. /// 批量插入数据
  119. /// </summary>
  120. /// <typeparam name="T">需要插入数据库的实体类型</typeparam>
  121. /// <param name="list">需要插入数据的列表</param>
  122. public bool Insert<T>(List<T> list)
  123. {
  124. //集合名称
  125. string collectionName = typeof(T).Name;
  126. return this.Insert<T>(list, collectionName);
  127. }
  128. /// <summary>
  129. /// 批量插入数据
  130. /// </summary>
  131. /// <typeparam name="T">需要插入数据库的实体类型</typeparam>
  132. /// <param name="list">需要插入数据的列表</param>
  133. /// <param name="collectionName">指定要插入的集合</param>
  134. public bool Insert<T>(List<T> list, string collectionName)
  135. {
  136. if (this._db == null)
  137. {
  138. return false;
  139. }
  140. try
  141. {
  142. MongoCollection<BsonDocument> mc = this._db.GetCollection<BsonDocument>(collectionName);
  143. //创建一个空间bson集合
  144. List<BsonDocument> bsonList = new List<BsonDocument>();
  145. //批量将数据转为bson格式 并且放进bson文档
  146. list.ForEach(t => bsonList.Add(t.ToBsonDocument()));
  147. //批量插入数据
  148. mc.InsertBatch(bsonList);
  149. return true;
  150. }
  151. catch (Exception ex)
  152. {
  153. return false;
  154. }
  155. }
  156. #endregion
  157. #region 查询数据
  158. #region 查询所有记录
  159. /// <summary>
  160. /// 查询一个集合中的所有数据
  161. /// </summary>
  162. /// <typeparam name="T">该集合数据的所属类型</typeparam>
  163. /// <param name="collectionName">指定集合的名称</param>
  164. /// <returns>返回一个List列表</returns>
  165. public List<T> FindAll<T>(string collectionName)
  166. {
  167. if (this._db == null)
  168. {
  169. return null;
  170. }
  171. try
  172. {
  173. MongoCollection<T> mc = this._db.GetCollection<T>(collectionName);
  174. //以实体方式取出其数据集合
  175. MongoCursor<T> mongoCursor = mc.FindAll();
  176. //直接转化为List返回
  177. return mongoCursor.ToList<T>();
  178. }
  179. catch (Exception ex)
  180. {
  181. return null;
  182. }
  183. }
  184. /// <summary>
  185. /// 查询一个集合中的所有数据 其集合的名称为T的名称
  186. /// </summary>
  187. /// <typeparam name="T">该集合数据的所属类型</typeparam>
  188. /// <returns>返回一个List列表</returns>
  189. public List<T> FindAll<T>()
  190. {
  191. string collectionName = typeof(T).Name;
  192. return FindAll<T>(collectionName);
  193. }
  194. #endregion
  195. #region 查询一条记录
  196. /// <summary>
  197. /// 查询索引最大的一条记录
  198. /// </summary>
  199. /// <typeparam name="T">该数据所属的类型</typeparam>
  200. /// <param name="collectionName">去指定查询的集合</param>
  201. /// <param name="sort">排序字段</param>
  202. /// <returns>返回一个实体类型</returns>
  203. public T FindOneToIndexMax<T>(string[] sort)
  204. {
  205. string collectionName = typeof(T).Name;
  206. return FindOneToIndexMax<T>(null, collectionName, sort);
  207. }
  208. /// <summary>
  209. /// 查询索引最大的一条记录
  210. /// </summary>
  211. /// <typeparam name="T">该数据所属的类型</typeparam>
  212. /// <param name="collectionName">去指定查询的集合</param>
  213. /// <param name="sort">排序字段</param>
  214. /// <returns>返回一个实体类型</returns>
  215. public T FindOneToIndexMax<T>(IMongoQuery query, string[] sort)
  216. {
  217. string collectionName = typeof(T).Name;
  218. return FindOneToIndexMax<T>(query, collectionName, sort);
  219. }
  220. /// <summary>
  221. /// 查询索引最大的一条记录
  222. /// </summary>
  223. /// <typeparam name="T">该数据所属的类型</typeparam>
  224. /// <param name="collectionName">去指定查询的集合</param>
  225. /// <param name="sort">排序字段</param>
  226. /// <returns>返回一个实体类型</returns>
  227. public T FindOneToIndexMax<T>(string collectionName, string[] sort)
  228. {
  229. return FindOneToIndexMax<T>(null, collectionName, sort);
  230. }
  231. /// <summary>
  232. /// 查询索引最大的一条记录
  233. /// </summary>
  234. /// <typeparam name="T">该数据所属的类型</typeparam>
  235. /// <param name="query">查询的条件 可以为空</param>
  236. /// <param name="collectionName">去指定查询的集合</param>
  237. /// <param name="sort">排序字段</param>
  238. /// <returns>返回一个实体类型</returns>
  239. public T FindOneToIndexMax<T>(IMongoQuery query, string collectionName, string[] sort)
  240. {
  241. if (this._db == null)
  242. {
  243. return default(T);
  244. }
  245. try
  246. {
  247. MongoCollection<T> mc = this._db.GetCollection<T>(collectionName);
  248. query = this.InitQuery(query);
  249. //var result = mc.Find(query).SetSkip(20).SetLimit(10).SetSortOrder(SortBy.Ascending("CreateOn")).ToList();
  250. T t = mc.Find(query).SetSortOrder(SortBy.Descending(sort)).FirstOrDefault<T>();
  251. return t;
  252. }
  253. catch (Exception ex)
  254. {
  255. return default(T);
  256. }
  257. }
  258. /// <summary>
  259. /// 查询索引最大的一条记录
  260. /// </summary>
  261. /// <typeparam name="T">该数据所属的类型</typeparam>
  262. /// <param name="query">查询的条件 可以为空</param>
  263. /// <param name="collectionName">去指定查询的集合</param>
  264. /// <param name="sort">排序字段</param>
  265. /// <returns>返回一个实体类型</returns>
  266. public T FindOneToIndexMaxAsc<T>(IMongoQuery query, string collectionName, string[] sort)
  267. {
  268. if (this._db == null)
  269. {
  270. return default(T);
  271. }
  272. try
  273. {
  274. MongoCollection<T> mc = this._db.GetCollection<T>(collectionName);
  275. query = this.InitQuery(query);
  276. //var result = mc.Find(query).SetSkip(20).SetLimit(10).SetSortOrder(SortBy.Ascending("CreateOn")).ToList();
  277. T t = mc.Find(query).SetSortOrder(SortBy.Ascending(sort)).FirstOrDefault<T>();
  278. return t;
  279. }
  280. catch (Exception ex)
  281. {
  282. return default(T);
  283. }
  284. }
  285. /// <summary>
  286. /// 查询一条记录
  287. /// </summary>
  288. /// <typeparam name="T">该数据所属的类型</typeparam>
  289. /// <param name="query">查询的条件 可以为空</param>
  290. /// <param name="collectionName">去指定查询的集合</param>
  291. /// <returns>返回一个实体类型</returns>
  292. public T FindOne<T>(IMongoQuery query, string collectionName)
  293. {
  294. if (this._db == null)
  295. {
  296. return default(T);
  297. }
  298. try
  299. {
  300. MongoCollection<T> mc = this._db.GetCollection<T>(collectionName);
  301. query = this.InitQuery(query);
  302. T t = mc.FindOne(query);
  303. return t;
  304. }
  305. catch (Exception ex)
  306. {
  307. return default(T);
  308. }
  309. }
  310. /// <summary>
  311. /// 查询一条记录
  312. /// </summary>
  313. /// <typeparam name="T">该数据所属的类型</typeparam>
  314. /// <param name="collectionName">去指定查询的集合</param>
  315. /// <returns>返回一个实体类型</returns>
  316. public T FindOne<T>(string collectionName)
  317. {
  318. return FindOne<T>(null, collectionName);
  319. }
  320. /// <summary>
  321. /// 查询一条记录
  322. /// </summary>
  323. /// <typeparam name="T">该数据所属的类型</typeparam>
  324. /// <returns>返回一个实体类型</returns>
  325. public T FindOne<T>()
  326. {
  327. string collectionName = typeof(T).Name;
  328. return FindOne<T>(null, collectionName);
  329. }
  330. /// <summary>
  331. /// 查询一条记录
  332. /// </summary>
  333. /// <typeparam name="T">该数据所属的类型</typeparam>
  334. /// <param name="query">查询的条件 可以为空</param>
  335. /// <returns>返回一个实体类型</returns>
  336. public T FindOne<T>(IMongoQuery query)
  337. {
  338. string collectionName = typeof(T).Name;
  339. return FindOne<T>(query, collectionName);
  340. }
  341. #endregion
  342. #region 普通的条件查询
  343. /// <summary>
  344. /// 根据指定条件查询集合中的数据
  345. /// </summary>
  346. /// <typeparam name="T">该集合数据的所属类型</typeparam>
  347. /// <param name="query">指定的查询条件 比如Query.And(Query.EQ("username","admin"),Query.EQ("password":"admin"))</param>
  348. /// <param name="collectionName">指定的集合的名称</param>
  349. /// <returns>返回一个List列表</returns>
  350. public List<T> Find<T>(IMongoQuery query, string collectionName)
  351. {
  352. if (this._db == null)
  353. {
  354. return null;
  355. }
  356. try
  357. {
  358. MongoCollection<T> mc = this._db.GetCollection<T>(collectionName);
  359. query = this.InitQuery(query);
  360. MongoCursor<T> mongoCursor = mc.Find(query);
  361. return mongoCursor.ToList<T>();
  362. }
  363. catch (Exception ex)
  364. {
  365. return null;
  366. }
  367. }
  368. /// <summary>
  369. /// 根据指定条件查询集合中的数据
  370. /// </summary>
  371. /// <typeparam name="T">该集合数据的所属类型</typeparam>
  372. /// <param name="query">指定的查询条件 比如Query.And(Query.EQ("username","admin"),Query.EQ("password":"admin"))</param>
  373. /// <returns>返回一个List列表</returns>
  374. public List<T> Find<T>(IMongoQuery query)
  375. {
  376. string collectionName = typeof(T).Name;
  377. return this.Find<T>(query, collectionName);
  378. }
  379. #endregion
  380. #region 分页查询 PageIndex和PageSize模式 在页数PageIndex大的情况下 效率明显变低
  381. /// <summary>
  382. /// 分页查询 PageIndex和PageSize模式 在页数PageIndex大的情况下 效率明显变低
  383. /// </summary>
  384. /// <typeparam name="T">所需查询的数据的实体类型</typeparam>
  385. /// <param name="query">查询的条件</param>
  386. /// <param name="pageIndex">当前的页数</param>
  387. /// <param name="pageSize">当前的尺寸</param>
  388. /// <param name="sortBy">排序方式</param>
  389. /// <param name="collectionName">集合名称</param>
  390. /// <returns>返回List列表</returns>
  391. public List<T> Find<T>(IMongoQuery query, int pageIndex, int pageSize, SortByDocument sortBy, string collectionName)
  392. {
  393. if (this._db == null)
  394. {
  395. return null;
  396. }
  397. try
  398. {
  399. MongoCollection<T> mc = this._db.GetCollection<T>(collectionName);
  400. MongoCursor<T> mongoCursor = null;
  401. query = this.InitQuery(query);
  402. sortBy = this.InitSortBy(sortBy, OBJECTID_KEY);
  403. //如页序号为0时初始化为1
  404. pageIndex = pageIndex == 0 ? 1 : pageIndex;
  405. //按条件查询 排序 跳数 取数
  406. mongoCursor = mc.Find(query).SetSortOrder(sortBy).SetSkip((pageIndex - 1) * pageSize).SetLimit(pageSize);
  407. return mongoCursor.ToList<T>();
  408. }
  409. catch (Exception ex)
  410. {
  411. return null;
  412. }
  413. }
  414. /// <summary>
  415. /// 分页查询 PageIndex和PageSize模式 在页数PageIndex大的情况下 效率明显变低
  416. /// </summary>
  417. /// <typeparam name="T">所需查询的数据的实体类型</typeparam>
  418. /// <param name="query">查询的条件</param>
  419. /// <param name="pageIndex">当前的页数</param>
  420. /// <param name="pageSize">当前的尺寸</param>
  421. /// <param name="sortBy">排序方式</param>
  422. /// <returns>返回List列表</returns>
  423. public List<T> Find<T>(IMongoQuery query, int pageIndex, int pageSize, SortByDocument sortBy)
  424. {
  425. string collectionName = typeof(T).Name;
  426. return this.Find<T>(query, pageIndex, pageSize, sortBy, collectionName);
  427. }
  428. #endregion
  429. #region 分页查询 指定索引最后项-PageSize模式
  430. /// <summary>
  431. /// 分页查询 指定索引最后项-PageSize模式
  432. /// </summary>
  433. /// <typeparam name="T">所需查询的数据的实体类型</typeparam>
  434. /// <param name="query">查询的条件 没有可以为null</param>
  435. /// <param name="indexName">索引名称</param>
  436. /// <param name="lastKeyValue">最后索引的值</param>
  437. /// <param name="pageSize">分页的尺寸</param>
  438. /// <param name="sortType">排序类型 1升序 -1降序 仅仅针对该索引</param>
  439. /// <param name="collectionName">指定的集合名称</param>
  440. /// <returns>返回一个List列表数据</returns>
  441. public List<T> Find<T>(IMongoQuery query, string indexName, object lastKeyValue, int pageSize, int sortType, string collectionName)
  442. {
  443. if (this._db == null)
  444. {
  445. return null;
  446. }
  447. try
  448. {
  449. MongoCollection<T> mc = this._db.GetCollection<T>(collectionName);
  450. MongoCursor<T> mongoCursor = null;
  451. query = this.InitQuery(query);
  452. //判断升降序后进行查询
  453. if (sortType > 0)
  454. {
  455. //升序
  456. if (lastKeyValue != null)
  457. {
  458. //有上一个主键的值传进来时才添加上一个主键的值的条件
  459. query = Query.And(query, Query.GT(indexName, BsonValue.Create(lastKeyValue)));
  460. }
  461. //先按条件查询 再排序 再取数
  462. mongoCursor = mc.Find(query).SetSortOrder(new SortByDocument(indexName, 1)).SetLimit(pageSize);
  463. }
  464. else
  465. {
  466. //降序
  467. if (lastKeyValue != null)
  468. {
  469. query = Query.And(query, Query.LT(indexName, BsonValue.Create(lastKeyValue)));
  470. }
  471. mongoCursor = mc.Find(query).SetSortOrder(new SortByDocument(indexName, -1)).SetLimit(pageSize);
  472. }
  473. return mongoCursor.ToList<T>();
  474. }
  475. catch (Exception ex)
  476. {
  477. return null;
  478. }
  479. }
  480. /// <summary>
  481. /// 分页查询 指定索引最后项-PageSize模式
  482. /// </summary>
  483. /// <typeparam name="T">所需查询的数据的实体类型</typeparam>
  484. /// <param name="query">查询的条件 没有可以为null</param>
  485. /// <param name="indexName">索引名称</param>
  486. /// <param name="lastKeyValue">最后索引的值</param>
  487. /// <param name="pageSize">分页的尺寸</param>
  488. /// <param name="sortType">排序类型 1升序 -1降序 仅仅针对该索引</param>
  489. /// <returns>返回一个List列表数据</returns>
  490. public List<T> Find<T>(IMongoQuery query, string indexName, object lastKeyValue, int pageSize, int sortType)
  491. {
  492. string collectionName = typeof(T).Name;
  493. return this.Find<T>(query, indexName, lastKeyValue, pageSize, sortType, collectionName);
  494. }
  495. /// <summary>
  496. /// 分页查询 指定ObjectId最后项-PageSize模式
  497. /// </summary>
  498. /// <typeparam name="T">所需查询的数据的实体类型</typeparam>
  499. /// <param name="query">查询的条件 没有可以为null</param>
  500. /// <param name="lastObjectId">上一条记录的ObjectId 没有可以为空</param>
  501. /// <param name="pageSize">每页尺寸</param>
  502. /// <param name="sortType">排序类型 1升序 -1降序 仅仅针对_id</param>
  503. /// <param name="collectionName">指定去查询集合的名称</param>
  504. /// <returns>返回一个List列表数据</returns>
  505. public List<T> Find<T>(IMongoQuery query, string lastObjectId, int pageSize, int sortType, string collectionName)
  506. {
  507. return this.Find<T>(query, OBJECTID_KEY, new ObjectId(lastObjectId), pageSize, sortType, collectionName);
  508. }
  509. /// <summary>
  510. /// 分页查询 指定ObjectId最后项-PageSize模式
  511. /// </summary>
  512. /// <typeparam name="T">所需查询的数据的实体类型</typeparam>
  513. /// <param name="query">查询的条件 没有可以为null</param>
  514. /// <param name="lastObjectId">上一条记录的ObjectId 没有可以为空</param>
  515. /// <param name="pageSize">每页尺寸</param>
  516. /// <param name="sortType">排序类型 1升序 -1降序 仅仅针对_id</param>
  517. /// <returns>返回一个List列表数据</returns>
  518. public List<T> Find<T>(IMongoQuery query, string lastObjectId, int pageSize, int sortType)
  519. {
  520. string collectionName = typeof(T).Name;
  521. return Find<T>(query, lastObjectId, pageSize, sortType, collectionName);
  522. }
  523. #endregion
  524. #endregion
  525. #region 更新数据
  526. /// <summary>
  527. /// 更新数据
  528. /// </summary>
  529. /// <typeparam name="T">更新的数据 所属的类型</typeparam>
  530. /// <param name="query">更新数据的查询</param>
  531. /// <param name="update">需要更新的文档</param>
  532. /// <param name="collectionName">指定更新集合的名称</param>
  533. public bool Update<T>(IMongoQuery query, IMongoUpdate update, string collectionName)
  534. {
  535. if (this._db == null)
  536. {
  537. return false;
  538. }
  539. try
  540. {
  541. MongoCollection<T> mc = this._db.GetCollection<T>(collectionName);
  542. query = this.InitQuery(query);
  543. //更新数据
  544. WriteConcernResult result = mc.Update(query, update, UpdateFlags.Multi);
  545. return true;
  546. }
  547. catch (Exception ex)
  548. {
  549. return false;
  550. }
  551. }
  552. ///// <summary>
  553. ///// 更新数据
  554. ///// </summary>
  555. //public bool Update<T>(string collectionName, Expression<Func<T, bool>> t)
  556. //{
  557. // var col = this._db.GetCollection<T>(collectionName);
  558. // //查出Name值为xumingxiang的第一条记录
  559. // T users = col.FindOne<T>(t);
  560. // object obj = new object();
  561. // //或者
  562. // //Users users = col.FindOne(new Document { { "Name", "xumingxiang" } });
  563. //}
  564. /// <summary>
  565. /// 更新数据
  566. /// </summary>
  567. /// <typeparam name="T">更新的数据 所属的类型</typeparam>
  568. /// <param name="query">更新数据的查询</param>
  569. /// <param name="update">需要更新的文档</param>
  570. public bool Update<T>(IMongoQuery query, IMongoUpdate update)
  571. {
  572. string collectionName = typeof(T).Name;
  573. return this.Update<T>(query, update, collectionName);
  574. }
  575. #endregion
  576. #region 移除/删除数据
  577. /// <summary>
  578. /// 移除指定的数据
  579. /// </summary>
  580. /// <typeparam name="T">移除的数据类型</typeparam>
  581. /// <param name="query">移除的数据条件</param>
  582. /// <param name="collectionName">指定的集合名词</param>
  583. public bool Remove<T>(IMongoQuery query, string collectionName)
  584. {
  585. if (this._db == null)
  586. {
  587. return false;
  588. }
  589. try
  590. {
  591. MongoCollection<T> mc = this._db.GetCollection<T>(collectionName);
  592. query = this.InitQuery(query);
  593. //根据指定查询移除数据
  594. mc.Remove(query);
  595. return true;
  596. }
  597. catch (Exception ex)
  598. {
  599. return false;
  600. }
  601. }
  602. /// <summary>
  603. /// 移除指定的数据
  604. /// </summary>
  605. /// <typeparam name="T">移除的数据类型</typeparam>
  606. /// <param name="query">移除的数据条件</param>
  607. public bool Remove<T>(IMongoQuery query)
  608. {
  609. string collectionName = typeof(T).Name;
  610. return this.Remove<T>(query, collectionName);
  611. }
  612. /// <summary>
  613. /// 移除实体里面所有的数据
  614. /// </summary>
  615. /// <typeparam name="T">移除的数据类型</typeparam>
  616. public bool ReomveAll<T>()
  617. {
  618. string collectionName = typeof(T).Name;
  619. return this.Remove<T>(null, collectionName);
  620. }
  621. /// <summary>
  622. /// 移除实体里面所有的数据
  623. /// </summary>
  624. /// <typeparam name="T">移除的数据类型</typeparam>
  625. /// <param name="collectionName">指定的集合名称</param>
  626. public bool RemoveAll<T>(string collectionName)
  627. {
  628. return this.Remove<T>(null, collectionName);
  629. }
  630. #endregion
  631. #region 创建索引
  632. /// <summary>
  633. /// 创建索引
  634. /// </summary>
  635. /// <typeparam name="T">需要创建索引的实体类型</typeparam>
  636. public bool CreateIndex<T>()
  637. {
  638. if (this._db == null)
  639. {
  640. return false;
  641. }
  642. try
  643. {
  644. string collectionName = typeof(T).Name;
  645. MongoCollection<BsonDocument> mc = this._db.GetCollection<BsonDocument>(collectionName);
  646. PropertyInfo[] propertys = typeof(T).GetProperties(BindingFlags.IgnoreCase | BindingFlags.Instance | BindingFlags.Public | BindingFlags.SetProperty);
  647. //得到该实体类型的属性
  648. foreach (PropertyInfo property in propertys)
  649. {
  650. //在各个属性中得到其特性
  651. foreach (object obj in property.GetCustomAttributes(true))
  652. {
  653. MongoDbFieldAttribute mongoField = obj as MongoDbFieldAttribute;
  654. if (mongoField != null)
  655. {// 此特性为mongodb的字段属性
  656. IndexKeysBuilder indexKey;
  657. if (mongoField.Ascending)
  658. {
  659. //升序 索引
  660. indexKey = IndexKeys.Ascending(property.Name);
  661. }
  662. else
  663. {
  664. //降序索引
  665. indexKey = IndexKeys.Descending(property.Name);
  666. }
  667. //创建该属性
  668. mc.CreateIndex(indexKey, IndexOptions.SetUnique(mongoField.Unique));
  669. }
  670. }
  671. }
  672. return true;
  673. }
  674. catch (Exception ex)
  675. {
  676. return false;
  677. }
  678. }
  679. #endregion
  680. #region 获取表的行数
  681. /// <summary>
  682. /// 获取数据表总行数
  683. /// </summary>
  684. /// <typeparam name="T"></typeparam>
  685. /// <param name="query"></param>
  686. /// <param name="collectionName"></param>
  687. /// <returns></returns>
  688. public long GetCount<T>(IMongoQuery query, string collectionName)
  689. {
  690. if (this._db == null)
  691. {
  692. return 0;
  693. }
  694. try
  695. {
  696. MongoCollection<T> mc = this._db.GetCollection<T>(collectionName);
  697. if (query == null)
  698. {
  699. return mc.Count();
  700. }
  701. else
  702. {
  703. return mc.Count(query);
  704. }
  705. }
  706. catch (Exception ex)
  707. {
  708. return 0;
  709. }
  710. }
  711. /// <summary>
  712. /// 获取数据表总行数
  713. /// </summary>
  714. /// <typeparam name="T"></typeparam>
  715. /// <param name="query"></param>
  716. /// <returns></returns>
  717. public long GetCount<T>(IMongoQuery query)
  718. {
  719. string collectionName = typeof(T).Name;
  720. return GetCount<T>(query, collectionName);
  721. }
  722. #endregion
  723. #region 获取集合的存储大小
  724. /// <summary>
  725. /// 获取集合的存储大小
  726. /// </summary>
  727. /// <typeparam name="T">该集合对应的实体类</typeparam>
  728. /// <returns>返回一个long型</returns>
  729. public long GetDataSize<T>()
  730. {
  731. string collectionName = typeof(T).Name;
  732. return GetDataSize(collectionName);
  733. }
  734. /// <summary>
  735. /// 获取集合的存储大小
  736. /// </summary>
  737. /// <param name="collectionName">该集合对应的名称</param>
  738. /// <returns>返回一个long型</returns>
  739. public long GetDataSize(string collectionName)
  740. {
  741. if (this._db == null)
  742. {
  743. return 0;
  744. }
  745. try
  746. {
  747. MongoCollection<BsonDocument> mc = this._db.GetCollection<BsonDocument>(collectionName);
  748. return mc.GetTotalStorageSize();
  749. }
  750. catch (Exception ex)
  751. {
  752. return 0;
  753. }
  754. }
  755. #endregion
  756. #region 私有的一些辅助方法
  757. /// <summary>
  758. /// 初始化查询记录 主要当该查询条件为空时 会附加一个恒真的查询条件,防止空查询报错
  759. /// </summary>
  760. /// <param name="query">查询的条件</param>
  761. /// <returns></returns>
  762. private IMongoQuery InitQuery(IMongoQuery query)
  763. {
  764. if (query == null)
  765. {
  766. //当查询为空时 附加恒真的条件 类似SQL:1=1的语法
  767. query = Query.Exists(OBJECTID_KEY);
  768. }
  769. return query;
  770. }
  771. /// <summary>
  772. /// 初始化排序条件 主要当条件为空时 会默认以ObjectId递增的一个排序
  773. /// </summary>
  774. /// <param name="sortBy"></param>
  775. /// <param name="sortByName"></param>
  776. /// <returns></returns>
  777. private SortByDocument InitSortBy(SortByDocument sortBy, string sortByName)
  778. {
  779. if (sortBy == null)
  780. {
  781. //默认ObjectId 递增
  782. sortBy = new SortByDocument(sortByName, -1);
  783. }
  784. return sortBy;
  785. }
  786. private UpdateDocument InitUpdateDocument(UpdateDocument update, string indexName)
  787. {
  788. if (update == null)
  789. {
  790. update = new UpdateDocument("$inc", new QueryDocument(indexName, 0));
  791. }
  792. return update;
  793. }
  794. #endregion
  795. }
  796. }