DEMOMySql/MicroServices/Business/Business.Core/MongoDBHelper/MongoDBTools.cs

using Business.Core.Attributes;
using Microsoft.Extensions.Options;
using MongoDB.Bson;
using MongoDB.Driver;
using MongoDB.Driver.Linq;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Linq.Expressions;
using System.Text;
using System.Threading.Tasks;
using Volo.Abp.Domain.Entities;

namespace Business.Core.MongoDBHelper
{
    /// <summary>
    /// MongoDB帮助类
    /// </summary>
    public class MongoDBTools<T> : IMongoDB<T> where T : Entity<long>
    {
        public readonly IMongoCollection<T> mongoCollection;
        public readonly IMongoDatabase dataBase;
        public IOptionsSnapshot<Config> _config;
        public CollectionNameAttribute collectonName;

        /// <summary>
        /// MongoDB链接
        /// </summary>
        /// <param name="config"></param>
        /// <exception cref="NotImplementedException"></exception>
        public MongoDBTools(IOptionsSnapshot<Config> config)
        {
            _config = config;
            //数据库链接
            MongoClient client = new MongoClient(config.Value.connectstring);

            collectonName = typeof(T).GetCustomAttributes(typeof(CollectionNameAttribute), true).FirstOrDefault() as CollectionNameAttribute;
            if (collectonName == null)
            {
                throw new NotImplementedException("请配置Attribute属性！");
            }
            //数据库
            dataBase = client.GetDatabase(collectonName.DatabaseName);
            //表名
            mongoCollection = dataBase.GetCollection<T>(collectonName.CollectionName);
        }

        /// <summary>
        /// 插入一条数据
        /// </summary>
        /// <param name="document"></param>
        /// <returns></returns>
        public Task InsertOne(T document)
        {
            return mongoCollection.InsertOneAsync(document);
        }

        /// <summary>
        /// 插入多条数据
        /// </summary>
        /// <param name="documents"></param>
        /// <returns></returns>
        /// <exception cref="NotImplementedException"></exception>
        public Task InsertMany(List<T> documents)
        {
           return mongoCollection.InsertManyAsync(documents,new InsertManyOptions() { IsOrdered = false});
        }

        /// <summary>
        /// 更新一条数据
        /// </summary>
        /// <param name="documents"></param>
        /// <param name="id"></param>
        /// <returns></returns>
        public Task<ReplaceOneResult> UpdateOne(T documents,long id)
        {
            return mongoCollection.ReplaceOneAsync(Builders<T>.Filter.Eq(p=>p.Id, id), documents);
        }

        /// <summary>
        /// 获取所有数据
        /// </summary>
        /// <returns></returns>
        /// <exception cref="NotImplementedException"></exception>
        public Task<List<T>> GetAll()
        {
            return mongoCollection.AsQueryable<T>().ToListAsync();
        }

        /// <summary>
        /// 跟据Id获取数据
        /// </summary>
        /// <param name="id"></param>
        /// <returns></returns>
        /// <exception cref="NotImplementedException"></exception>
        public Task<T> GetOneByID(long id)
        {
            return mongoCollection.Find(p => p.Id == id).FirstOrDefaultAsync();
        }

        /// <summary>
        /// 根据条件获取数据
        /// </summary>
        /// <returns></returns>
        public Task<List<T>> GetManyByCondition(Expression<Func<T, bool>> filter) 
        {
            return mongoCollection.Find(filter).ToListAsync();
        }

        /// <summary>
        /// 根据条件获取数据
        /// </summary>
        /// <returns></returns>
        public Task<List<T>> GetManyByIds(FilterDefinition<T> filter)
        {
            return mongoCollection.Find(filter).ToListAsync();
        }

        /// <summary>
        /// 根据id删除对象
        /// </summary>
        /// <param name="id"></param>
        public Task DeleteById(long id)
        {
            return mongoCollection.DeleteManyAsync(s => id==s.Id);
        }

        /// <summary>
        /// 根据id列表批量删除对象
        /// </summary>
        /// <param name="ids">id列表</param>
        public Task DeleteByIds(IEnumerable<long> ids)
        {
            return mongoCollection.DeleteManyAsync(s => ids.Contains(s.Id));
        }

        /// <summary>
        /// 删除数据
        /// </summary>
        /// <param name="expression"></param>
        /// <param name="isOne"></param>
        /// <returns></returns>
        public Task<DeleteResult> Delete(Expression<Func<T, bool>> expression, bool isOne = false)
        {
            if (isOne)
                return mongoCollection.DeleteOneAsync(expression);
            else
                return mongoCollection.DeleteManyAsync(expression);
        }

        /// <summary>
        /// 删除数据
        /// </summary>
        /// <param name="filter"></param>
        /// <param name="isOne"></param>
        /// <returns></returns>
        public Task<DeleteResult> Delete(FilterDefinition<T> filter, bool isOne = false)
        {
            if (isOne)
                return mongoCollection.DeleteOneAsync(filter);
            else
                return mongoCollection.DeleteManyAsync(filter);
        }

        /// <summary>
        /// 删除数据-直接删除集合，慎用，慎用，慎用
        /// </summary>
        /// <returns></returns>
        public Task DeleteAll()
        {
            return dataBase.DropCollectionAsync(collectonName.CollectionName);
        }
    }
}