db_card.cpp 6.1 KB

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