db_card.cpp 6.4 KB

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