349 lines
14 KiB
C#
Executable File
349 lines
14 KiB
C#
Executable File
using Microsoft.Extensions.Hosting;
|
||
using Microsoft.Extensions.Logging;
|
||
using ZB.MOM.WW.CBDDC.Core;
|
||
using ZB.MOM.WW.CBDDC.Core.Cache;
|
||
using ZB.MOM.WW.CBDDC.Core.Diagnostics;
|
||
using ZB.MOM.WW.CBDDC.Core.Sync;
|
||
using ZB.MOM.WW.CBDDC.Core.Storage;
|
||
using ZB.MOM.WW.CBDDC.Network;
|
||
using ZB.MOM.WW.CBDDC.Persistence.BLite;
|
||
using Microsoft.Extensions.DependencyInjection; // For IServiceProvider if needed
|
||
using Serilog.Context;
|
||
using ZB.MOM.WW.CBDDC.Sample.Console;
|
||
using ZB.MOM.WW.CBDDC.Core.Network;
|
||
|
||
namespace ZB.MOM.WW.CBDDC.Sample.Console;
|
||
|
||
public class ConsoleInteractiveService : BackgroundService
|
||
{
|
||
private readonly ILogger<ConsoleInteractiveService> _logger;
|
||
private readonly SampleDbContext _db;
|
||
private readonly ICBDDCNode _node;
|
||
private readonly IHostApplicationLifetime _lifetime;
|
||
|
||
|
||
// Auxiliary services for status/commands
|
||
private readonly IDocumentCache _cache;
|
||
private readonly IOfflineQueue _queue;
|
||
private readonly ICBDDCHealthCheck _healthCheck;
|
||
private readonly ISyncStatusTracker _syncTracker;
|
||
private readonly IServiceProvider _serviceProvider;
|
||
private readonly IPeerNodeConfigurationProvider _configProvider;
|
||
|
||
/// <summary>
|
||
/// Initializes a new instance of the <see cref="ConsoleInteractiveService"/> class.
|
||
/// </summary>
|
||
/// <param name="logger">The logger used by the interactive service.</param>
|
||
/// <param name="db">The sample database context.</param>
|
||
/// <param name="node">The active CBDDC node instance.</param>
|
||
/// <param name="lifetime">The application lifetime controller.</param>
|
||
/// <param name="cache">The document cache service.</param>
|
||
/// <param name="queue">The offline queue service.</param>
|
||
/// <param name="healthCheck">The health check service.</param>
|
||
/// <param name="syncTracker">The sync status tracker.</param>
|
||
/// <param name="serviceProvider">The service provider for resolving optional services.</param>
|
||
/// <param name="peerNodeConfigurationProvider">The provider for peer node configuration.</param>
|
||
public ConsoleInteractiveService(
|
||
ILogger<ConsoleInteractiveService> logger,
|
||
SampleDbContext db,
|
||
ICBDDCNode node,
|
||
IHostApplicationLifetime lifetime,
|
||
IDocumentCache cache,
|
||
IOfflineQueue queue,
|
||
ICBDDCHealthCheck healthCheck,
|
||
ISyncStatusTracker syncTracker,
|
||
IServiceProvider serviceProvider,
|
||
IPeerNodeConfigurationProvider peerNodeConfigurationProvider)
|
||
{
|
||
_logger = logger;
|
||
_db = db;
|
||
_node = node;
|
||
_lifetime = lifetime;
|
||
_cache = cache;
|
||
_queue = queue;
|
||
_healthCheck = healthCheck;
|
||
_syncTracker = syncTracker;
|
||
_serviceProvider = serviceProvider;
|
||
_configProvider = peerNodeConfigurationProvider;
|
||
}
|
||
|
||
/// <inheritdoc />
|
||
protected override async Task ExecuteAsync(CancellationToken stoppingToken)
|
||
{
|
||
var config = await _configProvider.GetConfiguration();
|
||
|
||
System.Console.WriteLine($"--- Interactive Console ---");
|
||
System.Console.WriteLine($"Node ID: {config.NodeId}");
|
||
PrintHelp();
|
||
|
||
while (!stoppingToken.IsCancellationRequested)
|
||
{
|
||
// Non-blocking read to allow cancellation check
|
||
if (!System.Console.KeyAvailable)
|
||
{
|
||
await Task.Delay(100, stoppingToken);
|
||
continue;
|
||
}
|
||
|
||
var input = System.Console.ReadLine();
|
||
if (string.IsNullOrEmpty(input)) continue;
|
||
|
||
try
|
||
{
|
||
using (LogContext.PushProperty("NodeId", config.NodeId))
|
||
using (LogContext.PushProperty("Command", input))
|
||
using (LogContext.PushProperty("OperationId", Guid.NewGuid().ToString("N")))
|
||
{
|
||
_logger.LogInformation("Handling interactive command {Command}", input);
|
||
await HandleInput(input);
|
||
}
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
System.Console.WriteLine($"Error: {ex.Message}");
|
||
}
|
||
|
||
if (input == "q" || input == "quit")
|
||
{
|
||
_lifetime.StopApplication();
|
||
break;
|
||
}
|
||
}
|
||
}
|
||
|
||
private void PrintHelp()
|
||
{
|
||
System.Console.WriteLine("Commands:");
|
||
System.Console.WriteLine(" [p]ut, [g]et, [d]elete, [f]ind, [l]ist peers, [q]uit");
|
||
System.Console.WriteLine(" [n]ew (auto), [s]pam (5x), [c]ount, [t]odos");
|
||
System.Console.WriteLine(" [h]ealth, cac[h]e");
|
||
System.Console.WriteLine(" [r]esolver [lww|merge], [demo] conflict");
|
||
}
|
||
|
||
private async Task HandleInput(string input)
|
||
{
|
||
var config = await _configProvider.GetConfiguration();
|
||
if (input.StartsWith("n"))
|
||
{
|
||
var ts = DateTime.Now.ToString("HH:mm:ss.fff");
|
||
var user = new User { Id = Guid.NewGuid().ToString(), Name = $"User-{ts}", Age = new Random().Next(18, 90), Address = new Address { City = "AutoCity" } };
|
||
await _db.Users.InsertAsync(user);
|
||
await _db.SaveChangesAsync();
|
||
System.Console.WriteLine($"[+] Created {user.Name} with Id: {user.Id}...");
|
||
}
|
||
else if (input.StartsWith("s"))
|
||
{
|
||
for (int i = 0; i < 5; i++)
|
||
{
|
||
var ts = DateTime.Now.ToString("HH:mm:ss.fff");
|
||
var user = new User { Id = Guid.NewGuid().ToString(), Name = $"User-{ts}", Age = new Random().Next(18, 90), Address = new Address { City = "SpamCity" } };
|
||
await _db.Users.InsertAsync(user);
|
||
System.Console.WriteLine($"[+] Created {user.Name} with Id: {user.Id}...");
|
||
await Task.Delay(100);
|
||
}
|
||
await _db.SaveChangesAsync();
|
||
}
|
||
else if (input.StartsWith("c"))
|
||
{
|
||
var userCount = _db.Users.FindAll().Count();
|
||
var todoCount = _db.TodoLists.FindAll().Count();
|
||
System.Console.WriteLine($"Collection 'Users': {userCount} documents");
|
||
System.Console.WriteLine($"Collection 'TodoLists': {todoCount} documents");
|
||
}
|
||
else if (input.StartsWith("p"))
|
||
{
|
||
var alice = new User { Id = Guid.NewGuid().ToString(), Name = "Alice", Age = 30, Address = new Address { City = "Paris" } };
|
||
var bob = new User { Id = Guid.NewGuid().ToString(), Name = "Bob", Age = 25, Address = new Address { City = "Rome" } };
|
||
await _db.Users.InsertAsync(alice);
|
||
await _db.Users.InsertAsync(bob);
|
||
await _db.SaveChangesAsync();
|
||
System.Console.WriteLine($"Put Alice ({alice.Id}) and Bob ({bob.Id})");
|
||
}
|
||
else if (input.StartsWith("g"))
|
||
{
|
||
System.Console.Write("Enter user Id: ");
|
||
var id = System.Console.ReadLine();
|
||
if (!string.IsNullOrEmpty(id))
|
||
{
|
||
var u = _db.Users.FindById(id);
|
||
System.Console.WriteLine(u != null ? $"Got: {u.Name}, Age {u.Age}, City: {u.Address?.City}" : "Not found");
|
||
}
|
||
}
|
||
else if (input.StartsWith("d"))
|
||
{
|
||
System.Console.Write("Enter user Id to delete: ");
|
||
var id = System.Console.ReadLine();
|
||
if (!string.IsNullOrEmpty(id))
|
||
{
|
||
await _db.Users.DeleteAsync(id);
|
||
await _db.SaveChangesAsync();
|
||
System.Console.WriteLine($"Deleted user {id}");
|
||
}
|
||
}
|
||
else if (input.StartsWith("l"))
|
||
{
|
||
var peers = _node.Discovery.GetActivePeers();
|
||
var handshakeSvc = _serviceProvider.GetService<ZB.MOM.WW.CBDDC.Network.Security.IPeerHandshakeService>();
|
||
var secureIcon = handshakeSvc != null ? "🔒" : "🔓";
|
||
|
||
System.Console.WriteLine($"Active Peers ({secureIcon}):");
|
||
foreach (var p in peers)
|
||
System.Console.WriteLine($" - {p.NodeId} at {p.Address}");
|
||
|
||
if (handshakeSvc != null)
|
||
System.Console.WriteLine("\nℹ️ Secure mode: Connections use ECDH + AES-256");
|
||
}
|
||
else if (input.StartsWith("f"))
|
||
{
|
||
System.Console.WriteLine("Query: Age > 28");
|
||
var results = _db.Users.Find(u => u.Age > 28);
|
||
foreach (var u in results) System.Console.WriteLine($"Found: {u.Name} ({u.Age})");
|
||
}
|
||
else if (input.StartsWith("h"))
|
||
{
|
||
var health = await _healthCheck.CheckAsync();
|
||
var syncStatus = _syncTracker.GetStatus();
|
||
var handshakeSvc = _serviceProvider.GetService<ZB.MOM.WW.CBDDC.Network.Security.IPeerHandshakeService>();
|
||
|
||
System.Console.WriteLine("=== Health Check ===");
|
||
System.Console.WriteLine($"Database: {(health.DatabaseHealthy ? "✓" : "✗")}");
|
||
System.Console.WriteLine($"Network: {(health.NetworkHealthy ? "✓" : "✗")}");
|
||
System.Console.WriteLine($"Security: {(handshakeSvc != null ? "🔒 Encrypted" : "🔓 Plaintext")}");
|
||
System.Console.WriteLine($"Connected Peers: {health.ConnectedPeers}");
|
||
System.Console.WriteLine($"Last Sync: {health.LastSyncTime?.ToString("HH:mm:ss") ?? "Never"}");
|
||
System.Console.WriteLine($"Total Synced: {syncStatus.TotalDocumentsSynced} docs");
|
||
|
||
if (health.Errors.Any())
|
||
{
|
||
System.Console.WriteLine("Errors:");
|
||
foreach (var err in health.Errors.Take(3)) System.Console.WriteLine($" - {err}");
|
||
}
|
||
}
|
||
else if (input.StartsWith("ch") || input == "cache")
|
||
{
|
||
var stats = _cache.GetStatistics();
|
||
System.Console.WriteLine($"=== Cache Stats ===\nSize: {stats.Size}\nHits: {stats.Hits}\nMisses: {stats.Misses}\nRate: {stats.HitRate:P1}");
|
||
}
|
||
else if (input.StartsWith("r") && input.Contains("resolver"))
|
||
{
|
||
var parts = input.Split(' ');
|
||
if (parts.Length > 1)
|
||
{
|
||
var newResolver = parts[1].ToLower() switch
|
||
{
|
||
"lww" => (IConflictResolver)new LastWriteWinsConflictResolver(),
|
||
"merge" => new RecursiveNodeMergeConflictResolver(),
|
||
_ => null
|
||
};
|
||
|
||
if (newResolver != null)
|
||
{
|
||
// Note: Requires restart to fully apply. For demo, we inform user.
|
||
System.Console.WriteLine($"⚠️ Resolver changed to {parts[1].ToUpper()}. Restart node to apply.");
|
||
System.Console.WriteLine($" (Current session continues with previous resolver)");
|
||
}
|
||
else
|
||
{
|
||
System.Console.WriteLine("Usage: resolver [lww|merge]");
|
||
}
|
||
}
|
||
}
|
||
else if (input == "demo")
|
||
{
|
||
await RunConflictDemo();
|
||
}
|
||
else if (input == "todos")
|
||
{
|
||
var lists = _db.TodoLists.FindAll();
|
||
|
||
System.Console.WriteLine("=== Todo Lists ===");
|
||
foreach (var list in lists)
|
||
{
|
||
System.Console.WriteLine($"📋 {list.Name} ({list.Items.Count} items)");
|
||
foreach (var item in list.Items)
|
||
{
|
||
var status = item.Completed ? "✓" : " ";
|
||
System.Console.WriteLine($" [{status}] {item.Task}");
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
private async Task RunConflictDemo()
|
||
{
|
||
System.Console.WriteLine("\n=== Conflict Resolution Demo ===");
|
||
System.Console.WriteLine("Simulating concurrent edits to a TodoList...\n");
|
||
|
||
// Create initial list
|
||
var list = new TodoList
|
||
{
|
||
Id = Guid.NewGuid().ToString(),
|
||
Name = "Shopping List",
|
||
Items = new List<TodoItem>
|
||
{
|
||
new TodoItem { Task = "Buy milk", Completed = false },
|
||
new TodoItem { Task = "Buy bread", Completed = false }
|
||
}
|
||
};
|
||
|
||
await _db.TodoLists.InsertAsync(list);
|
||
await _db.SaveChangesAsync();
|
||
System.Console.WriteLine($"✓ Created list '{list.Name}' with {list.Items.Count} items");
|
||
await Task.Delay(100);
|
||
|
||
// Simulate Node A edit: Mark item as completed, add new item
|
||
var listA = _db.TodoLists.FindById(list.Id);
|
||
if (listA != null)
|
||
{
|
||
listA.Items[0].Completed = true; // Mark milk as done
|
||
listA.Items.Add(new TodoItem { Task = "Buy eggs", Completed = false });
|
||
await _db.TodoLists.UpdateAsync(listA);
|
||
await _db.SaveChangesAsync();
|
||
System.Console.WriteLine("📝 Node A: Marked 'Buy milk' complete, added 'Buy eggs'");
|
||
}
|
||
|
||
await Task.Delay(100);
|
||
|
||
// Simulate Node B edit: Mark different item, add different item
|
||
var listB = _db.TodoLists.FindById(list.Id);
|
||
if (listB != null)
|
||
{
|
||
listB.Items[1].Completed = true; // Mark bread as done
|
||
listB.Items.Add(new TodoItem { Task = "Buy cheese", Completed = false });
|
||
await _db.TodoLists.UpdateAsync(listB);
|
||
await _db.SaveChangesAsync();
|
||
System.Console.WriteLine("📝 Node B: Marked 'Buy bread' complete, added 'Buy cheese'");
|
||
}
|
||
|
||
await Task.Delay(200);
|
||
|
||
// Show final merged state
|
||
var merged = _db.TodoLists.FindById(list.Id);
|
||
if (merged != null)
|
||
{
|
||
System.Console.WriteLine("\n🔀 Merged Result:");
|
||
System.Console.WriteLine($" List: {merged.Name}");
|
||
foreach (var item in merged.Items)
|
||
{
|
||
var status = item.Completed ? "✓" : " ";
|
||
System.Console.WriteLine($" [{status}] {item.Task}");
|
||
}
|
||
|
||
var resolver = _serviceProvider.GetRequiredService<IConflictResolver>();
|
||
var resolverType = resolver.GetType().Name;
|
||
System.Console.WriteLine($"\nℹ️ Resolution Strategy: {resolverType}");
|
||
|
||
if (resolverType.Contains("Recursive"))
|
||
{
|
||
System.Console.WriteLine(" → Items merged by 'id', both edits preserved");
|
||
}
|
||
else
|
||
{
|
||
System.Console.WriteLine(" → Last write wins, Node B changes override Node A");
|
||
}
|
||
}
|
||
|
||
System.Console.WriteLine("\n✓ Demo complete. Run 'todos' to see all lists.\n");
|
||
}
|
||
}
|