HikvisionBaseDataTaskTest.java 35 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765
  1. package com.xjrsoft.module.job;
  2. import cn.hutool.db.Db;
  3. import cn.hutool.db.Entity;
  4. import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
  5. import com.google.gson.JsonArray;
  6. import com.google.gson.JsonElement;
  7. import com.google.gson.JsonObject;
  8. import com.google.gson.JsonParser;
  9. import com.xjrsoft.XjrSoftApplication;
  10. import com.xjrsoft.common.mybatis.SqlRunnerAdapter;
  11. import com.xjrsoft.common.utils.SqlRunnerAdapterUtil;
  12. import com.xjrsoft.module.base.entity.BaseClass;
  13. import com.xjrsoft.module.evaluate.controller.EvaluateItemController;
  14. import com.xjrsoft.module.hikvision.entity.HikvisionData;
  15. import com.xjrsoft.module.hikvision.mapper.HikvisionDataMapper;
  16. import com.xjrsoft.module.hikvision.util.ApiUtil;
  17. import com.xjrsoft.module.hikvision.util.DataUtil;
  18. import com.xjrsoft.module.hikvision.util.OutInRecordUtil;
  19. import com.xjrsoft.module.organization.entity.Department;
  20. import com.xjrsoft.module.teacher.entity.XjrUser;
  21. import com.xjrsoft.module.teacher.mapper.FaceImportMapper;
  22. import com.yomahub.liteflow.util.JsonUtil;
  23. import org.junit.jupiter.api.Test;
  24. import org.junit.runner.RunWith;
  25. import org.springframework.beans.factory.annotation.Autowired;
  26. import org.springframework.boot.test.context.SpringBootTest;
  27. import org.springframework.test.context.junit4.SpringRunner;
  28. import java.sql.SQLException;
  29. import java.text.SimpleDateFormat;
  30. import java.util.ArrayList;
  31. import java.util.Date;
  32. import java.util.HashMap;
  33. import java.util.HashSet;
  34. import java.util.List;
  35. import java.util.Map;
  36. import java.util.Set;
  37. /**
  38. * @author dzx
  39. * @date 2024/5/14
  40. */
  41. @RunWith(SpringRunner.class)
  42. @SpringBootTest(classes = XjrSoftApplication.class)
  43. class HikvisionBaseDataTaskTest {
  44. @Autowired
  45. private FaceImportMapper faceImportMapper;
  46. @Autowired
  47. EvaluateItemController evaluateItemController;
  48. OutInRecordUtil outInRecordUtil = new OutInRecordUtil();
  49. @Autowired
  50. private HikvisionDataMapper hikvisionDataMapper;
  51. @Test
  52. void test2(){
  53. String sql = "select * from base_class";
  54. List<Map<String, Object>> maps = SqlRunnerAdapter.db().selectList(sql, BaseClass.class);
  55. for (Map<String, Object> map : maps) {
  56. BaseClass baseClass = SqlRunnerAdapterUtil.convertMapToEntity(map, BaseClass.class);
  57. System.out.println(JsonUtil.toJsonString(baseClass));
  58. }
  59. sql = "SELECT DISTINCT LENGTH(hierarchy) FROM xjr_department WHERE delete_mark = 0";
  60. List<Map<String, Object>> list = SqlRunnerAdapter.db().selectList(sql);
  61. Set<Integer> levelSet = new HashSet<>();
  62. for (Map<String, Object> strings : list) {
  63. Integer[] strings1 = SqlRunnerAdapterUtil.convertMapToIntegerArray(strings);
  64. System.out.println(strings1.toString());
  65. levelSet.add(strings1[0]);
  66. }
  67. }
  68. @Test
  69. void syncStudentData() throws InterruptedException {
  70. DataUtil dataUtil = new DataUtil();
  71. String tableName = "base_student";
  72. List<HikvisionData> studentList = hikvisionDataMapper.selectList(
  73. new QueryWrapper<HikvisionData>().lambda().eq(HikvisionData::getTableName, tableName)
  74. );
  75. Map<String, String> stringStringMap = dataUtil.insertStudentOne(tableName, null, studentList, 1827234730185633794L);
  76. System.out.println(JsonUtil.toJsonString(stringStringMap));
  77. }
  78. @Test
  79. void selectOrg(){
  80. String apiPath = "/api/resource/v1/org/orgList";
  81. ApiUtil apiUtil = new ApiUtil();
  82. JsonObject jsonObject = new JsonObject();
  83. jsonObject.addProperty("pageNo", 1);
  84. jsonObject.addProperty("pageSize", 500);
  85. String doPost = apiUtil.doPost(apiPath, jsonObject.toString(), null);
  86. System.out.println(doPost);
  87. String sql = "SELECT * FROM hikvision_data WHERE 1 = 1 and table_name = 'base_class'";
  88. List<Map<String, Object>> maps = SqlRunnerAdapter.db().selectList(sql);
  89. List<String> sourceIds = new ArrayList<>();
  90. for (Map<String, Object> map : maps) {
  91. sourceIds.add(map.get("source_id").toString());
  92. }
  93. JsonParser jsonParser = new JsonParser();
  94. JsonElement parse = jsonParser.parse(doPost);
  95. JsonObject resultJson = parse.getAsJsonObject();
  96. String tableName = "hikvision_data";
  97. SimpleDateFormat sdf = new SimpleDateFormat("yyy-MM-dd HH:mm:ss");
  98. if("0".equals(resultJson.get("code").getAsString()) && "success".equals(resultJson.get("msg").getAsString())){
  99. JsonArray array = resultJson.get("data").getAsJsonObject().get("list").getAsJsonArray();
  100. for (JsonElement jsonElement : array) {
  101. JsonObject json = jsonElement.getAsJsonObject();
  102. String orgIndexCode = json.get("orgIndexCode").getAsString();
  103. if(sourceIds.contains(orgIndexCode)){
  104. continue;
  105. }
  106. Entity entity = Entity.create(tableName);
  107. entity.set("source_id", orgIndexCode);
  108. entity.set("table_name", "base_class");
  109. entity.set("hikvision_id", orgIndexCode);
  110. entity.set("create_date", sdf.format(new Date()));
  111. SqlRunnerAdapter.db().dynamicInsert(tableName, entity);
  112. }
  113. }
  114. }
  115. @Test
  116. void test() throws Exception {
  117. String sql = "SELECT distinct table_name FROM hikvision_data WHERE 1 = 1";
  118. List<Map<String, Object>> maps = SqlRunnerAdapter.db().selectList(sql, HikvisionData.class);
  119. Set<String> tables = new HashSet<>();
  120. for (Map<String, Object> map : maps) {
  121. String[] strings = SqlRunnerAdapterUtil.convertMapToStringArray(map);
  122. tables.add(strings[0]);
  123. }
  124. sql = "SELECT * FROM hikvision_data WHERE 0 = 0";
  125. List<Map<String, Object>> list = SqlRunnerAdapter.db().selectList(sql, HikvisionData.class);
  126. Map<String, Map<String, String>> dataMap = new HashMap<>();
  127. for (String table : tables) {
  128. Map<String, String> tableData = new HashMap<>();
  129. for (Map<String, Object> hikvision : list) {
  130. if(!table.equals(hikvision.get("table_name").toString())){
  131. continue;
  132. }
  133. tableData.put(hikvision.get("source_id").toString(), hikvision.get("hikvision_id").toString());
  134. }
  135. dataMap.put(table, tableData);
  136. }
  137. DataUtil dataUtil = new DataUtil();
  138. String tableName = "xjr_department";
  139. Map<String, String> department = dataMap.get(tableName);//dataUtil.insertDepartment(use, tableName, dataMap.get(tableName));
  140. if(department.isEmpty() && dataMap.get(tableName) != null){
  141. department = dataMap.get(tableName);
  142. }
  143. //
  144. // //推送教职工
  145. tableName = "base_teacher";
  146. // insertTeacher(tableName, dataMap.get(tableName), department);
  147. //
  148. //推送车辆
  149. // String carTableName = "car_message_apply";
  150. // Map<String, String> baseClass = dataMap.get(carTableName);
  151. // dataUtil.insertCar(use, carTableName,baseClass, faceImportMapper);
  152. tableName = "base_class";
  153. // Map<String, String> baseClass = dataMap.get(tableName);
  154. // Map<String, String> classes = dataUtil.insertClass(tableName, baseClass);
  155. tableName = "base_student";
  156. List<HikvisionData> studentList = hikvisionDataMapper.selectList(
  157. new QueryWrapper<HikvisionData>().lambda().eq(HikvisionData::getTableName, tableName)
  158. );
  159. Map<String, String> baseStudent = dataMap.get(tableName);
  160. insertStudentOne(tableName, baseStudent, studentList);
  161. // sql = "SELECT * FROM wf_room_applicant WHERE delete_mark = 0 AND STATUS = 1";
  162. // List<Map<String, Object>> wfRoomApplicantList = SqlRunnerAdapter.db().selectList(sql);
  163. //
  164. // Map<String, String> baseStudent = dataMap.get(tableName);
  165. // for (Map<String, Object> objectMap : wfRoomApplicantList) {
  166. // Long applicantUserId = Long.parseLong(objectMap.get("applicant_user_id").toString());
  167. // Map<String, String> student = dataUtil.insertStudentOne(tableName, baseStudent, studentList, applicantUserId);
  168. // }
  169. // selectOrg(use, "base_class");
  170. //查询车辆
  171. // String carTableName = "car_message_apply";
  172. // selectCar(use, carTableName);
  173. // outInRecordUtil.GetVehicleRecord(faceImportMapper);
  174. // outInRecordUtil.GetTeacherAndStudentRecords(faceImportMapper);
  175. //// selecAllPersonById(use);
  176. // selectResource(use);
  177. }
  178. @Test
  179. void selectPerson(){
  180. String sql = "SELECT * FROM xjr_user WHERE id IN (\n" +
  181. "SELECT source_id FROM hikvision_data WHERE hikvision_id LIKE '%-%' AND table_name = 'base_teacher'\n" +
  182. ")";
  183. List<Map<String, Object>> list = SqlRunnerAdapter.db().selectList(sql, String[].class);
  184. }
  185. @Test
  186. void testCarInsert() throws Exception {
  187. //outInRecordUtil.GetVehicleRecord(faceImportMapper);
  188. }
  189. public static void selectCar(Db db, String tableName) throws SQLException {
  190. String sql = "SELECT * FROM " + tableName + " WHERE delete_mark = 0";
  191. List<Department> dataList = db.query(sql, Department.class);
  192. String apiPath = "/api/resource/v2/vehicle/advance/vehicleList";
  193. JsonObject jsonObject = new JsonObject();
  194. jsonObject.addProperty("pageNo", 1);
  195. jsonObject.addProperty("pageSize", 500);
  196. ApiUtil apiUtil = new ApiUtil();
  197. String result = apiUtil.doPost(apiPath, jsonObject.toString(), null);
  198. System.out.println("-----------------------------------------------------");
  199. System.out.println(result);
  200. }
  201. public static void insertRecord(Db db, String tableName, Map<?, String> idsMap) throws SQLException {
  202. if (idsMap.isEmpty()) {
  203. return;
  204. }
  205. var sqls = new ArrayList<String>();
  206. for (var entry : idsMap.entrySet()) {
  207. var sourceId = entry.getKey();
  208. var sql = "INSERT INTO hikvision_data(create_date, table_name, source_id, hikvision_id) " +
  209. "VALUES(now(), '" + tableName + "', '" + sourceId + "', '" + entry.getValue() + "')";
  210. sqls.add(sql);
  211. }
  212. db.executeBatch(sqls);
  213. }
  214. public Map<String, String> insertStudentOne(String tableName, Map<String, String> tableData, List<HikvisionData> studentList) throws Exception {
  215. String sql = "SELECT t1.id,t1.name,CONCAT(t3.id,'-',replace(t4.name,'级',''),'-',t3.class_type,'-',t2.stduy_status) as orgIndexCode," +
  216. " t1.user_name,t1.gender,DATE_FORMAT(t1.birth_date, '%Y-%m-%d') as birthday,t1.mobile,t1.email,t1.credential_type,t1.credential_number FROM xjr_user t1" +
  217. " INNER JOIN base_student_school_roll t2 ON t1.id = t2.user_id" +
  218. " INNER JOIN base_class t3 ON t2.class_id = t3.id" +
  219. " INNER JOIN base_grade t4 ON t3.grade_id = t4.id" +
  220. " WHERE t1.delete_mark = 0 AND t2.delete_mark = 0" +
  221. " AND t3.class_type IS NOT NULL AND t3.is_graduate = 1" +
  222. " and t1.credential_number IN (" +
  223. " SELECT t3.credential_number\n" +
  224. " FROM student_report_record t1\n" +
  225. " INNER JOIN base_new_student t3 ON t3.id = t1.user_id\n" +
  226. " LEFT JOIN banding_task_class_student t2 ON t1.user_id = t2.new_student_id AND t2.delete_mark = 0\n" +
  227. " LEFT JOIN banding_task_class t5 ON t2.banding_task_class_id = t5.id\n" +
  228. " LEFT JOIN xjr_user t6 ON t5.teacher_id = t6.id\n" +
  229. " LEFT JOIN xjr_dictionary_detail t7 ON t3.gender = t7.code AND t7.delete_mark = 0\n" +
  230. " LEFT JOIN xjr_dictionary_detail t8 ON t3.source = t8.code AND t8.delete_mark = 0\n" +
  231. " LEFT JOIN xjr_dictionary_detail t9 ON t3.stduy_status = t9.code AND t9.delete_mark = 0\n" +
  232. " LEFT JOIN banding_task t10 ON t5.banding_task_id = t10.id\n" +
  233. " LEFT JOIN enrollment_plan t11 ON t11.id = t3.enrollment_plan_id\n" +
  234. " LEFT JOIN base_grade t4 ON t11.grade_id = t4.id\n" +
  235. " LEFT JOIN student_report_plan t12 ON t12.id = t1.student_report_plan_id\n" +
  236. " LEFT JOIN base_major_set t13 ON t13.id = t5.major_set_id\n" +
  237. " LEFT JOIN xjr_dictionary_detail t14 ON t5.class_type = t14.code AND t14.delete_mark = 0\n" +
  238. " LEFT JOIN xjr_department t15 ON t15.id = t13.department_id\n" +
  239. " WHERE t1.delete_mark = 0 AND t1.enabled_mark = 1\n" +
  240. " AND t12.status IN(1,2)\n" +
  241. " AND t3.enrollment_plan_id = 1897950348853186562\n" +
  242. " AND t12.category = 2\n" +
  243. " AND t1.report_time IS NOT NULL" +
  244. " )";
  245. List<Map<String, Object>> list = SqlRunnerAdapter.db().selectList(sql, String[].class);
  246. Map<Integer, String> clientMap = new HashMap<>();
  247. List<Entity> insertList = new ArrayList<>();
  248. Map<String, String> historyMap = new HashMap<>();
  249. for (HikvisionData hikvisionData : studentList) {
  250. historyMap.put(hikvisionData.getSourceId(), hikvisionData.getSendData());
  251. }
  252. Map<String, String> idMap = new HashMap<>();
  253. JsonParser jsonParser = new JsonParser();
  254. ApiUtil apiUtil = new ApiUtil();
  255. Map<String, String> header = new HashMap<>();
  256. header.put("tagId", "insert_student");
  257. List<JsonObject> addList = new ArrayList<>();
  258. List<JsonObject> updateList = new ArrayList<>();
  259. for(int i = 0; i < list.size(); i ++){
  260. Map<String, Object> objectMap = list.get(i);
  261. String id = objectMap.get("id").toString();
  262. JsonObject paramJson = new JsonObject();
  263. paramJson.addProperty("clientId", i);
  264. paramJson.addProperty("personId", id);
  265. paramJson.addProperty("personName", objectMap.get("name").toString());
  266. paramJson.addProperty("orgIndexCode", objectMap.get("orgIndexCode").toString());
  267. paramJson.addProperty("phoneNo", objectMap.get("mobile")==null?"":objectMap.get("mobile").toString());
  268. paramJson.addProperty("jobNo", objectMap.get("user_name").toString());
  269. paramJson.addProperty("birthday", objectMap.get("birthday")==null?"":objectMap.get("birthday").toString());
  270. paramJson.addProperty("personType", 2);
  271. int gender = 0;
  272. String genderData = objectMap.get("gender") == null ? "" : objectMap.get("gender").toString();
  273. switch (genderData) {
  274. case "SB10001":
  275. gender = 1;
  276. break;
  277. case "SB10002":
  278. gender = 2;
  279. break;
  280. default:
  281. gender = 0;
  282. break;
  283. }
  284. paramJson.addProperty("gender", gender);
  285. if(historyMap.containsKey(id)){
  286. if(historyMap.get(id) ==null || !paramJson.toString().equals(historyMap.get(id))){
  287. updateList.add(paramJson);
  288. }
  289. continue;
  290. }
  291. addList.add(paramJson);
  292. }
  293. String apiPath = "/api/resource/v1/person/single/add";
  294. SimpleDateFormat sdf = new SimpleDateFormat("yyy-MM-dd HH:mm:ss");
  295. for (JsonObject paramJson : addList) {
  296. String result = apiUtil.doPost(apiPath, paramJson.toString(), null, header);
  297. JsonElement parse = jsonParser.parse(result);
  298. JsonObject resultJson = parse.getAsJsonObject();
  299. if("0".equals(resultJson.get("code").getAsString()) && "success".equals(resultJson.get("msg").getAsString())){
  300. Entity entity = Entity.create(tableName);
  301. entity.set("table_name", tableName);
  302. entity.set("create_date", sdf.format(new Date()));
  303. entity.set("source_id", clientMap.get(paramJson.get("clientId").getAsInt()));
  304. entity.set("hikvision_id", paramJson.get("personId").getAsString());
  305. entity.set("send_data", paramJson.toString());
  306. insertList.add(entity);
  307. }else if("0x00052102".equals(resultJson.get("code").getAsString()) && "PersonId Already In Db".equals(resultJson.get("msg").getAsString())){
  308. Entity entity = Entity.create(tableName);
  309. entity.set("table_name", tableName);
  310. entity.set("create_date", sdf.format(new Date()));
  311. entity.set("source_id", paramJson.get("personId").getAsString());
  312. entity.set("hikvision_id", paramJson.get("personId").getAsString());
  313. entity.set("send_data", paramJson.toString());
  314. insertList.add(entity);
  315. }
  316. }
  317. tableName = "hikvision_data";
  318. if(!insertList.isEmpty()){
  319. SqlRunnerAdapter.db().dynamicInsertBatch(tableName, insertList);
  320. }
  321. apiPath = "/api/resource/v1/person/single/update";
  322. for (JsonObject paramJson : updateList) {
  323. Thread.sleep(500);
  324. String result = apiUtil.doPost(apiPath, paramJson.toString(), null, header);
  325. JsonElement parse = jsonParser.parse(result);
  326. JsonObject resultJson = parse.getAsJsonObject();
  327. if("0".equals(resultJson.get("code").getAsString()) && "success".equals(resultJson.get("msg").getAsString())){
  328. Entity entity = Entity.create(tableName);
  329. entity.set("send_data", paramJson.toString());
  330. Entity where = Entity.create(tableName);
  331. where.set("source_id", paramJson.get("personId").getAsString());
  332. SqlRunnerAdapter.db().dynamicUpdate(tableName, entity, where);
  333. }
  334. }
  335. //插入记录表
  336. return idMap;
  337. }
  338. @Test
  339. void deleteClass() {
  340. String apiPath = "/api/resource/v1/org/batch/delete";
  341. JsonObject jsonObject = new JsonObject();
  342. ApiUtil apiUtil = new ApiUtil();
  343. JsonArray dataArray = new JsonArray();
  344. dataArray.add("2025-job_class-FB3002");
  345. JsonObject paramJson = new JsonObject();
  346. paramJson.add("indexCodes", dataArray);
  347. Map<String, String> headerMap = new HashMap<>();
  348. headerMap.put("tagId", "deleteOrg");
  349. String result = apiUtil.doPost(apiPath, paramJson.toString(), null, headerMap);
  350. System.out.println(result);
  351. }
  352. /**
  353. * 查询资源信息
  354. * @param db
  355. * @throws SQLException
  356. */
  357. void selectResource(Db db) throws SQLException {
  358. String apiPath = "/api/irds/v2/resource/resourcesByParams";
  359. JsonObject jsonObject = new JsonObject();
  360. jsonObject.addProperty("pageNo", 1);
  361. jsonObject.addProperty("pageSize", 500);
  362. jsonObject.addProperty("resourceType", "door");
  363. ApiUtil apiUtil = new ApiUtil();
  364. String result = apiUtil.doPost(apiPath, jsonObject.toString(), null, null);
  365. System.out.println("------------------------------------------------------------------------");
  366. System.out.println(result);
  367. }
  368. @Test
  369. void selecAllPersonById() throws SQLException {
  370. String apiPath = "/api/resource/v2/person/advance/personList";
  371. String sql = "SELECT t1.id, t1.name,t1.user_name,t2.dept_id FROM xjr_user t1\n" +
  372. "INNER JOIN xjr_user_dept_relation t2 ON t1.id = t2.user_id\n" +
  373. " WHERE t1.id IN (\n" +
  374. "SELECT source_id FROM hikvision_data WHERE hikvision_id LIKE '%-%' AND table_name = 'base_teacher'\n" +
  375. ")";
  376. List<Map<String, Object>> maps = SqlRunnerAdapter.db().selectList(sql);
  377. ApiUtil apiUtil = new ApiUtil();
  378. Map<String, String> userInfoMap = new HashMap<>();
  379. JsonParser parser = new JsonParser();
  380. for (Map<String, Object> map : maps) {
  381. String name = map.get("name").toString();
  382. JsonObject jsonObject = new JsonObject();
  383. jsonObject.addProperty("pageNo", 1);
  384. jsonObject.addProperty("pageSize", 500);
  385. jsonObject.addProperty("personName", name);
  386. jsonObject.addProperty("orgIndexCodes", map.get("dept_id").toString());
  387. String result = apiUtil.doPost(apiPath, jsonObject.toString(), null, null);
  388. JsonObject asJsonObject = parser.parse(result).getAsJsonObject();
  389. if("0".equals(asJsonObject.get("code").getAsString())){
  390. JsonArray jsonArray = asJsonObject.get("data").getAsJsonObject().get("list").getAsJsonArray();
  391. if(jsonArray.size() == 0){
  392. continue;
  393. }
  394. JsonObject object = jsonArray.get(0).getAsJsonObject();
  395. if(name.equals(object.get("personName").getAsString()) && map.get("user_name").toString().equals(object.get("jobNo").getAsString())){
  396. userInfoMap.put(map.get("id").toString(), object.get("personId").getAsString());
  397. }
  398. }
  399. }
  400. System.out.println(userInfoMap);
  401. }
  402. void selecAllPerson(Db db) throws SQLException {
  403. String apiPath = "/api/resource/v2/person/personList";
  404. JsonObject jsonObject = new JsonObject();
  405. jsonObject.addProperty("pageNo", 1);
  406. jsonObject.addProperty("pageSize", 500);
  407. ApiUtil apiUtil = new ApiUtil();
  408. String result = apiUtil.doPost(apiPath, jsonObject.toString(), null, null);
  409. System.out.println(result);
  410. String sql = "SELECT t1.* FROM xjr_user t1" +
  411. " INNER JOIN base_teacher t2 ON t1.id = t2.user_id" +
  412. " WHERE t1.delete_mark = 0 AND t2.delete_mark = 0";
  413. List<XjrUser> dataList = db.query(sql, XjrUser.class);
  414. Map<String, Long> phoneMap = new HashMap<>();
  415. for (XjrUser xjrUser : dataList) {
  416. phoneMap.put(xjrUser.getMobile(), xjrUser.getId());
  417. }
  418. JsonParser jsonParser = new JsonParser();
  419. JsonObject resultJson = jsonParser.parse(result).getAsJsonObject();
  420. JsonArray array = resultJson.get("data").getAsJsonObject().get("list").getAsJsonArray();
  421. Map<Long, String> idMap = new HashMap<>();
  422. for (JsonElement jsonElement : array) {
  423. JsonObject element = jsonElement.getAsJsonObject();
  424. if(!element.has("phoneNo") || element.get("phoneNo").isJsonNull()){
  425. System.out.println(element.toString());
  426. continue;
  427. }
  428. String phoneNo = element.get("phoneNo").getAsString();
  429. idMap.put(phoneMap.get(phoneNo), element.get("personId").getAsString());
  430. }
  431. insertRecord(db, "base_teacher", idMap);
  432. }
  433. void selectOrg(Db db, String tableName) throws SQLException {
  434. String sql = "SELECT * FROM " + tableName + " WHERE delete_mark = 0";
  435. List<Department> dataList = db.query(sql, Department.class);
  436. String apiPath = "/api/resource/v1/org/orgList";
  437. JsonObject jsonObject = new JsonObject();
  438. jsonObject.addProperty("pageNo", 1);
  439. jsonObject.addProperty("pageSize", 500);
  440. ApiUtil apiUtil = new ApiUtil();
  441. String result = apiUtil.doPost(apiPath, jsonObject.toString(), null);
  442. System.out.println("-----------------------------------------------------");
  443. System.out.println(result);
  444. }
  445. @Test
  446. void deleteOrg(Db db, String tableName) throws SQLException {
  447. String sql = "SELECT * FROM " + tableName + " WHERE delete_mark = 0 ORDER BY LENGTH(hierarchy)";
  448. List<Department> dataList = db.query(sql, Department.class);
  449. String apiPath = "/api/resource/v1/org/batch/delete";
  450. Map<Long, String> idMap = new HashMap<>();
  451. JsonParser jsonParser = new JsonParser();
  452. ApiUtil apiUtil = new ApiUtil();
  453. JsonArray dataArray = new JsonArray();
  454. for (Department department : dataList) {
  455. dataArray.add(department.getId().toString());
  456. }
  457. JsonObject paramJson = new JsonObject();
  458. paramJson.add("indexCodes", dataArray);
  459. Map<String, String> headerMap = new HashMap<>();
  460. headerMap.put("tagId", "deleteOrg");
  461. String result = apiUtil.doPost(apiPath, paramJson.toString(), null, headerMap);
  462. System.out.println(result);
  463. }
  464. public Map<String, String> insertTeacher(String tableName, Map<String, String> tableData, Map<String, String> deptMap){
  465. String sql = "SELECT distinct t1.id,t1.name,ifnull(t1.gender,'SB10001') as gender,t3.dept_id,t1.birth_date,t1.mobile,t1.email,t1.credential_type,t1.credential_number,t1.user_name FROM xjr_user t1" +
  466. " INNER JOIN " + tableName + " t2 ON t1.id = t2.user_id" +
  467. " INNER JOIN xjr_user_dept_relation t3 ON t1.id = t3.user_id" +
  468. " INNER JOIN xjr_department t4 ON t3.dept_id = t4.id " +
  469. " WHERE t1.delete_mark = 0 AND t2.delete_mark = 0 AND t4.delete_mark = 0" +
  470. " AND t1.name = '徐海洋'";
  471. List<Map<String, Object>> list = SqlRunnerAdapter.db().selectList(sql, String[].class);
  472. JsonArray paramArray = new JsonArray();
  473. Map<Integer, String> clientMap = new HashMap<>();
  474. for(int i = 0; i < list.size(); i ++){
  475. Map<String, Object> objectMap = list.get(i);
  476. if(tableData != null && tableData.containsKey(objectMap.get("id").toString())){
  477. //continue;
  478. }
  479. JsonObject paramJson = new JsonObject();
  480. paramJson.addProperty("clientId", i);
  481. paramJson.addProperty("personId", objectMap.get("id").toString() + i);
  482. paramJson.addProperty("personName", objectMap.get("name").toString());
  483. paramJson.addProperty("orgIndexCode", objectMap.get("dept_id").toString());
  484. paramJson.addProperty("phoneNo", objectMap.get("mobile")==null?"":objectMap.get("mobile").toString());
  485. paramJson.addProperty("jobNo", objectMap.get("user_name").toString());
  486. paramJson.addProperty("birthday", objectMap.get("birth_date")==null?"":objectMap.get("birth_date").toString());
  487. paramJson.addProperty("personType", 1);
  488. int gender = 1;
  489. switch (objectMap.get("gender").toString()) {
  490. case "SB10001":
  491. gender = 1;
  492. break;
  493. case "SB10002":
  494. gender = 2;
  495. break;
  496. default:
  497. gender = 0;
  498. break;
  499. }
  500. paramJson.addProperty("gender", gender);
  501. paramArray.add(paramJson);
  502. clientMap.put(i, objectMap.get("id").toString());
  503. }
  504. String apiPath = "/api/resource/v1/person/batch/add";
  505. Map<String, String> idMap = new HashMap<>();
  506. JsonParser jsonParser = new JsonParser();
  507. ApiUtil apiUtil = new ApiUtil();
  508. Map<String, String> header = new HashMap<>();
  509. header.put("tagId", "insert_teacher");
  510. String result = apiUtil.doPost(apiPath, paramArray.toString(), null, header);
  511. JsonElement parse = jsonParser.parse(result);
  512. JsonObject resultJson = parse.getAsJsonObject();
  513. if("0".equals(resultJson.get("code").getAsString()) && "success".equals(resultJson.get("msg").getAsString())){
  514. JsonArray array = resultJson.get("data").getAsJsonObject().get("successes").getAsJsonArray();
  515. for (JsonElement jsonElement : array) {
  516. JsonObject jsonObject = jsonElement.getAsJsonObject();
  517. idMap.put(clientMap.get(jsonObject.get("clientId").getAsInt()), jsonObject.get("personId").getAsString());
  518. }
  519. }else{
  520. }
  521. //插入记录表
  522. insertRecord(tableName, idMap);
  523. return idMap;
  524. }
  525. void insertRecord(String tableName, Map<?, String> idsMap){
  526. if (idsMap.isEmpty()) {
  527. return;
  528. }
  529. for (var entry : idsMap.entrySet()) {
  530. var sourceId = entry.getKey();
  531. String sql = "INSERT INTO hikvision_data(create_date, table_name, source_id, hikvision_id) " +
  532. "VALUES(now(), '" + tableName + "', '" + sourceId + "', '" + entry.getValue() + "')";
  533. SqlRunnerAdapter.db().insert(sql);
  534. }
  535. }
  536. @Test
  537. void deletePerson() throws SQLException {
  538. String sql = "SELECT * FROM xjr_user WHERE id IN (\n" +
  539. "SELECT source_id FROM hikvision_data WHERE table_name IN( 'base_teacher','base_student')\n" +
  540. "AND hikvision_id LIKE '%-%'\n" +
  541. "AND hikvision_id NOT IN('14954805813957-14','14954812448325-19','14954812339909-17','14954812392774-18')\n" +
  542. ") AND id NOT IN ('14954807016389', '14954806910277')\n";
  543. List<Map<String, Object>> maps = SqlRunnerAdapter.db().selectList(sql);
  544. String apiPath = "/api/resource/v1/person/batch/delete";
  545. JsonObject paramJson = new JsonObject();
  546. JsonArray personIds = new JsonArray();
  547. for (Map<String, Object> map : maps) {
  548. personIds.add(map.get("id").toString());
  549. }
  550. paramJson.add("personIds", personIds);
  551. ApiUtil apiUtil = new ApiUtil();
  552. Map<String, String> header = new HashMap<>();
  553. header.put("tagId", "delete_teacher");
  554. String result = apiUtil.doPost(apiPath, paramJson.toString(), null, header);
  555. System.out.println(result);
  556. }
  557. /**
  558. * 同步海康学生信息到hikvision_data表中
  559. */
  560. @Test
  561. void syncStudentId(){
  562. String sql = "SELECT t1.* FROM hikvision_data t1" +
  563. " INNER JOIN base_class t2 ON t1.source_id LIKE CONCAT(t2.id, '%')" +
  564. " WHERE t1.table_name = 'base_class'" +
  565. " AND t2.grade_id = 1868532405673439232";
  566. List<Map<String, Object>> list = SqlRunnerAdapter.db().selectList(sql);
  567. List<String> orgIndexCodes = new ArrayList<>();
  568. for (int i = 0; i < list.size(); i ++){
  569. Map<String, Object> dataMap = list.get(i);
  570. orgIndexCodes.add(dataMap.get("hikvision_id").toString());
  571. }
  572. JsonParser jsonParser = new JsonParser();
  573. ApiUtil apiUtil = new ApiUtil();
  574. String apiPath = "/api/resource/v2/person/advance/personList";
  575. Set<String> personIds = new HashSet<>();
  576. for (String orgIndexCode : orgIndexCodes) {
  577. JsonObject jsonObject = new JsonObject();
  578. jsonObject.addProperty("pageNo", 1);
  579. jsonObject.addProperty("pageSize", 500);
  580. jsonObject.addProperty("orgIndexCodes", orgIndexCode);
  581. String result = apiUtil.doPost(apiPath, jsonObject.toString(), null, null);
  582. JsonObject resultJson = jsonParser.parse(result).getAsJsonObject();
  583. JsonArray array = resultJson.get("data").getAsJsonObject().get("list").getAsJsonArray();
  584. for (JsonElement jsonElement : array) {
  585. JsonObject element = jsonElement.getAsJsonObject();
  586. String personId = element.get("personId").getAsString();
  587. personIds.add(personId);
  588. }
  589. }
  590. for (String personId : personIds) {
  591. String insertsql = "INSERT INTO hikvision_data(create_date,TABLE_NAME,source_id,hikvision_id) " +
  592. " SELECT NOW(),'base_student','" + personId + "','" + personId + "'" +
  593. " where not EXISTS (" +
  594. " SELECT 1 FROM hikvision_data " +
  595. " WHERE source_id = '" + personId + "'" +
  596. " );";
  597. SqlRunnerAdapter.db().insert(insertsql);
  598. }
  599. }
  600. @Test
  601. void selecAllStudent() throws SQLException {
  602. ApiUtil apiUtil = new ApiUtil();
  603. String apiPath = "/api/resource/v2/person/orgIndexCode/personList";
  604. String sql = "SELECT CONCAT(t1.id, '-', REPLACE(t2.title,'年',''), '-', t1.class_type,'-FB3001') as id FROM base_class t1" +
  605. " INNER JOIN base_grade t2 ON t1.grade_id = t2.id" +
  606. " WHERE t1.delete_mark = 0 AND t1.is_graduate = 1 and t2.id = 1868532405673439232";
  607. List<Map<String, Object>> list = SqlRunnerAdapter.db().selectList(sql);
  608. for (Map<String, Object> objectMap : list) {
  609. JsonObject jsonObject = new JsonObject();
  610. jsonObject.addProperty("pageNo", 1);
  611. jsonObject.addProperty("pageSize", 500);
  612. jsonObject.addProperty("orgIndexCode", objectMap.get("id").toString());
  613. String result = apiUtil.doPost(apiPath, jsonObject.toString(), null, null);
  614. JsonParser jsonParser = new JsonParser();
  615. JsonObject resultJson = jsonParser.parse(result).getAsJsonObject();
  616. JsonArray array = resultJson.get("data").getAsJsonObject().get("list").getAsJsonArray();
  617. Map<Long, String> idMap = new HashMap<>();
  618. for (JsonElement jsonElement : array) {
  619. JsonObject element = jsonElement.getAsJsonObject();
  620. if(!element.has("phoneNo") || element.get("phoneNo").isJsonNull()){
  621. System.out.println(element.toString());
  622. continue;
  623. }
  624. String personId = element.get("personId").getAsString();
  625. String insSql = "INSERT INTO hikvision_data(create_date,TABLE_NAME,source_id,hikvision_id) " +
  626. "SELECT NOW(),'base_student', '" + personId + "','" + personId + "' WHERE NOT EXISTS (SELECT 1 FROM hikvision_data WHERE hikvision_id = '" + personId + "')";
  627. SqlRunnerAdapter.db().insert(insSql);
  628. }
  629. }
  630. sql = "SELECT CONCAT(t1.id, '-', REPLACE(t2.title,'年',''), '-', t1.class_type,'-FB3003') as id FROM base_class t1" +
  631. " INNER JOIN base_grade t2 ON t1.grade_id = t2.id" +
  632. " WHERE t1.delete_mark = 0 AND t1.is_graduate = 1 and t2.id = 1868532405673439232";
  633. list = SqlRunnerAdapter.db().selectList(sql);
  634. for (Map<String, Object> objectMap : list) {
  635. JsonObject jsonObject = new JsonObject();
  636. jsonObject.addProperty("pageNo", 1);
  637. jsonObject.addProperty("pageSize", 500);
  638. jsonObject.addProperty("orgIndexCode", objectMap.get("id").toString());
  639. String result = apiUtil.doPost(apiPath, jsonObject.toString(), null, null);
  640. JsonParser jsonParser = new JsonParser();
  641. JsonObject resultJson = jsonParser.parse(result).getAsJsonObject();
  642. JsonArray array = resultJson.get("data").getAsJsonObject().get("list").getAsJsonArray();
  643. Map<Long, String> idMap = new HashMap<>();
  644. for (JsonElement jsonElement : array) {
  645. JsonObject element = jsonElement.getAsJsonObject();
  646. if(!element.has("phoneNo") || element.get("phoneNo").isJsonNull()){
  647. System.out.println(element.toString());
  648. continue;
  649. }
  650. String personId = element.get("personId").getAsString();
  651. String insSql = "INSERT INTO hikvision_data(create_date,TABLE_NAME,source_id,hikvision_id) " +
  652. "SELECT NOW(),'base_student', '" + personId + "','" + personId + "' WHERE NOT EXISTS (SELECT 1 FROM teacher_out_in_record WHERE hikvision_id = '" + personId + "')";
  653. SqlRunnerAdapter.db().insert(insSql);
  654. }
  655. }
  656. }
  657. }