BasicConfig.h 5.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199
  1. /*
  2. * @Description:
  3. * @Version: 1.0
  4. * @Autor: lishengyin
  5. * @Date: 2022-03-28 09:03:02
  6. * @LastEditors: lishengyin
  7. * @LastEditTime: 2022-08-23 16:01:50
  8. */
  9. #ifndef __BASICECONFIG_H_
  10. #define __BASICECONFIG_H_
  11. #include <iostream>
  12. #include <vector>
  13. #include <rapidjson/document.h>
  14. #include <rapidjson/rapidjson.h>
  15. #include <rapidjson/stringbuffer.h>
  16. #include <rapidjson/writer.h>
  17. using namespace std;
  18. class BasicConfig
  19. {
  20. public:
  21. // usr
  22. std::string usr;
  23. std::string pwd;
  24. // NettyIp;
  25. std::string NettyIP;
  26. int NettyPort;
  27. std::string mysql_IP;
  28. int mysql_port;
  29. double DiskUsageThreshold;
  30. int HardDiskCleanupTime;
  31. int InferDataRetentionTime;
  32. int interval;
  33. int sensitivity;
  34. int TimeOut;
  35. int debug;
  36. int getHistoryVideoFlag;
  37. int ClearPastRecords;
  38. int LowVersion = 0;
  39. double bboxSize = 0.0;
  40. int ExpelInterval = 0;
  41. std::string com_port = "";
  42. int ExpelPort = 19002;
  43. int InferInterval = 2000;
  44. std::string payServerIP = "10.81.160.141";
  45. int filterLevel = 2;
  46. public:
  47. BasicConfig(){}
  48. ~BasicConfig(){}
  49. /**
  50. * @description: json转换为object
  51. * @param {string&} json
  52. * @return {*}
  53. */
  54. bool JsonToObject(std::string& json){
  55. rapidjson::Document doc;
  56. if (doc.Parse<rapidjson::kParseCommentsFlag>(json.c_str()).HasParseError()) {
  57. return false;
  58. }
  59. // get members
  60. const auto end = doc.MemberEnd();
  61. // json_type
  62. if(end == doc.FindMember("usr") || !doc["usr"].IsString()) {
  63. return false;
  64. }else{
  65. usr = doc["usr"].GetString();
  66. }
  67. if(end == doc.FindMember("pwd") || !doc["pwd"].IsString()) {
  68. return false;
  69. }else{
  70. pwd = doc["pwd"].GetString();
  71. }
  72. if(end == doc.FindMember("NettyIP") || !doc["NettyIP"].IsString()) {
  73. return false;
  74. }else{
  75. NettyIP = doc["NettyIP"].GetString();
  76. }
  77. if(end == doc.FindMember("NettyPort") || !doc["NettyPort"].IsString()) {
  78. return false;
  79. }else{
  80. NettyPort = std::atoi(doc["NettyPort"].GetString());
  81. }
  82. if(end == doc.FindMember("DiskUsageThreshold") || !doc["DiskUsageThreshold"].IsString()) {
  83. return false;
  84. }else{
  85. DiskUsageThreshold = std::atof(doc["DiskUsageThreshold"].GetString());
  86. }
  87. if(end == doc.FindMember("HardDiskCleanupTime") || !doc["HardDiskCleanupTime"].IsString()) {
  88. return false;
  89. }else{
  90. HardDiskCleanupTime = std::atoi(doc["HardDiskCleanupTime"].GetString());
  91. }
  92. if(end == doc.FindMember("InferDataRetentionTime") || !doc["InferDataRetentionTime"].IsString()) {
  93. return false;
  94. }else{
  95. InferDataRetentionTime = std::atoi(doc["InferDataRetentionTime"].GetString());
  96. }
  97. if(end == doc.FindMember("interval") || !doc["interval"].IsString()) {
  98. return false;
  99. }else{
  100. interval = std::atoi(doc["interval"].GetString());
  101. }
  102. if(end == doc.FindMember("sensitivity") || !doc["sensitivity"].IsString()) {
  103. return false;
  104. }else{
  105. sensitivity = std::atoi(doc["sensitivity"].GetString());
  106. }
  107. if(end == doc.FindMember("TimeOut") || !doc["TimeOut"].IsString()) {
  108. return false;
  109. }else{
  110. TimeOut = std::atoi(doc["TimeOut"].GetString());
  111. }
  112. if(end == doc.FindMember("mysql_IP") || !doc["mysql_IP"].IsString()) {
  113. return false;
  114. }else{
  115. mysql_IP = doc["mysql_IP"].GetString();
  116. }
  117. if(end == doc.FindMember("mysql_port") || !doc["mysql_port"].IsString()) {
  118. return false;
  119. }else{
  120. mysql_port = std::atoi(doc["mysql_port"].GetString());
  121. }
  122. if(end != doc.FindMember("com_port") && doc["com_port"].IsString()) {
  123. com_port = doc["com_port"].GetString();
  124. }
  125. if(end != doc.FindMember("Debug") && doc["Debug"].IsString()) {
  126. debug = std::atoi(doc["Debug"].GetString());
  127. }
  128. if(end != doc.FindMember("getHistoryVideoFlag") && doc["getHistoryVideoFlag"].IsString()) {
  129. getHistoryVideoFlag = std::atoi(doc["getHistoryVideoFlag"].GetString());
  130. }
  131. if(end != doc.FindMember("ClearPastRecords") && doc["ClearPastRecords"].IsString()) {
  132. ClearPastRecords = std::atoi(doc["ClearPastRecords"].GetString());
  133. }
  134. if(end != doc.FindMember("bboxSize") && doc["bboxSize"].IsString()) {
  135. bboxSize = std::atof(doc["bboxSize"].GetString());
  136. }
  137. if(end != doc.FindMember("ExpelInterval") && doc["ExpelInterval"].IsString()) {
  138. ExpelInterval = std::atoi(doc["ExpelInterval"].GetString());
  139. }
  140. if(end != doc.FindMember("LowVersion") && doc["LowVersion"].IsString()){
  141. LowVersion = std::atoi(doc["LowVersion"].GetString());
  142. }
  143. if(end != doc.FindMember("ExpelPort") && doc["ExpelPort"].IsString()){
  144. ExpelPort = std::atoi(doc["ExpelPort"].GetString());
  145. }
  146. if(end != doc.FindMember("InferInterval") && doc["InferInterval"].IsString()){
  147. InferInterval = std::atoi(doc["InferInterval"].GetString());
  148. }
  149. if(end != doc.FindMember("payServerIP") && doc["payServerIP"].IsString()){
  150. payServerIP = doc["payServerIP"].GetString();
  151. }
  152. if(end != doc.FindMember("filterLevel") && doc["filterLevel"].IsString()){
  153. filterLevel = std::atoi(doc["filterLevel"].GetString());
  154. }
  155. return true;
  156. }
  157. };
  158. #endif