using MessagePack;
using NetMQ;
using NetMQ.Sockets;

namespace Azaion.CommonSecurity;

public static class ZeroMqExtensions
{
    public static T? Get<T>(this DealerSocket dealer, Func<byte[], bool>? shouldInterceptFn = null) where T : class
    {
        if (!dealer.TryReceiveFrameBytes(TimeSpan.FromMinutes(2), out var bytes))
            throw new Exception($"Unable to get {typeof(T).Name}");
        if (shouldInterceptFn != null && shouldInterceptFn(bytes))
            return null;
        return MessagePackSerializer.Deserialize<T>(bytes);
    }
}