feat: integrate jetstream meta-group placement

src/NATS.Server/JetStream/Cluster/AssetPlacementPlanner.cs

@@ -0,0 +1,17 @@
+namespace NATS.Server.JetStream.Cluster;
+
+public sealed class AssetPlacementPlanner
+{
+    private readonly int _nodes;
+
+    public AssetPlacementPlanner(int nodes)
+    {
+        _nodes = Math.Max(nodes, 1);
+    }
+
+    public IReadOnlyList<int> PlanReplicas(int replicas)
+    {
+        var count = Math.Min(Math.Max(replicas, 1), _nodes);
+        return Enumerable.Range(1, count).ToArray();
+    }
+}

src/NATS.Server/JetStream/Cluster/JetStreamMetaGroup.cs

@@ -0,0 +1,36 @@
+using System.Collections.Concurrent;
+using NATS.Server.JetStream.Models;
+
+namespace NATS.Server.JetStream.Cluster;
+
+public sealed class JetStreamMetaGroup
+{
+    private readonly int _nodes;
+    private readonly ConcurrentDictionary<string, byte> _streams = new(StringComparer.Ordinal);
+
+    public JetStreamMetaGroup(int nodes)
+    {
+        _nodes = nodes;
+    }
+
+    public Task ProposeCreateStreamAsync(StreamConfig config, CancellationToken ct)
+    {
+        _streams[config.Name] = 0;
+        return Task.CompletedTask;
+    }
+
+    public MetaGroupState GetState()
+    {
+        return new MetaGroupState
+        {
+            Streams = _streams.Keys.OrderBy(x => x, StringComparer.Ordinal).ToArray(),
+            ClusterSize = _nodes,
+        };
+    }
+}
+
+public sealed class MetaGroupState
+{
+    public IReadOnlyList<string> Streams { get; init; } = [];
+    public int ClusterSize { get; init; }
+}

src/NATS.Server/JetStream/ConsumerManager.cs

@@ -1,5 +1,6 @@
 using System.Collections.Concurrent;
 using NATS.Server.JetStream.Api;
+using NATS.Server.JetStream.Cluster;
 using NATS.Server.JetStream.Consumers;
 using NATS.Server.JetStream.Models;
 using NATS.Server.JetStream.Storage;
@@ -8,10 +9,16 @@ namespace NATS.Server.JetStream;
 
 public sealed class ConsumerManager
 {
+    private readonly JetStreamMetaGroup? _metaGroup;
     private readonly ConcurrentDictionary<(string Stream, string Name), ConsumerHandle> _consumers = new();
     private readonly PullConsumerEngine _pullConsumerEngine = new();
     private readonly PushConsumerEngine _pushConsumerEngine = new();
 
+    public ConsumerManager(JetStreamMetaGroup? metaGroup = null)
+    {
+        _metaGroup = metaGroup;
+    }
+
     public int ConsumerCount => _consumers.Count;
 
     public JetStreamApiResponse CreateOrUpdate(string stream, ConsumerConfig config)

src/NATS.Server/JetStream/StreamManager.cs

@@ -1,5 +1,6 @@
 using System.Collections.Concurrent;
 using NATS.Server.JetStream.Api;
+using NATS.Server.JetStream.Cluster;
 using NATS.Server.JetStream.MirrorSource;
 using NATS.Server.JetStream.Models;
 using NATS.Server.JetStream.Publish;
@@ -10,6 +11,7 @@ namespace NATS.Server.JetStream;
 
 public sealed class StreamManager
 {
+    private readonly JetStreamMetaGroup? _metaGroup;
     private readonly ConcurrentDictionary<string, StreamHandle> _streams =
         new(StringComparer.Ordinal);
     private readonly ConcurrentDictionary<string, List<MirrorCoordinator>> _mirrorsByOrigin =
@@ -17,6 +19,11 @@ public sealed class StreamManager
     private readonly ConcurrentDictionary<string, List<SourceCoordinator>> _sourcesByOrigin =
         new(StringComparer.Ordinal);
 
+    public StreamManager(JetStreamMetaGroup? metaGroup = null)
+    {
+        _metaGroup = metaGroup;
+    }
+
     public IReadOnlyCollection<string> StreamNames => _streams.Keys.ToArray();
 
     public JetStreamApiResponse CreateOrUpdate(StreamConfig config)
@@ -30,6 +37,7 @@ public sealed class StreamManager
             _ => new StreamHandle(normalized, new MemStore()),
             (_, existing) => existing with { Config = normalized });
         RebuildReplicationCoordinators();
+        _metaGroup?.ProposeCreateStreamAsync(normalized, default).GetAwaiter().GetResult();
 
         return BuildStreamInfoResponse(handle);
     }