🔨 Reconfigured to use new discovery

DysonNetwork.Shared/Registry/ServiceRegistrar.cs

@@ -3,6 +3,7 @@ using Etcdserverpb;
 using Google.Protobuf;
 using Microsoft.Extensions.Configuration;
 using Microsoft.Extensions.Hosting;
+using Microsoft.Extensions.Options;
 
 namespace DysonNetwork.Shared.Registry;
 
@@ -12,14 +13,21 @@ public class ServiceRegistrar(EtcdClient etcd)
     private string? _serviceKey;
     private long _leaseId;
     private long _ttlSeconds;
+    private readonly Dictionary<string, int> _roundRobinCounters = new();
 
     /// <summary>
     /// Register the service in etcd with a TTL lease.
     /// </summary>
-    public async Task RegisterAsync(string serviceName, string servicePart, string instanceId, string host, int port, long ttlSeconds = 30)
+    public async Task RegisterAsync(
+        string serviceName,
+        string servicePart,
+        string instanceId,
+        string host,
+        int port,
+        long ttlSeconds = 30)
     {
         _ttlSeconds = ttlSeconds;
-        _serviceKey = $"/services/{serviceName}/${servicePart}/{instanceId}";
+        _serviceKey = $"/services/{serviceName}/{servicePart}/{instanceId}";
         var serviceValue = $"{host}:{port}";
 
         // Create and store TTL lease
@@ -70,20 +78,66 @@ public class ServiceRegistrar(EtcdClient etcd)
             }
         }
     }
+
+    /// <summary>
+    /// Get all service instances for a specific service name and part.
+    /// </summary>
+    public async Task<List<string>> GetServiceInstancesAsync(string serviceName, string servicePart)
+    {
+        var prefix = $"/services/{serviceName}/{servicePart}/";
+        var request = new RangeRequest
+        {
+            Key = ByteString.CopyFromUtf8(prefix),
+            RangeEnd = ByteString.CopyFromUtf8(prefix + "\0")
+        };
+        var response = await etcd.GetAsync(request);
+        var instances = response.Kvs.Select(kv => kv.Value.ToStringUtf8()).ToList();
+        return instances;
+    }
+
+    /// <summary>
+    /// Get a single service instance with load balancing (round-robin).
+    /// </summary>
+    public async Task<string> GetServiceInstanceAsync(string serviceName, string servicePart)
+    {
+        var instances = await GetServiceInstancesAsync(serviceName, servicePart);
+        if (instances.Count == 0)
+            throw new InvalidOperationException($"No instances found for service '{serviceName}' part '{servicePart}'");
+        var key = $"{serviceName}/{servicePart}";
+        if (!_roundRobinCounters.ContainsKey(key))
+            _roundRobinCounters[key] = 0;
+        var instance = instances[_roundRobinCounters[key] % instances.Count];
+        _roundRobinCounters[key] = (_roundRobinCounters[key] + 1) % int.MaxValue;
+        return instance;
+    }
 }
 
-public class ServiceRegistrarHostedService(ServiceRegistrar registrar, IConfiguration config) : IHostedService
+public sealed class ServiceRegistrationOptions
 {
+    public string Name { get; set; } = null!;
+    public string Host { get; set; } = "127.0.0.1";
+    public string InstanceId { get; set; } = Guid.NewGuid().ToString("N");
+}
+
+public class ServiceRegistrarHostedService(
+    ServiceRegistrar registrar,
+    IConfiguration configuration,
+    IOptions<ServiceRegistrationOptions> options
+)
+    : IHostedService
+{
+    private readonly ServiceRegistrationOptions _opts = options.Value;
+
     public async Task StartAsync(CancellationToken cancellationToken)
     {
-        var name = config["Service:Name"];
-        var host = config["Service:Host"];
-        var grpcPort = int.Parse(config["Service:GrpcPort"]!);
-        var httpPort = int.Parse(config["Service:HttpPort"]!);
-        var instanceId = config["Service:InstanceId"] ?? Guid.NewGuid().ToString("N");
+        var grpcPort = int.Parse(configuration.GetValue("GRPC_PORT", "5000"));
+        await registrar.RegisterAsync(_opts.Name, "grpc", _opts.InstanceId, _opts.Host, grpcPort);
 
-        await registrar.RegisterAsync(name, "grpc", instanceId, host, grpcPort);
-        await registrar.RegisterAsync(name, "http", instanceId, host, httpPort);
+        var httpPorts = configuration.GetValue("HTTP_PORTS", "6000")
+            .Split(',', StringSplitOptions.RemoveEmptyEntries)
+            .Select(p => int.Parse(p.Trim()))
+            .ToArray();
+        await registrar.RegisterAsync(_opts.Name, "http", _opts.InstanceId, _opts.Host, httpPorts.First());
     }
 
     public async Task StopAsync(CancellationToken cancellationToken)