mine_business.cpp 9.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361
  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::fetch_add()
  152. {
  153. m_staffer_num_ptr->fetch_add();
  154. }
  155. void mine_business::record_staffer_num()
  156. {
  157. m_staffer_num_ptr->record_staffer_num();
  158. }
  159. void mine_business::load()
  160. {
  161. m_reverse_alarm_ptr->load_sensor();
  162. }
  163. void mine_business::handle_reverse_alarm()
  164. {
  165. m_reverse_alarm_ptr->run_sensor();
  166. }
  167. void mine_business::make_reverse_condition(uint64_t type,uint32_t id,int32_t antid,uint32_t ct,uint64_t tof,uint32_t sid)
  168. {
  169. m_reverse_alarm_ptr->make_condition(type,id,antid,ct,tof,sid);
  170. }
  171. void staffer_num_business::record_staffer_num()
  172. {
  173. static uint32_t min_num=0,max_num=0;
  174. std::time_t t = time(NULL);
  175. char ti[64] = { 0 };
  176. strftime(ti,sizeof(ti),"%Y/%m/%d %H:%M:%S",localtime(&t));
  177. std::string sti(ti);
  178. int minute = atoi(sti.substr(sti.find_first_of(':')+1,2).c_str());
  179. if(m_record_staffer_timeval==0)
  180. {
  181. min_num=max_num=m_staff_num;
  182. m_record_staffer_timeval = t;
  183. }
  184. if(t-m_record_staffer_timeval>=120 && minute%2 == 0)
  185. {
  186. std::stringstream ss;
  187. ss<< "INSERT INTO his_staff_number(max_num,min_num,ave_num) VALUES("<<max_num<<','<<min_num<<','<<std::lround((max_num+min_num)/2)<<");";
  188. logn_info(2,"staff_number:%s",ss.str().c_str());
  189. db_tool::PushAsync(ss.str().c_str());
  190. min_num=max_num=m_staff_num;
  191. m_record_staffer_timeval = t;
  192. }
  193. else
  194. {
  195. if (min_num>m_staff_num) min_num = m_staff_num;
  196. if (max_num < m_staff_num) max_num = m_staff_num;
  197. }
  198. reset();
  199. return;
  200. }
  201. //前端修改数据库表,dat_sensor dat_drivingface,dat_drivingface_vehicle的时候,需要加载.
  202. void reverse_alarm_business::load_sensor()
  203. {
  204. 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;";
  205. std::string Error;
  206. YADB::CDBResultSet DBRes;
  207. sDBConnPool.Query(sql.c_str(),DBRes,Error);
  208. int nCount = DBRes.GetRecordCount( Error );
  209. if (nCount < 1)
  210. log_error("init_sensor..failed[%s]", sql.c_str());
  211. while ( DBRes.GetNextRecod(Error) )
  212. {
  213. int32_t sensor_id = 0;
  214. DBRes.GetField( "sensor_id",sensor_id, Error );
  215. int32_t data = 0;
  216. DBRes.GetField( "data_source",data, Error );
  217. int32_t workfaceid = 0;
  218. DBRes.GetField( "work_face_id",workfaceid, Error );
  219. std::string card_id;
  220. DBRes.GetField( "card_id",card_id, Error );
  221. double bx,by;
  222. DBRes.GetField( "base_point_x",bx, Error );
  223. DBRes.GetField( "base_point_y",by, Error );
  224. 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);
  225. if(data <= 0) continue;
  226. uint64_t cid = tool_other::card_id_to_u64(card_id);
  227. std::shared_ptr<card_sensor_mgr> tm=nullptr;
  228. auto iter = m_sensor_mgr.find(cid);
  229. if(iter != m_sensor_mgr.end())
  230. tm=iter->second;
  231. else
  232. {
  233. tm = std::make_shared<card_sensor_mgr>();
  234. m_sensor_mgr.insert({cid,tm});
  235. }
  236. point p(bx,by);
  237. tm->push(workfaceid,sensor_id,p,card_id);
  238. }
  239. }
  240. void reverse_alarm_business::run_sensor()
  241. {
  242. static time_t time_interval=0;
  243. time_t t = time(NULL);
  244. if (t-time_interval>3*60)
  245. {
  246. time_interval =t;
  247. printf("reverse_alarm......\n");
  248. for(const auto &x:m_sensor_mgr)
  249. x.second->for_each();
  250. }
  251. }
  252. void card_sensor_mgr::for_each()
  253. {
  254. for(const auto & cst:m_mgr)
  255. {
  256. auto cs = cst.second;
  257. if(cs->sensorId_list.empty())
  258. continue;
  259. char strsql[512]={0};
  260. 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]);
  261. std::string Error;
  262. YADB::CDBResultSet DBRes;
  263. sDBConnPool.Query(strsql,DBRes,Error);
  264. int nCount = DBRes.GetRecordCount( Error );
  265. if (nCount < 1)
  266. {
  267. log_error("load_sensor_data..failed[%s]", strsql);
  268. continue;
  269. }
  270. double prenum=-1;
  271. double nrow=0;double sum=0;bool f=false;
  272. while ( DBRes.GetNextRecod(Error) )
  273. {
  274. double num = 0;
  275. DBRes.GetField( "data_value",num, Error );
  276. if (point::eq(num,0,1e-10))continue; //0异常数据抛掉
  277. if(point::eq(prenum,-1,1e-10)){ //如果没有参考数据,则判断,如果有历史数据,用历史,没有则用第一个数据。
  278. if(cs->empty()) prenum=num; else prenum=cs->back();
  279. }
  280. if (prenum-num>0.1){ //异常数据 如果之前存在正常数据则继续,如果没有则本次不作参考。0.1暂定
  281. if(cs->empty()){
  282. f=true;break;
  283. }
  284. else continue;
  285. }
  286. nrow++;sum+=num;
  287. }
  288. if (!f && nrow!=0){
  289. double result=sum/nrow;
  290. log_info("reverse_alarm_push %s,%.3f",cs->card_id.c_str(),result);
  291. if(cs->push(result)){//满足三个并且呈现上升趋势
  292. log_info("reverse_alarm_up %s",cs->card_id.c_str());
  293. if (!cs->warning())
  294. cs->set_true();//置true,主进程开始抓取数据。
  295. }
  296. }
  297. if (cs->timeout()){//时间满足,判断是否告警
  298. auto reader_it = sit_list::instance()->get(cs->reader_id());
  299. if (!reader_it){
  300. cs->clear_sensor();
  301. continue;
  302. }
  303. std::string msg;
  304. if(cs->check_location_reversal(msg))//满足告警条件,查看是否已经存在告警,没有产生告警
  305. {
  306. log_info("reverse_alarm_create:%s",cs->card_id.c_str());
  307. //产生告警。
  308. //deal_readers_event(ET_READER_LOCATION_REVERSAL,true,reader_it->second);
  309. }
  310. else //不满足告警条件,查看是否已经存在告警,存在则取消告警。
  311. {
  312. log_info("reverse_alarm_delete:%s",cs->card_id.c_str());
  313. //补
  314. //deal_readers_event(ET_READER_LOCATION_REVERSAL,false,reader_it->second);
  315. }
  316. log_info("%s",msg.c_str());
  317. cs->clear_sensor();//清除数据,重新判断
  318. }
  319. }
  320. }
  321. void reverse_alarm_business::make_condition(uint64_t type,uint32_t id,int32_t antid,uint32_t ct,uint64_t tof,uint32_t sid)
  322. {
  323. uint64_t cid = tool_other::type_id_to_u64(type,id);
  324. auto it = m_sensor_mgr.find(cid);
  325. if(it != m_sensor_mgr.end())
  326. it->second->make_condition(antid,ct,tof,sid);
  327. }
  328. void card_sensor_mgr::make_condition(int32_t antid,uint32_t ct,uint64_t tof,uint32_t sid)
  329. {
  330. for(const auto x:m_mgr)
  331. {
  332. auto pdc =x.second;
  333. if (pdc->warning()){
  334. auto sit = sit_list::instance()->get(sid);
  335. point p0=(*sit)[0];point p1=(*sit)[1];
  336. if (!((*pdc)[0]==p0 && (*pdc)[1]==p1)){
  337. pdc->set(p0,p1,sid);
  338. log_info("[reverse_alarm_start]%s,bigger:%d",pdc->card_id.c_str(),pdc->bigger);
  339. }
  340. log_info("[reverse_alarm_working]%s,ct:%d,antid:%d,dist:%.2f",pdc->card_id.c_str(),ct,antid,tof);
  341. pdc->make_reverse_condition(antid,ct,tof);
  342. }
  343. }
  344. }