封装的 mongodb的helper

http://blog.csdn.net/sgear/article/details/7624130

直接上代码吧。不知道为什么插入代码会提示出错，直接黏贴吧。-_-!!!  

此项目已开源，开源地址是：

 

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

{

    /// <summary>

    /// mongodb的封装类。

    /// add by yancong2008@gmail.com 2011.05.14

    /// </summary>

    public sealed class MongoDBHelper

    //public sealed class MongoDBHelper<T>

    //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<T>(string collectionName, T entity)

        {

            return MongoDBHelper.InsertOne<T>(MongoDBHelper.connectionString_Default, MongoDBHelper.database_Default, collectionName, entity);

        }

        public static SafeModeResult InsertOne<T>(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<BsonDocument> myCollection = database.GetCollection<BsonDocument>(collectionName);

                result = myCollection.Insert(entity);

            }

            return result;

        }

        public static IEnumerable<SafeModeResult> InsertAll<T>(string collectionName, IEnumerable<T> entitys)

        {

            return MongoDBHelper.InsertAll<T>(MongoDBHelper.connectionString_Default, MongoDBHelper.database_Default, collectionName, entitys);

        }

        public static IEnumerable<SafeModeResult> InsertAll<T>(string connectionString, string databaseName, string collectionName, IEnumerable<T> entitys)

        {

            IEnumerable<SafeModeResult> result = null;

            if (null == entitys)

            {

                return null;

            }

            MongoServer server = MongoServer.Create(connectionString);

            //获取数据库或者创建数据库（不存在的话）。

            MongoDatabase database = server.GetDatabase(databaseName);

            using (server.RequestStart(database))//开始连接数据库。

            {

                MongoCollection<BsonDocument> myCollection = database.GetCollection<BsonDocument>(collectionName);

                result = myCollection.InsertBatch(entitys);

            }

            return result;

        }

        #endregion

        #region 修改

        public static SafeModeResult UpdateOne<T>(string collectionName, T entity)

        {

            return MongoDBHelper.UpdateOne<T>(MongoDBHelper.connectionString_Default, MongoDBHelper.database_Default, collectionName, entity);

        }

        public static SafeModeResult UpdateOne<T>(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<BsonDocument> myCollection = database.GetCollection<BsonDocument>(collectionName);

                result = myCollection.Save(entity);

            }

            return result;

        }

        /// <summary>

        /// 

        /// </summary>

        /// <typeparam name="T"></typeparam>

        /// <param name="collectionName"></param>

        /// <param name="query">条件查询。 调用示例：Query.Matches("Title", "感冒") 或者 Query.EQ("Title", "感冒") 或者Query.And(Query.Matches("Title", "感冒"),Query.EQ("Author", "yanc")) 等等</param>

        /// <param name="update">更新设置。调用示例：Update.Set("Title", "yanc") 或者 Update.Set("Title", "yanc").Set("Author", "yanc2") 等等</param>

        /// <returns></returns>

        public static SafeModeResult UpdateAll<T>(string collectionName, IMongoQuery query, IMongoUpdate update)

        {

            return MongoDBHelper.UpdateAll<T>(MongoDBHelper.connectionString_Default, MongoDBHelper.database_Default, collectionName, query, update);

        }

        /// <summary>

        /// 

        /// </summary>

        /// <typeparam name="T"></typeparam>

        /// <param name="connectionString"></param>

        /// <param name="databaseName"></param>

        /// <param name="collectionName"></param>

        /// <param name="query">条件查询。 调用示例：Query.Matches("Title", "感冒") 或者 Query.EQ("Title", "感冒") 或者Query.And(Query.Matches("Title", "感冒"),Query.EQ("Author", "yanc")) 等等</param>

        /// <param name="update">更新设置。调用示例：Update.Set("Title", "yanc") 或者 Update.Set("Title", "yanc").Set("Author", "yanc2") 等等</param>

        /// <returns></returns>

        public static SafeModeResult UpdateAll<T>(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<BsonDocument> myCollection = database.GetCollection<BsonDocument>(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<BsonDocument> myCollection = database.GetCollection<BsonDocument>(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);

        }

        /// <summary>

        /// 

        /// </summary>

        /// <param name="collectionName"></param>

        /// <param name="query">条件查询。 调用示例：Query.Matches("Title", "感冒") 或者 Query.EQ("Title", "感冒") 或者Query.And(Query.Matches("Title", "感冒"),Query.EQ("Author", "yanc")) 等等</param>

        /// <returns></returns>

        public static SafeModeResult DeleteAll(string collectionName, IMongoQuery query)

        {

            return MongoDBHelper.DeleteAll(MongoDBHelper.connectionString_Default, MongoDBHelper.database_Default, collectionName, query);

        }

        /// <summary>

        /// 

        /// </summary>

        /// <param name="connectionString"></param>

        /// <param name="databaseName"></param>

        /// <param name="collectionName"></param>

        /// <param name="query">条件查询。 调用示例：Query.Matches("Title", "感冒") 或者 Query.EQ("Title", "感冒") 或者Query.And(Query.Matches("Title", "感冒"),Query.EQ("Author", "yanc")) 等等</param>

        /// <returns></returns>

        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<BsonDocument> myCollection = database.GetCollection<BsonDocument>(collectionName);

                if (null == query)

                {

                    result = myCollection.RemoveAll();

                }

                else

                {

                    result = myCollection.Remove(query);

                }

            }

            return result;

        }

        #endregion

        #region 获取单条信息

        public static T GetOne<T>(string collectionName, string _id)

        {

            return MongoDBHelper.GetOne<T>(MongoDBHelper.connectionString_Default, MongoDBHelper.database_Default, collectionName, _id);

        }

        public static T GetOne<T>(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<BsonDocument> myCollection = database.GetCollection<BsonDocument>(collectionName);

                result = myCollection.FindOneAs<T>(Query.EQ("_id", id));

            }

            return result;

        }

        /// <summary>

        /// 

        /// </summary>

        /// <typeparam name="T"></typeparam>

        /// <param name="collectionName"></param>

        /// <param name="query">条件查询。 调用示例：Query.Matches("Title", "感冒") 或者 Query.EQ("Title", "感冒") 或者Query.And(Query.Matches("Title", "感冒"),Query.EQ("Author", "yanc")) 等等</param>

        /// <returns></returns>

        public static T GetOne<T>(string collectionName, IMongoQuery query)

        {

            return GetOne<T>(MongoDBHelper.connectionString_Default, MongoDBHelper.database_Default, collectionName, query);

        }

        /// <summary>

        /// 

        /// </summary>

        /// <typeparam name="T"></typeparam>

        /// <param name="connectionString"></param>

        /// <param name="databaseName"></param>

        /// <param name="collectionName"></param>

        /// <param name="query">条件查询。 调用示例：Query.Matches("Title", "感冒") 或者 Query.EQ("Title", "感冒") 或者Query.And(Query.Matches("Title", "感冒"),Query.EQ("Author", "yanc")) 等等</param>

        /// <returns></returns>

        public static T GetOne<T>(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<BsonDocument> myCollection = database.GetCollection<BsonDocument>(collectionName);

                if (null == query)

                {

                    result = myCollection.FindOneAs<T>();

                }

                else

                {

                    result = myCollection.FindOneAs<T>(query);

                }

            }

            return result;

        }

        #endregion

        #region 获取多个

        public static List<T> GetAll<T>(string collectionName)

        {

            return MongoDBHelper.GetAll<T>(MongoDBHelper.connectionString_Default, MongoDBHelper.database_Default, collectionName);

        }

        /// <summary>

        /// 如果不清楚具体的数量，一般不要用这个函数。

        /// </summary>

        /// <typeparam name="T"></typeparam>

        /// <param name="collectionName"></param>

        /// <returns></returns>

        public static List<T> GetAll<T>(string connectionString, string databaseName, string collectionName)

        {

            MongoServer server = MongoServer.Create(connectionString);

            //获取数据库或者创建数据库（不存在的话）。

            MongoDatabase database = server.GetDatabase(databaseName);

            List<T> result = new List<T>();

            using (server.RequestStart(database))//开始连接数据库。

            {

                MongoCollection<BsonDocument> myCollection = database.GetCollection<BsonDocument>(collectionName);

                foreach (T entity in myCollection.FindAllAs<T>())

                {

                    result.Add(entity);

                }

            }

            return result;

        }

        public static List<T> GetAll<T>(string collectionName, int count)

        {

            return MongoDBHelper.GetAll<T>(collectionName, count, null, null);

        }

        /// <summary>

        /// 

        /// </summary>

        /// <typeparam name="T"></typeparam>

        /// <param name="collectionName"></param>

        /// <param name="count"></param>

        /// <param name="query">条件查询。 调用示例：Query.Matches("Title", "感冒") 或者 Query.EQ("Title", "感冒") 或者Query.And(Query.Matches("Title", "感冒"),Query.EQ("Author", "yanc")) 等等</param>

        /// <returns></returns>

        public static List<T> GetAll<T>(string collectionName, int count, IMongoQuery query)

        {

            return MongoDBHelper.GetAll<T>(collectionName, count, query, null);

        }

        /// <summary>

        /// 

        /// </summary>

        /// <typeparam name="T"></typeparam>

        /// <param name="collectionName"></param>

        /// <param name="count"></param>

        /// <param name="sortBy">排序用的。调用示例：SortBy.Descending("Title") 或者 SortBy.Descending("Title").Ascending("Author")等等</param>

        /// <returns></returns>

        public static List<T> GetAll<T>(string collectionName, int count, IMongoSortBy sortBy)

        {

            return MongoDBHelper.GetAll<T>(collectionName, count, null, sortBy);

        }

        /// <summary>

        /// 

        /// </summary>

        /// <typeparam name="T"></typeparam>

        /// <param name="collectionName"></param>

        /// <param name="count"></param>

        /// <param name="query">条件查询。 调用示例：Query.Matches("Title", "感冒") 或者 Query.EQ("Title", "感冒") 或者Query.And(Query.Matches("Title", "感冒"),Query.EQ("Author", "yanc")) 等等</param>

        /// <param name="sortBy">排序用的。调用示例：SortBy.Descending("Title") 或者 SortBy.Descending("Title").Ascending("Author")等等</param>

        /// <param name="fields">只返回所需要的字段的数据。调用示例："Title" 或者 new string[] { "Title", "Author" }等等</param>

        /// <returns></returns>

        public static List<T> GetAll<T>(string collectionName, int count, IMongoQuery query, IMongoSortBy sortBy, params string[] fields)

        {

            PagerInfo pagerInfo = new PagerInfo();

            pagerInfo.Page = 1;

            pagerInfo.PageSize = count;

            return MongoDBHelper.GetAll<T>(MongoDBHelper.connectionString_Default, MongoDBHelper.database_Default, collectionName, query, pagerInfo, sortBy, fields);

        }

        /// <summary>

        /// 

        /// </summary>

        /// <typeparam name="T"></typeparam>

        /// <param name="collectionName"></param>

        /// <param name="query">条件查询。 调用示例：Query.Matches("Title", "感冒") 或者 Query.EQ("Title", "感冒") 或者Query.And(Query.Matches("Title", "感冒"),Query.EQ("Author", "yanc")) 等等</param>

        /// <param name="pagerInfo"></param>

        /// <returns></returns>

        public static List<T> GetAll<T>(string collectionName, IMongoQuery query, PagerInfo pagerInfo)

        {

            return MongoDBHelper.GetAll<T>(MongoDBHelper.connectionString_Default, MongoDBHelper.database_Default, collectionName, query, pagerInfo, null);

        }

        /// <summary>

        /// 

        /// </summary>

        /// <typeparam name="T"></typeparam>

        /// <param name="collectionName"></param>

        /// <param name="query">条件查询。 调用示例：Query.Matches("Title", "感冒") 或者 Query.EQ("Title", "感冒") 或者Query.And(Query.Matches("Title", "感冒"),Query.EQ("Author", "yanc")) 等等</param>

        /// <param name="pagerInfo"></param>

        /// <param name="sortBy">排序用的。调用示例：SortBy.Descending("Title") 或者 SortBy.Descending("Title").Ascending("Author")等等</param>

        /// <returns></returns>

        public static List<T> GetAll<T>(string collectionName, IMongoQuery query, PagerInfo pagerInfo, IMongoSortBy sortBy)

        {

            return MongoDBHelper.GetAll<T>(MongoDBHelper.connectionString_Default, MongoDBHelper.database_Default, collectionName, query, pagerInfo, sortBy);

        }

        /// <summary>

        /// 

        /// </summary>

        /// <typeparam name="T"></typeparam>

        /// <param name="collectionName"></param>

        /// <param name="query">条件查询。 调用示例：Query.Matches("Title", "感冒") 或者 Query.EQ("Title", "感冒") 或者Query.And(Query.Matches("Title", "感冒"),Query.EQ("Author", "yanc")) 等等</param>

        /// <param name="pagerInfo"></param>

        /// <param name="fields">只返回所需要的字段的数据。调用示例："Title" 或者 new string[] { "Title", "Author" }等等</param>

        /// <returns></returns>

        public static List<T> GetAll<T>(string collectionName, IMongoQuery query, PagerInfo pagerInfo, params string[] fields)

        {

            return MongoDBHelper.GetAll<T>(MongoDBHelper.connectionString_Default, MongoDBHelper.database_Default, collectionName, query, pagerInfo, null, fields);

        }

        /// <summary>

        /// 

        /// </summary>

        /// <typeparam name="T"></typeparam>

        /// <param name="collectionName"></param>

        /// <param name="query">条件查询。 调用示例：Query.Matches("Title", "感冒") 或者 Query.EQ("Title", "感冒") 或者Query.And(Query.Matches("Title", "感冒"),Query.EQ("Author", "yanc")) 等等</param>

        /// <param name="pagerInfo"></param>

        /// <param name="sortBy">排序用的。调用示例：SortBy.Descending("Title") 或者 SortBy.Descending("Title").Ascending("Author")等等</param>

        /// <param name="fields">只返回所需要的字段的数据。调用示例："Title" 或者 new string[] { "Title", "Author" }等等</param>

        /// <returns></returns>

        public static List<T> GetAll<T>(string collectionName, IMongoQuery query, PagerInfo pagerInfo, IMongoSortBy sortBy, params string[] fields)

        {

            return MongoDBHelper.GetAll<T>(MongoDBHelper.connectionString_Default, MongoDBHelper.database_Default, collectionName, query, pagerInfo, sortBy, fields);

        }

        /// <summary>

        /// 

        /// </summary>

        /// <typeparam name="T"></typeparam>

        /// <param name="connectionString"></param>

        /// <param name="databaseName"></param>

        /// <param name="collectionName"></param>

        /// <param name="query">条件查询。 调用示例：Query.Matches("Title", "感冒") 或者 Query.EQ("Title", "感冒") 或者Query.And(Query.Matches("Title", "感冒"),Query.EQ("Author", "yanc")) 等等</param>

        /// <param name="pagerInfo"></param>

        /// <param name="sortBy">排序用的。调用示例：SortBy.Descending("Title") 或者 SortBy.Descending("Title").Ascending("Author")等等</param>

        /// <param name="fields">只返回所需要的字段的数据。调用示例："Title" 或者 new string[] { "Title", "Author" }等等</param>

        /// <returns></returns>

        public static List<T> GetAll<T>(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<T> result = new List<T>();

            using (server.RequestStart(database))//开始连接数据库。

            {

                MongoCollection<BsonDocument> myCollection = database.GetCollection<BsonDocument>(collectionName);

                MongoCursor<T> myCursor;

                if (null == query)

                {

                    myCursor = myCollection.FindAllAs<T>();

                }

                else

                {

                    myCursor = myCollection.FindAs<T>(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<BsonDocument> myCollection = database.GetCollection<BsonDocument>(collectionName);

                if (!myCollection.IndexExists(keyNames))

                {

                    myCollection.EnsureIndex(keyNames);

                }

            }

        }

        #endregion

    }

}