123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864 |
- using MongoDB.Bson;
- using MongoDB.Driver;
- using MongoDB.Driver.Builders;
- using System;
- using System.Collections.Generic;
- using System.Linq;
- using System.Linq.Expressions;
- using System.Reflection;
- using System.Text;
- using System.Threading.Tasks;
- namespace Ant.Service.Mongodb
- {
- /// <summary>
- /// Mongo db的数据库帮助类
- /// </summary>
- public class MongoDbHelper
- {
- /// <summary>
- /// 数据库的实例
- /// </summary>
- private MongoDatabase _db;
- /// <summary>
- /// ObjectId的键
- /// </summary>
- private readonly string OBJECTID_KEY = "_id";
- public MongoDbHelper(string host, string timeOut)
- {
- this._db = new MongoDb(host, timeOut).GetDataBase();
- }
- public MongoDbHelper(MongoDatabase db)
- {
- this._db = db;
- }
- #region 查询实例
- //Query.All(“name”, “a”, “b”);//通过多个元素来匹配数组
- //Query.And(Query.EQ(“name”, “a”), Query.EQ(“title”, “t”));//同时满足多个条件
- //Query.EQ(“name”, “a”);//等于
- //Query.Exists(“type”, true);//判断键值是否存在
- //Query.GT(“value”, 2);//大于>
- //Query.GTE(“value”, 3);//大于等于>=
- //Query.In(“name”, “a”, “b”);//包括指定的所有值,可以指定不同类型的条件和值
- //Query.LT(“value”, 9);//小于<
- //Query.LTE(“value”, 8);//小于等于<=
- //Query.Mod(“value”, 3, 1);//将查询值除以第一个给定值,若余数等于第二个给定值则返回该结果
- //Query.NE(“name”, “c”);//不等于
- //Query.Nor(Array);//不包括数组中的值
- //Query.Not(“name”);//元素条件语句
- //Query.NotIn(“name”, “a”, 2);//返回与数组中所有条件都不匹配的文档
- //Query.Or(Query.EQ(“name”, “a”), Query.EQ(“title”, “t”));//满足其中一个条件
- //Query.Size(“name”, 2);//给定键的长度
- //Query.Type(“_id”, BsonType.ObjectId);//给定键的类型
- //Query.Where(BsonJavaScript);//执行JavaScript
- //Query.Matches(“Title”, str);//模糊查询 相当于sql中like – str可包含正则表达式
- #endregion
- public T GetNextSequence<T>(IMongoQuery query, SortByDocument sortBy, UpdateDocument update, string collectionName, string indexName)
- {
- if (this._db == null)
- {
- return default(T);
- }
- try
- {
- MongoCollection<T> mc = this._db.GetCollection<T>(collectionName);
- query = this.InitQuery(query);
- sortBy = this.InitSortBy(sortBy, OBJECTID_KEY);
- update = this.InitUpdateDocument(update, indexName);
- var ido = mc.FindAndModify(query, sortBy, update, true, false);
- return ido.GetModifiedDocumentAs<T>();
- }
- catch (Exception ex)
- {
- return default(T);
- }
- }
- #region 插入数据
- /// <summary>
- /// 将数据插入进数据库
- /// </summary>
- /// <typeparam name="T">需要插入数据的类型</typeparam>
- /// <param name="t">需要插入的具体实体</param>
- public bool Insert<T>(T t)
- {
- //集合名称
- string collectionName = typeof(T).Name;
- return Insert<T>(t, collectionName);
- }
- /// <summary>
- /// 将数据插入进数据库
- /// </summary>
- /// <typeparam name="T">需要插入数据库的实体类型</typeparam>
- /// <param name="t">需要插入数据库的具体实体</param>
- /// <param name="collectionName">指定插入的集合</param>
- public bool Insert<T>(T t, string collectionName)
- {
- if (this._db == null)
- {
- return false;
- }
- try
- {
- MongoCollection<BsonDocument> mc = this._db.GetCollection<BsonDocument>(collectionName);
- //将实体转换为bson文档
- BsonDocument bd = t.ToBsonDocument();
- //进行插入操作
- WriteConcernResult result = mc.Insert(bd);
- if (!string.IsNullOrEmpty(result.ErrorMessage))
- {
- return false;
- }
- return true;
- }
- catch (Exception ex)
- {
- return false;
- }
- }
- /// <summary>
- /// 批量插入数据
- /// </summary>
- /// <typeparam name="T">需要插入数据库的实体类型</typeparam>
- /// <param name="list">需要插入数据的列表</param>
- public bool Insert<T>(List<T> list)
- {
- //集合名称
- string collectionName = typeof(T).Name;
- return this.Insert<T>(list, collectionName);
- }
- /// <summary>
- /// 批量插入数据
- /// </summary>
- /// <typeparam name="T">需要插入数据库的实体类型</typeparam>
- /// <param name="list">需要插入数据的列表</param>
- /// <param name="collectionName">指定要插入的集合</param>
- public bool Insert<T>(List<T> list, string collectionName)
- {
- if (this._db == null)
- {
- return false;
- }
- try
- {
- MongoCollection<BsonDocument> mc = this._db.GetCollection<BsonDocument>(collectionName);
- //创建一个空间bson集合
- List<BsonDocument> bsonList = new List<BsonDocument>();
- //批量将数据转为bson格式 并且放进bson文档
- list.ForEach(t => bsonList.Add(t.ToBsonDocument()));
- //批量插入数据
- mc.InsertBatch(bsonList);
- return true;
- }
- catch (Exception ex)
- {
- return false;
- }
- }
- #endregion
- #region 查询数据
- #region 查询所有记录
- /// <summary>
- /// 查询一个集合中的所有数据
- /// </summary>
- /// <typeparam name="T">该集合数据的所属类型</typeparam>
- /// <param name="collectionName">指定集合的名称</param>
- /// <returns>返回一个List列表</returns>
- public List<T> FindAll<T>(string collectionName)
- {
- if (this._db == null)
- {
- return null;
- }
- try
- {
- MongoCollection<T> mc = this._db.GetCollection<T>(collectionName);
- //以实体方式取出其数据集合
- MongoCursor<T> mongoCursor = mc.FindAll();
- //直接转化为List返回
- return mongoCursor.ToList<T>();
- }
- catch (Exception ex)
- {
- return null;
- }
- }
- /// <summary>
- /// 查询一个集合中的所有数据 其集合的名称为T的名称
- /// </summary>
- /// <typeparam name="T">该集合数据的所属类型</typeparam>
- /// <returns>返回一个List列表</returns>
- public List<T> FindAll<T>()
- {
- string collectionName = typeof(T).Name;
- return FindAll<T>(collectionName);
- }
- #endregion
- #region 查询一条记录
- /// <summary>
- /// 查询索引最大的一条记录
- /// </summary>
- /// <typeparam name="T">该数据所属的类型</typeparam>
- /// <param name="collectionName">去指定查询的集合</param>
- /// <param name="sort">排序字段</param>
- /// <returns>返回一个实体类型</returns>
- public T FindOneToIndexMax<T>(string[] sort)
- {
- string collectionName = typeof(T).Name;
- return FindOneToIndexMax<T>(null, collectionName, sort);
- }
- /// <summary>
- /// 查询索引最大的一条记录
- /// </summary>
- /// <typeparam name="T">该数据所属的类型</typeparam>
- /// <param name="collectionName">去指定查询的集合</param>
- /// <param name="sort">排序字段</param>
- /// <returns>返回一个实体类型</returns>
- public T FindOneToIndexMax<T>(IMongoQuery query, string[] sort)
- {
- string collectionName = typeof(T).Name;
- return FindOneToIndexMax<T>(query, collectionName, sort);
- }
- /// <summary>
- /// 查询索引最大的一条记录
- /// </summary>
- /// <typeparam name="T">该数据所属的类型</typeparam>
- /// <param name="collectionName">去指定查询的集合</param>
- /// <param name="sort">排序字段</param>
- /// <returns>返回一个实体类型</returns>
- public T FindOneToIndexMax<T>(string collectionName, string[] sort)
- {
- return FindOneToIndexMax<T>(null, collectionName, sort);
- }
- /// <summary>
- /// 查询索引最大的一条记录
- /// </summary>
- /// <typeparam name="T">该数据所属的类型</typeparam>
- /// <param name="query">查询的条件 可以为空</param>
- /// <param name="collectionName">去指定查询的集合</param>
- /// <param name="sort">排序字段</param>
- /// <returns>返回一个实体类型</returns>
- public T FindOneToIndexMax<T>(IMongoQuery query, string collectionName, string[] sort)
- {
- if (this._db == null)
- {
- return default(T);
- }
- try
- {
- MongoCollection<T> mc = this._db.GetCollection<T>(collectionName);
- query = this.InitQuery(query);
- //var result = mc.Find(query).SetSkip(20).SetLimit(10).SetSortOrder(SortBy.Ascending("CreateOn")).ToList();
- T t = mc.Find(query).SetSortOrder(SortBy.Descending(sort)).FirstOrDefault<T>();
- return t;
- }
- catch (Exception ex)
- {
- return default(T);
- }
- }
- /// <summary>
- /// 查询索引最大的一条记录
- /// </summary>
- /// <typeparam name="T">该数据所属的类型</typeparam>
- /// <param name="query">查询的条件 可以为空</param>
- /// <param name="collectionName">去指定查询的集合</param>
- /// <param name="sort">排序字段</param>
- /// <returns>返回一个实体类型</returns>
- public T FindOneToIndexMaxAsc<T>(IMongoQuery query, string collectionName, string[] sort)
- {
- if (this._db == null)
- {
- return default(T);
- }
- try
- {
- MongoCollection<T> mc = this._db.GetCollection<T>(collectionName);
- query = this.InitQuery(query);
- //var result = mc.Find(query).SetSkip(20).SetLimit(10).SetSortOrder(SortBy.Ascending("CreateOn")).ToList();
- T t = mc.Find(query).SetSortOrder(SortBy.Ascending(sort)).FirstOrDefault<T>();
- return t;
- }
- catch (Exception ex)
- {
- return default(T);
- }
- }
- /// <summary>
- /// 查询一条记录
- /// </summary>
- /// <typeparam name="T">该数据所属的类型</typeparam>
- /// <param name="query">查询的条件 可以为空</param>
- /// <param name="collectionName">去指定查询的集合</param>
- /// <returns>返回一个实体类型</returns>
- public T FindOne<T>(IMongoQuery query, string collectionName)
- {
- if (this._db == null)
- {
- return default(T);
- }
- try
- {
- MongoCollection<T> mc = this._db.GetCollection<T>(collectionName);
- query = this.InitQuery(query);
- T t = mc.FindOne(query);
- return t;
- }
- catch (Exception ex)
- {
- return default(T);
- }
- }
- /// <summary>
- /// 查询一条记录
- /// </summary>
- /// <typeparam name="T">该数据所属的类型</typeparam>
- /// <param name="collectionName">去指定查询的集合</param>
- /// <returns>返回一个实体类型</returns>
- public T FindOne<T>(string collectionName)
- {
- return FindOne<T>(null, collectionName);
- }
- /// <summary>
- /// 查询一条记录
- /// </summary>
- /// <typeparam name="T">该数据所属的类型</typeparam>
- /// <returns>返回一个实体类型</returns>
- public T FindOne<T>()
- {
- string collectionName = typeof(T).Name;
- return FindOne<T>(null, collectionName);
- }
- /// <summary>
- /// 查询一条记录
- /// </summary>
- /// <typeparam name="T">该数据所属的类型</typeparam>
- /// <param name="query">查询的条件 可以为空</param>
- /// <returns>返回一个实体类型</returns>
- public T FindOne<T>(IMongoQuery query)
- {
- string collectionName = typeof(T).Name;
- return FindOne<T>(query, collectionName);
- }
- #endregion
- #region 普通的条件查询
- /// <summary>
- /// 根据指定条件查询集合中的数据
- /// </summary>
- /// <typeparam name="T">该集合数据的所属类型</typeparam>
- /// <param name="query">指定的查询条件 比如Query.And(Query.EQ("username","admin"),Query.EQ("password":"admin"))</param>
- /// <param name="collectionName">指定的集合的名称</param>
- /// <returns>返回一个List列表</returns>
- public List<T> Find<T>(IMongoQuery query, string collectionName)
- {
- if (this._db == null)
- {
- return null;
- }
- try
- {
- MongoCollection<T> mc = this._db.GetCollection<T>(collectionName);
- query = this.InitQuery(query);
- MongoCursor<T> mongoCursor = mc.Find(query);
- return mongoCursor.ToList<T>();
- }
- catch (Exception ex)
- {
- return null;
- }
- }
- /// <summary>
- /// 根据指定条件查询集合中的数据
- /// </summary>
- /// <typeparam name="T">该集合数据的所属类型</typeparam>
- /// <param name="query">指定的查询条件 比如Query.And(Query.EQ("username","admin"),Query.EQ("password":"admin"))</param>
- /// <returns>返回一个List列表</returns>
- public List<T> Find<T>(IMongoQuery query)
- {
- string collectionName = typeof(T).Name;
- return this.Find<T>(query, collectionName);
- }
- #endregion
- #region 分页查询 PageIndex和PageSize模式 在页数PageIndex大的情况下 效率明显变低
- /// <summary>
- /// 分页查询 PageIndex和PageSize模式 在页数PageIndex大的情况下 效率明显变低
- /// </summary>
- /// <typeparam name="T">所需查询的数据的实体类型</typeparam>
- /// <param name="query">查询的条件</param>
- /// <param name="pageIndex">当前的页数</param>
- /// <param name="pageSize">当前的尺寸</param>
- /// <param name="sortBy">排序方式</param>
- /// <param name="collectionName">集合名称</param>
- /// <returns>返回List列表</returns>
- public List<T> Find<T>(IMongoQuery query, int pageIndex, int pageSize, SortByDocument sortBy, string collectionName)
- {
- if (this._db == null)
- {
- return null;
- }
- try
- {
- MongoCollection<T> mc = this._db.GetCollection<T>(collectionName);
- MongoCursor<T> mongoCursor = null;
- query = this.InitQuery(query);
- sortBy = this.InitSortBy(sortBy, OBJECTID_KEY);
- //如页序号为0时初始化为1
- pageIndex = pageIndex == 0 ? 1 : pageIndex;
- //按条件查询 排序 跳数 取数
- mongoCursor = mc.Find(query).SetSortOrder(sortBy).SetSkip((pageIndex - 1) * pageSize).SetLimit(pageSize);
- return mongoCursor.ToList<T>();
- }
- catch (Exception ex)
- {
- return null;
- }
- }
- /// <summary>
- /// 分页查询 PageIndex和PageSize模式 在页数PageIndex大的情况下 效率明显变低
- /// </summary>
- /// <typeparam name="T">所需查询的数据的实体类型</typeparam>
- /// <param name="query">查询的条件</param>
- /// <param name="pageIndex">当前的页数</param>
- /// <param name="pageSize">当前的尺寸</param>
- /// <param name="sortBy">排序方式</param>
- /// <returns>返回List列表</returns>
- public List<T> Find<T>(IMongoQuery query, int pageIndex, int pageSize, SortByDocument sortBy)
- {
- string collectionName = typeof(T).Name;
- return this.Find<T>(query, pageIndex, pageSize, sortBy, collectionName);
- }
- #endregion
- #region 分页查询 指定索引最后项-PageSize模式
- /// <summary>
- /// 分页查询 指定索引最后项-PageSize模式
- /// </summary>
- /// <typeparam name="T">所需查询的数据的实体类型</typeparam>
- /// <param name="query">查询的条件 没有可以为null</param>
- /// <param name="indexName">索引名称</param>
- /// <param name="lastKeyValue">最后索引的值</param>
- /// <param name="pageSize">分页的尺寸</param>
- /// <param name="sortType">排序类型 1升序 -1降序 仅仅针对该索引</param>
- /// <param name="collectionName">指定的集合名称</param>
- /// <returns>返回一个List列表数据</returns>
- public List<T> Find<T>(IMongoQuery query, string indexName, object lastKeyValue, int pageSize, int sortType, string collectionName)
- {
- if (this._db == null)
- {
- return null;
- }
- try
- {
- MongoCollection<T> mc = this._db.GetCollection<T>(collectionName);
- MongoCursor<T> mongoCursor = null;
- query = this.InitQuery(query);
- //判断升降序后进行查询
- if (sortType > 0)
- {
- //升序
- if (lastKeyValue != null)
- {
- //有上一个主键的值传进来时才添加上一个主键的值的条件
- query = Query.And(query, Query.GT(indexName, BsonValue.Create(lastKeyValue)));
- }
- //先按条件查询 再排序 再取数
- mongoCursor = mc.Find(query).SetSortOrder(new SortByDocument(indexName, 1)).SetLimit(pageSize);
- }
- else
- {
- //降序
- if (lastKeyValue != null)
- {
- query = Query.And(query, Query.LT(indexName, BsonValue.Create(lastKeyValue)));
- }
- mongoCursor = mc.Find(query).SetSortOrder(new SortByDocument(indexName, -1)).SetLimit(pageSize);
- }
- return mongoCursor.ToList<T>();
- }
- catch (Exception ex)
- {
- return null;
- }
- }
- /// <summary>
- /// 分页查询 指定索引最后项-PageSize模式
- /// </summary>
- /// <typeparam name="T">所需查询的数据的实体类型</typeparam>
- /// <param name="query">查询的条件 没有可以为null</param>
- /// <param name="indexName">索引名称</param>
- /// <param name="lastKeyValue">最后索引的值</param>
- /// <param name="pageSize">分页的尺寸</param>
- /// <param name="sortType">排序类型 1升序 -1降序 仅仅针对该索引</param>
- /// <returns>返回一个List列表数据</returns>
- public List<T> Find<T>(IMongoQuery query, string indexName, object lastKeyValue, int pageSize, int sortType)
- {
- string collectionName = typeof(T).Name;
- return this.Find<T>(query, indexName, lastKeyValue, pageSize, sortType, collectionName);
- }
- /// <summary>
- /// 分页查询 指定ObjectId最后项-PageSize模式
- /// </summary>
- /// <typeparam name="T">所需查询的数据的实体类型</typeparam>
- /// <param name="query">查询的条件 没有可以为null</param>
- /// <param name="lastObjectId">上一条记录的ObjectId 没有可以为空</param>
- /// <param name="pageSize">每页尺寸</param>
- /// <param name="sortType">排序类型 1升序 -1降序 仅仅针对_id</param>
- /// <param name="collectionName">指定去查询集合的名称</param>
- /// <returns>返回一个List列表数据</returns>
- public List<T> Find<T>(IMongoQuery query, string lastObjectId, int pageSize, int sortType, string collectionName)
- {
- return this.Find<T>(query, OBJECTID_KEY, new ObjectId(lastObjectId), pageSize, sortType, collectionName);
- }
- /// <summary>
- /// 分页查询 指定ObjectId最后项-PageSize模式
- /// </summary>
- /// <typeparam name="T">所需查询的数据的实体类型</typeparam>
- /// <param name="query">查询的条件 没有可以为null</param>
- /// <param name="lastObjectId">上一条记录的ObjectId 没有可以为空</param>
- /// <param name="pageSize">每页尺寸</param>
- /// <param name="sortType">排序类型 1升序 -1降序 仅仅针对_id</param>
- /// <returns>返回一个List列表数据</returns>
- public List<T> Find<T>(IMongoQuery query, string lastObjectId, int pageSize, int sortType)
- {
- string collectionName = typeof(T).Name;
- return Find<T>(query, lastObjectId, pageSize, sortType, collectionName);
- }
- #endregion
- #endregion
- #region 更新数据
- /// <summary>
- /// 更新数据
- /// </summary>
- /// <typeparam name="T">更新的数据 所属的类型</typeparam>
- /// <param name="query">更新数据的查询</param>
- /// <param name="update">需要更新的文档</param>
- /// <param name="collectionName">指定更新集合的名称</param>
- public bool Update<T>(IMongoQuery query, IMongoUpdate update, string collectionName)
- {
- if (this._db == null)
- {
- return false;
- }
- try
- {
- MongoCollection<T> mc = this._db.GetCollection<T>(collectionName);
- query = this.InitQuery(query);
- //更新数据
- WriteConcernResult result = mc.Update(query, update, UpdateFlags.Multi);
- return true;
- }
- catch (Exception ex)
- {
- return false;
- }
- }
- ///// <summary>
- ///// 更新数据
- ///// </summary>
- //public bool Update<T>(string collectionName, Expression<Func<T, bool>> t)
- //{
- // var col = this._db.GetCollection<T>(collectionName);
- // //查出Name值为xumingxiang的第一条记录
- // T users = col.FindOne<T>(t);
- // object obj = new object();
- // //或者
- // //Users users = col.FindOne(new Document { { "Name", "xumingxiang" } });
- //}
- /// <summary>
- /// 更新数据
- /// </summary>
- /// <typeparam name="T">更新的数据 所属的类型</typeparam>
- /// <param name="query">更新数据的查询</param>
- /// <param name="update">需要更新的文档</param>
- public bool Update<T>(IMongoQuery query, IMongoUpdate update)
- {
- string collectionName = typeof(T).Name;
- return this.Update<T>(query, update, collectionName);
- }
- #endregion
- #region 移除/删除数据
- /// <summary>
- /// 移除指定的数据
- /// </summary>
- /// <typeparam name="T">移除的数据类型</typeparam>
- /// <param name="query">移除的数据条件</param>
- /// <param name="collectionName">指定的集合名词</param>
- public bool Remove<T>(IMongoQuery query, string collectionName)
- {
- if (this._db == null)
- {
- return false;
- }
- try
- {
- MongoCollection<T> mc = this._db.GetCollection<T>(collectionName);
- query = this.InitQuery(query);
- //根据指定查询移除数据
- mc.Remove(query);
- return true;
- }
- catch (Exception ex)
- {
- return false;
- }
- }
- /// <summary>
- /// 移除指定的数据
- /// </summary>
- /// <typeparam name="T">移除的数据类型</typeparam>
- /// <param name="query">移除的数据条件</param>
- public bool Remove<T>(IMongoQuery query)
- {
- string collectionName = typeof(T).Name;
- return this.Remove<T>(query, collectionName);
- }
- /// <summary>
- /// 移除实体里面所有的数据
- /// </summary>
- /// <typeparam name="T">移除的数据类型</typeparam>
- public bool ReomveAll<T>()
- {
- string collectionName = typeof(T).Name;
- return this.Remove<T>(null, collectionName);
- }
- /// <summary>
- /// 移除实体里面所有的数据
- /// </summary>
- /// <typeparam name="T">移除的数据类型</typeparam>
- /// <param name="collectionName">指定的集合名称</param>
- public bool RemoveAll<T>(string collectionName)
- {
- return this.Remove<T>(null, collectionName);
- }
- #endregion
- #region 创建索引
- /// <summary>
- /// 创建索引
- /// </summary>
- /// <typeparam name="T">需要创建索引的实体类型</typeparam>
- public bool CreateIndex<T>()
- {
- if (this._db == null)
- {
- return false;
- }
- try
- {
- string collectionName = typeof(T).Name;
- MongoCollection<BsonDocument> mc = this._db.GetCollection<BsonDocument>(collectionName);
- PropertyInfo[] propertys = typeof(T).GetProperties(BindingFlags.IgnoreCase | BindingFlags.Instance | BindingFlags.Public | BindingFlags.SetProperty);
- //得到该实体类型的属性
- foreach (PropertyInfo property in propertys)
- {
- //在各个属性中得到其特性
- foreach (object obj in property.GetCustomAttributes(true))
- {
- MongoDbFieldAttribute mongoField = obj as MongoDbFieldAttribute;
- if (mongoField != null)
- {// 此特性为mongodb的字段属性
- IndexKeysBuilder indexKey;
- if (mongoField.Ascending)
- {
- //升序 索引
- indexKey = IndexKeys.Ascending(property.Name);
- }
- else
- {
- //降序索引
- indexKey = IndexKeys.Descending(property.Name);
- }
- //创建该属性
- mc.CreateIndex(indexKey, IndexOptions.SetUnique(mongoField.Unique));
- }
- }
- }
- return true;
- }
- catch (Exception ex)
- {
- return false;
- }
- }
- #endregion
- #region 获取表的行数
- /// <summary>
- /// 获取数据表总行数
- /// </summary>
- /// <typeparam name="T"></typeparam>
- /// <param name="query"></param>
- /// <param name="collectionName"></param>
- /// <returns></returns>
- public long GetCount<T>(IMongoQuery query, string collectionName)
- {
- if (this._db == null)
- {
- return 0;
- }
- try
- {
- MongoCollection<T> mc = this._db.GetCollection<T>(collectionName);
- if (query == null)
- {
- return mc.Count();
- }
- else
- {
- return mc.Count(query);
- }
- }
- catch (Exception ex)
- {
- return 0;
- }
- }
- /// <summary>
- /// 获取数据表总行数
- /// </summary>
- /// <typeparam name="T"></typeparam>
- /// <param name="query"></param>
- /// <returns></returns>
- public long GetCount<T>(IMongoQuery query)
- {
- string collectionName = typeof(T).Name;
- return GetCount<T>(query, collectionName);
- }
- #endregion
- #region 获取集合的存储大小
- /// <summary>
- /// 获取集合的存储大小
- /// </summary>
- /// <typeparam name="T">该集合对应的实体类</typeparam>
- /// <returns>返回一个long型</returns>
- public long GetDataSize<T>()
- {
- string collectionName = typeof(T).Name;
- return GetDataSize(collectionName);
- }
- /// <summary>
- /// 获取集合的存储大小
- /// </summary>
- /// <param name="collectionName">该集合对应的名称</param>
- /// <returns>返回一个long型</returns>
- public long GetDataSize(string collectionName)
- {
- if (this._db == null)
- {
- return 0;
- }
- try
- {
- MongoCollection<BsonDocument> mc = this._db.GetCollection<BsonDocument>(collectionName);
- return mc.GetTotalStorageSize();
- }
- catch (Exception ex)
- {
- return 0;
- }
- }
- #endregion
- #region 私有的一些辅助方法
- /// <summary>
- /// 初始化查询记录 主要当该查询条件为空时 会附加一个恒真的查询条件,防止空查询报错
- /// </summary>
- /// <param name="query">查询的条件</param>
- /// <returns></returns>
- private IMongoQuery InitQuery(IMongoQuery query)
- {
- if (query == null)
- {
- //当查询为空时 附加恒真的条件 类似SQL:1=1的语法
- query = Query.Exists(OBJECTID_KEY);
- }
- return query;
- }
- /// <summary>
- /// 初始化排序条件 主要当条件为空时 会默认以ObjectId递增的一个排序
- /// </summary>
- /// <param name="sortBy"></param>
- /// <param name="sortByName"></param>
- /// <returns></returns>
- private SortByDocument InitSortBy(SortByDocument sortBy, string sortByName)
- {
- if (sortBy == null)
- {
- //默认ObjectId 递增
- sortBy = new SortByDocument(sortByName, -1);
- }
- return sortBy;
- }
- private UpdateDocument InitUpdateDocument(UpdateDocument update, string indexName)
- {
- if (update == null)
- {
- update = new UpdateDocument("$inc", new QueryDocument(indexName, 0));
- }
- return update;
- }
- #endregion
- }
- }
|