db_card.cpp 6.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209
  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 "../common.h"
  9. #include "../card_person.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. }
  50. std::string Error;
  51. YADB::CDBResultSet DBRes;
  52. sDBConnPool.Query(sql.c_str(),DBRes,Error);
  53. int nCount = DBRes.GetRecordCount( Error );
  54. if (nCount < 1)
  55. {
  56. log_error("增加或修改失败,数据库中找不到: sql=%s", sql.c_str());
  57. return map;
  58. }
  59. while ( DBRes.GetNextRecod(Error) )
  60. {
  61. unsigned int vehicle_id = 0;
  62. DBRes.GetField( "vehicle_id",vehicle_id, Error );
  63. std::string card_id;
  64. DBRes.GetField( "card_id",card_id, Error );
  65. uint32_t vsid = atoi(card_id.substr(3).c_str());
  66. unsigned int card_type_id = 0;
  67. DBRes.GetField( "card_type_id",card_type_id, Error );
  68. int dept_id = 0;
  69. DBRes.GetField( "dept_id",dept_id, Error );
  70. int group_id = 0;
  71. DBRes.GetField( "group_id",group_id, Error );
  72. int vehicle_type_id = 0;
  73. DBRes.GetField( "vehicle_type_id",vehicle_type_id, Error );
  74. int vehicle_level_id = 0;
  75. DBRes.GetField( "vehicle_level_id",vehicle_level_id, Error );
  76. int need_display = 0;
  77. DBRes.GetField( "need_display",need_display, Error );
  78. int power_alarm = 0;
  79. DBRes.GetField( "power_alarm",power_alarm, Error );
  80. int vehicle_category_id = 0;
  81. DBRes.GetField( "vehicle_category_id",vehicle_category_id, Error );
  82. int bigger_car_flag= 0;
  83. DBRes.GetField( "bigger_car_flag",bigger_car_flag, Error );
  84. double over_speed= 0;
  85. DBRes.GetField( "over_speed",over_speed, Error );
  86. std::string strategy;
  87. if(card_type_id ==CT_VEHICLE)
  88. strategy = config.get("car.strategy","CS_1");
  89. else
  90. strategy = config.get("coal_winning.strategy","WS_1");
  91. auto clb = card_location_base::make_car(strategy,vsid,need_display,card_type_id,
  92. dept_id,vehicle_category_id, vehicle_type_id,vehicle_level_id,vehicle_id);
  93. uint64_t cardid = getId(vsid,card_type_id);
  94. log_info("cardId:%llu,id:%d dept_id:%d,need_display:%d-cardid:%s,categoryid:%d,vchile_id:%d,type:%d",
  95. cardid,vsid,dept_id,need_display,card_id.c_str(),vehicle_category_id,vehicle_id,card_type_id);
  96. map.insert({cardid,clb});
  97. }
  98. return map;
  99. }
  100. std::unordered_map<uint64_t,std::shared_ptr<card_location_base>> load_person(int64_t id64,const std::string&strategy)
  101. {
  102. std::string sql = "SELECT s.staff_id, s.card_id, c.card_type_id, s.dept_id, s.group_id, s.occupation_id, \
  103. ol.occupation_level_id,s.worktype_id,s.need_display,s.work_line\
  104. ,ds.name as staffer_name,dd.name as dept_name\
  105. FROM dat_staff_extend s \
  106. LEFT JOIN dat_staff ds ON ds.staff_id = s.staff_id \
  107. LEFT JOIN dat_dept dd ON dd.dept_id = s.dept_id \
  108. LEFT JOIN dat_card c ON s.card_id = c.card_id \
  109. LEFT JOIN dat_occupation o ON s.occupation_id = o.occupation_id \
  110. LEFT JOIN dat_occupation_level ol ON ol.occupation_level_id = o.occupation_level_id \
  111. WHERE s.duty_id = 0 AND c.state_id = 0";
  112. std::unordered_map<uint64_t,std::shared_ptr<card_location_base>> map;
  113. std::string card_id_str = to13str(id64);
  114. if(-1 == id64)
  115. {
  116. sql.append(";");
  117. }
  118. else
  119. {
  120. sql.append(" AND s.card_id = '");
  121. sql.append(card_id_str);
  122. sql.append("';");
  123. }
  124. std::string Error;
  125. YADB::CDBResultSet DBRes;
  126. sDBConnPool.Query(sql.c_str(),DBRes,Error);
  127. int nCount = DBRes.GetRecordCount( Error );
  128. if (nCount < 1)
  129. {
  130. log_error("增加或修改失败,数据库中找不到: sql", sql.c_str());
  131. return map;
  132. }
  133. while ( DBRes.GetNextRecod(Error) )
  134. {
  135. unsigned int staff_id = 0;
  136. DBRes.GetField( "staff_id",staff_id, Error );
  137. std::string card_id;
  138. DBRes.GetField( "card_id",card_id, Error );
  139. uint32_t vsid = atoi(card_id.substr(3).c_str());
  140. unsigned int card_type_id = 0;
  141. DBRes.GetField( "card_type_id",card_type_id, Error );
  142. int dept_id = 0;
  143. DBRes.GetField( "dept_id",dept_id, Error );
  144. int group_id = 0;
  145. DBRes.GetField( "group_id",group_id, Error );
  146. int occupation_id = 0;
  147. DBRes.GetField( "occupation_id",occupation_id, Error );
  148. int occupation_level_id = 0;
  149. DBRes.GetField( "occupation_level_id",occupation_level_id, Error );
  150. int need_display = 0;
  151. DBRes.GetField( "need_display",need_display, Error );
  152. int work_line = 0;
  153. DBRes.GetField( "work_line",work_line, Error );
  154. std::string staffer_name;
  155. DBRes.GetField( "staffer_name",staffer_name, Error );
  156. std::string dept_name;
  157. DBRes.GetField( "dept_name",dept_name, Error );
  158. std::shared_ptr<card_location_base> clb =
  159. std::make_shared<person>(strategy,vsid,need_display,card_type_id,dept_id,occupation_level_id,staff_id,work_line,staffer_name,dept_name);
  160. uint64_t cardid = tool_other::type_id_to_u64(vsid,card_type_id);
  161. 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",
  162. 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());
  163. map.insert({cardid,clb});
  164. }
  165. return map;
  166. }
  167. }