zsyzhou 发表于 2015-7-6 11:16:18

一个mongodb的helper,封装了一些简单常用的功能

直接上代码吧。不知道为什么插入代码会提示出错,直接黏贴吧。-_-!!!  此项目已开源,开源地址是:
http://mongodbhelper-csharp.googlecode.com/svn/trunk/
  


using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;

using MongoDB.Driver;
using MongoDB.Bson;
using MongoDB.Driver.Builders;

namespace MongoDBHelper
{

    ///
    /// mongodb的封装类。
    /// add by yancong2008@gmail.com 2011.05.14
    ///

    public sealed class MongoDBHelper
    //public sealed class MongoDBHelper
    //where T :class
    {

      public static readonly string connectionString_Default = System.Configuration.ConfigurationManager.AppSettings["ConnectionString_mongoDB"];
      public static readonly string database_Default = System.Configuration.ConfigurationManager.AppSettings["Database_mongoDB"];

      #region 新增

      public static SafeModeResult InsertOne(string collectionName, T entity)
      {
            return MongoDBHelper.InsertOne(MongoDBHelper.connectionString_Default, MongoDBHelper.database_Default, collectionName, entity);
      }

      public static SafeModeResult InsertOne(string connectionString, string databaseName, string collectionName, T entity)
      {
            SafeModeResult result = new SafeModeResult();

            if (null == entity)
            {
                return null;
            }

            MongoServer server = MongoServer.Create(connectionString);

            //获取数据库或者创建数据库(不存在的话)。
            MongoDatabase database = server.GetDatabase(databaseName);



            using (server.RequestStart(database))//开始连接数据库。
            {
                MongoCollection myCollection = database.GetCollection(collectionName);
                result = myCollection.Insert(entity);
            }

            return result;
      }

      public static IEnumerable InsertAll(string collectionName, IEnumerable entitys)
      {
            return MongoDBHelper.InsertAll(MongoDBHelper.connectionString_Default, MongoDBHelper.database_Default, collectionName, entitys);
      }

      public static IEnumerable InsertAll(string connectionString, string databaseName, string collectionName, IEnumerable entitys)
      {
            IEnumerable result = null;

            if (null == entitys)
            {
                return null;
            }

            MongoServer server = MongoServer.Create(connectionString);

            //获取数据库或者创建数据库(不存在的话)。
            MongoDatabase database = server.GetDatabase(databaseName);



            using (server.RequestStart(database))//开始连接数据库。
            {
                MongoCollection myCollection = database.GetCollection(collectionName);
                result = myCollection.InsertBatch(entitys);
            }

            return result;
      }

      #endregion


      #region 修改

      public static SafeModeResult UpdateOne(string collectionName, T entity)
      {
            return MongoDBHelper.UpdateOne(MongoDBHelper.connectionString_Default, MongoDBHelper.database_Default, collectionName, entity);
      }

      public static SafeModeResult UpdateOne(string connectionString, string databaseName, string collectionName, T entity)
      {
            MongoServer server = MongoServer.Create(connectionString);

            //获取数据库或者创建数据库(不存在的话)。
            MongoDatabase database = server.GetDatabase(databaseName);

            SafeModeResult result;

            using (server.RequestStart(database))//开始连接数据库。
            {
                MongoCollection myCollection = database.GetCollection(collectionName);

                result = myCollection.Save(entity);

            }

            return result;
      }

      ///
      ///
      ///
      ///
      ///
      /// 条件查询。 调用示例:Query.Matches("Title", "感冒") 或者 Query.EQ("Title", "感冒") 或者Query.And(Query.Matches("Title", "感冒"),Query.EQ("Author", "yanc")) 等等
      /// 更新设置。调用示例:Update.Set("Title", "yanc") 或者 Update.Set("Title", "yanc").Set("Author", "yanc2") 等等
      ///
      public static SafeModeResult UpdateAll(string collectionName, IMongoQuery query, IMongoUpdate update)
      {
            return MongoDBHelper.UpdateAll(MongoDBHelper.connectionString_Default, MongoDBHelper.database_Default, collectionName, query, update);
      }

      ///
      ///
      ///
      ///
      ///
      ///
      ///
      /// 条件查询。 调用示例:Query.Matches("Title", "感冒") 或者 Query.EQ("Title", "感冒") 或者Query.And(Query.Matches("Title", "感冒"),Query.EQ("Author", "yanc")) 等等
      /// 更新设置。调用示例:Update.Set("Title", "yanc") 或者 Update.Set("Title", "yanc").Set("Author", "yanc2") 等等
      ///
      public static SafeModeResult UpdateAll(string connectionString, string databaseName, string collectionName, IMongoQuery query, IMongoUpdate update)
      {
            SafeModeResult result;

            if (null == query || null == update)
            {
                return null;
            }


            MongoServer server = MongoServer.Create(connectionString);

            //获取数据库或者创建数据库(不存在的话)。
            MongoDatabase database = server.GetDatabase(databaseName);



            using (server.RequestStart(database))//开始连接数据库。
            {
                MongoCollection myCollection = database.GetCollection(collectionName);

                result = myCollection.Update(query, update, UpdateFlags.Multi);
            }

            return result;
      }

      #endregion


      #region 删除

      public static SafeModeResult Delete(string collectionName, string _id)
      {
            return MongoDBHelper.Delete(MongoDBHelper.connectionString_Default, MongoDBHelper.database_Default, collectionName, _id);
      }

      public static SafeModeResult Delete(string connectionString, string databaseName, string collectionName, string _id)
      {
            SafeModeResult result;
            ObjectId id;
            if (!ObjectId.TryParse(_id, out id))
            {
                return null;
            }



            MongoServer server = MongoServer.Create(connectionString);

            //获取数据库或者创建数据库(不存在的话)。
            MongoDatabase database = server.GetDatabase(databaseName);



            using (server.RequestStart(database))//开始连接数据库。
            {
                MongoCollection myCollection = database.GetCollection(collectionName);

                result = myCollection.Remove(Query.EQ("_id", id));
            }

            return result;

      }

      public static SafeModeResult DeleteAll(string collectionName)
      {
            return MongoDBHelper.DeleteAll(MongoDBHelper.connectionString_Default, MongoDBHelper.database_Default, collectionName, null);
      }

      ///
      ///
      ///
      ///
      /// 条件查询。 调用示例:Query.Matches("Title", "感冒") 或者 Query.EQ("Title", "感冒") 或者Query.And(Query.Matches("Title", "感冒"),Query.EQ("Author", "yanc")) 等等
      ///
      public static SafeModeResult DeleteAll(string collectionName, IMongoQuery query)
      {
            return MongoDBHelper.DeleteAll(MongoDBHelper.connectionString_Default, MongoDBHelper.database_Default, collectionName, query);
      }

      ///
      ///
      ///
      ///
      ///
      ///
      /// 条件查询。 调用示例:Query.Matches("Title", "感冒") 或者 Query.EQ("Title", "感冒") 或者Query.And(Query.Matches("Title", "感冒"),Query.EQ("Author", "yanc")) 等等
      ///
      public static SafeModeResult DeleteAll(string connectionString, string databaseName, string collectionName, IMongoQuery query)
      {
            MongoServer server = MongoServer.Create(connectionString);

            //获取数据库或者创建数据库(不存在的话)。
            MongoDatabase database = server.GetDatabase(databaseName);

            SafeModeResult result;

            using (server.RequestStart(database))//开始连接数据库。
            {
                MongoCollection myCollection = database.GetCollection(collectionName);

                if (null == query)
                {
                  result = myCollection.RemoveAll();
                }
                else
                {
                  result = myCollection.Remove(query);
                }
            }

            return result;

      }

      #endregion


      #region 获取单条信息

      public static T GetOne(string collectionName, string _id)
      {
            return MongoDBHelper.GetOne(MongoDBHelper.connectionString_Default, MongoDBHelper.database_Default, collectionName, _id);
      }

      public static T GetOne(string connectionString, string databaseName, string collectionName, string _id)
      {
            T result = default(T);
            ObjectId id;
            if (!ObjectId.TryParse(_id, out id))
            {
                return default(T);
            }

            MongoServer server = MongoServer.Create(connectionString);

            //获取数据库或者创建数据库(不存在的话)。
            MongoDatabase database = server.GetDatabase(databaseName);



            using (server.RequestStart(database))//开始连接数据库。
            {
                MongoCollection myCollection = database.GetCollection(collectionName);


                result = myCollection.FindOneAs(Query.EQ("_id", id));
            }

            return result;
      }

      ///
      ///
      ///
      ///
      ///
      /// 条件查询。 调用示例:Query.Matches("Title", "感冒") 或者 Query.EQ("Title", "感冒") 或者Query.And(Query.Matches("Title", "感冒"),Query.EQ("Author", "yanc")) 等等
      ///
      public static T GetOne(string collectionName, IMongoQuery query)
      {
            return GetOne(MongoDBHelper.connectionString_Default, MongoDBHelper.database_Default, collectionName, query);
      }

      ///
      ///
      ///
      ///
      ///
      ///
      ///
      /// 条件查询。 调用示例:Query.Matches("Title", "感冒") 或者 Query.EQ("Title", "感冒") 或者Query.And(Query.Matches("Title", "感冒"),Query.EQ("Author", "yanc")) 等等
      ///
      public static T GetOne(string connectionString, string databaseName, string collectionName, IMongoQuery query)
      {
            MongoServer server = MongoServer.Create(connectionString);

            //获取数据库或者创建数据库(不存在的话)。
            MongoDatabase database = server.GetDatabase(databaseName);

            T result = default(T);

            using (server.RequestStart(database))//开始连接数据库。
            {
                MongoCollection myCollection = database.GetCollection(collectionName);

                if (null == query)
                {
                  result = myCollection.FindOneAs();
                }
                else
                {
                  result = myCollection.FindOneAs(query);
                }
            }

            return result;
      }

      #endregion


      #region 获取多个

      public static List GetAll(string collectionName)
      {
            return MongoDBHelper.GetAll(MongoDBHelper.connectionString_Default, MongoDBHelper.database_Default, collectionName);
      }

      ///
      /// 如果不清楚具体的数量,一般不要用这个函数。
      ///
      ///
      ///
      ///
      public static List GetAll(string connectionString, string databaseName, string collectionName)
      {
            MongoServer server = MongoServer.Create(connectionString);

            //获取数据库或者创建数据库(不存在的话)。
            MongoDatabase database = server.GetDatabase(databaseName);

            List result = new List();

            using (server.RequestStart(database))//开始连接数据库。
            {
                MongoCollection myCollection = database.GetCollection(collectionName);

                foreach (T entity in myCollection.FindAllAs())
                {
                  result.Add(entity);
                }
            }

            return result;
      }

      public static List GetAll(string collectionName, int count)
      {
            return MongoDBHelper.GetAll(collectionName, count, null, null);
      }

      ///
      ///
      ///
      ///
      ///
      ///
      /// 条件查询。 调用示例:Query.Matches("Title", "感冒") 或者 Query.EQ("Title", "感冒") 或者Query.And(Query.Matches("Title", "感冒"),Query.EQ("Author", "yanc")) 等等
      ///
      public static List GetAll(string collectionName, int count, IMongoQuery query)
      {
            return MongoDBHelper.GetAll(collectionName, count, query, null);
      }

      ///
      ///
      ///
      ///
      ///
      ///
      /// 排序用的。调用示例:SortBy.Descending("Title") 或者 SortBy.Descending("Title").Ascending("Author")等等
      ///
      public static List GetAll(string collectionName, int count, IMongoSortBy sortBy)
      {
            return MongoDBHelper.GetAll(collectionName, count, null, sortBy);
      }

      ///
      ///
      ///
      ///
      ///
      ///
      /// 条件查询。 调用示例:Query.Matches("Title", "感冒") 或者 Query.EQ("Title", "感冒") 或者Query.And(Query.Matches("Title", "感冒"),Query.EQ("Author", "yanc")) 等等
      /// 排序用的。调用示例:SortBy.Descending("Title") 或者 SortBy.Descending("Title").Ascending("Author")等等
      /// 只返回所需要的字段的数据。调用示例:"Title" 或者 new string[] { "Title", "Author" }等等
      ///
      public static List GetAll(string collectionName, int count, IMongoQuery query, IMongoSortBy sortBy, params string[] fields)
      {
            PagerInfo pagerInfo = new PagerInfo();
            pagerInfo.Page = 1;
            pagerInfo.PageSize = count;
            return MongoDBHelper.GetAll(MongoDBHelper.connectionString_Default, MongoDBHelper.database_Default, collectionName, query, pagerInfo, sortBy, fields);
      }

      ///
      ///
      ///
      ///
      ///
      /// 条件查询。 调用示例:Query.Matches("Title", "感冒") 或者 Query.EQ("Title", "感冒") 或者Query.And(Query.Matches("Title", "感冒"),Query.EQ("Author", "yanc")) 等等
      ///
      ///
      public static List GetAll(string collectionName, IMongoQuery query, PagerInfo pagerInfo)
      {
            return MongoDBHelper.GetAll(MongoDBHelper.connectionString_Default, MongoDBHelper.database_Default, collectionName, query, pagerInfo, null);
      }

      ///
      ///
      ///
      ///
      ///
      /// 条件查询。 调用示例:Query.Matches("Title", "感冒") 或者 Query.EQ("Title", "感冒") 或者Query.And(Query.Matches("Title", "感冒"),Query.EQ("Author", "yanc")) 等等
      ///
      /// 排序用的。调用示例:SortBy.Descending("Title") 或者 SortBy.Descending("Title").Ascending("Author")等等
      ///
      public static List GetAll(string collectionName, IMongoQuery query, PagerInfo pagerInfo, IMongoSortBy sortBy)
      {
            return MongoDBHelper.GetAll(MongoDBHelper.connectionString_Default, MongoDBHelper.database_Default, collectionName, query, pagerInfo, sortBy);
      }

      ///
      ///
      ///
      ///
      ///
      /// 条件查询。 调用示例:Query.Matches("Title", "感冒") 或者 Query.EQ("Title", "感冒") 或者Query.And(Query.Matches("Title", "感冒"),Query.EQ("Author", "yanc")) 等等
      ///
      /// 只返回所需要的字段的数据。调用示例:"Title" 或者 new string[] { "Title", "Author" }等等
      ///
      public static List GetAll(string collectionName, IMongoQuery query, PagerInfo pagerInfo, params string[] fields)
      {
            return MongoDBHelper.GetAll(MongoDBHelper.connectionString_Default, MongoDBHelper.database_Default, collectionName, query, pagerInfo, null, fields);
      }

      ///
      ///
      ///
      ///
      ///
      /// 条件查询。 调用示例:Query.Matches("Title", "感冒") 或者 Query.EQ("Title", "感冒") 或者Query.And(Query.Matches("Title", "感冒"),Query.EQ("Author", "yanc")) 等等
      ///
      /// 排序用的。调用示例:SortBy.Descending("Title") 或者 SortBy.Descending("Title").Ascending("Author")等等
      /// 只返回所需要的字段的数据。调用示例:"Title" 或者 new string[] { "Title", "Author" }等等
      ///
      public static List GetAll(string collectionName, IMongoQuery query, PagerInfo pagerInfo, IMongoSortBy sortBy, params string[] fields)
      {
            return MongoDBHelper.GetAll(MongoDBHelper.connectionString_Default, MongoDBHelper.database_Default, collectionName, query, pagerInfo, sortBy, fields);
      }

      ///
      ///
      ///
      ///
      ///
      ///
      ///
      /// 条件查询。 调用示例:Query.Matches("Title", "感冒") 或者 Query.EQ("Title", "感冒") 或者Query.And(Query.Matches("Title", "感冒"),Query.EQ("Author", "yanc")) 等等
      ///
      /// 排序用的。调用示例:SortBy.Descending("Title") 或者 SortBy.Descending("Title").Ascending("Author")等等
      /// 只返回所需要的字段的数据。调用示例:"Title" 或者 new string[] { "Title", "Author" }等等
      ///
      public static List GetAll(string connectionString, string databaseName, string collectionName, IMongoQuery query, PagerInfo pagerInfo, IMongoSortBy sortBy, params string[] fields)
      {
            MongoServer server = MongoServer.Create(connectionString);

            //获取数据库或者创建数据库(不存在的话)。
            MongoDatabase database = server.GetDatabase(databaseName);

            List result = new List();

            using (server.RequestStart(database))//开始连接数据库。
            {
                MongoCollection myCollection = database.GetCollection(collectionName);

                MongoCursor myCursor;

                if (null == query)
                {
                  myCursor = myCollection.FindAllAs();
                }
                else
                {
                  myCursor = myCollection.FindAs(query);
                }

                if (null != sortBy)
                {
                  myCursor.SetSortOrder(sortBy);
                }

                if (null != fields)
                {
                  myCursor.SetFields(fields);
                }

                foreach (T entity in myCursor.SetSkip((pagerInfo.Page - 1) * pagerInfo.PageSize).SetLimit(pagerInfo.PageSize))//.SetSkip(100).SetLimit(10)是指读取第一百条后的10条数据。
                {
                  result.Add(entity);
                }


            }

            return result;
      }

      #endregion


      #region 索引
      public static void CreateIndex(string collectionName, params string[] keyNames)
      {
            MongoDBHelper.CreateIndex(MongoDBHelper.connectionString_Default, MongoDBHelper.database_Default, collectionName, keyNames);
      }

      public static void CreateIndex(string connectionString, string databaseName, string collectionName, params string[] keyNames)
      {
            SafeModeResult result = new SafeModeResult();

            if (null == keyNames)
            {
                return;
            }

            MongoServer server = MongoServer.Create(connectionString);

            //获取数据库或者创建数据库(不存在的话)。
            MongoDatabase database = server.GetDatabase(databaseName);



            using (server.RequestStart(database))//开始连接数据库。
            {
                MongoCollection myCollection = database.GetCollection(collectionName);
                if (!myCollection.IndexExists(keyNames))
                {
                  myCollection.EnsureIndex(keyNames);
                }
            }

      }
      #endregion




    }
}
  
  
  
页: [1]
查看完整版本: 一个mongodb的helper,封装了一些简单常用的功能