mine_business.cpp 10 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370
  1. #include <cmath>
  2. #include <sstream>
  3. #include <vector>
  4. #include <list>
  5. #include <array>
  6. #include <map>
  7. #include "mine_business.h"
  8. #include "log.h"
  9. #include "point.h"
  10. #include "db/db_api/CDBConnPool.h"
  11. #include "db/db_tool.h"
  12. #include "common_tool.h"
  13. #include "ant.h"
  14. struct card_sensor
  15. {
  16. card_sensor(int32_t wid,const point &p,const std::string &cid)
  17. :card_id(cid)
  18. ,sid(0)
  19. ,id(wid)
  20. ,bigger(-1),count(0),sum(0),ct(-1),stime(0),flag(false)
  21. {
  22. base_point=p;
  23. }
  24. std::string card_id;
  25. int32_t sid;
  26. int32_t id;
  27. std::vector<int> sensorId_list;
  28. point base_point;
  29. std::array<point,2> ant;
  30. int8_t bigger;
  31. uint32_t count,sum;
  32. uint16_t ct;
  33. uint64_t stime;
  34. bool flag;
  35. std::list<double> his_data;
  36. std::array<uint64_t,2> dist;
  37. void clear()
  38. {
  39. count=sum=0;stime=0;flag=false;
  40. }
  41. bool timeout()
  42. {
  43. time_t t=time(NULL);
  44. return flag && t-stime>10*60;
  45. }
  46. bool check_location_reversal(std::string & str)
  47. {
  48. time_t t=time(NULL);
  49. time_t tval= t-stime;
  50. bool f=false;
  51. if (tval !=0 && sum*1.0/tval > 0.8 && count*1.0/sum <0.2)
  52. f=true;
  53. char buf[1024]={0};
  54. sprintf(buf,"[reverse_alarm_checking]%s,count:%d,sum:%d,val:%ld,s:%.2f,v:%.2f,alarm:%d",
  55. card_id.c_str(),count,sum,tval,(sum*1.0/tval),(count*1.0/sum),f);
  56. str=buf;
  57. return f;
  58. }
  59. void make_reverse_condition(int16_t antid,uint16_t ct,double d)
  60. {
  61. dist[antid]=d;
  62. if(ct == ct){
  63. sum++;
  64. int b=-1;
  65. if (dist[0]>dist[1]) b=0 ;else b =1;
  66. if(b==bigger) count++; //合格的点
  67. }
  68. ct = ct;
  69. }
  70. void set(const point &p0,const point &p1,int id)
  71. {
  72. int8_t b=0;
  73. ant[0].set(p0);ant[1].set(p1);
  74. double d0=base_point.dist(p0);double d1=base_point.dist(p1);
  75. if(d0>d1)b=1;else b=0;
  76. if (bigger!=-1 && b!=bigger){
  77. count=sum=0;stime=time(0);
  78. }
  79. bigger=b;sid=id;
  80. }
  81. bool push(double d){
  82. bool f=true;
  83. his_data.push_back(d);
  84. if (his_data.size()==3){
  85. double da=0;
  86. for (const auto &d:his_data){
  87. if(da>d || d-da<0.004){f=false;break;}
  88. da=d;
  89. }
  90. his_data.pop_front();
  91. }
  92. else f=false;
  93. return f;
  94. }
  95. point &operator[](int i){return ant[i];}
  96. const point &operator[](int i)const {return ant[i];}
  97. void set_true(){flag=true;stime=time(NULL);}
  98. bool warning(){return flag;}
  99. bool empty(){return his_data.empty();}
  100. double back(){return his_data.back();}
  101. void clear_sensor(){clear();}
  102. int reader_id(){return sid;}
  103. void push_back(int32_t sensor_id){sensorId_list.push_back(sensor_id);}
  104. };
  105. struct card_sensor_mgr
  106. {
  107. void push(int32_t wid,int32_t sid,const point &p,const std::string & cardid)
  108. {
  109. std::shared_ptr<card_sensor> cs=nullptr;
  110. auto it=m_mgr.find(wid);
  111. if(it!=m_mgr.end())
  112. cs=it->second;
  113. else
  114. {
  115. cs=std::make_shared<card_sensor>(wid,p,cardid);
  116. m_mgr.insert({wid,cs});
  117. }
  118. cs->push_back(sid);
  119. }
  120. void for_each();
  121. void make_condition(int32_t antid,uint32_t ct,uint64_t tof,uint32_t sid);
  122. std::map<int32_t,std::shared_ptr<card_sensor>> m_mgr;
  123. };
  124. struct staffer_num_business
  125. {
  126. void record_staffer_num();
  127. void fetch_add(){m_staff_num++;}
  128. private:
  129. void reset(){m_staff_num=0;}
  130. std::atomic<uint32_t> m_staff_num;
  131. std::time_t m_record_staffer_timeval;
  132. };
  133. struct reverse_alarm_business
  134. {
  135. void load_sensor();
  136. void run_sensor();
  137. void make_condition(uint64_t type,uint32_t id,int32_t antid,uint32_t ct,uint64_t tof,uint32_t sid);
  138. private:
  139. std::map<uint64_t,std::shared_ptr<card_sensor_mgr>> m_sensor_mgr;
  140. };
  141. mine_business::mine_business()
  142. {
  143. m_staffer_num_ptr.reset(new staffer_num_business);
  144. m_reverse_alarm_ptr.reset(new reverse_alarm_business);
  145. }
  146. mine_business* mine_business::inst()
  147. {
  148. static mine_business ct;
  149. return &ct;
  150. }
  151. void mine_business::run_business()
  152. {
  153. mine_business::inst()->record_staffer_num();
  154. mine_business::inst()->handle_reverse_alarm();
  155. }
  156. void mine_business::fetch_add()
  157. {
  158. m_staffer_num_ptr->fetch_add();
  159. }
  160. void mine_business::record_staffer_num()
  161. {
  162. m_staffer_num_ptr->record_staffer_num();
  163. }
  164. void mine_business::load()
  165. {
  166. m_reverse_alarm_ptr->load_sensor();
  167. }
  168. void mine_business::handle_reverse_alarm()
  169. {
  170. m_reverse_alarm_ptr->run_sensor();
  171. }
  172. void mine_business::make_reverse_condition(uint64_t type,uint32_t id,int32_t antid,uint32_t ct,uint64_t tof,uint32_t sid)
  173. {
  174. m_reverse_alarm_ptr->make_condition(type,id,antid,ct,tof,sid);
  175. }
  176. ///////staffer_num_business
  177. /**********************************
  178. //每两分钟把人员数量入库。
  179. //不往前端推送的人员,不算在数量范围内
  180. *********************************/
  181. void staffer_num_business::record_staffer_num()
  182. {
  183. static uint32_t min_num=0,max_num=0;
  184. std::time_t t = time(NULL);
  185. char ti[64] = { 0 };
  186. strftime(ti,sizeof(ti),"%Y/%m/%d %H:%M:%S",localtime(&t));
  187. std::string sti(ti);
  188. int minute = atoi(sti.substr(sti.find_first_of(':')+1,2).c_str());
  189. if(m_record_staffer_timeval==0)
  190. {
  191. min_num=max_num=m_staff_num;
  192. m_record_staffer_timeval = t;
  193. }
  194. if(t-m_record_staffer_timeval>=120 && minute%2 == 0)
  195. {
  196. std::stringstream ss;
  197. ss<< "INSERT INTO his_staff_number(max_num,min_num,ave_num) VALUES("<<max_num<<','<<min_num<<','<<std::lround((max_num+min_num)/2)<<");";
  198. logn_info(2,"staff_number:%s",ss.str().c_str());
  199. db_tool::PushAsync(ss.str().c_str());
  200. min_num=max_num=m_staff_num;
  201. m_record_staffer_timeval = t;
  202. }
  203. else
  204. {
  205. if (min_num>m_staff_num) min_num = m_staff_num;
  206. if (max_num < m_staff_num) max_num = m_staff_num;
  207. }
  208. reset();
  209. return;
  210. }
  211. //前端修改数据库表,dat_sensor dat_drivingface,dat_drivingface_vehicle的时候,需要加载.
  212. void reverse_alarm_business::load_sensor()
  213. {
  214. std::string sql = "SELECT sensor_id,data_source,work_face_id,v.vehicle_id,ve.card_id,d.base_point_x,d.base_point_y FROM dat_sensor ,dat_drivingface_vehicle v,dat_drivingface d,dat_vehicle_extend ve WHERE sensor_type_id = 1 AND work_face_id = v.drivingface_id AND work_face_id = d.drivingface_id AND ve.vehicle_id = v.vehicle_id;";
  215. std::string Error;
  216. YADB::CDBResultSet DBRes;
  217. sDBConnPool.Query(sql.c_str(),DBRes,Error);
  218. int nCount = DBRes.GetRecordCount( Error );
  219. if (nCount < 1)
  220. log_error("init_sensor..failed[%s]", sql.c_str());
  221. while ( DBRes.GetNextRecod(Error) )
  222. {
  223. int32_t sensor_id = 0;
  224. DBRes.GetField( "sensor_id",sensor_id, Error );
  225. int32_t data = 0;
  226. DBRes.GetField( "data_source",data, Error );
  227. int32_t workfaceid = 0;
  228. DBRes.GetField( "work_face_id",workfaceid, Error );
  229. std::string card_id;
  230. DBRes.GetField( "card_id",card_id, Error );
  231. double bx,by;
  232. DBRes.GetField( "base_point_x",bx, Error );
  233. DBRes.GetField( "base_point_y",by, Error );
  234. log_info("init sensor :workfaceid:%d,sensor_id:%d,data:%d,card:%s,(%.2f,%.2f)",workfaceid,sensor_id,data,card_id.c_str(),bx,by);
  235. if(data <= 0) continue;
  236. uint64_t cid = tool_other::card_id_to_u64(card_id);
  237. std::shared_ptr<card_sensor_mgr> tm=nullptr;
  238. auto iter = m_sensor_mgr.find(cid);
  239. if(iter != m_sensor_mgr.end())
  240. tm=iter->second;
  241. else
  242. {
  243. tm = std::make_shared<card_sensor_mgr>();
  244. m_sensor_mgr.insert({cid,tm});
  245. }
  246. point p(bx,by);
  247. tm->push(workfaceid,sensor_id,p,card_id);
  248. }
  249. }
  250. void reverse_alarm_business::run_sensor()
  251. {
  252. static time_t time_interval=0;
  253. time_t t = time(NULL);
  254. if (t-time_interval>3*60)
  255. {
  256. time_interval =t;
  257. printf("reverse_alarm......\n");
  258. for(const auto &x:m_sensor_mgr)
  259. x.second->for_each();
  260. }
  261. }
  262. void card_sensor_mgr::for_each()
  263. {
  264. for(const auto & cst:m_mgr)
  265. {
  266. auto cs = cst.second;
  267. if(cs->sensorId_list.empty())
  268. continue;
  269. char strsql[512]={0};
  270. snprintf(strsql,512, "SELECT data_value FROM his_sensor_data WHERE sensor_id = %d ORDER BY write_time DESC LIMIT 3;",cs->sensorId_list[0]);
  271. std::string Error;
  272. YADB::CDBResultSet DBRes;
  273. sDBConnPool.Query(strsql,DBRes,Error);
  274. int nCount = DBRes.GetRecordCount( Error );
  275. if (nCount < 1)
  276. {
  277. log_error("load_sensor_data..failed[%s]", strsql);
  278. continue;
  279. }
  280. double prenum=-1;
  281. double nrow=0;double sum=0;bool f=false;
  282. while ( DBRes.GetNextRecod(Error) )
  283. {
  284. double num = 0;
  285. DBRes.GetField( "data_value",num, Error );
  286. if (point::eq(num,0,1e-10))continue; //0异常数据抛掉
  287. if(point::eq(prenum,-1,1e-10)){ //如果没有参考数据,则判断,如果有历史数据,用历史,没有则用第一个数据。
  288. if(cs->empty()) prenum=num; else prenum=cs->back();
  289. }
  290. if (prenum-num>0.1){ //异常数据 如果之前存在正常数据则继续,如果没有则本次不作参考。0.1暂定
  291. if(cs->empty()){
  292. f=true;break;
  293. }
  294. else continue;
  295. }
  296. nrow++;sum+=num;
  297. }
  298. if (!f && nrow!=0){
  299. double result=sum/nrow;
  300. log_info("reverse_alarm_push %s,%.3f",cs->card_id.c_str(),result);
  301. if(cs->push(result)){//满足三个并且呈现上升趋势
  302. log_info("reverse_alarm_up %s",cs->card_id.c_str());
  303. if (!cs->warning())
  304. cs->set_true();//置true,主进程开始抓取数据。
  305. }
  306. }
  307. if (cs->timeout()){//时间满足,判断是否告警
  308. auto reader_it = sit_list::instance()->get(cs->reader_id());
  309. if (!reader_it){
  310. cs->clear_sensor();
  311. continue;
  312. }
  313. std::string msg;
  314. if(cs->check_location_reversal(msg))//满足告警条件,查看是否已经存在告警,没有产生告警
  315. {
  316. log_info("reverse_alarm_create:%s",cs->card_id.c_str());
  317. //产生告警。
  318. //deal_readers_event(ET_READER_LOCATION_REVERSAL,true,reader_it->second);
  319. }
  320. else //不满足告警条件,查看是否已经存在告警,存在则取消告警。
  321. {
  322. log_info("reverse_alarm_delete:%s",cs->card_id.c_str());
  323. //补
  324. //deal_readers_event(ET_READER_LOCATION_REVERSAL,false,reader_it->second);
  325. }
  326. log_info("%s",msg.c_str());
  327. cs->clear_sensor();//清除数据,重新判断
  328. }
  329. }
  330. }
  331. void reverse_alarm_business::make_condition(uint64_t type,uint32_t id,int32_t antid,uint32_t ct,uint64_t tof,uint32_t sid)
  332. {
  333. uint64_t cid = tool_other::type_id_to_u64(type,id);
  334. auto it = m_sensor_mgr.find(cid);
  335. if(it != m_sensor_mgr.end())
  336. it->second->make_condition(antid,ct,tof,sid);
  337. }
  338. void card_sensor_mgr::make_condition(int32_t antid,uint32_t ct,uint64_t tof,uint32_t sid)
  339. {
  340. for(const auto x:m_mgr)
  341. {
  342. auto pdc =x.second;
  343. if (pdc->warning()){
  344. auto sit = sit_list::instance()->get(sid);
  345. point p0=(*sit)[0];point p1=(*sit)[1];
  346. if (!((*pdc)[0]==p0 && (*pdc)[1]==p1)){
  347. pdc->set(p0,p1,sid);
  348. log_info("[reverse_alarm_start]%s,bigger:%d",pdc->card_id.c_str(),pdc->bigger);
  349. }
  350. log_info("[reverse_alarm_working]%s,ct:%d,antid:%d,dist:%.2f",pdc->card_id.c_str(),ct,antid,tof);
  351. pdc->make_reverse_condition(antid,ct,tof);
  352. }
  353. }
  354. }