From f317709fbc7de95fa2e1b7f41d01c85a7fec8238 Mon Sep 17 00:00:00 2001 From: zs <zhousong@weben-smart.com> Date: 周四, 21 11月 2024 10:50:06 +0800 Subject: [PATCH] 注册表方式启动 --- DataCapture_MA/Form1.cs | 70 +++++++++++++++++++++++++++++----- 1 files changed, 59 insertions(+), 11 deletions(-) diff --git a/DataCapture_MA/Form1.cs b/DataCapture_MA/Form1.cs index 74496c1..f6a8d4a 100644 --- a/DataCapture_MA/Form1.cs +++ b/DataCapture_MA/Form1.cs @@ -17,11 +17,13 @@ using DataCapture_MA.AutoMapperConfig; using System.IO; using IWshRuntimeLibrary; +using Microsoft.Win32; namespace DataCapture_MA { public partial class CaptrueForm : Form { + private CancellationTokenSource _cts = new CancellationTokenSource(); bool RunningFlag = false; Socket skt = WZ.Useful.Commons.NetworkUtil.CreateUdpSocket(); EndPoint RemotePoint = null; @@ -63,22 +65,68 @@ cfg.AddProfile<MapperConfiguration>(); }); - // 鍒涘缓蹇嵎鏂瑰紡 - string shortcutName = "MaDataCapture"; // 蹇嵎鏂瑰紡鍚嶇О - string targetPath = Application.ExecutablePath; // 绋嬪簭璺緞 - string startupPath = Environment.GetFolderPath(Environment.SpecialFolder.Startup); // 鍚姩鏂囦欢澶硅矾寰� - CreateShortcut(shortcutName, targetPath, startupPath); // 鍒涘缓蹇嵎鏂瑰紡 + ////鍒涘缓蹇嵎鏂瑰紡 + //string shortcutName = "MaDataCapture"; // 蹇嵎鏂瑰紡鍚嶇О + //string targetPath = Application.ExecutablePath; // 绋嬪簭璺緞 + //string startupPath = Environment.GetFolderPath(Environment.SpecialFolder.Startup); // 鍚姩鏂囦欢澶硅矾寰� + //CreateShortcut(shortcutName, targetPath, startupPath); // 鍒涘缓蹇嵎鏂瑰紡 - if (ipSelect.Items.Count > 0) - { - StartReceive(); // 鍚姩绋嬪簭 - CreateDatabase(); // 鍒濆鍖栨暟鎹簱 - } + SetRegistryAutoStart("MaDataCapture", Application.ExecutablePath); // 娉ㄥ唽琛ㄦ柟寮忓疄鐜拌嚜鍚姩 + StartReceive(); // 鍚姩绋嬪簭 + CreateDatabase(); // 鍒濆鍖栨暟鎹簱 + + Log4NetHelper.WriteErrorLog(LogType.Receive, "绋嬪簭鍚姩:" + DateTime.Now); // 鍒犻櫎涓�涓湀鍓嶇殑鏁版嵁 new Thread(AutoDeleteHistoryDate.Handler).Start(); + + this.FormClosing += CaptrueForm_FormClosing; + this.FormClosed += CaptrueForm_FormClosed; + } + /// <summary> + /// 璁剧疆娉ㄥ唽琛ㄥ疄鐜拌嚜鍚姩 + /// </summary> + /// <param name="appName">搴旂敤绋嬪簭鍚嶇О</param> + /// <param name="appPath">搴旂敤绋嬪簭璺緞</param> + private void SetRegistryAutoStart(string appName, string appPath) + { + try + { + // 鎵撳紑娉ㄥ唽琛ㄨ矾寰� + RegistryKey key = Registry.CurrentUser.OpenSubKey(@"Software\Microsoft\Windows\CurrentVersion\Run", true); + + if (key == null) + { + //throw new Exception("鏃犳硶璁块棶娉ㄥ唽琛ㄨ矾寰勶紒"); + Log4NetHelper.WriteErrorLog(LogType.Receive, "鏃犳硶璁块棶娉ㄥ唽琛ㄨ矾寰�:" + DateTime.Now); + } + + // 璁剧疆娉ㄥ唽琛ㄩ敭鍊� + key.SetValue(appName, appPath); + + key.Close(); + } + catch (Exception ex) + { + MessageBox.Show($"璁剧疆寮�鏈鸿嚜鍚姩澶辫触锛歿ex.Message}锛岃杩涜鎵嬪姩鍚姩", "閿欒", MessageBoxButtons.OK, MessageBoxIcon.Error); + } + } + private void CaptrueForm_FormClosing(object sender, FormClosingEventArgs e) + { + Log4NetHelper.WriteErrorLog(LogType.Receive, "绋嬪簭鍏抽棴:" + DateTime.Now); + // 閫氱煡鍚庡彴浠诲姟鍙栨秷 + _cts.Cancel(); + + // 纭繚浠诲姟瀹屾垚 + Task.WaitAll(Task.WhenAll()); } + private void CaptrueForm_FormClosed(object sender, FormClosedEventArgs e) + { + Log4NetHelper.WriteErrorLog(LogType.Receive, "绋嬪簭閫�鍑�:" + DateTime.Now); + // 纭繚绋嬪簭閫�鍑� + Environment.Exit(0); + } public static void CreateShortcut(string shortcutName, string targetPath, string startupPath) { @@ -92,7 +140,7 @@ shortcut.WorkingDirectory = Path.GetDirectoryName(targetPath); shortcut.Save(); } - + private void ReceiveHandle() { //鎺ユ敹鏁版嵁澶勭悊绾跨▼ -- Gitblit v1.9.3