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

using Business.Core.Attributes;
using Microsoft.Extensions.Options;
using MongoDB.Bson;
using MongoDB.Driver;
using System;
using System.Collections.Generic;
using System.Linq;
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 IOptionsSnapshot<Config> _config;

        /// <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);

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

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

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

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

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

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

    }
}