Skip to content
  • P
    Projects
  • G
    Groups
  • S
    Snippets
  • Help

丁松杰 / Pole

  • This project
    • Loading...
  • Sign in
Go to a project
  • Project
  • Repository
  • Issues 0
  • Merge Requests 0
  • Pipelines
  • Wiki
  • Snippets
  • Members
  • Activity
  • Graph
  • Charts
  • Create a new issue
  • Jobs
  • Commits
  • Issue Boards
  • Files
  • Commits
  • Branches
  • Tags
  • Contributors
  • Graph
  • Compare
  • Charts
Switch branch/tag
  • Pole
  • src
  • Pole.Application
  • EventBus
  • ReliableMessageTransactionWorker.cs
Find file
BlameHistoryPermalink
  • dingsongjie's avatar
    fix bug · cdda9bcf
    dingsongjie committed 5 years ago
    cdda9bcf
ReliableMessageTransactionWorker.cs 2.78 KB
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76
using Microsoft.Extensions.Logging;
using Pole.Domain.UnitOfWork;
using Pole.ReliableMessage.Abstraction;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading;
using System.Threading.Tasks;

namespace Pole.Application.EventBus
{
    public class ReliableMessageTransactionWorker : IWorker
    {
        private readonly IReliableMessageScopedBuffer _reliableMessageScopedBuffer;
        private readonly IReliableBus _reliableBus;
        private readonly ILogger<ReliableMessageTransactionWorker> _logger;

        public ReliableMessageTransactionWorker(IReliableMessageScopedBuffer reliableMessageScopedBuffer, IReliableBus reliableBus, ILogger<ReliableMessageTransactionWorker> logger)
        {
            _reliableMessageScopedBuffer = reliableMessageScopedBuffer;
            _reliableBus = reliableBus;
            _logger = logger;
        }

        public int Order => 200;

        public WorkerStatus WorkerStatus { get; set; }

        public async Task Commit(CancellationToken cancellationToken = default)
        {
            var events = _reliableMessageScopedBuffer.GetAll();
            try
            {
                var tasks = events.Select(async @event =>
                {
                    await _reliableBus.Publish(@event.Event, @event.PrePublishEventId, cancellationToken);
                });
                await Task.WhenAll(tasks);
            }
            catch (Exception ex)
            {
                _logger.LogError(ex, "ReliableMessageTransactionWorker.Commit error");
                // 此时 预发送成功 ,数据库事务提交成功 ,发送消息至消息队列失败 ,任然返回成功 ,因为预发送消息 的重试机制会让 消息发送成功
            }
            WorkerStatus = WorkerStatus.Commited;
            return;
        }

        public void Dispose()
        {

        }

        public async Task PreCommit(CancellationToken cancellationToken = default)
        {
            var events = _reliableMessageScopedBuffer.GetAll();
            foreach (var @event in events)
            {
                @event.PrePublishEventId = await _reliableBus.PrePublish(@event.Event, @event.EventType, @event.CallbackParemeter, cancellationToken);
            }
            WorkerStatus = WorkerStatus.PreCommited;
        }

        public Task Rollback(CancellationToken cancellationToken = default)
        {
            var events = _reliableMessageScopedBuffer.GetAll();
            events.Where(m => !string.IsNullOrEmpty(m.PrePublishEventId)).ToList().ForEach(async @event =>
            {
                await _reliableBus.Cancel(@event.PrePublishEventId, cancellationToken);
            });
            WorkerStatus = WorkerStatus.Rollbacked;
            return Task.FromResult(1);
        }
    }
}