db_card.cpp 6.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217
  1. #include <memory>
  2. #include <unordered_map>
  3. #include "log.h"
  4. #include "config_file.h"
  5. #include "db_api/CDBResultSet.h"
  6. #include "db_api/CDBSingletonDefine.h"
  7. #include "../card_base.h"
  8. #include "../card_person.h"
  9. #include "../common.h"
  10. //#include "../common_tool.h"
  11. extern config_file config;
  12. namespace db_card
  13. {
  14. static std::string to13str(uint64_t data)
  15. {
  16. char ss[20]={0};
  17. sprintf(ss, "%013ld", data);
  18. return std::string(ss);
  19. }
  20. static uint64_t getId(uint32_t cardid,uint64_t type)
  21. {
  22. return type<<32|cardid;
  23. }
  24. std::unordered_map<uint64_t,std::shared_ptr<card_location_base>> load_car(int64_t id64)
  25. {
  26. std::string sql = "SELECT ve.vehicle_id, ve.card_id, c.card_type_id, \
  27. ve.dept_id, ve.group_id, v.vehicle_type_id, vt.vehicle_level_id, \
  28. vt.is_railroad AS vt_is_railroad,ve.need_display ,ve.power_alarm,\
  29. vt.vehicle_category_id,v.bigger_car_flag,vc.over_speed \
  30. FROM dat_vehicle_extend ve \
  31. LEFT JOIN dat_vehicle v ON ve.vehicle_id = v.vehicle_id \
  32. LEFT JOIN dat_card c ON ve.card_id = c.card_id \
  33. LEFT JOIN dat_dept d ON ve.dept_id = d.dept_id \
  34. LEFT JOIN dat_group g ON ve.group_id = g.group_id \
  35. LEFT JOIN dat_vehicle_type vt ON v.vehicle_type_id = vt.vehicle_type_id \
  36. LEFT JOIN dat_vehicle_category vc ON vc.vehicle_category_id = vt.vehicle_category_id \
  37. WHERE c.state_id = 0";
  38. std::unordered_map<uint64_t,std::shared_ptr<card_location_base>> map;
  39. std::string card_id_str = to13str(id64);
  40. if(-1 == id64)
  41. {
  42. sql.append(";");
  43. }
  44. else
  45. {
  46. sql.append(" AND ve.card_id ='");
  47. sql.append(card_id_str);
  48. sql.append("';");
  49. log_info("基础数据 增加或修改车卡 sql=%s", sql.c_str());
  50. }
  51. std::string Error;
  52. YADB::CDBResultSet DBRes;
  53. sDBConnPool.Query(sql.c_str(),DBRes,Error);
  54. int nCount = DBRes.GetRecordCount( Error );
  55. if (nCount < 1)
  56. {
  57. log_error("增加或修改失败,数据库中找不到: card_id=%s", card_id_str.c_str());
  58. return map;
  59. }
  60. while ( DBRes.GetNextRecod(Error) )
  61. {
  62. unsigned int vehicle_id = 0;
  63. DBRes.GetField( "vehicle_id",vehicle_id, Error );
  64. std::string card_id;
  65. DBRes.GetField( "card_id",card_id, Error );
  66. uint32_t vsid = atoi(card_id.substr(3).c_str());
  67. unsigned int card_type_id = 0;
  68. DBRes.GetField( "card_type_id",card_type_id, Error );
  69. int dept_id = 0;
  70. DBRes.GetField( "dept_id",dept_id, Error );
  71. int group_id = 0;
  72. DBRes.GetField( "group_id",group_id, Error );
  73. int vehicle_type_id = 0;
  74. DBRes.GetField( "vehicle_type_id",vehicle_type_id, Error );
  75. int vehicle_level_id = 0;
  76. DBRes.GetField( "vehicle_level_id",vehicle_level_id, Error );
  77. int need_display = 0;
  78. DBRes.GetField( "need_display",need_display, Error );
  79. int power_alarm = 0;
  80. DBRes.GetField( "power_alarm",power_alarm, Error );
  81. int vehicle_category_id = 0;
  82. DBRes.GetField( "vehicle_category_id",vehicle_category_id, Error );
  83. int bigger_car_flag= 0;
  84. DBRes.GetField( "bigger_car_flag",bigger_car_flag, Error );
  85. double over_speed= 0;
  86. DBRes.GetField( "over_speed",over_speed, Error );
  87. std::string strategy;
  88. if(card_type_id ==CT_VEHICLE)
  89. strategy = config.get("car.strategy","CS_1");
  90. else
  91. strategy = config.get("coal_winning.strategy","WS_1");
  92. auto clb = card_location_base::make_car(strategy,vsid,need_display,card_type_id,
  93. dept_id,vehicle_category_id, vehicle_type_id,vehicle_level_id,vehicle_id);
  94. uint64_t cardid = getId(vsid,card_type_id);
  95. log_info("cardId:%llu,id:%d dept_id:%d,need_display:%d-cardid:%s,categoryid:%d,vchile_id:%d,type:%d",
  96. cardid,vsid,dept_id,need_display,card_id.c_str(),vehicle_category_id,vehicle_id,card_type_id);
  97. map.insert({cardid,clb});
  98. log_info("基础数据 增加或修改车卡成功:卡id:%ld, dept_id:%d,need_display:%d,categoryid:%d",
  99. id64,dept_id,need_display,vehicle_category_id);
  100. }
  101. return map;
  102. }
  103. std::unordered_map<uint64_t,std::shared_ptr<card_location_base>> load_person(int64_t id64,const std::string&strategy)
  104. {
  105. std::string sql = "SELECT staff_id, s.card_id, c.card_type_id, s.dept_id, s.group_id, s.occupation_id, \
  106. ol.occupation_level_id,s.worktype_id,s.need_display,s.work_line\
  107. ,ds.name as staffer_name,dd.name as dept_name\
  108. FROM dat_staff_extend s \
  109. LEFT JOIN dat_staff ds ON ds.staff_id = s.staff_id \
  110. LEFT JOIN dat_dept dd ON dd.dept_id = s.dept_id \
  111. LEFT JOIN dat_card c ON s.card_id = c.card_id \
  112. LEFT JOIN dat_occupation o ON s.occupation_id = o.occupation_id \
  113. LEFT JOIN dat_occupation_level ol ON ol.occupation_level_id = o.occupation_level_id \
  114. WHERE s.duty_id = 0 AND c.state_id = 0";
  115. std::unordered_map<uint64_t,std::shared_ptr<card_location_base>> map;
  116. std::string card_id_str = to13str(id64);
  117. if(-1 == id64)
  118. {
  119. sql.append(";");
  120. }
  121. else
  122. {
  123. sql.append(" AND s.card_id = '");
  124. sql.append(card_id_str);
  125. sql.append("';");
  126. log_info("基础数据 增加或修改人卡 sql=%s", sql.c_str());
  127. }
  128. std::string Error;
  129. YADB::CDBResultSet DBRes;
  130. sDBConnPool.Query(sql.c_str(),DBRes,Error);
  131. int nCount = DBRes.GetRecordCount( Error );
  132. if (nCount < 1)
  133. {
  134. log_error("增加或修改失败,数据库中找不到: card_id=%s", card_id_str.c_str());
  135. return map;
  136. }
  137. while ( DBRes.GetNextRecod(Error) )
  138. {
  139. unsigned int staff_id = 0;
  140. DBRes.GetField( "staff_id",staff_id, Error );
  141. std::string card_id;
  142. DBRes.GetField( "card_id",card_id, Error );
  143. uint32_t vsid = atoi(card_id.substr(3).c_str());
  144. unsigned int card_type_id = 0;
  145. DBRes.GetField( "card_type_id",card_type_id, Error );
  146. int dept_id = 0;
  147. DBRes.GetField( "dept_id",dept_id, Error );
  148. int group_id = 0;
  149. DBRes.GetField( "group_id",group_id, Error );
  150. int occupation_id = 0;
  151. DBRes.GetField( "occupation_id",occupation_id, Error );
  152. int occupation_level_id = 0;
  153. DBRes.GetField( "occupation_level_id",occupation_level_id, Error );
  154. int need_display = 0;
  155. DBRes.GetField( "need_display",need_display, Error );
  156. int work_line = 0;
  157. DBRes.GetField( "work_line",work_line, Error );
  158. std::string staffer_name;
  159. DBRes.GetField( "staffer_name",staffer_name, Error );
  160. std::string dept_name;
  161. DBRes.GetField( "dept_name",dept_name, Error );
  162. std::shared_ptr<card_location_base> clb =
  163. std::make_shared<person>(strategy,vsid,need_display,card_type_id,dept_id,occupation_level_id,staff_id,work_line,staffer_name,dept_name);
  164. uint64_t cardid = getId(vsid,card_type_id);
  165. log_info("cardId:%llu,id:%d dept_id:%d,need_display:%d,card:%s:work_line:%d,staff_id:%d,type:%d,staffer_name:%s,dept_name:%s",
  166. cardid,vsid,dept_id,need_display,card_id.c_str(),work_line,staff_id,card_type_id,staffer_name.c_str(),dept_name.c_str());
  167. map.insert({cardid,clb});
  168. }
  169. return map;
  170. }
  171. }