RawJsonProtocolAdapter.cs

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

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

public sealed class RawJsonProtocolAdapter : IMessageProtocolAdapter
{
    private readonly Encoding _encoding;
    private readonly string _defaultRoute;
    private readonly string _defaultName;

    public RawJsonProtocolAdapter()
        : this(new RawJsonProtocolOptions())
    {
    }

    public RawJsonProtocolAdapter(string defaultRoute, string defaultName)
        : this(new RawJsonProtocolOptions
        {
            DefaultRoute = defaultRoute,
            DefaultName = defaultName
        })
    {
    }

    public RawJsonProtocolAdapter(
        Encoding encoding,
        string defaultRoute,
        string defaultName)
        : this(new RawJsonProtocolOptions
        {
            Encoding = encoding ?? throw new ArgumentNullException(nameof(encoding)),
            DefaultRoute = defaultRoute,
            DefaultName = defaultName
        })
    {
    }

    public RawJsonProtocolAdapter(RawJsonProtocolOptions options)
    {
        ArgumentNullException.ThrowIfNull(options);
        ArgumentException.ThrowIfNullOrWhiteSpace(options.DefaultRoute);
        ArgumentException.ThrowIfNullOrWhiteSpace(options.DefaultName);
 
        _encoding = options.Encoding ?? throw new ArgumentNullException(nameof(options.Encoding));
        _defaultRoute = options.DefaultRoute;
        _defaultName = options.DefaultName;
        NormalizePayloadToUtf8 = options.NormalizePayloadToUtf8;
    }

    public bool NormalizePayloadToUtf8 { get; }

    public MessageEnvelope Decode(byte[] payload)
    {
        ArgumentNullException.ThrowIfNull(payload);
 
        var json = _encoding.GetString(payload);
        string route;
        string name;
 
        using (var document = JsonDocument.Parse(json))
        {
            var root = document.RootElement;
 
            route = TryGetString(root, "route")
                    ?? TryGetString(root, "Route")
                    ?? TryGetString(root, "routingKey")
                    ?? TryGetString(root, "topic")
                    ?? TryGetString(root, "type")
                    ?? TryGetString(root, "cmd")
                    ?? _defaultRoute;
 
            name = TryGetString(root, "name")
                   ?? TryGetString(root, "Name")
                   ?? TryGetString(root, "type")
                   ?? TryGetString(root, "cmd")
                   ?? _defaultName;
        }
 
        return new MessageEnvelope
        {
            Name = name,
            Route = route,
            Payload = NormalizePayloadToUtf8
                ? Encoding.UTF8.GetBytes(json)
                : _encoding.GetBytes(json),
            Headers = new Dictionary<string, string>
            {
                ["ContentType"] = "application/json",
                ["Protocol"] = "RawJson",
                ["ExternalEncoding"] = _encoding.WebName
            }
        };
    }

    public byte[] Encode(MessageEnvelope message)
    {
        ArgumentNullException.ThrowIfNull(message);
 
        if (message.Payload.Length > 0)
        {
            var json = Encoding.UTF8.GetString(message.Payload);
            return _encoding.GetBytes(json);
        }
 
        var generatedJson = JsonSerializer.Serialize(new
        {
            name = message.Name,
            route = message.Route
        });
 
        return _encoding.GetBytes(generatedJson);
    }

    private static string? TryGetString(JsonElement root, string propertyName)
    {
        if (!root.TryGetProperty(propertyName, out var property))
        {
            return null;
        }
 
        return property.ValueKind == JsonValueKind.String
            ? property.GetString()
            : property.ToString();
    }
}