Sanhe.Abp.Framework/modules/common/Sanhe.Abp.Notifications/Sanhe/Abp/Notifications/Internal/NotificationSender.cs

using Microsoft.Extensions.Logging;
using Microsoft.Extensions.Logging.Abstractions;
using Microsoft.Extensions.Options;
using Sanhe.Abp.IdGenerator;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Threading.Tasks;
using Volo.Abp.DependencyInjection;
using Volo.Abp.EventBus.Distributed;
using Volo.Abp.Uow;

namespace Sanhe.Abp.Notifications.Internal
{
    /// <summary>
    /// 默认实现通过分布式事件发送通知
    /// 可替换实现来发送实时通知
    /// </summary>
    public class NotificationSender : INotificationSender, ITransientDependency
    {
        /// <summary>
        /// Reference to <see cref="ILogger{NotificationSender}"/>.
        /// </summary>
        public ILogger<NotificationSender> Logger { get; set; }
        /// <summary>
        /// Reference to <see cref="IDistributedEventBus"/>.
        /// </summary>
        public IDistributedEventBus DistributedEventBus { get; }
        /// <summary>
        /// Reference to <see cref="IDistributedIdGenerator"/>.
        /// </summary>
        protected IDistributedIdGenerator DistributedIdGenerator { get; }
        /// <summary>
        /// Reference to <see cref="IUnitOfWorkManager"/>.
        /// </summary>
        protected IUnitOfWorkManager UnitOfWorkManager { get; }
        /// <summary>
        /// 通知选项
        /// </summary>
        protected AbpNotificationsOptions Options { get; }

        public NotificationSender(
           IDistributedEventBus distributedEventBus,
           IDistributedIdGenerator distributedIdGenerator,
           IUnitOfWorkManager unitOfWorkManager,
           IOptions<AbpNotificationsOptions> optionsAccessor)
        {
            Options = optionsAccessor.Value;
            DistributedEventBus = distributedEventBus;
            DistributedIdGenerator = distributedIdGenerator;
            UnitOfWorkManager = unitOfWorkManager;
            Logger = NullLogger<NotificationSender>.Instance;
        }

        public async Task<string> SendNofiterAsync(
            string name,
            NotificationData data,
            UserIdentifier user = null,
            Guid? tenantId = null,
            NotificationSeverity severity = NotificationSeverity.Info)
        {
            if (user == null)
            {
                return await PublishNofiterAsync(name, data, null, tenantId, severity);

            }
            else
            {
                return await PublishNofiterAsync(name, data, new List<UserIdentifier> { user }, tenantId, severity);
            }
        }

        public async Task<string> SendNofitersAsync(
            string name,
            NotificationData data,
            IEnumerable<UserIdentifier> users = null,
            Guid? tenantId = null,
            NotificationSeverity severity = NotificationSeverity.Info)
        {
            return await PublishNofiterAsync(name, data, users, tenantId, severity);
        }

        protected async Task<string> PublishNofiterAsync(
            string name,
            NotificationData data,
            IEnumerable<UserIdentifier> users = null,
            Guid? tenantId = null,
            NotificationSeverity severity = NotificationSeverity.Info)
        {
            var eto = new NotificationEto<NotificationData>(data)
            {
                Id = DistributedIdGenerator.Create(),
                TenantId = tenantId,
                Users = users?.ToList(),
                Name = name,
                CreationTime = DateTime.Now,
                Severity = severity
            };

            if (UnitOfWorkManager.Current != null)
            {
                UnitOfWorkManager.Current.OnCompleted(async () =>
                {
                    await DistributedEventBus.PublishAsync(eto);
                });
            }
            else
            {
                await DistributedEventBus.PublishAsync(eto);
            }

            return eto.Id.ToString();
        }
    }
}