GuardInfo.cpp 8.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379
  1. #include "stdafx.h"
  2. #include "GuardInfo.h"
  3. #include "Global.h"
  4. IMPLEMENT_SERIAL(CGuardInfo, CObject, 0)
  5. CGuardInfo::CGuardInfo(void)
  6. {
  7. m_StudioMac = _T("");
  8. m_StudioDomain = _T("");
  9. m_StudioGuid = _T("");
  10. m_bUseYtUsb = FALSE;
  11. m_StudioName = _T("");
  12. m_StudioOwner = _T("");
  13. m_StudioAddress = _T("");
  14. m_StudioTel = _T("");
  15. m_StudioInfo = _T("");
  16. m_StudioVer = _T("");
  17. m_Uptodate = _T("");
  18. m_ExecResult = _T("");
  19. }
  20. CGuardInfo::~CGuardInfo(void)
  21. {
  22. }
  23. CGuardInfo &CGuardInfo::operator=(CGuardInfo& member)
  24. {
  25. m_StudioMac = member.m_StudioMac;
  26. m_StudioDomain = member.m_StudioDomain;
  27. m_StudioGuid = member.m_StudioGuid;
  28. m_bUseYtUsb = member.m_bUseYtUsb;
  29. m_StudioName = member.m_StudioName;
  30. m_StudioOwner = member.m_StudioOwner;
  31. m_StudioAddress = member.m_StudioAddress;
  32. m_StudioTel = member.m_StudioTel;
  33. m_StudioInfo = member.m_StudioInfo;
  34. m_StudioVer = member.m_StudioVer;
  35. m_Uptodate = member.m_Uptodate;
  36. m_ExecResult = member.m_ExecResult;
  37. return *this;
  38. }
  39. void CGuardInfo::Serialize(CArchive& ar)
  40. {
  41. CObject::Serialize(ar);
  42. if( ar.IsStoring() )
  43. {
  44. ar << m_StudioMac;
  45. ar << m_StudioGuid;
  46. ar << m_bUseYtUsb;
  47. ar << m_StudioName;
  48. ar << m_StudioDomain;
  49. ar << m_StudioOwner;
  50. ar << m_StudioAddress;
  51. ar << m_StudioTel;
  52. ar << m_StudioInfo;
  53. ar << m_StudioVer;
  54. ar << m_Uptodate;
  55. ar << m_ExecResult;
  56. }
  57. else
  58. {
  59. ar >> m_StudioMac;
  60. ar >> m_StudioGuid;
  61. ar >> m_bUseYtUsb;
  62. ar >> m_StudioName;
  63. ar >> m_StudioDomain;
  64. ar >> m_StudioOwner;
  65. ar >> m_StudioAddress;
  66. ar >> m_StudioTel;
  67. ar >> m_StudioInfo;
  68. ar >> m_StudioVer;
  69. ar >> m_Uptodate;
  70. ar >> m_ExecResult;
  71. }
  72. }
  73. //////////////////////////////////////////////////////////////////////////
  74. IMPLEMENT_SERIAL(CGuardCommand, CObject, 0)
  75. CGuardCommand::CGuardCommand(void)
  76. {
  77. m_byTrigger = 0;
  78. m_nClientCount = -1;
  79. m_nOrderCount = -1;
  80. m_nHistoryOrderCount = -1;
  81. m_nExecCommand = -1;
  82. m_strExecTime = _T("");
  83. m_strOldTableName = _T("");
  84. m_strNewTableName = _T("");
  85. m_strOldFieldName = _T("");
  86. m_strNewFieldName = _T("");
  87. m_strEncryptWord = _T("");
  88. m_nEncrypOrderCount = -1;
  89. m_nGasserDataLen = -1;
  90. }
  91. CGuardCommand::~CGuardCommand(void)
  92. {
  93. }
  94. CGuardCommand &CGuardCommand::operator=(CGuardCommand& member)
  95. {
  96. m_byTrigger = member.m_byTrigger;
  97. m_nClientCount = member.m_nClientCount;
  98. m_nOrderCount = member.m_nOrderCount;
  99. m_nHistoryOrderCount = member.m_nHistoryOrderCount;
  100. m_strExecTime = member.m_strExecTime;
  101. m_nExecCommand = member.m_nExecCommand;
  102. m_strOldTableName = member.m_strOldTableName;
  103. m_strNewTableName = member.m_strNewTableName;
  104. m_strOldFieldName = member.m_strOldFieldName;
  105. m_strNewFieldName = member.m_strNewFieldName;
  106. m_strEncryptWord = member.m_strEncryptWord;
  107. m_nEncrypOrderCount = member.m_nEncrypOrderCount;
  108. m_nGasserDataLen = member.m_nGasserDataLen;
  109. return *this;
  110. }
  111. void CGuardCommand::Serialize(CArchive& ar)
  112. {
  113. CObject::Serialize(ar);
  114. if( ar.IsStoring() )
  115. {
  116. ar << m_byTrigger;
  117. ar << m_nClientCount;
  118. ar << m_nOrderCount;
  119. ar << m_nHistoryOrderCount;
  120. ar << m_strExecTime;
  121. ar << m_nExecCommand;
  122. ar << m_strOldTableName;
  123. ar << m_strNewTableName;
  124. ar << m_strOldFieldName;
  125. ar << m_strNewFieldName;
  126. ar << m_strEncryptWord;
  127. ar << m_nEncrypOrderCount;
  128. ar << m_nGasserDataLen;
  129. }
  130. else
  131. {
  132. ar >> m_byTrigger;
  133. ar >> m_nClientCount;
  134. ar >> m_nOrderCount;
  135. ar >> m_nHistoryOrderCount;
  136. ar >> m_strExecTime;
  137. ar >> m_nExecCommand;
  138. ar >> m_strOldTableName;
  139. ar >> m_strNewTableName;
  140. ar >> m_strOldFieldName;
  141. ar >> m_strNewFieldName;
  142. ar >> m_strEncryptWord;
  143. ar >> m_nEncrypOrderCount;
  144. ar >> m_nGasserDataLen;
  145. }
  146. }
  147. //////////////////////////////////////////////////////////////////////////
  148. IMPLEMENT_SERIAL(CGuardCommandEx, CGuardCommand, 0)
  149. HANDLE CGuardCommandEx::m_hGuardEvent = NULL;
  150. HANDLE CGuardCommandEx::m_hGuardThread = NULL;
  151. CGuardCommandEx::CGuardCommandEx(void)
  152. {
  153. m_bTrigger = FALSE;
  154. m_bFinished = FALSE;
  155. m_strLastTriggerTime = _T("");
  156. m_strLastBakupTime = _T("");
  157. m_strLastStartTime = _T("");
  158. }
  159. CGuardCommandEx::~CGuardCommandEx(void)
  160. {
  161. }
  162. CGuardCommandEx &CGuardCommandEx::operator=(CGuardCommandEx& member)
  163. {
  164. *(CGuardCommand*)this = *((CGuardCommand*)&member);
  165. m_bTrigger = member.m_bTrigger;
  166. m_bFinished = member.m_bFinished;
  167. m_strLastTriggerTime = member.m_strLastTriggerTime;
  168. m_strLastBakupTime = member.m_strLastBakupTime;
  169. m_strLastStartTime = member.m_strLastStartTime;
  170. return *this;
  171. }
  172. void CGuardCommandEx::Serialize(CArchive& ar)
  173. {
  174. CGuardCommand::Serialize(ar);
  175. if( ar.IsStoring() )
  176. {
  177. ar << m_bTrigger;
  178. ar << m_bFinished;
  179. ar << m_strLastTriggerTime;
  180. ar << m_strLastBakupTime;
  181. ar << m_strLastStartTime;
  182. }
  183. else
  184. {
  185. ar >> m_bTrigger;
  186. ar >> m_bFinished;
  187. ar >> m_strLastTriggerTime;
  188. ar >> m_strLastBakupTime;
  189. ar >> m_strLastStartTime;
  190. }
  191. }
  192. //////////////////////////////////////////////////////////////////////////
  193. void CGuardCommandEx::ProCmd_ModifyTableName(IN LPCTSTR lpOldTableName, IN LPCTSTR lpNewTableName)
  194. {
  195. CDatabase *pdb = NULL;
  196. ODBCConnGuard ConnGuard(pdb, -1, 3000);
  197. if ( pdb )
  198. {
  199. CString strSQL;
  200. strSQL.Format(_T("execute sp_rename '%s', '%s'"), lpOldTableName, lpNewTableName);
  201. try
  202. {
  203. pdb->ExecuteSQL(strSQL);
  204. }
  205. catch (CDBException* e)
  206. {
  207. e->Delete();
  208. #ifdef _DEBUG
  209. WriteTextLog(_T("%s,%d,%s:%s"), __FILE__, __LINE__, __FUNCTION__, e->m_strError);
  210. #endif // _DEBUG
  211. }
  212. }
  213. }
  214. void CGuardCommandEx::ProCmd_ModifyFieldName(IN LPCTSTR lpTableName, IN LPCTSTR lpOldFieldName, IN LPCTSTR lpNewFieldName)
  215. {
  216. CDatabase *pdb = NULL;
  217. ODBCConnGuard ConnGuard(pdb, -1, 3000);
  218. if ( pdb )
  219. {
  220. CString strSQL;
  221. strSQL.Format(_T("execute sp_rename '[%s].[%s]', '%s'"), lpTableName, lpOldFieldName, lpNewFieldName);
  222. try
  223. {
  224. pdb->ExecuteSQL(strSQL);
  225. }
  226. catch (CDBException* e)
  227. {
  228. e->Delete();
  229. #ifdef _DEBUG
  230. WriteTextLog(_T("%s,%d,%s:%s"), __FILE__, __LINE__, __FUNCTION__, e->m_strError);
  231. #endif // _DEBUG
  232. }
  233. }
  234. }
  235. //////////////////////////////////////////////////////////////////////////
  236. void CGuardCommandEx::Pro_NextTimeTrigger(IN CGuardCommandEx* pGuardCmdEx)
  237. {
  238. switch( pGuardCmdEx->m_nExecCommand )
  239. {
  240. default:
  241. break;
  242. }
  243. }
  244. void CGuardCommandEx::Pro_NextTimeBarkup(IN CGuardCommandEx* pGuardCmdEx)
  245. {
  246. }
  247. void CGuardCommandEx::Pro_NextTimeStart(IN CGuardCommandEx* pGuardCmdEx)
  248. {
  249. }
  250. void CGuardCommandEx::Pro_ClientOpen(IN CGuardCommandEx* pGuardCmdEx)
  251. {
  252. }
  253. void CGuardCommandEx::Pro_OrderCount(IN CGuardCommandEx* pGuardCmdEx)
  254. {
  255. }
  256. void CGuardCommandEx::Pro_HistoryOrderCount(IN CGuardCommandEx* pGuardCmdEx)
  257. {
  258. }
  259. void CGuardCommandEx::Pro_SpecifyTime(IN CGuardCommandEx* pGuardCmdEx)
  260. {
  261. }
  262. BOOL CGuardCommandEx::StartGuardThread()
  263. {
  264. if ( m_hGuardEvent == NULL )
  265. m_hGuardEvent = CreateEvent(NULL, TRUE, FALSE, NULL);
  266. if ( m_hGuardEvent == NULL )
  267. return FALSE;
  268. if ( m_hGuardThread == NULL )
  269. m_hGuardThread = CreateThread(NULL, 0, GuardThread, this, 0, NULL);
  270. if ( m_hGuardThread == NULL )
  271. return FALSE;
  272. }
  273. void CGuardCommandEx::EndOfGuardThread()
  274. {
  275. if ( m_hGuardEvent )
  276. SetEvent(m_hGuardEvent);
  277. if ( m_hGuardThread )
  278. {
  279. WaitForSingleObject(m_hGuardThread, INFINITE);
  280. CloseHandle(m_hGuardThread);
  281. m_hGuardThread = NULL;
  282. }
  283. if ( m_hGuardEvent )
  284. {
  285. CloseHandle(m_hGuardEvent);
  286. m_hGuardEvent = NULL;
  287. }
  288. }
  289. DWORD CGuardCommandEx::GuardThread(LPVOID lpParamter)
  290. {
  291. CGuardCommandEx *pthis = (CGuardCommandEx*)lpParamter;
  292. do
  293. {
  294. if ( pthis->m_bFinished == FALSE)
  295. {
  296. // 触发类型;
  297. switch( pthis->m_byTrigger )
  298. {
  299. // 下次触发时触发;
  300. case GTE_NextTimeTrigger:
  301. pthis->Pro_NextTimeTrigger(pthis);
  302. break;
  303. // 下次备份时触发;
  304. case GTE_NextTimeBarkup:
  305. pthis->Pro_NextTimeBarkup(pthis);
  306. break;
  307. // 下次启动时触发;
  308. case GTE_NextTimeStart:
  309. pthis->Pro_NextTimeStart(pthis);
  310. break;
  311. // 客户端启动第n个时;
  312. case GTE_ClientOpen:
  313. pthis->Pro_ClientOpen(pthis);
  314. break;
  315. // 订单超过n个时;
  316. case GTE_OrderCount:
  317. pthis->Pro_OrderCount(pthis);
  318. break;
  319. // 历史订单超过n个时;
  320. case GTE_HistoryOrderCount:
  321. pthis->Pro_HistoryOrderCount(pthis);
  322. break;
  323. // 指定具体日期时间;
  324. case GTE_SpecifyTime:
  325. pthis->Pro_SpecifyTime(pthis);
  326. break;
  327. default:
  328. break;
  329. }
  330. pthis->m_bFinished = TRUE;
  331. }
  332. } while ( WaitForSingleObject(m_hGuardEvent, 5000) == WAIT_TIMEOUT );
  333. return 0;
  334. }