解决快捷键可能延迟响应的问题

JiuLing-zhang · JiuLing-zhang · commit 8e0d845eaed7 · 2025-10-21T21:52:59.000+08:00
diff --git a/src/ComputerLock/App.xaml.cs b/src/ComputerLock/App.xaml.cs
@@ -1,12 +1,12 @@
-﻿using System.Globalization;
-using System.IO;
-using Microsoft.Extensions.DependencyInjection;
-using MudBlazor.Services;
-using System.Windows;
-using ComputerLock.Interfaces;
+﻿using ComputerLock.Interfaces;
 using ComputerLock.Update;
 using JiuLing.TitleBarKit;
+using Microsoft.Extensions.DependencyInjection;
+using MudBlazor.Services;
 using MudExtensions.Services;
+using System.Globalization;
+using System.IO;
+using System.Windows;
 using Application = System.Windows.Application;
 
 namespace ComputerLock;
@@ -17,8 +17,11 @@ public partial class App : Application
 {
     private static Mutex _mutex = default!;
     private WindowMain? _mainWindow;
+    private ILogger _logger;
     protected override void OnStartup(StartupEventArgs e)
     {
+        base.OnStartup(e);
+
         _mutex = new Mutex(true, AppBase.FriendlyName);
         if (!_mutex.WaitOne(0, false))
         {
@@ -30,7 +33,6 @@ protected override void OnStartup(StartupEventArgs e)
         Environment.CurrentDirectory = Path.GetDirectoryName(AppBase.ExecutablePath);
 
         Init();
-        base.OnStartup(e);
     }
 
     private void Init()
@@ -96,11 +98,36 @@ private void Init()
             System.Windows.Media.RenderOptions.ProcessRenderMode = System.Windows.Interop.RenderMode.SoftwareOnly;
         }
 
+        _logger = sp.GetRequiredService<ILogger>();
+        // 确保 HotkeyHook 已被创建
+        sp.GetRequiredService<HotkeyHook>();
+        KeepMessageLive();
+
         _mainWindow = sp.GetRequiredService<WindowMain>();
         Application.Current.MainWindow = _mainWindow;
         _mainWindow.Show();
     }
 
+    private void KeepMessageLive()
+    {
+        // 启动后台消息循环，防止主线程被挂起
+        Task.Run(() =>
+        {
+            try
+            {
+                while (WinApi.GetMessage(out var msg, IntPtr.Zero, 0, 0))
+                {
+                    WinApi.TranslateMessage(ref msg);
+                    WinApi.DispatchMessage(ref msg);
+                }
+            }
+            catch (Exception ex)
+            {
+                _logger.Error("热键消息保活服务异常", ex);
+            }
+        });
+    }
+
     protected override void OnExit(ExitEventArgs e)
     {
         _mainWindow?.Dispose();
diff --git a/src/ComputerLock/Interfaces/IWindowsMessageBox.cs b/src/ComputerLock/Interfaces/IWindowsMessageBox.cs
@@ -1,6 +1,6 @@
 ﻿namespace ComputerLock.Interfaces;
 
-internal interface IWindowsMessageBox
+public interface IWindowsMessageBox
 {
     void Show(string message);
 }
diff --git a/src/ComputerLock/Platforms/WinApi.cs b/src/ComputerLock/Platforms/WinApi.cs
@@ -146,4 +146,26 @@ public struct LUID_AND_ATTRIBUTES
     public const uint SE_PRIVILEGE_ENABLED = 0x00000002;
 
     #endregion
+
+    #region 消息相关
+    [StructLayout(LayoutKind.Sequential)]
+    public struct ApiMessage
+    {
+        public IntPtr hwnd;
+        public uint message;
+        public IntPtr wParam;
+        public IntPtr lParam;
+        public uint time;
+        public Point pt;
+    }
+
+    [DllImport("user32.dll")]
+    public static extern bool GetMessage(out ApiMessage lpMsg, IntPtr hWnd, uint wMsgFilterMin, uint wMsgFilterMax);
+
+    [DllImport("user32.dll")]
+    public static extern bool TranslateMessage([In] ref ApiMessage lpMsg);
+
+    [DllImport("user32.dll")]
+    public static extern IntPtr DispatchMessage([In] ref ApiMessage lpMsg);
+    #endregion
 }
diff --git a/src/ComputerLock/Shared/MainLayout.razor.cs b/src/ComputerLock/Shared/MainLayout.razor.cs
@@ -3,6 +3,7 @@
 using JiuLing.TitleBarKit;
 
 namespace ComputerLock.Shared;
+
 public partial class MainLayout
 {
     private bool _isDarkMode;
@@ -31,9 +32,6 @@ public partial class MainLayout
     [Inject]
     private IGlobalLockService GlobalLockService { get; set; } = null!;
 
-    [Inject]
-    private HotkeyHook HotkeyHook { get; set; } = null!;
-
     [Inject]
     private ILogger Logger { get; set; } = null!;
 
@@ -144,39 +142,6 @@ private void InitializeEventBinding()
         }
         _isInitialized = true;
 
-        if (AppSettings.LockHotkeyString.IsNotEmpty())
-        {
-            RegisterLockHotkey();
-        }
-
-        HotkeyHook.HotkeyPressed += (id) =>
-        {
-            if (id == (int)HotkeyType.Lock)
-            {
-                if (!GlobalLockService.IsLocked)
-                {
-                    Logger.Info("快捷键锁定");
-                    GlobalLockService.Lock();
-                }
-                else
-                {
-                    if (AppSettings.ScreenUnlockMethod == ScreenUnlockMethods.Hotkey && AppSettings.IsUnlockUseLockHotkey)
-                    {
-                        Logger.Info("快捷键解锁");
-                        GlobalLockService.Unlock();
-                    }
-                }
-            }
-            else if (id == (int)HotkeyType.Unlock)
-            {
-                if (GlobalLockService.IsLocked)
-                {
-                    Logger.Info("快捷键解锁（独立解锁）");
-                    GlobalLockService.Unlock();
-                }
-            }
-        };
-
         ThemeSwitchService.OnThemeChanged += async theme =>
         {
             await SwitchThemeAsync(theme);
@@ -194,20 +159,5 @@ private async Task OpenDonationDialog()
         await Dialog.ShowAsync<Donation>("", options);
     }
 
-    public void RegisterLockHotkey()
-    {
-        try
-        {
-            if (AppSettings.LockHotkey != null)
-            {
-                Logger.Info("注册锁屏热键");
-                HotkeyHook.Register((int)HotkeyType.Lock, AppSettings.LockHotkey);
-            }
-        }
-        catch (Exception ex)
-        {
-            Logger.Error($"绑定锁屏热键失败", ex);
-            Snackbar.Add($"{Lang["ExRegistFailed"]}{ex.Message}", Severity.Error);
-        }
-    }
+
 }
diff --git a/src/ComputerLock/WindowMain.xaml.cs b/src/ComputerLock/WindowMain.xaml.cs
@@ -3,31 +3,72 @@
 using ComputerLock.Interfaces;
 
 namespace ComputerLock;
+
 public partial class WindowMain : Window, IDisposable
 {
     private readonly AppSettings _appSettings;
     private readonly IGlobalLockService _globalLockService;
     private readonly ILogger _logger;
+    private readonly HotkeyHook _hotkeyHook;
+    private readonly IStringLocalizer<Lang> _lang;
+    private readonly IWindowsMessageBox _windowsMessageBox;
 
     private readonly NotifyIcon _notifyIcon = new();
     private readonly ContextMenuStrip _contextMenuStrip = new();
 
-    public WindowMain(AppSettings appSettings, IGlobalLockService globalLockService, ILogger logger)
+    public WindowMain(AppSettings appSettings, IGlobalLockService globalLockService, ILogger logger, HotkeyHook hotkeyHook, IStringLocalizer<Lang> lang, IWindowsMessageBox windowsMessageBox)
     {
         InitializeComponent();
 
         _appSettings = appSettings;
         _globalLockService = globalLockService;
         _logger = logger;
+        _hotkeyHook = hotkeyHook;
+        _lang = lang;
+        _windowsMessageBox = windowsMessageBox;
 
         InitializeNotifyIcon();
+
         _logger.Info("系统启动");
 
+        if (_appSettings.LockHotkeyString.IsNotEmpty())
+        {
+            RegisterLockHotkey();
+        }
+        _hotkeyHook.HotkeyPressed += (id) =>
+        {
+            if (id == (int)HotkeyType.Lock)
+            {
+                if (!_globalLockService.IsLocked)
+                {
+                    _logger.Info("快捷键锁定");
+                    _globalLockService.Lock();
+                }
+                else
+                {
+                    if (_appSettings.ScreenUnlockMethod == ScreenUnlockMethods.Hotkey && _appSettings.IsUnlockUseLockHotkey)
+                    {
+                        _logger.Info("快捷键解锁");
+                        _globalLockService.Unlock();
+                    }
+                }
+            }
+            else if (id == (int)HotkeyType.Unlock)
+            {
+                if (_globalLockService.IsLocked)
+                {
+                    _logger.Info("快捷键解锁（独立解锁）");
+                    _globalLockService.Unlock();
+                }
+            }
+        };
+
         if (_appSettings.LockOnStartup)
         {
             _logger.Info("程序启动时锁定屏幕");
             _globalLockService.Lock();
         }
+
     }
 
     private void Window_Loaded(object sender, RoutedEventArgs e)
@@ -109,6 +150,23 @@ private void Window_KeyDown(object sender, System.Windows.Input.KeyEventArgs e)
             }
         }
     }
+
+    private void RegisterLockHotkey()
+    {
+        try
+        {
+            if (_appSettings.LockHotkey != null)
+            {
+                _logger.Info("注册锁屏热键");
+                _hotkeyHook.Register((int)HotkeyType.Lock, _appSettings.LockHotkey);
+            }
+        }
+        catch (Exception ex)
+        {
+            _logger.Error($"绑定锁屏热键失败", ex);
+            _windowsMessageBox.Show($"{_lang["ExRegistFailed"]}{ex.Message}");
+        }
+    }
     public void Dispose()
     {
         _logger.Info("系统资源释放，系统关闭");

Original file line number	Diff line number	Diff line change
`@@ -1,6 +1,6 @@`
`1`	`1`	`namespace ComputerLock.Interfaces;`
`2`	`2`
`3`		`-internal interface IWindowsMessageBox`
	`3`	`+public interface IWindowsMessageBox`
`4`	`4`	`{`
`5`	`5`	`void Show(string message);`
`6`	`6`	`}`