main.cpp 6.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204
  1. #include <ev++.h>
  2. #include <log.h>
  3. #include <net-service.h>
  4. #include "db_api/CDBSingletonDefine.h"
  5. #include "websocket/wsClientMgr.h"
  6. #include "websocket/wsTimerThread.h"
  7. #include "card.h"
  8. #include "ant.h"
  9. #include "area.h"
  10. #include "card_path.h"
  11. #include "landmark.h"
  12. #include "special_area.h"
  13. #include "mine.h"
  14. #include "event.h"
  15. #include"module_service/module_mgr.h"
  16. #include <config_file.h>
  17. #include "three_rates.h"
  18. #include "mine_business.h"
  19. #include "main_test.h"
  20. #include "ya_setting.h"
  21. #include "websocket/web_connect.h"
  22. #include "forbid_staff_down_mine.h"
  23. config_file config;
  24. void handlereader(uint32_t readerid,bool duration,uint32_t t)
  25. {}
  26. //三率模块中告警回调
  27. void Handle_ThreeRates_Event_Callback(const int evType, const int evId, uint64_t id
  28. , double limitVal, double curVal, bool bFalg)
  29. {
  30. if (evId >= EVENT_TYPE::CARD_EVENT_COUNT_MAX || evType > OBJECT_TYPE::OT_CARD)
  31. {
  32. return ;
  33. }
  34. event_tool::instance()->handle_event((OBJECT_TYPE)evType,(EVENT_TYPE)evId,id,limitVal,curVal,bFalg);
  35. }
  36. struct Init_Setting
  37. {
  38. void init()
  39. {
  40. YADB::_DB_POOL_SETTING_ DBSetting;
  41. DBSetting.Host = config.get("db.host","127.0.0.1");
  42. DBSetting.User = config.get("db.user","root");
  43. DBSetting.PWD = config.get("db.passwd","123456");
  44. DBSetting.DBName = config.get("db.dbname","yaxt");
  45. DBSetting.CharSet = config.get("db.charset","utf8");
  46. DBSetting.TimeOut = config.get("db.conn_timeout",5);
  47. DBSetting.PoolSize = config.get("db.pool_size",30);
  48. DBSetting.Port = config.get("db.port",3306);
  49. if(!_mysql_init(DBSetting))
  50. {
  51. std_info("连接DB[%s:%s] 失败,采集服务器无法启动!",DBSetting.Host.c_str(),DBSetting.User.c_str());
  52. log_warn("连接DB[%s:%s] 失败,采集服务器无法启动!",DBSetting.Host.c_str(),DBSetting.User.c_str());
  53. exit(0);
  54. }
  55. log_info("数据库线程池创建成功");
  56. std::string url=config.get("service.websocket_url","ws://127.0.0.1:8086");
  57. int32_t send_interval =config.get("service.interval_send_json_postion",1);
  58. std_info("json_interval:%d",send_interval);
  59. std::vector<std::string> url_list;
  60. url_list.push_back(url);
  61. if(!wsClientMgr_init(url_list,send_interval))
  62. {
  63. std_info("连接webServer[%s] 失败,采集服务器无法启动!",url.c_str());
  64. log_warn("连接webServer[%s] 失败!",url.c_str());
  65. }
  66. CYaSetting::Init_sys_setting();
  67. sit_list::instance()->load_from_db();
  68. card_list::instance()->init_card_from_db();
  69. area_list::instance()->init_from_db();
  70. forbid_staff_down_mine::instance()->init_forbid_staff();
  71. //point pt(3348,100);
  72. //int id = area_list::instance()->get_area(pt)->id();
  73. //std_info("test area:%d",id);
  74. Landmark_list::instance()->init_from_db();
  75. //auto a = Landmark_list::instance()->get(5,id,pt);
  76. //std_info("test landmark:id:%d,dir:%d,dis:%.2f",std::get<0>(a),std::get<1>(a),std::get<2>(a));
  77. special_area_list::instance()->init_from_db();
  78. //auto s = special_area_list::instance()->get_special_id(1013,point(4727,-254),2);
  79. //std_info("test special_area:%d",s);
  80. mine_business::inst()->load();
  81. event_list::instance()->load_his_data_from_db();
  82. card_list::instance()->load_his_card_postion_from_db();
  83. if(three_rates_flag){
  84. db_para dp;
  85. dp.Host=DBSetting.Host;
  86. dp.User=DBSetting.User;
  87. dp.PWD=DBSetting.PWD ;
  88. dp.DBName=DBSetting.DBName;
  89. dp.CharSet=DBSetting.CharSet ;
  90. dp.TimeOut=DBSetting.TimeOut;
  91. dp.PoolSize=DBSetting.PoolSize;
  92. init_three_rates(dp);
  93. }
  94. log_info("Init_Setting::init Success. \n" );
  95. }
  96. void init_three_rates(const db_para& dbs)
  97. {
  98. log_info("init three_rates ...................");
  99. init_para ip;
  100. ip.send_pt = handlereader;
  101. ip.driving_face_alarm = Handle_ThreeRates_Event_Callback;
  102. three_rates::get_instance()->init(ip,dbs);
  103. three_rates::get_instance()->start();
  104. std_info("加载三率模块成功");
  105. }
  106. bool _mysql_init(YADB::_DB_POOL_SETTING_ &dps)
  107. {
  108. std::string szError = "";
  109. bool flag = true;
  110. //创建连接池
  111. if ( !sDBConnPool.Create( dps, szError ) )
  112. {
  113. log_error("数据库线程池创建失败,Err=%s", szError.c_str());
  114. std_error("数据库线程池创建失败,Err=%s", szError.c_str());
  115. flag = false;
  116. }
  117. return flag;
  118. }
  119. bool wsClientMgr_init(const std::vector<std::string>&uri_list,int32_t st)
  120. {
  121. std::map<std::string, YA::MSG_HANDLE_FUNC_TYPE> MsgFuncList;
  122. module_mgr::init(config, MsgFuncList);
  123. web_connect::init(uri_list, st, MsgFuncList);
  124. return web_connect::connect();
  125. }
  126. };
  127. void cleanup()
  128. {
  129. module_mgr::stop();
  130. web_connect::close();
  131. }
  132. void usage(char ** argv)
  133. {
  134. std::string cmd(argv[1]);
  135. if(cmd=="--help" || cmd == "-H")
  136. {
  137. printf("用法:yals [选项]文件...\n");
  138. printf("选项:\n");
  139. printf("--help 显示此帮助说明\n");
  140. printf("-v|--version 显示采集程序版本信息\n");
  141. }
  142. else if(cmd == "--version" || cmd == "-v")
  143. {
  144. printf("yals (采集程序) 2.0.0 build 11\n");
  145. }
  146. else
  147. {
  148. printf("Unknown option:%s\n",cmd.c_str());
  149. printf( "usage:yals [--version][--help][-v][-H]\n");
  150. }
  151. }
  152. int main(int argc ,char * argv[])
  153. {
  154. if(argc >1)
  155. {
  156. usage(argv);
  157. return 0;
  158. }
  159. log_init("../etc/log.ini");
  160. if(config.open("../etc/config.ini"))
  161. return -1;
  162. Init_Setting is;
  163. is.init();
  164. module_mgr::start();
  165. web_connect::start_beatheart_monitor();
  166. atexit(&cleanup);
  167. net_service mh;
  168. int port=config.get("service.port",4000);
  169. log_info("service_handle::instance(&mh)->run(%d)",port);
  170. std_info("service_handle::instance(&mh)->run(%d)",port);
  171. service_handle::instance(&mh)->run(port);
  172. sDBConnPool.Close();
  173. sleep(1);
  174. printf("YA-SERV Server Exit ...\n");
  175. return 0;
  176. }