src/DotPulsar/MessageMetadata.cs

/*
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

namespace DotPulsar;

using DotPulsar.Internal.Extensions;
using DotPulsar.Internal.PulsarApi;

/// <summary>
/// The message metadata builder.
/// </summary>
public sealed class MessageMetadata
{
    /// <summary>
    /// Initializes a new instance of the message metadata builder.
    /// </summary>
    public MessageMetadata()
        => Metadata = new Internal.PulsarApi.MessageMetadata();

    internal Internal.PulsarApi.MessageMetadata Metadata { get; }

    /// <summary>
    /// Manually set the compression information.
    /// </summary>
    public void SetCompressionInfo(CompressionType compressionType, uint uncompressedSize)
    {
        Metadata.Compression = (Internal.PulsarApi.CompressionType) compressionType;
        Metadata.UncompressedSize = uncompressedSize;
    }

    /// <summary>
    /// The delivery time of the message as unix time in milliseconds.
    /// </summary>
    public long DeliverAtTime
    {
        get => Metadata.DeliverAtTime;
        set => Metadata.DeliverAtTime = value;
    }

    /// <summary>
    /// The delivery time of the message as an UTC DateTime.
    /// </summary>
    public DateTime DeliverAtTimeAsDateTime
    {
        get => Metadata.GetDeliverAtTimeAsDateTime();
        set => Metadata.SetDeliverAtTime(value);
    }

    /// <summary>
    /// The delivery time of the message as a DateTimeOffset.
    /// </summary>
    public DateTimeOffset DeliverAtTimeAsDateTimeOffset
    {
        get => Metadata.GetDeliverAtTimeAsDateTimeOffset();
        set => Metadata.SetDeliverAtTime(value);
    }

    /// <summary>
    /// The event time of the message as unix time in milliseconds.
    /// </summary>
    public ulong EventTime
    {
        get => Metadata.EventTime;
        set => Metadata.EventTime = value;
    }

    /// <summary>
    /// The event time of the message as an UTC DateTime.
    /// </summary>
    public DateTime EventTimeAsDateTime
    {
        get => Metadata.GetEventTimeAsDateTime();
        set => Metadata.SetEventTime(value);
    }

    /// <summary>
    /// The event time of the message as a DateTimeOffset.
    /// </summary>
    public DateTimeOffset EventTimeAsDateTimeOffset
    {
        get => Metadata.GetEventTimeAsDateTimeOffset();
        set => Metadata.SetEventTime(value);
    }

    /// <summary>
    /// The key of the message as a string.
    /// </summary>
    public string? Key
    {
        get => Metadata.PartitionKey;
        set => Metadata.SetKey(value);
    }

    /// <summary>
    /// The key of the message as bytes.
    /// </summary>
    public byte[]? KeyBytes
    {
        get => Metadata.GetKeyAsBytes();
        set => Metadata.SetKey(value);
    }

    /// <summary>
    /// The ordering key of the message.
    /// </summary>
    public byte[]? OrderingKey
    {
        get => Metadata.OrderingKey;
        set => Metadata.OrderingKey = value;
    }

    /// <summary>
    /// The properties of the message.
    /// </summary>
    public string? this[string key]
    {
        get
        {
            for (var i = 0; i < Metadata.Properties.Count; ++i)
            {
                var prop = Metadata.Properties[i];

                if (prop.Key == key)
                    return prop.Value;
            }

            return null;
        }
        set
        {
            if (key is null)
                throw new ArgumentNullException(nameof(key), $"The '{nameof(key)}' cannot be null");

            if (value is null)
                throw new ArgumentNullException(nameof(value), $"The '{nameof(value)}' cannot be null");

            for (var i = 0; i < Metadata.Properties.Count; ++i)
            {
                var prop = Metadata.Properties[i];

                if (prop.Key != key)
                    continue;

                prop.Value = value;

                return;
            }

            Metadata.Properties.Add(new KeyValue { Key = key, Value = value });
        }
    }

    /// <summary>
    /// The schema version of the message.
    /// </summary>
    public byte[]? SchemaVersion
    {
        get => Metadata.SchemaVersion;
        set => Metadata.SchemaVersion = value;
    }

    /// <summary>
    /// The sequence id of the message.
    /// </summary>
    public ulong SequenceId
    {
        get => Metadata.SequenceId;
        set => Metadata.SequenceId = value;
    }
}