PlainTextProtocolAdapter.cs

이 파일의 문서화 페이지로 가기

using System.Text;
using Dreamine.Communication.Abstractions.Interfaces;
using Dreamine.Communication.Abstractions.Models;
 
namespace Dreamine.Communication.Core.Protocols;

public sealed class PlainTextProtocolAdapter : IMessageProtocolAdapter
{
    private readonly Encoding _encoding;
    private readonly string _route;
    private readonly string _name;

    public PlainTextProtocolAdapter()
        : this(new PlainTextProtocolOptions())
    {
    }

    public PlainTextProtocolAdapter(
        Encoding encoding,
        string route,
        string name)
        : this(new PlainTextProtocolOptions
        {
            Encoding = encoding ?? throw new ArgumentNullException(nameof(encoding)),
            Route = route,
            Name = name
        })
    {
    }

    public PlainTextProtocolAdapter(PlainTextProtocolOptions options)
    {
        ArgumentNullException.ThrowIfNull(options);
        ArgumentException.ThrowIfNullOrWhiteSpace(options.Route);
        ArgumentException.ThrowIfNullOrWhiteSpace(options.Name);
 
        _encoding = options.Encoding ?? throw new ArgumentNullException(nameof(options.Encoding));
        _route = options.Route;
        _name = options.Name;
        NormalizePayloadToUtf8 = options.NormalizePayloadToUtf8;
    }

    public bool NormalizePayloadToUtf8 { get; }

    public MessageEnvelope Decode(byte[] payload)
    {
        ArgumentNullException.ThrowIfNull(payload);
 
        var text = _encoding.GetString(payload);
 
        return new MessageEnvelope
        {
            Name = _name,
            Route = _route,
            Payload = NormalizePayloadToUtf8
                ? Encoding.UTF8.GetBytes(text)
                : _encoding.GetBytes(text),
            Headers = new Dictionary<string, string>
            {
                ["ContentType"] = "text/plain",
                ["Protocol"] = "PlainText",
                ["ExternalEncoding"] = _encoding.WebName
            }
        };
    }

    public byte[] Encode(MessageEnvelope message)
    {
        ArgumentNullException.ThrowIfNull(message);
 
        if (message.Payload.Length == 0)
        {
            return [];
        }
 
        var text = Encoding.UTF8.GetString(message.Payload);
        return _encoding.GetBytes(text);
    }
}