This commit is contained in:
unknown 2025-08-25 18:49:43 +08:00
parent b87779ade7
commit 4b30df49e2
2 changed files with 87 additions and 77 deletions

View File

@ -11,15 +11,15 @@ namespace Hi.Webapi.Hubs
/// </summary> /// </summary>
public class RenderingHub : Hub public class RenderingHub : Hub
{ {
private readonly RenderingService _renderingService;
private readonly ILogger<RenderingHub> _logger;
private readonly Dictionary<string, byte[]> _lastFrameCache = []; private readonly Dictionary<string, byte[]> _lastFrameCache = [];
private readonly Dictionary<string, Mat4d> _sketchViewCache = []; private readonly Dictionary<string, Mat4d> _sketchViewCache = [];
public RenderingService RenderingService { get; }
public ILogger<RenderingHub> Logger { get; }
public RenderingHub(RenderingService renderingService, ILogger<RenderingHub> logger) public RenderingHub(RenderingService renderingService, ILogger<RenderingHub> logger)
{ {
_renderingService = renderingService; RenderingService = renderingService;
_logger = logger; Logger = logger;
} }
/// <summary> /// <summary>
@ -28,18 +28,17 @@ namespace Hi.Webapi.Hubs
public async Task InitializeCanvas(int width, int height) public async Task InitializeCanvas(int width, int height)
{ {
var sessionId = Context.ConnectionId; var sessionId = Context.ConnectionId;
_logger.LogInformation($"InitializeCanvas called - SessionId: {sessionId}, Width: {width}, Height: {height}"); Logger.LogInformation($"InitializeCanvas called - SessionId: {sessionId}, Width: {width}, Height: {height}");
var engine = _renderingService.GetOrCreateEngine(sessionId); var engine = RenderingService.GetOrCreateEngine(sessionId);
_logger.LogInformation($"Engine created/retrieved - SessionId: {sessionId}"); Logger.LogInformation($"Engine created/retrieved - SessionId: {sessionId}");
// 啟動引擎(必須在設置回調之前)
engine.Start(width, height); engine.Start(width, height);
_logger.LogInformation($"Engine started with size {width}x{height} - SessionId: {sessionId}"); Logger.LogInformation($"Engine started with size {width}x{height} - SessionId: {sessionId}");
// 在設置回調之前,先捕獲客戶端代理 // 在設置回調之前,先捕獲客戶端代理
var clientProxy = Clients.Caller; var clientProxy = Clients.Caller;
var logger = _logger; var logger = Logger;
var frameCount = 0; var frameCount = 0;
// 設置渲染回調 // 設置渲染回調
@ -52,11 +51,8 @@ namespace Hi.Webapi.Hubs
frameCount++; frameCount++;
if (frameCount <= 5) // 只記錄前5幀 if (frameCount <= 5) // 只記錄前5幀
{
logger.LogInformation($"Frame {frameCount} rendered - Size: {w}x{h}, SessionId: {sessionId}"); logger.LogInformation($"Frame {frameCount} rendered - Size: {w}x{h}, SessionId: {sessionId}");
}
// 檢查寬度和高度是否有效
if (w <= 0 || h <= 0) if (w <= 0 || h <= 0)
{ {
logger.LogWarning($"Invalid image size: {w}x{h} - SessionId: {sessionId}"); logger.LogWarning($"Invalid image size: {w}x{h} - SessionId: {sessionId}");
@ -65,7 +61,6 @@ namespace Hi.Webapi.Hubs
try try
{ {
// 在回調中同步處理圖像數據,轉換為 byte[]
int wh = w * h; int wh = w * h;
byte[] rgba = new byte[wh * 4]; byte[] rgba = new byte[wh * 4];
for (int i = 0; i < wh; i++) for (int i = 0; i < wh; i++)
@ -85,11 +80,8 @@ namespace Hi.Webapi.Hubs
} }
_lastFrameCache[sessionId] = rgba; _lastFrameCache[sessionId] = rgba;
if (frameCount <= 5) if (frameCount <= 5)
{
logger.LogInformation($"Frame {frameCount} sending - Size: {w}x{h}, Data: {rgba.Length} bytes"); logger.LogInformation($"Frame {frameCount} sending - Size: {w}x{h}, Data: {rgba.Length} bytes");
}
using MemoryStream dstMemoryStream = new MemoryStream(); using MemoryStream dstMemoryStream = new MemoryStream();
using GZipStream gZipStream = new GZipStream(dstMemoryStream, CompressionMode.Compress); using GZipStream gZipStream = new GZipStream(dstMemoryStream, CompressionMode.Compress);
@ -104,9 +96,7 @@ namespace Hi.Webapi.Hubs
var compressionRatio = (float)compressedRgbaArray.Length / rgba.Length * 100; var compressionRatio = (float)compressedRgbaArray.Length / rgba.Length * 100;
if (frameCount <= 5) if (frameCount <= 5)
{
logger.LogInformation($"Compressed: {compressedRgbaArray.Length} bytes ({compressionRatio:F1}% of original)"); logger.LogInformation($"Compressed: {compressedRgbaArray.Length} bytes ({compressionRatio:F1}% of original)");
}
// 發送壓縮數據到客戶端(注意:第二個參數應該是原始長度,不是壓縮後的長度) // 發送壓縮數據到客戶端(注意:第二個參數應該是原始長度,不是壓縮後的長度)
_ = clientProxy.SendAsync("ImageUpdate", _ = clientProxy.SendAsync("ImageUpdate",
@ -144,14 +134,14 @@ namespace Hi.Webapi.Hubs
engine.SetViewToHomeView(); engine.SetViewToHomeView();
_sketchViewCache[sessionId] = engine.SketchView; _sketchViewCache[sessionId] = engine.SketchView;
} }
_logger.LogInformation($"View initialized - SessionId: {sessionId}"); Logger.LogInformation($"View initialized - SessionId: {sessionId}");
// 確保引擎可見並開始渲染 // 確保引擎可見並開始渲染
engine.IsVisible = true; engine.IsVisible = true;
_logger.LogInformation($"Engine visibility set to true - SessionId: {sessionId}"); Logger.LogInformation($"Engine visibility set to true - SessionId: {sessionId}");
await Clients.Caller.SendAsync("CanvasInitialized", sessionId); await Clients.Caller.SendAsync("CanvasInitialized", sessionId);
_logger.LogInformation($"Canvas initialized for session: {sessionId}"); Logger.LogInformation($"Canvas initialized for session: {sessionId}");
} }
/// <summary> /// <summary>
@ -160,9 +150,9 @@ namespace Hi.Webapi.Hubs
public Task HandleMouseMove(double x, double y, int buttonMask) public Task HandleMouseMove(double x, double y, int buttonMask)
{ {
var sessionId = Context.ConnectionId; var sessionId = Context.ConnectionId;
_logger.LogDebug($"HandleMouseMove - SessionId: {sessionId}, X: {x}, Y: {y}, ButtonMask: {buttonMask}"); Logger.LogDebug($"HandleMouseMove - SessionId: {sessionId}, X: {x}, Y: {y}, ButtonMask: {buttonMask}");
var engine = _renderingService.GetOrCreateEngine(sessionId); var engine = RenderingService.GetOrCreateEngine(sessionId);
var p = new Vec2i((int)x, (int)y); var p = new Vec2i((int)x, (int)y);
// 移動鼠標 // 移動鼠標
@ -189,9 +179,9 @@ namespace Hi.Webapi.Hubs
public Task HandleMouseDown(double x, double y, int button) public Task HandleMouseDown(double x, double y, int button)
{ {
var sessionId = Context.ConnectionId; var sessionId = Context.ConnectionId;
_logger.LogInformation($"HandleMouseDown - SessionId: {sessionId}, X: {x}, Y: {y}, Button: {button}"); Logger.LogInformation($"HandleMouseDown - SessionId: {sessionId}, X: {x}, Y: {y}, Button: {button}");
var engine = _renderingService.GetOrCreateEngine(sessionId); var engine = RenderingService.GetOrCreateEngine(sessionId);
engine.MouseButtonDown(button); engine.MouseButtonDown(button);
return Task.CompletedTask; return Task.CompletedTask;
} }
@ -202,9 +192,9 @@ namespace Hi.Webapi.Hubs
public Task HandleMouseUp(double x, double y, int button) public Task HandleMouseUp(double x, double y, int button)
{ {
var sessionId = Context.ConnectionId; var sessionId = Context.ConnectionId;
_logger.LogInformation($"HandleMouseUp - SessionId: {sessionId}, X: {x}, Y: {y}, Button: {button}"); Logger.LogInformation($"HandleMouseUp - SessionId: {sessionId}, X: {x}, Y: {y}, Button: {button}");
var engine = _renderingService.GetOrCreateEngine(sessionId); var engine = RenderingService.GetOrCreateEngine(sessionId);
engine.MouseButtonUp(button); engine.MouseButtonUp(button);
return Task.CompletedTask; return Task.CompletedTask;
} }
@ -215,9 +205,9 @@ namespace Hi.Webapi.Hubs
public Task HandleMouseWheel(double x, double y, double deltaX, double deltaY, string browserBrand = "chrome") public Task HandleMouseWheel(double x, double y, double deltaX, double deltaY, string browserBrand = "chrome")
{ {
var sessionId = Context.ConnectionId; var sessionId = Context.ConnectionId;
_logger.LogInformation($"HandleMouseWheel - SessionId: {sessionId}, X: {x}, Y: {y}, DeltaX: {deltaX}, DeltaY: {deltaY}, Browser: {browserBrand}"); Logger.LogInformation($"HandleMouseWheel - SessionId: {sessionId}, X: {x}, Y: {y}, DeltaX: {deltaX}, DeltaY: {deltaY}, Browser: {browserBrand}");
var engine = _renderingService.GetOrCreateEngine(sessionId); var engine = RenderingService.GetOrCreateEngine(sessionId);
// 根據瀏覽器類型獲取縮放比例 // 根據瀏覽器類型獲取縮放比例
double scale = GetWheelScaling(browserBrand); double scale = GetWheelScaling(browserBrand);
@ -239,9 +229,9 @@ namespace Hi.Webapi.Hubs
public Task HandleResize(int width, int height) public Task HandleResize(int width, int height)
{ {
var sessionId = Context.ConnectionId; var sessionId = Context.ConnectionId;
_logger.LogInformation($"HandleResize - SessionId: {sessionId}, Width: {width}, Height: {height}"); Logger.LogInformation($"HandleResize - SessionId: {sessionId}, Width: {width}, Height: {height}");
var engine = _renderingService.GetOrCreateEngine(sessionId); var engine = RenderingService.GetOrCreateEngine(sessionId);
// 重置圖像緩存以強制重新渲染 // 重置圖像緩存以強制重新渲染
if (_lastFrameCache.ContainsKey(sessionId)) if (_lastFrameCache.ContainsKey(sessionId))
@ -251,7 +241,7 @@ namespace Hi.Webapi.Hubs
engine.Resize(width, height); engine.Resize(width, height);
_logger.LogInformation($"Resize completed - SessionId: {sessionId}"); Logger.LogInformation($"Resize completed - SessionId: {sessionId}");
return Task.CompletedTask; return Task.CompletedTask;
} }
@ -261,9 +251,9 @@ namespace Hi.Webapi.Hubs
public Task HandleVisibilityChange(string visibilityState) public Task HandleVisibilityChange(string visibilityState)
{ {
var sessionId = Context.ConnectionId; var sessionId = Context.ConnectionId;
_logger.LogInformation($"HandleVisibilityChange - SessionId: {sessionId}, State: {visibilityState}"); Logger.LogInformation($"HandleVisibilityChange - SessionId: {sessionId}, State: {visibilityState}");
var engine = _renderingService.GetOrCreateEngine(sessionId); var engine = RenderingService.GetOrCreateEngine(sessionId);
engine.IsVisible = visibilityState == "visible"; engine.IsVisible = visibilityState == "visible";
return Task.CompletedTask; return Task.CompletedTask;
@ -275,9 +265,9 @@ namespace Hi.Webapi.Hubs
public Task SetView(string viewType) public Task SetView(string viewType)
{ {
var sessionId = Context.ConnectionId; var sessionId = Context.ConnectionId;
_logger.LogInformation($"SetView called - SessionId: {sessionId}, ViewType: {viewType}"); Logger.LogInformation($"SetView called - SessionId: {sessionId}, ViewType: {viewType}");
var engine = _renderingService.GetOrCreateEngine(sessionId); var engine = RenderingService.GetOrCreateEngine(sessionId);
switch (viewType.ToLower()) switch (viewType.ToLower())
{ {
@ -309,7 +299,7 @@ namespace Hi.Webapi.Hubs
engine.SetViewToHomeView(); engine.SetViewToHomeView();
break; break;
default: default:
_logger.LogWarning($"Unknown view type: {viewType} - SessionId: {sessionId}"); Logger.LogWarning($"Unknown view type: {viewType} - SessionId: {sessionId}");
break; break;
} }
@ -324,9 +314,9 @@ namespace Hi.Webapi.Hubs
public Task HandleKeyDown(string key, string code, bool ctrlKey, bool shiftKey, bool altKey) public Task HandleKeyDown(string key, string code, bool ctrlKey, bool shiftKey, bool altKey)
{ {
var sessionId = Context.ConnectionId; var sessionId = Context.ConnectionId;
_logger.LogInformation($"HandleKeyDown - SessionId: {sessionId}, Key: {key}, Code: {code}, Ctrl: {ctrlKey}, Shift: {shiftKey}, Alt: {altKey}"); Logger.LogInformation($"HandleKeyDown - SessionId: {sessionId}, Key: {key}, Code: {code}, Ctrl: {ctrlKey}, Shift: {shiftKey}, Alt: {altKey}");
var engine = _renderingService.GetOrCreateEngine(sessionId); var engine = RenderingService.GetOrCreateEngine(sessionId);
// 使用 key 的 HashCode與 Blazor 版本一致) // 使用 key 的 HashCode與 Blazor 版本一致)
long keyCode = key.ToLower().GetHashCode(); long keyCode = key.ToLower().GetHashCode();
@ -360,9 +350,9 @@ namespace Hi.Webapi.Hubs
public Task HandleKeyUp(string key, string code, bool ctrlKey, bool shiftKey, bool altKey) public Task HandleKeyUp(string key, string code, bool ctrlKey, bool shiftKey, bool altKey)
{ {
var sessionId = Context.ConnectionId; var sessionId = Context.ConnectionId;
_logger.LogInformation($"HandleKeyUp - SessionId: {sessionId}, Key: {key}, Code: {code}, Ctrl: {ctrlKey}, Shift: {shiftKey}, Alt: {altKey}"); Logger.LogInformation($"HandleKeyUp - SessionId: {sessionId}, Key: {key}, Code: {code}, Ctrl: {ctrlKey}, Shift: {shiftKey}, Alt: {altKey}");
var engine = _renderingService.GetOrCreateEngine(sessionId); var engine = RenderingService.GetOrCreateEngine(sessionId);
long keyCode = key.ToLower().GetHashCode(); long keyCode = key.ToLower().GetHashCode();
engine.KeyUp(keyCode); engine.KeyUp(keyCode);
@ -376,9 +366,9 @@ namespace Hi.Webapi.Hubs
public Task HandleTouchDown(int pointerId, double x, double y) public Task HandleTouchDown(int pointerId, double x, double y)
{ {
var sessionId = Context.ConnectionId; var sessionId = Context.ConnectionId;
_logger.LogInformation($"HandleTouchDown - SessionId: {sessionId}, PointerId: {pointerId}, X: {x}, Y: {y}"); Logger.LogInformation($"HandleTouchDown - SessionId: {sessionId}, PointerId: {pointerId}, X: {x}, Y: {y}");
var engine = _renderingService.GetOrCreateEngine(sessionId); var engine = RenderingService.GetOrCreateEngine(sessionId);
engine.TouchDown(pointerId, (int)x, (int)y); engine.TouchDown(pointerId, (int)x, (int)y);
return Task.CompletedTask; return Task.CompletedTask;
@ -390,9 +380,9 @@ namespace Hi.Webapi.Hubs
public Task HandleTouchMove(int pointerId, double x, double y) public Task HandleTouchMove(int pointerId, double x, double y)
{ {
var sessionId = Context.ConnectionId; var sessionId = Context.ConnectionId;
_logger.LogDebug($"HandleTouchMove - SessionId: {sessionId}, PointerId: {pointerId}, X: {x}, Y: {y}"); Logger.LogDebug($"HandleTouchMove - SessionId: {sessionId}, PointerId: {pointerId}, X: {x}, Y: {y}");
var engine = _renderingService.GetOrCreateEngine(sessionId); var engine = RenderingService.GetOrCreateEngine(sessionId);
engine.TouchMove(pointerId, (int)x, (int)y); engine.TouchMove(pointerId, (int)x, (int)y);
return Task.CompletedTask; return Task.CompletedTask;
@ -404,9 +394,9 @@ namespace Hi.Webapi.Hubs
public Task HandleTouchUp(int pointerId) public Task HandleTouchUp(int pointerId)
{ {
var sessionId = Context.ConnectionId; var sessionId = Context.ConnectionId;
_logger.LogInformation($"HandleTouchUp - SessionId: {sessionId}, PointerId: {pointerId}"); Logger.LogInformation($"HandleTouchUp - SessionId: {sessionId}, PointerId: {pointerId}");
var engine = _renderingService.GetOrCreateEngine(sessionId); var engine = RenderingService.GetOrCreateEngine(sessionId);
engine.TouchUp(pointerId); engine.TouchUp(pointerId);
return Task.CompletedTask; return Task.CompletedTask;
@ -440,8 +430,8 @@ namespace Hi.Webapi.Hubs
_lastFrameCache.Remove(sessionId); _lastFrameCache.Remove(sessionId);
_sketchViewCache.Remove(sessionId); _sketchViewCache.Remove(sessionId);
_renderingService.RemoveEngine(sessionId); RenderingService.RemoveEngine(sessionId);
_logger.LogInformation($"Client disconnected: {sessionId}"); Logger.LogInformation($"Client disconnected: {sessionId}");
await base.OnDisconnectedAsync(exception); await base.OnDisconnectedAsync(exception);
} }
} }

View File

@ -6,16 +6,21 @@ using System.Collections.Concurrent;
namespace Hi.Webapi.Services namespace Hi.Webapi.Services
{ {
/// <summary> /// <summary>
/// 管理 DispEngine 實例和渲染操作的服務 /// Singleton Service for managing <see cref="DispEngine"/>.
/// </summary> /// </summary>
public class RenderingService : IDisposable public class RenderingService : IDisposable
{ {
private readonly ConcurrentDictionary<string, DispEngine> _engines = new(); /// <summary>
private readonly ILogger<RenderingService> _logger; /// Engine Dictionary.
/// Key is sessionID.
/// </summary>
public ConcurrentDictionary<string, DispEngine> EngineDictionary { get; } = new();
ILogger<RenderingService> Logger { get; }
private bool disposedValue;
public RenderingService(ILogger<RenderingService> logger) public RenderingService(ILogger<RenderingService> logger)
{ {
_logger = logger; Logger = logger;
} }
/// <summary> /// <summary>
@ -23,9 +28,9 @@ namespace Hi.Webapi.Services
/// </summary> /// </summary>
public DispEngine GetOrCreateEngine(string sessionId) public DispEngine GetOrCreateEngine(string sessionId)
{ {
return _engines.GetOrAdd(sessionId, id => return EngineDictionary.GetOrAdd(sessionId, id =>
{ {
_logger.LogInformation($"創建新的 DispEngineSessionId: {id}"); Logger.LogInformation($"創建新的 DispEngineSessionId: {id}");
var engine = new DispEngine(); var engine = new DispEngine();
engine.BackgroundColor = new Vec3d(0.1, 0.1, 0.5); engine.BackgroundColor = new Vec3d(0.1, 0.1, 0.5);
engine.BackgroundOpacity = 0.1; engine.BackgroundOpacity = 0.1;
@ -38,9 +43,9 @@ namespace Hi.Webapi.Services
/// </summary> /// </summary>
public bool RemoveEngine(string sessionId) public bool RemoveEngine(string sessionId)
{ {
if (_engines.TryRemove(sessionId, out var engine)) if (EngineDictionary.TryRemove(sessionId, out var engine))
{ {
_logger.LogInformation($"移除 DispEngineSessionId: {sessionId}"); Logger.LogInformation($"移除 DispEngineSessionId: {sessionId}");
try try
{ {
@ -52,7 +57,7 @@ namespace Hi.Webapi.Services
} }
catch (Exception ex) catch (Exception ex)
{ {
_logger.LogError(ex, $"清理 DispEngine 時發生錯誤SessionId: {sessionId}"); Logger.LogError(ex, $"清理 DispEngine 時發生錯誤SessionId: {sessionId}");
} }
return true; return true;
@ -63,11 +68,15 @@ namespace Hi.Webapi.Services
/// <summary> /// <summary>
/// 獲取當前活動的引擎數量 /// 獲取當前活動的引擎數量
/// </summary> /// </summary>
public int GetActiveEngineCount() => _engines.Count; public int GetActiveEngineCount() => EngineDictionary.Count;
/// <inheritdoc/>
public void Dispose() protected virtual void Dispose(bool disposing)
{ {
foreach (var kvp in _engines) if (!disposedValue)
{
if (disposing)
{
foreach (var kvp in EngineDictionary)
{ {
try try
{ {
@ -76,10 +85,21 @@ namespace Hi.Webapi.Services
} }
catch (Exception ex) catch (Exception ex)
{ {
_logger.LogError(ex, $"Dispose 時清理引擎錯誤SessionId: {kvp.Key}"); Logger.LogError(ex, $"Dispose 時清理引擎錯誤SessionId: {kvp.Key}");
} }
} }
_engines.Clear(); EngineDictionary.Clear();
} }
disposedValue = true;
}
}
/// <inheritdoc/>
public void Dispose()
{
Dispose(disposing: true);
GC.SuppressFinalize(this);
}
} }
} }