feat: implement jetstream push delivery and heartbeat

src/NATS.Server/JetStream/Api/Handlers/ConsumerApiHandlers.cs

@@ -62,6 +62,12 @@ public static class ConsumerApiHandlers
             if (root.TryGetProperty("filter_subject", out var filterEl))
                 config.FilterSubject = filterEl.GetString();
 
+            if (root.TryGetProperty("push", out var pushEl) && pushEl.ValueKind == JsonValueKind.True)
+                config.Push = true;
+
+            if (root.TryGetProperty("heartbeat_ms", out var hbEl) && hbEl.TryGetInt32(out var hbMs))
+                config.HeartbeatMs = hbMs;
+
             return config;
         }
         catch (JsonException)

src/NATS.Server/JetStream/ConsumerManager.cs

@@ -10,6 +10,7 @@ public sealed class ConsumerManager
 {
     private readonly ConcurrentDictionary<(string Stream, string Name), ConsumerHandle> _consumers = new();
     private readonly PullConsumerEngine _pullConsumerEngine = new();
+    private readonly PushConsumerEngine _pushConsumerEngine = new();
 
     public int ConsumerCount => _consumers.Count;
 
@@ -61,10 +62,28 @@ public sealed class ConsumerManager
 
         return await _pullConsumerEngine.FetchAsync(streamHandle, consumer, batch, ct);
     }
+
+    public void OnPublished(string stream, StoredMessage message)
+    {
+        foreach (var handle in _consumers.Values.Where(c => c.Stream == stream && c.Config.Push))
+            _pushConsumerEngine.Enqueue(handle.PushFrames, message, handle.Config);
+    }
+
+    public PushFrame? ReadPushFrame(string stream, string durableName)
+    {
+        if (!_consumers.TryGetValue((stream, durableName), out var consumer))
+            return null;
+
+        if (consumer.PushFrames.Count == 0)
+            return null;
+
+        return consumer.PushFrames.Dequeue();
+    }
 }
 
 public sealed record ConsumerHandle(string Stream, ConsumerConfig Config)
 {
     public ulong NextSequence { get; set; } = 1;
     public Queue<StoredMessage> Pending { get; } = new();
+    public Queue<PushFrame> PushFrames { get; } = new();
 }

src/NATS.Server/JetStream/Consumers/PushConsumerEngine.cs

@@ -0,0 +1,31 @@
+using NATS.Server.JetStream.Models;
+using NATS.Server.JetStream.Storage;
+
+namespace NATS.Server.JetStream.Consumers;
+
+public sealed class PushConsumerEngine
+{
+    public void Enqueue(Queue<PushFrame> queue, StoredMessage message, ConsumerConfig config)
+    {
+        queue.Enqueue(new PushFrame
+        {
+            IsData = true,
+            Message = message,
+        });
+
+        if (config.HeartbeatMs > 0)
+        {
+            queue.Enqueue(new PushFrame
+            {
+                IsHeartbeat = true,
+            });
+        }
+    }
+}
+
+public sealed class PushFrame
+{
+    public bool IsData { get; init; }
+    public bool IsHeartbeat { get; init; }
+    public StoredMessage? Message { get; init; }
+}

src/NATS.Server/NatsServer.cs

@@ -100,8 +100,20 @@ public sealed class NatsServer : IMessageRouter, ISubListAccess, IDisposable
     public bool HasRemoteInterest(string subject) => _globalAccount.SubList.HasRemoteInterest(subject);
     public bool TryCaptureJetStreamPublish(string subject, ReadOnlyMemory<byte> payload, out PubAck ack)
     {
-        if (_jetStreamPublisher != null)
-            return _jetStreamPublisher.TryCapture(subject, payload, out ack);
+        if (_jetStreamPublisher != null && _jetStreamPublisher.TryCapture(subject, payload, out ack))
+        {
+            if (ack.ErrorCode == null
+                && _jetStreamConsumerManager != null
+                && _jetStreamStreamManager != null
+                && _jetStreamStreamManager.TryGet(ack.Stream, out var streamHandle))
+            {
+                var stored = streamHandle.Store.LoadAsync(ack.Seq, default).GetAwaiter().GetResult();
+                if (stored != null)
+                    _jetStreamConsumerManager.OnPublished(ack.Stream, stored);
+            }
+
+            return true;
+        }
 
         ack = new PubAck();
         return false;