TimeSyncServerDlg.cpp 8.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336
  1. // TimeSyncServerDlg.cpp : 实现文件
  2. //
  3. #include "stdafx.h"
  4. #include "TimeSyncServer.h"
  5. #include "TimeSyncServerDlg.h"
  6. #include "afxdialogex.h"
  7. #include "Simplelog.h"
  8. #include "MonitorObject.h"
  9. #include <ODBC/DBConnectPool.h>
  10. #include "Device.h"
  11. #include "AppService.h"
  12. #ifdef _DEBUG
  13. #define new DEBUG_NEW
  14. #endif
  15. // 用于应用程序“关于”菜单项的 CAboutDlg 对话框
  16. class CAboutDlg : public CDialogEx
  17. {
  18. public:
  19. CAboutDlg();
  20. // 对话框数据
  21. #ifdef AFX_DESIGN_TIME
  22. enum { IDD = IDD_ABOUTBOX };
  23. #endif
  24. protected:
  25. virtual void DoDataExchange(CDataExchange* pDX); // DDX/DDV 支持
  26. // 实现
  27. protected:
  28. DECLARE_MESSAGE_MAP()
  29. };
  30. CAboutDlg::CAboutDlg() : CDialogEx(IDD_ABOUTBOX)
  31. {
  32. }
  33. void CAboutDlg::DoDataExchange(CDataExchange* pDX)
  34. {
  35. CDialogEx::DoDataExchange(pDX);
  36. }
  37. BEGIN_MESSAGE_MAP(CAboutDlg, CDialogEx)
  38. END_MESSAGE_MAP()
  39. // CTimeSyncServerDlg 对话框
  40. CTimeSyncServerDlg::CTimeSyncServerDlg(CWnd* pParent /*=NULL*/)
  41. : CDialogEx(IDD_TIMESYNCSERVER_DIALOG, pParent)
  42. {
  43. m_hIcon = AfxGetApp()->LoadIcon(IDI_ICON1);
  44. m_strSvrPath = CSimpleLog::GetAppDir() + AfxGetAppName() + ".exe";
  45. }
  46. void CTimeSyncServerDlg::DoDataExchange(CDataExchange* pDX)
  47. {
  48. CDialogEx::DoDataExchange(pDX);
  49. }
  50. void CTimeSyncServerDlg::ToTray()
  51. {
  52. m_nid.cbSize = (DWORD)sizeof(NOTIFYICONDATA);
  53. m_nid.hWnd = this->m_hWnd;
  54. m_nid.uID = IDI_ICON1;//图标的ID
  55. m_nid.uFlags = NIF_ICON | NIF_MESSAGE | NIF_TIP;
  56. m_nid.uCallbackMessage = WM_SHOWTASK;//自定义的消息名称,在推盘图标上任意动作就产生此消息
  57. m_nid.hIcon = LoadIcon(AfxGetInstanceHandle(), MAKEINTRESOURCE(IDI_ICON1));
  58. strcpy_s(m_nid.szTip, "数据中心"); //信息提示条
  59. Shell_NotifyIcon(NIM_ADD, &m_nid); //在托盘区添加图标
  60. ShowWindow(SW_HIDE); //隐藏主窗口
  61. m_bIsShow = false;
  62. }
  63. void CTimeSyncServerDlg::DeleteTray()
  64. {
  65. m_nid.cbSize = (DWORD)sizeof(NOTIFYICONDATA);
  66. m_nid.hWnd = this->m_hWnd;
  67. m_nid.uID = IDI_ICON1;//删除图标,仅使用ID即可(我们使用的是图标的ID)
  68. //nid.uFlags=NIF_ICON;//删除图标,仅使用ICON即可
  69. /*nid.uFlags=NIF_ICON|NIF_MESSAGE|NIF_TIP ;
  70. nid.uCallbackMessage=WM_SHOWTASK;//自定义的消息名称
  71. nid.hIcon=LoadIcon(AfxGetInstanceHandle(),MAKEINTRESOURCE(IDR_MAINFRAME));
  72. strcpy(nid.szTip,"程序名称"); //信息提示条为"计划任务提醒"
  73. */
  74. Shell_NotifyIcon(NIM_DELETE, &m_nid); //在托盘区删除图标
  75. }
  76. VOID CTimeSyncServerDlg::OnTrayMenuShowMainWnd()
  77. {
  78. //ShowWindow(SW_SHOW);
  79. ::SetWindowPos(
  80. GetSafeHwnd(),
  81. HWND_TOP,
  82. 0,
  83. 0,
  84. 0,
  85. 0,
  86. SWP_NOMOVE | SWP_NOSIZE | SWP_NOACTIVATE);
  87. ShowWindow(SW_RESTORE);
  88. // 置顶
  89. SetForegroundWindow();
  90. m_bIsShow = true;
  91. }
  92. void CTimeSyncServerDlg::OnTrayMenuCloseClient()
  93. {
  94. DeleteTray();//删除托盘图标
  95. SPDLOG_ERROR("主动退出程序");
  96. TerminateProcess(INVALID_HANDLE_VALUE, 0);
  97. CAppService::Instance()->Stop();
  98. CTimeSyncServerDlg::OnCancel();
  99. }
  100. LRESULT CTimeSyncServerDlg::OnShowTask(WPARAM wParam, LPARAM lParam)
  101. {
  102. if (wParam != IDI_ICON1)//不是通过前面注册的图标响应的消息的话,退出
  103. return 1;
  104. switch (lParam)
  105. {
  106. case WM_RBUTTONDOWN://右键按下时弹出快捷菜单
  107. {
  108. POINT point;
  109. ::GetCursorPos(&point);//得到鼠标位置
  110. CMenu Popupmenu;
  111. if (!Popupmenu.CreatePopupMenu())
  112. return 0;
  113. Popupmenu.AppendMenu(MF_STRING, ID_POPMENU_SHOW_MAIN_WND, _T("显示主界面"));
  114. Popupmenu.AppendMenu(MF_STRING, ID_POPMENU_CLOSE_CLIENT, _T("退出服务"));
  115. Popupmenu.TrackPopupMenu(TPM_LEFTBUTTON | TPM_LEFTALIGN, point.x, point.y, this);
  116. Popupmenu.DestroyMenu();
  117. }
  118. break;
  119. case WM_LBUTTONDBLCLK://双击左键的处理
  120. {
  121. OnTrayMenuShowMainWnd();
  122. //this->ShowWindow(SW_SHOW);//简单的显示主窗口
  123. //DeleteTray();//删除窗口
  124. }
  125. break;
  126. default:
  127. break;
  128. }
  129. return 0;
  130. }
  131. BEGIN_MESSAGE_MAP(CTimeSyncServerDlg, CDialogEx)
  132. ON_WM_SYSCOMMAND()
  133. ON_WM_PAINT()
  134. ON_WM_QUERYDRAGICON()
  135. ON_BN_CLICKED(IDC_BUTTON1, &CTimeSyncServerDlg::OnBnClickedButton1)
  136. ON_BN_CLICKED(IDC_BUTTON2, &CTimeSyncServerDlg::OnBnClickedButton2)
  137. ON_BN_CLICKED(IDC_BUTTON3, &CTimeSyncServerDlg::OnBnClickedButton3)
  138. ON_BN_CLICKED(IDC_BUTTON4, &CTimeSyncServerDlg::OnBnClickedButton4)
  139. ON_BN_CLICKED(IDC_BTN_READ, &CTimeSyncServerDlg::OnBnClickedBtnRead)
  140. ON_MESSAGE(WM_SHOWTASK, OnShowTask)
  141. ON_COMMAND(ID_POPMENU_SHOW_MAIN_WND, OnTrayMenuShowMainWnd)
  142. ON_COMMAND(ID_POPMENU_CLOSE_CLIENT, OnTrayMenuCloseClient)
  143. END_MESSAGE_MAP()
  144. // CTimeSyncServerDlg 消息处理程序
  145. BOOL CTimeSyncServerDlg::OnInitDialog()
  146. {
  147. CDialogEx::OnInitDialog();
  148. // 将“关于...”菜单项添加到系统菜单中。
  149. // IDM_ABOUTBOX 必须在系统命令范围内。
  150. ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
  151. ASSERT(IDM_ABOUTBOX < 0xF000);
  152. CMenu* pSysMenu = GetSystemMenu(FALSE);
  153. if (pSysMenu != NULL)
  154. {
  155. BOOL bNameValid;
  156. CString strAboutMenu;
  157. bNameValid = strAboutMenu.LoadString(IDS_ABOUTBOX);
  158. ASSERT(bNameValid);
  159. if (!strAboutMenu.IsEmpty())
  160. {
  161. pSysMenu->AppendMenu(MF_SEPARATOR);
  162. pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
  163. }
  164. }
  165. // 设置此对话框的图标。 当应用程序主窗口不是对话框时,框架将自动
  166. // 执行此操作
  167. SetIcon(m_hIcon, TRUE); // 设置大图标
  168. SetIcon(m_hIcon, FALSE); // 设置小图标
  169. SetWindowText(AfxGetAppName());
  170. //托盘图标
  171. m_nid.cbSize = (DWORD)sizeof(NOTIFYICONDATA);
  172. m_nid.hWnd = this->m_hWnd;
  173. m_nid.uID = IDI_ICON1;//图标的ID
  174. m_nid.uFlags = NIF_ICON | NIF_MESSAGE | NIF_TIP;
  175. m_nid.uCallbackMessage = WM_SHOWTASK;//自定义的消息名称,在推盘图标上任意动作就产生此消息
  176. m_nid.hIcon = LoadIcon(AfxGetInstanceHandle(), MAKEINTRESOURCE(IDI_ICON1));
  177. strcpy_s(m_nid.szTip, "数据中心");
  178. Shell_NotifyIcon(NIM_ADD, &m_nid); //在托盘区添加图标
  179. ShowWindow(SW_SHOW); //隐藏主窗口
  180. CenterWindow();
  181. return TRUE; // 除非将焦点设置到控件,否则返回 TRUE
  182. }
  183. void CTimeSyncServerDlg::OnSysCommand(UINT nID, LPARAM lParam)
  184. {
  185. if ((nID & 0xFFF0) == IDM_ABOUTBOX)
  186. {
  187. CAboutDlg dlgAbout;
  188. dlgAbout.DoModal();
  189. }
  190. if (nID == SC_MINIMIZE)//点击标题栏的“最小化”
  191. {
  192. ToTray(); //最小化到托盘的函数
  193. return;
  194. }
  195. if (nID == SC_CLOSE)
  196. {
  197. ToTray(); //最小化到托盘的函数
  198. return;
  199. }
  200. else
  201. {
  202. CDialogEx::OnSysCommand(nID, lParam);
  203. }
  204. }
  205. // 如果向对话框添加最小化按钮,则需要下面的代码
  206. // 来绘制该图标。 对于使用文档/视图模型的 MFC 应用程序,
  207. // 这将由框架自动完成。
  208. void CTimeSyncServerDlg::OnPaint()
  209. {
  210. if (IsIconic())
  211. {
  212. CPaintDC dc(this); // 用于绘制的设备上下文
  213. SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0);
  214. // 使图标在工作区矩形中居中
  215. int cxIcon = GetSystemMetrics(SM_CXICON);
  216. int cyIcon = GetSystemMetrics(SM_CYICON);
  217. CRect rect;
  218. GetClientRect(&rect);
  219. int x = (rect.Width() - cxIcon + 1) / 2;
  220. int y = (rect.Height() - cyIcon + 1) / 2;
  221. // 绘制图标
  222. dc.DrawIcon(x, y, m_hIcon);
  223. }
  224. else
  225. {
  226. CDialogEx::OnPaint();
  227. }
  228. }
  229. //当用户拖动最小化窗口时系统调用此函数取得光标
  230. //显示。
  231. HCURSOR CTimeSyncServerDlg::OnQueryDragIcon()
  232. {
  233. return static_cast<HCURSOR>(m_hIcon);
  234. }
  235. void CTimeSyncServerDlg::OnBnClickedButton1()
  236. {
  237. UpdateData(TRUE);//读取编辑框变量m_strSvrPath的值
  238. if (InstallService(m_strSvrPath))
  239. {
  240. //printf("/n/nService Installed Sucessfully/n");
  241. MessageBox("Service Installed Sucessfully", "note", MB_OK | MB_ICONINFORMATION);
  242. }
  243. else
  244. {
  245. //printf("/n/nError Installing Service/n");
  246. MessageBox("Error Installing Service", "note", MB_OK | MB_ICONERROR);
  247. }
  248. }
  249. void CTimeSyncServerDlg::OnBnClickedButton2()
  250. {
  251. if (DeleteService())
  252. {
  253. //printf("/n/nService UnInstalled Sucessfully/n");
  254. MessageBox("Service UnInstalled Sucessfully", "note", MB_OK | MB_ICONINFORMATION);
  255. }
  256. else
  257. {
  258. //printf("/n/nError UnInstalling Service/n");
  259. MessageBox("Error UnInstalling Service", "note", MB_OK | MB_ICONWARNING);
  260. }
  261. }
  262. void CTimeSyncServerDlg::OnBnClickedButton3()
  263. {
  264. //打开服务管理器
  265. ::ShellExecute(NULL, "open", "cmd.exe", "/c services.msc", NULL, SW_HIDE);
  266. }
  267. void CTimeSyncServerDlg::OnBnClickedButton4()
  268. {
  269. //打开启动项
  270. ::ShellExecute(NULL, "open", "cmd.exe", "/c msconfig", NULL, SW_HIDE);
  271. }
  272. void CTimeSyncServerDlg::OnBnClickedBtnRead()
  273. {
  274. CDBConnectPool::Instance()->DBExecuteSQL(CString("TRUNCATE TABLE rm_alarm_set"));
  275. CDBConnectPool::Instance()->DBExecuteSQL(CString("DELETE rm_mo WHERE [id]!= '100000'"));
  276. CDBConnectPool::Instance()->DBExecuteSQL(CString("TRUNCATE TABLE rm_map"));
  277. //加载设备
  278. CDeviceMng::Instance()->LoadDevice();
  279. CMonitorObjectMng::Instance()->LoadMonitorTree();
  280. CMonitorObjectMng::Instance()->LoadHistoryData();
  281. CAppService::Instance()->GetHandle()->SendAllClientSync();
  282. }