SPAServer.cpp 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690
  1. #include "SPAServer.h"
  2. #include "spdlog/spdlog.h"
  3. #include <filesystem>
  4. #include "ThreadPool/ThreadPool.h"
  5. #include "GlobalInfo/GlobalVariable.h"
  6. #include "GlobalInfo/GlobalConfig.h"
  7. #include "UniversalFunc.h"
  8. #include "FuncOrdinary.h"
  9. #include "FuncIllegalInvasion.h"
  10. #include "FuncOnAndOffWork.h"
  11. #include "FuncRegionalPerson.h"
  12. #include "GlobalFuncThread.h"
  13. #include <QCoreApplication>
  14. #include <QTcpSocket>
  15. SPAServer::SPAServer()
  16. {
  17. m_logger = spdlog::get("SPAServer");
  18. if(m_logger == nullptr)
  19. {
  20. SPDLOG_ERROR("APAServer logger is nullptr");
  21. return;
  22. }
  23. m_threadRunning = true;
  24. /* 初始化WebAPI */
  25. // m_fromWebAPI.initWebApi("http://192.1.3.133:31000/v6/", "", "4c2f9fc91c22dd98331e47af2e2964f4");
  26. /* 模拟违禁品算法ID,后续需要动态调整 */
  27. // GVariable.ActContraband = "OD210_026_005246_001-IZRTKyEx";
  28. }
  29. SPAServer::~SPAServer()
  30. {
  31. }
  32. /* 启动服务 */
  33. void SPAServer::startServer()
  34. {
  35. /* 获取TCP监听的端口,创建TCP连接 */
  36. short Port = GConfig.listenPort();
  37. /* 连接信号和槽 */
  38. m_tcpServer.listen(QHostAddress::Any, Port);
  39. connect(&m_tcpServer, SIGNAL(newConnection()), this, SLOT(do_newConnection()));
  40. /* 设置全局运行标志位 */
  41. GThreadInfo.setRunning(true);
  42. /* 添加获取基础信息的线程 */
  43. CPPTP.add_task(&SPAServer::threadFromSuperBrain, this);
  44. // threadFromSuperBrain();
  45. /* 延时启动任务分配线程 */
  46. std::this_thread::sleep_for(std::chrono::seconds(2));
  47. /* 添加分配任务的线程 */
  48. CPPTP.add_task(&SPAServer::threadDistribution, this);
  49. // threadDistribution();
  50. /* 测试Redis读取并解析数据线程 */
  51. }
  52. /* 接收到新的TCP连接 */
  53. void SPAServer::do_newConnection()
  54. {
  55. // SPDLOG_LOGGER_INFO(m_logger, "接收到一个新的TCP连接");
  56. QTcpSocket* pTcpSocket = m_tcpServer.nextPendingConnection();
  57. if(pTcpSocket == nullptr)
  58. {
  59. SPDLOG_LOGGER_ERROR(m_logger, "TCP Socket is nullptr");
  60. return;
  61. }
  62. // SPDLOG_LOGGER_INFO(m_logger, "接收到一个TCP连接,IP:{}, Port:{}", pTcpSocket->peerAddress().toString().toStdString(), pTcpSocket->peerPort());
  63. // connect(pTcpSocket, SIGNAL(readyRead()), this, SLOT(do_recvTCPData()));
  64. /* 删掉连接 */
  65. pTcpSocket->close();
  66. pTcpSocket->deleteLater();
  67. }
  68. /**
  69. * @brief 从基础平台获取算法信息和设备信息的线程函数
  70. * 从基础平台获取算法信息和设备信息,然后更新到EQM数据库所对应的表中
  71. *
  72. */
  73. void SPAServer::threadFromSuperBrain()
  74. {
  75. SPDLOG_LOGGER_INFO(m_logger, "开启 fromSuperBrainThread 线程, 获取算法和设备信息");
  76. /* 创建变量 */
  77. std::vector<AlgorithmInfo> vecAlgNewInfo;
  78. std::vector<DeviceInfo> vecDevNewInfo;
  79. /* 初始化WebAPI */
  80. if(!m_fromWebAPIUseSB.initWebApi(GConfig.webapiUrl(), GConfig.webapiKey(), GConfig.webapiAppType()))
  81. {
  82. SPDLOG_LOGGER_ERROR(m_logger, "threadFromSuperBrain线程初始化WebAPI失败");
  83. return;
  84. }
  85. /* 初始化SuperBrain */
  86. m_fromSuperBrain.initSuperBrain(GConfig.superBrainUrl(), GConfig.superBrainAppKey(), GConfig.superBrainAppSecret());
  87. /* 获取一次Token,这玩意好像不是必须的 */
  88. if(!m_fromSuperBrain.getToken())
  89. {
  90. SPDLOG_LOGGER_ERROR(m_logger, "FromSuperBrain线程获取Token失败");
  91. return;
  92. }
  93. while (m_threadRunning)
  94. {
  95. SPDLOG_LOGGER_INFO(m_logger, "----------------- 刷新算法和设备信息 -----------------");
  96. /* 先更新数据库的信息,防止从其他地方更改了数据库,这里没有刷新本地缓存 */
  97. m_fromWebAPIUseSB.getAlgorithmInfo(m_vecEqmAlgInfo);
  98. m_fromWebAPIUseSB.getDeviceInfo(m_vecEqmDevInfo);
  99. m_fromWebAPIUseSB.getDeviceAlgorithmInfo(m_vecEqmDevInfo, m_listDevIDDelete);
  100. /* 从超脑获取基础信息 */
  101. // m_fromSuperBrain.getTaskTypeList(vecAlgNewInfo);
  102. m_fromSuperBrain.getDeviceList(vecDevNewInfo);
  103. // SPDLOG_LOGGER_INFO(m_logger, "获取到的设备信息数量: {}", vecDevNewInfo.size());
  104. /* 将设备返回的算法ID替换成带有后缀的全ID(现在不需要后缀了) */
  105. // checkAlgorithmID(vecDevNewInfo);
  106. /* 处理算法信息,算法信息不需要在服务里处理了,在外部手动处理 */
  107. // bool algIsUpdate = processAlgorithmInfo(vecAlgNewInfo);
  108. /* 处理设备信息 */
  109. bool devIsUpdate = processDeviceInfo(vecDevNewInfo);
  110. if(!devIsUpdate)
  111. {
  112. SPDLOG_LOGGER_INFO(m_logger, "设备信息和算法信息没有更新");
  113. }
  114. vecAlgNewInfo.clear();
  115. vecDevNewInfo.clear();
  116. /* 更新算法详细信息 */
  117. m_mutexActionInfo.lock();
  118. m_fromWebAPIUseSB.getActionInfo(m_listActionInfo);
  119. m_mutexActionInfo.unlock();
  120. /* 20秒更新一次 */
  121. std::this_thread::sleep_for(std::chrono::seconds(GVariable.CheckDeviceTime));
  122. }
  123. SPDLOG_LOGGER_INFO(m_logger, "退出 fromSuperBrainThread 线程");
  124. }
  125. /* 替换算法ID为全ID */
  126. // void SPAServer::checkAlgorithmID(std::vector<DeviceInfo>& vecDevInfo)
  127. // {
  128. // /* 替换算法ID为全ID */
  129. // for(auto& dev : vecDevInfo)
  130. // {
  131. // for(auto act = dev.vecAlgorithmInfo.begin(); act != dev.vecAlgorithmInfo.end(); act++)
  132. // {
  133. // /* 替换算法ID为全ID */
  134. // std::string strFullID = GVariable.getFullActionID(act->ActionID);
  135. // if(strFullID.empty())
  136. // {
  137. // SPDLOG_LOGGER_INFO(m_logger, "未知的算法ID: {}", act->ActionID);
  138. // /* 删除算法ID */
  139. // // act = dev.vecAlgorithmInfo.erase(act);
  140. // }else {
  141. // /* 替换算法ID */
  142. // act->ActionID = strFullID;
  143. // }
  144. // }
  145. // }
  146. // }
  147. /* 处理算法信息,返回值为true,说明有改变,需要重新读取 */
  148. bool SPAServer::processAlgorithmInfo(std::vector<AlgorithmInfo> vecNewAlgInfo)
  149. {
  150. std::vector<AlgorithmInfo> vecAlgUpdate;
  151. std::vector<AlgorithmInfo> vecAlgDelete;
  152. /* 对比数据库表格信息,这里只有插入和删除,没有更新 */
  153. compareAlgorithmInfo(vecNewAlgInfo, vecAlgUpdate, vecAlgDelete);
  154. /* 更新数据库,先删除,再写入刷新 */
  155. bool isUpdate = false;
  156. if(vecAlgDelete.size() > 0)
  157. {
  158. SPDLOG_LOGGER_DEBUG(m_logger, "删除算法信息");
  159. m_fromWebAPIUseSB.deleteAlgorithmInfo(vecAlgDelete);
  160. isUpdate = true;
  161. }
  162. if(vecAlgUpdate.size() > 0)
  163. {
  164. SPDLOG_LOGGER_DEBUG(m_logger, "写入算法信息");
  165. m_fromWebAPIUseSB.writeAlgorithmInfo(vecAlgUpdate);
  166. isUpdate = true;
  167. }
  168. return isUpdate;
  169. }
  170. /**
  171. * @brief 处理设备信息
  172. *
  173. * @param vecNewDevInfo 传入新获取到的值
  174. * @return true 需要重新读取数据库,获取新的数据
  175. * @return false 无需读取数据库
  176. */
  177. bool SPAServer::processDeviceInfo(std::vector<DeviceInfo>& vecNewDevInfo)
  178. {
  179. std::vector<DeviceInfo> vecDevInsert;
  180. std::vector<DeviceInfo> vecDevUpdate;
  181. std::vector<DeviceInfo> vecDevDelete;
  182. /*-------------------------------------------------------------------------
  183. ****** 这里只对比设备信息,不对比设备的算法信息,算法信息在下面单独对比 *******
  184. *------------------------------------------------------------------------*/
  185. /* 先自身查重 */
  186. // for(auto it = m_vecEqmDevInfo.begin(); it != m_vecEqmDevInfo.end(); it++)
  187. // {
  188. // for(auto it0 = it + 1; it0 != m_vecEqmDevInfo.end();)
  189. // {
  190. // if(it->DeviceID == it0->DeviceID)
  191. // {
  192. // SPDLOG_LOGGER_DEBUG(m_logger, "设备ID: {} 重复", it->DeviceID);
  193. // it0 = m_vecEqmDevInfo.erase(it0);
  194. // vecDevDelete.push_back(*it0);
  195. // }else {
  196. // ++it0;
  197. // }
  198. // }
  199. // }
  200. /* 如果本地缓存没有数据,那么就全部插入 */
  201. if(m_vecEqmDevInfo.size() > 0)
  202. {
  203. for(auto& DevInfo : vecNewDevInfo)
  204. {
  205. bool isExist = false;
  206. for(auto& it0 : m_vecEqmDevInfo)
  207. {
  208. if(DevInfo.DeviceID == it0.DeviceID)
  209. {
  210. isExist = true;
  211. /* 对比其他项是否相等,不相等就更新 */
  212. if(DevInfo == it0)
  213. {
  214. continue;
  215. }else {
  216. vecDevUpdate.push_back(DevInfo);
  217. }
  218. break;
  219. }
  220. }
  221. if(!isExist)
  222. {
  223. vecDevInsert.push_back(DevInfo);
  224. }
  225. }
  226. }else {
  227. vecDevInsert = vecNewDevInfo;
  228. }
  229. /* 获取删除列表 */
  230. if(vecNewDevInfo.size() > 0)
  231. {
  232. bool isExist = false;
  233. for(const auto& it : m_vecEqmDevInfo)
  234. {
  235. isExist = false;
  236. for(const auto& it0 : vecNewDevInfo)
  237. {
  238. if(it.DeviceID == it0.DeviceID)
  239. {
  240. isExist = true;
  241. break;
  242. }
  243. }
  244. if(!isExist)
  245. {
  246. vecDevDelete.push_back(it);
  247. }
  248. }
  249. }else {
  250. vecDevDelete = m_vecEqmDevInfo;
  251. }
  252. bool isUpdate = false;
  253. /* 先删除多余的数据 */
  254. if(vecDevDelete.size() > 0)
  255. {
  256. if(m_fromWebAPIUseSB.deleteDeviceInfo(vecDevDelete))
  257. {
  258. SPDLOG_LOGGER_DEBUG(m_logger,"删除设备信息到 tCamerInfo, 数量:{}",vecDevDelete.size());
  259. isUpdate = true;
  260. }
  261. }
  262. /* 更新数据 */
  263. if(vecDevUpdate.size() > 0)
  264. {
  265. if(m_fromWebAPIUseSB.updateDeviceInfo(vecDevUpdate))
  266. {
  267. SPDLOG_LOGGER_DEBUG(m_logger, "更新设备信息到 tCamerInfo, 数量:{}", vecDevUpdate.size());
  268. isUpdate = true;
  269. }
  270. }
  271. /* 插入数据 */
  272. if(vecDevInsert.size() > 0)
  273. {
  274. if(m_fromWebAPIUseSB.insertDeviceInfo(vecDevInsert))
  275. {
  276. SPDLOG_LOGGER_DEBUG(m_logger, "插入设备信息到 tCamerInfo, 数量:{}", vecDevInsert.size());
  277. isUpdate = true;
  278. }
  279. }
  280. // for(auto& it : vecNewDevInfo)
  281. // {
  282. // for(auto& it0 : it.vecAlgorithmInfo)
  283. // {
  284. // SPDLOG_LOGGER_DEBUG(m_logger, "设备ID: {}, 算法ID: {}", it.DeviceID, it0.ActionID);
  285. // }
  286. // }
  287. /*-------------------------------------------------------------------------
  288. ************* 处理设备和算子关联的表格,单独对比设备的算法信息 *************
  289. *------------------------------------------------------------------------*/
  290. /* 插入新的设备信息 */
  291. // if(vecDevInsert.size() > 0)
  292. // {
  293. // if(m_fromWebAPIUseSB.insertDeviceAlgorithmInfo(vecDevInsert))
  294. // {
  295. // SPDLOG_LOGGER_DEBUG(m_logger, "插入设备和算法关联表(tActionCamer), 数量: {}", vecDevInsert.size());
  296. // isUpdate = true;
  297. // }
  298. // }
  299. vecDevUpdate.clear();
  300. /* 对比现有的设备是否需要更新算法 */
  301. compareDeviceAlgorithmInfo(vecNewDevInfo, vecDevUpdate);
  302. if(vecDevUpdate.size() > 0)
  303. {
  304. if(m_fromWebAPIUseSB.updateDeviceAlgorithmInfo(vecDevUpdate))
  305. {
  306. SPDLOG_LOGGER_DEBUG(m_logger, "更新设备和算法关联表(tActionCamer), 更新设备数目:{}", vecDevUpdate.size());
  307. isUpdate = true;
  308. }
  309. }
  310. /* 删除tActionCamer表中消失的设备信息 */
  311. if(m_listDevIDDelete.size() > 0)
  312. {
  313. if(m_fromWebAPIUseSB.deleteDeviceAlgorithmInfo(m_listDevIDDelete))
  314. {
  315. SPDLOG_LOGGER_DEBUG(m_logger, "删除消失的设备关联的算法(tActionCamer), 数目: {}", m_listDevIDDelete.size());
  316. isUpdate = true;
  317. }
  318. }
  319. return isUpdate;
  320. }
  321. /* 对比现有的数据和新获取到的数据,取出要删除和添加的数据 */
  322. void SPAServer::compareAlgorithmInfo(const std::vector<AlgorithmInfo>& vecNewInfo, std::vector<AlgorithmInfo>& vecAlgUpdate, std::vector<AlgorithmInfo>& vecAlgDelete)
  323. {
  324. /* 取出要添加的,如果本地缓存是0,那么全部都要添加 */
  325. if(m_vecEqmAlgInfo.size() > 0)
  326. {
  327. for(const auto& it : vecNewInfo)
  328. {
  329. bool isExist = false;
  330. for(const auto& it0 : m_vecEqmAlgInfo)
  331. {
  332. /* 如果存在就退出循环 */
  333. if(it.ActionID == it0.ActionID)
  334. {
  335. isExist = true;
  336. break;
  337. }
  338. }
  339. if(!isExist)
  340. {
  341. vecAlgUpdate.push_back(it);
  342. }
  343. }
  344. }else {
  345. vecAlgUpdate = vecNewInfo;
  346. }
  347. /* 取出要删除的,如果新的数据是0,那么全部都要删除 */
  348. if(vecNewInfo.size() > 0)
  349. {
  350. bool isExist = false;
  351. for(const auto& it : m_vecEqmAlgInfo)
  352. {
  353. isExist = false;
  354. for(const auto& it0 : vecNewInfo)
  355. {
  356. if(it.ActionID == it0.ActionID)
  357. {
  358. isExist = true;
  359. break;
  360. }
  361. }
  362. if(!isExist)
  363. {
  364. vecAlgDelete.push_back(it);
  365. }
  366. }
  367. }else {
  368. vecAlgDelete = m_vecEqmAlgInfo;
  369. }
  370. }
  371. /**
  372. * @brief 对比设备和算法关联表是否需要更新
  373. * 对比规则:
  374. * 1、这里只对比已有的设备ID,需要删除的ID在获取到tActionCamer表是就已经取出来了
  375. * 2、如果设备ID相等,那么进一步对比算法信息是否相等
  376. * 3、如果设备ID相等,但是算法信息数目不相等,那么直接加入更新列表
  377. * 4、如果设备ID相等,算法信息数目相等,进一步对比算法信息
  378. *
  379. * @param vecNewInfo
  380. * @param vecDevUpdate
  381. */
  382. void SPAServer::compareDeviceAlgorithmInfo(const std::vector<DeviceInfo>& vecNewInfo, std::vector<DeviceInfo>& vecDevUpdate)
  383. {
  384. vecDevUpdate.clear();
  385. for(const auto& it0 : vecNewInfo)
  386. {
  387. for(const auto& it1 : m_vecEqmDevInfo)
  388. {
  389. if(it0.DeviceID == it1.DeviceID)
  390. {
  391. /* 设备的算法信息数目不相等,直接加入更新列表 */
  392. if(it0.vecAlgorithmInfo.size() != it1.vecAlgorithmInfo.size())
  393. {
  394. vecDevUpdate.push_back(it0);
  395. break;
  396. }
  397. /* 设备的算法信息数目相等,进一步对比算法信息 */
  398. bool isEquality = true;
  399. for(const auto& it2 : it0.vecAlgorithmInfo)
  400. {
  401. bool isEq2 = false;
  402. for(const auto& it3 : it1.vecAlgorithmInfo)
  403. {
  404. /* 这里只对比算法ID */
  405. if(it2.ActionID != it3.ActionID)
  406. {
  407. continue;
  408. }else {
  409. isEq2 = true;
  410. break;
  411. }
  412. }
  413. if(!isEq2)
  414. {
  415. isEquality = false;
  416. break;
  417. }
  418. }
  419. if(!isEquality)
  420. {
  421. vecDevUpdate.push_back(it0);
  422. break;
  423. }
  424. }
  425. }
  426. }
  427. }
  428. /**
  429. * @brief 分派任务的线程
  430. 1、 线程定时刷新房间和摄像机的关联关系,以及和算法Action的关联关系
  431. 2、 将算法信息加入到不同的列表中
  432. 需要多个摄像机配合的加入到 m_runListRoomActionInfo 列表
  433. 不需要多个摄像机配合的加入到 m_runListActionInfo 列表
  434. 3、 每次刷新都会清空ActionInfo或者RoomActionInfo的摄像机列表,但是不会动其他基本信息,如果是列表中没有对应的信息,
  435. 就会创建新的ActionInfo,那么RunState的状态是INIT,需要开启新的线程
  436. 如果刷新完成后,算法对应的摄像机列表为空,那么对应的线程就会停止运行,将RunState设置为STOP,本线程的下一轮
  437. 循环就会删除这个ActionInfo
  438. 4、在这里定期删除一年前的报警数据
  439. */
  440. void SPAServer::threadDistribution()
  441. {
  442. SPDLOG_LOGGER_INFO(m_logger, "**************** 开启分派任务线程 ****************");
  443. /* 房间相机关联信息 */
  444. std::list<RoomCameraInfo> listRC;
  445. /* 存储获取到的应用和启用时间的信息 */
  446. std::list<ChannelAppInfo> listAppAndTime;
  447. /* 初始化WebAPI */
  448. std::shared_ptr<FromWebAPI> pFromWebAPI = std::make_shared<FromWebAPI>();
  449. pFromWebAPI->initWebApi(GConfig.webapiUrl(), GConfig.webapiKey(), GConfig.webapiAppType());
  450. while (m_threadRunning)
  451. {
  452. /* ======================================================= */
  453. /* 更新通道名称和摄像机名称信息 */
  454. std::map<int, std::string> mapChannelName;
  455. pFromWebAPI->getChannelInfo(mapChannelName);
  456. GConfig.setChannelInfo(mapChannelName);
  457. std::map<int, std::string> mapCameraName;
  458. pFromWebAPI->getCameraInfo(mapCameraName);
  459. GConfig.setCameraInfo(mapCameraName);
  460. /* ======================================================= */
  461. GThreadInfo.lockRunFAI();
  462. /* 先清理已经退出的线程所用到的Action或者RoomAction */
  463. GThreadInfo.clearNoneFuncThreadInfo();
  464. /* 清空已经停止运行的功能类实例 */
  465. clearNoneFuncThreadInfo();
  466. /* ======================================================= */
  467. /* 创建应用信息,根据从EQM数据库读取到的配置的应用信息创建 */
  468. pFromWebAPI->getAlarmAppInfo(listAppAndTime);
  469. for(const auto& it : listAppAndTime)
  470. {
  471. /* 创建应用信息,如果已有该应用,就更新时间
  472. * 这里只创建应用信息块,没有对应的房间、算法信息 */
  473. GThreadInfo.addFuncThreadInfo(it);
  474. }
  475. /* 设置应用对应的频率信息 */
  476. // for(auto& func : GThreadInfo.getList())
  477. // {
  478. // /* 设置频率名称 */
  479. // func->strChannelName = GConfig.getChannelName(func->ChannelID);
  480. // }
  481. /* 设置对应的摄像机名称列表 */
  482. GThreadInfo.setCameraInfo(GConfig.getCameraInfoMap());
  483. /* 先获取EQM数据库信息,取出房间和摄像机关联信息,包括所在的频率 */
  484. m_mutexActionInfo.lock();
  485. pFromWebAPI->getActionInfo(m_listActionInfo);
  486. /* 将算法信息加入到不同的功能列表中,先清空功能对应的算法设备列表
  487. * 算法信息带有摄像机和房间信息,在从数据库读取的时候就读取出来了
  488. * 如果是新创建的应用,有房间信息就将状态由NONE转换成INIT */
  489. GThreadInfo.clearActionList();
  490. for(const auto& it : m_listActionInfo.getData())
  491. {
  492. GThreadInfo.addActionInfo(*it);
  493. }
  494. /* 检查算法信息的状态,频率里的应用没有配置摄像机就设置为线程运行状态为停止,退出该线程 */
  495. GThreadInfo.setNoneCameraFuncStop();
  496. m_mutexActionInfo.unlock();
  497. /* 开启线程 */
  498. for(const auto& func : GThreadInfo.getList())
  499. {
  500. /* 只有状态是 RUN_STATE_INIT 的才创建线程实例 */
  501. if(func->RunState == RunTimeState::RUN_STATE_INIT)
  502. {
  503. /* 创建实例 */
  504. FuncBase* pFunc = createFuncInstance(*func);
  505. if(pFunc == nullptr)
  506. {
  507. SPDLOG_LOGGER_ERROR(m_logger, "创建功能实例失败:{}", func->strFunctionName);
  508. continue;
  509. }
  510. CPPTP.add_task(&FuncBase::thread_task, pFunc);
  511. m_listFuncBase.push_back(pFunc);
  512. /* 睡眠50ms,让这个线程初始化完成 */
  513. std::this_thread::sleep_for(std::chrono::milliseconds(50));
  514. }
  515. }
  516. GThreadInfo.unlockRunFAI();
  517. /* 定期删除一年前的报警数据 */
  518. pFromWebAPI->deleteOverOneYearAlarm();
  519. /* 休眠n秒,默认应该是300秒 */
  520. std::this_thread::sleep_for(std::chrono::seconds(GVariable.CheckAppSet));
  521. // std::this_thread::sleep_for(std::chrono::seconds(3));
  522. }
  523. SPDLOG_LOGGER_INFO(m_logger, "分派任务线程退出");
  524. }
  525. /**
  526. * @brief 创建任务实例的函数,根据不同的功能创建出不同的实例
  527. *
  528. * @param info
  529. * @return FuncBase*
  530. */
  531. FuncBase* SPAServer::createFuncInstance(FuncThreadInfo& info)
  532. {
  533. if(info.appFunction == AppFunction::APP_NONE)
  534. {
  535. return nullptr;
  536. }
  537. FuncBase* pFunc = nullptr;
  538. /* 普通的功能,只用一个摄像机一个算法的功能 */
  539. if( info.appFunction == AppFunction::APP_AllDown || info.appFunction == AppFunction::APP_NoMask ||
  540. info.appFunction == AppFunction::APP_PlayPhone || info.appFunction == AppFunction::APP_Mouse ||
  541. info.appFunction == AppFunction::APP_Fatigue || info.appFunction == AppFunction::APP_Contraband )
  542. {
  543. /* 普通检测功能 */
  544. auto tmpFunc = new FuncOrdinary();
  545. tmpFunc->setFuncThreadInfo(info);
  546. pFunc = tmpFunc;
  547. }
  548. else if(info.appFunction == AppFunction::APP_OnWork)
  549. {
  550. /* 人员在岗检测 */
  551. // auto tmpFunc = new FuncOnAndOffWork();
  552. // tmpFunc->setFuncThreadInfo(info);
  553. // pFunc = tmpFunc;
  554. }
  555. else if(info.appFunction == AppFunction::APP_Illegal)
  556. {
  557. /* 非法入侵 */
  558. // auto tmpFunc = new FuncIllegalInvasion();
  559. // tmpFunc->setFuncThreadInfo(info);
  560. // pFunc = tmpFunc;
  561. }
  562. else if(info.appFunction == AppFunction::APP_Regional)
  563. {
  564. /* 区域人员检测 */
  565. auto tmpFunc = new FuncRegionalPersonCount();
  566. tmpFunc->setFuncThreadInfo(info);
  567. pFunc = tmpFunc;
  568. }
  569. return pFunc;
  570. }
  571. /* 清理没有在运行的线程实例 */
  572. void SPAServer::clearNoneFuncThreadInfo()
  573. {
  574. for(auto it = m_listFuncBase.begin(); it != m_listFuncBase.end(); )
  575. {
  576. if(!(*it)->getThreadRunning())
  577. {
  578. /* 需要转换成相应的子类实例才能删除 */
  579. if( (*it)->getApp() == AppFunction::APP_AllDown || (*it)->getApp() == AppFunction::APP_NoMask ||
  580. (*it)->getApp() == AppFunction::APP_PlayPhone || (*it)->getApp() == AppFunction::APP_Mouse ||
  581. (*it)->getApp() == AppFunction::APP_Fatigue || (*it)->getApp() == AppFunction::APP_Contraband )
  582. {
  583. FuncOrdinary* p = dynamic_cast<FuncOrdinary*>(*it);
  584. delete p;
  585. }
  586. else if((*it)->getApp() == AppFunction::APP_OnWork)
  587. {
  588. /* 人员在岗检测 */
  589. FuncOnAndOffWork* p = dynamic_cast<FuncOnAndOffWork*>(*it);
  590. delete p;
  591. }
  592. else if((*it)->getApp() == AppFunction::APP_Illegal)
  593. {
  594. /* 非法入侵 */
  595. FuncIllegalInvasion* p = dynamic_cast<FuncIllegalInvasion*>(*it);
  596. delete p;
  597. }
  598. else if((*it)->getApp() == AppFunction::APP_Regional)
  599. {
  600. /* 区域人员检测 */
  601. FuncRegionalPersonCount* p = dynamic_cast<FuncRegionalPersonCount*>(*it);
  602. delete p;
  603. }
  604. it = m_listFuncBase.erase(it);
  605. }else {
  606. ++it;
  607. }
  608. }
  609. }