src/MassTransit.Newtonsoft/Serialization/EncryptMessageBodyV2.cs - develop - vshmidt/masstransit

src/MassTransit.Newtonsoft/Serialization/EncryptMessageBodyV2.cs

66 строк2 KB

Chris Patterson

Binary serializers (BSON, encrypted) can now be stored as Base64 strings, making them compatible with the message scheduler

21 фев 2022, 19:28

21 фев 2022, 19:286a4853a

#nullable enable
namespace MassTransit.Serialization
{
    using System;
    using System.IO;
    using System.Runtime.Serialization;


    public class EncryptMessageBodyV2<T> :
        MessageBody
        where T : class
    {
        readonly MessageBody _clearMessageBody;
        readonly SendContext<T> _context;
        readonly ICryptoStreamProviderV2 _provider;
        byte[]? _bytes;

        public EncryptMessageBodyV2(ICryptoStreamProviderV2 provider, SendContext<T> context, MessageBody clearMessageBody)
        {
            _provider = provider;
            _context = context;
            _clearMessageBody = clearMessageBody;
        }

        public long? Length => _bytes?.Length;

        public Stream GetStream()
        {
            return new MemoryStream(GetBytes(), false);
        }

        public byte[] GetBytes()
        {
            if (_bytes != null)
                return _bytes;

            try
            {
                using var stream = new MemoryStream();
                using (var cryptoStream = _provider.GetEncryptStream(stream, _context.Headers))
                {
                    var bytes = _clearMessageBody.GetBytes();

                    cryptoStream.Write(bytes, 0, bytes.Length);
                }

                _bytes = stream.ToArray();

                return _bytes;
            }
            catch (SerializationException)
            {
                throw;
            }
            catch (Exception ex)
            {
                throw new SerializationException("Failed to serialize message", ex);
            }
        }

        public string GetString()
        {
            return Convert.ToBase64String(GetBytes());
        }
    }
}