HikvisionBaseDataTaskTest.java 29 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667
  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 selectOrg(){
  70. String apiPath = "/api/resource/v1/org/orgList";
  71. ApiUtil apiUtil = new ApiUtil();
  72. JsonObject jsonObject = new JsonObject();
  73. jsonObject.addProperty("pageNo", 1);
  74. jsonObject.addProperty("pageSize", 500);
  75. String doPost = apiUtil.doPost(apiPath, jsonObject.toString(), null);
  76. System.out.println(doPost);
  77. String sql = "SELECT * FROM hikvision_data WHERE 1 = 1 and table_name = 'base_class'";
  78. List<Map<String, Object>> maps = SqlRunnerAdapter.db().selectList(sql);
  79. List<String> sourceIds = new ArrayList<>();
  80. for (Map<String, Object> map : maps) {
  81. sourceIds.add(map.get("source_id").toString());
  82. }
  83. JsonParser jsonParser = new JsonParser();
  84. JsonElement parse = jsonParser.parse(doPost);
  85. JsonObject resultJson = parse.getAsJsonObject();
  86. String tableName = "hikvision_data";
  87. SimpleDateFormat sdf = new SimpleDateFormat("yyy-MM-dd HH:mm:ss");
  88. if("0".equals(resultJson.get("code").getAsString()) && "success".equals(resultJson.get("msg").getAsString())){
  89. JsonArray array = resultJson.get("data").getAsJsonObject().get("list").getAsJsonArray();
  90. for (JsonElement jsonElement : array) {
  91. JsonObject json = jsonElement.getAsJsonObject();
  92. String orgIndexCode = json.get("orgIndexCode").getAsString();
  93. if(sourceIds.contains(orgIndexCode)){
  94. continue;
  95. }
  96. Entity entity = Entity.create(tableName);
  97. entity.set("source_id", orgIndexCode);
  98. entity.set("table_name", "base_class");
  99. entity.set("hikvision_id", orgIndexCode);
  100. entity.set("create_date", sdf.format(new Date()));
  101. SqlRunnerAdapter.db().dynamicInsert(tableName, entity);
  102. }
  103. }
  104. }
  105. @Test
  106. void test() throws Exception {
  107. String sql = "SELECT distinct table_name FROM hikvision_data WHERE 1 = 1";
  108. List<Map<String, Object>> maps = SqlRunnerAdapter.db().selectList(sql, HikvisionData.class);
  109. Set<String> tables = new HashSet<>();
  110. for (Map<String, Object> map : maps) {
  111. String[] strings = SqlRunnerAdapterUtil.convertMapToStringArray(map);
  112. tables.add(strings[0]);
  113. }
  114. sql = "SELECT * FROM hikvision_data WHERE 0 = 0";
  115. List<Map<String, Object>> list = SqlRunnerAdapter.db().selectList(sql, HikvisionData.class);
  116. Map<String, Map<String, String>> dataMap = new HashMap<>();
  117. for (String table : tables) {
  118. Map<String, String> tableData = new HashMap<>();
  119. for (Map<String, Object> hikvision : list) {
  120. if(!table.equals(hikvision.get("table_name").toString())){
  121. continue;
  122. }
  123. tableData.put(hikvision.get("source_id").toString(), hikvision.get("hikvision_id").toString());
  124. }
  125. dataMap.put(table, tableData);
  126. }
  127. DataUtil dataUtil = new DataUtil();
  128. String tableName = "xjr_department";
  129. Map<String, String> department = dataMap.get(tableName);//dataUtil.insertDepartment(use, tableName, dataMap.get(tableName));
  130. if(department.isEmpty() && dataMap.get(tableName) != null){
  131. department = dataMap.get(tableName);
  132. }
  133. //
  134. // //推送教职工
  135. tableName = "base_teacher";
  136. // insertTeacher(tableName, dataMap.get(tableName), department);
  137. //
  138. //推送车辆
  139. // String carTableName = "car_message_apply";
  140. // Map<String, String> baseClass = dataMap.get(carTableName);
  141. // dataUtil.insertCar(use, carTableName,baseClass, faceImportMapper);
  142. // tableName = "base_class";
  143. // Map<String, String> baseClass = dataMap.get(tableName);
  144. // Map<String, String> classes = dataUtil.insertClass(tableName, baseClass);
  145. tableName = "base_student";
  146. List<HikvisionData> studentList = hikvisionDataMapper.selectList(
  147. new QueryWrapper<HikvisionData>().lambda().eq(HikvisionData::getTableName, tableName)
  148. );
  149. Map<String, String> baseStudent = dataMap.get(tableName);
  150. insertStudentOne(tableName, baseStudent, studentList);
  151. // sql = "SELECT * FROM wf_room_applicant WHERE delete_mark = 0 AND STATUS = 1";
  152. // List<Map<String, Object>> wfRoomApplicantList = SqlRunnerAdapter.db().selectList(sql);
  153. //
  154. // Map<String, String> baseStudent = dataMap.get(tableName);
  155. // for (Map<String, Object> objectMap : wfRoomApplicantList) {
  156. // Long applicantUserId = Long.parseLong(objectMap.get("applicant_user_id").toString());
  157. // Map<String, String> student = dataUtil.insertStudentOne(tableName, baseStudent, studentList, applicantUserId);
  158. // }
  159. // selectOrg(use, "base_class");
  160. //查询车辆
  161. // String carTableName = "car_message_apply";
  162. // selectCar(use, carTableName);
  163. // outInRecordUtil.GetVehicleRecord(faceImportMapper);
  164. // outInRecordUtil.GetTeacherAndStudentRecords(faceImportMapper);
  165. //// selecAllPersonById(use);
  166. // selectResource(use);
  167. }
  168. @Test
  169. void selectPerson(){
  170. String sql = "SELECT * FROM xjr_user WHERE id IN (\n" +
  171. "SELECT source_id FROM hikvision_data WHERE hikvision_id LIKE '%-%' AND table_name = 'base_teacher'\n" +
  172. ")";
  173. List<Map<String, Object>> list = SqlRunnerAdapter.db().selectList(sql, String[].class);
  174. }
  175. @Test
  176. void testCarInsert() throws Exception {
  177. //outInRecordUtil.GetVehicleRecord(faceImportMapper);
  178. }
  179. public static void selectCar(Db db, String tableName) throws SQLException {
  180. String sql = "SELECT * FROM " + tableName + " WHERE delete_mark = 0";
  181. List<Department> dataList = db.query(sql, Department.class);
  182. String apiPath = "/api/resource/v2/vehicle/advance/vehicleList";
  183. JsonObject jsonObject = new JsonObject();
  184. jsonObject.addProperty("pageNo", 1);
  185. jsonObject.addProperty("pageSize", 500);
  186. ApiUtil apiUtil = new ApiUtil();
  187. String result = apiUtil.doPost(apiPath, jsonObject.toString(), null);
  188. System.out.println("-----------------------------------------------------");
  189. System.out.println(result);
  190. }
  191. public static void insertRecord(Db db, String tableName, Map<?, String> idsMap) throws SQLException {
  192. if (idsMap.isEmpty()) {
  193. return;
  194. }
  195. var sqls = new ArrayList<String>();
  196. for (var entry : idsMap.entrySet()) {
  197. var sourceId = entry.getKey();
  198. var sql = "INSERT INTO hikvision_data(create_date, table_name, source_id, hikvision_id) " +
  199. "VALUES(now(), '" + tableName + "', '" + sourceId + "', '" + entry.getValue() + "')";
  200. sqls.add(sql);
  201. }
  202. db.executeBatch(sqls);
  203. }
  204. public Map<String, String> insertStudentOne(String tableName, Map<String, String> tableData, List<HikvisionData> studentList) throws Exception {
  205. String sql = "SELECT t1.id,t1.name,CONCAT(t3.id,'-',replace(t4.name,'级',''),'-',t3.class_type,'-',t2.stduy_status) as orgIndexCode," +
  206. " 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" +
  207. " INNER JOIN base_student_school_roll t2 ON t1.id = t2.user_id" +
  208. " INNER JOIN base_class t3 ON t2.class_id = t3.id" +
  209. " INNER JOIN base_grade t4 ON t3.grade_id = t4.id" +
  210. " WHERE t1.delete_mark = 0 AND t2.delete_mark = 0" +
  211. " AND t3.class_type IS NOT NULL AND t3.is_graduate = 1";
  212. List<Map<String, Object>> list = SqlRunnerAdapter.db().selectList(sql, String[].class);
  213. Map<Integer, String> clientMap = new HashMap<>();
  214. List<Entity> insertList = new ArrayList<>();
  215. Map<String, String> historyMap = new HashMap<>();
  216. for (HikvisionData hikvisionData : studentList) {
  217. historyMap.put(hikvisionData.getSourceId(), hikvisionData.getSendData());
  218. }
  219. Map<String, String> idMap = new HashMap<>();
  220. JsonParser jsonParser = new JsonParser();
  221. ApiUtil apiUtil = new ApiUtil();
  222. Map<String, String> header = new HashMap<>();
  223. header.put("tagId", "insert_student");
  224. List<JsonObject> addList = new ArrayList<>();
  225. List<JsonObject> updateList = new ArrayList<>();
  226. for(int i = 0; i < list.size(); i ++){
  227. Map<String, Object> objectMap = list.get(i);
  228. String id = objectMap.get("id").toString();
  229. JsonObject paramJson = new JsonObject();
  230. paramJson.addProperty("clientId", i);
  231. paramJson.addProperty("personId", id);
  232. paramJson.addProperty("personName", objectMap.get("name").toString());
  233. paramJson.addProperty("orgIndexCode", objectMap.get("orgIndexCode").toString());
  234. paramJson.addProperty("phoneNo", objectMap.get("mobile")==null?"":objectMap.get("mobile").toString());
  235. paramJson.addProperty("jobNo", objectMap.get("user_name").toString());
  236. paramJson.addProperty("birthday", objectMap.get("birthday")==null?"":objectMap.get("birthday").toString());
  237. paramJson.addProperty("personType", 2);
  238. int gender = 0;
  239. String genderData = objectMap.get("gender") == null ? "" : objectMap.get("gender").toString();
  240. switch (genderData) {
  241. case "SB10001":
  242. gender = 1;
  243. break;
  244. case "SB10002":
  245. gender = 2;
  246. break;
  247. default:
  248. gender = 0;
  249. break;
  250. }
  251. paramJson.addProperty("gender", gender);
  252. if(historyMap.containsKey(id)){
  253. if(historyMap.get(id) ==null || !paramJson.toString().equals(historyMap.get(id))){
  254. updateList.add(paramJson);
  255. }
  256. continue;
  257. }
  258. addList.add(paramJson);
  259. }
  260. String apiPath = "/api/resource/v1/person/single/add";
  261. SimpleDateFormat sdf = new SimpleDateFormat("yyy-MM-dd HH:mm:ss");
  262. for (JsonObject paramJson : addList) {
  263. String result = apiUtil.doPost(apiPath, paramJson.toString(), null, header);
  264. JsonElement parse = jsonParser.parse(result);
  265. JsonObject resultJson = parse.getAsJsonObject();
  266. if("0".equals(resultJson.get("code").getAsString()) && "success".equals(resultJson.get("msg").getAsString())){
  267. Entity entity = Entity.create(tableName);
  268. entity.set("table_name", tableName);
  269. entity.set("create_date", sdf.format(new Date()));
  270. entity.set("source_id", clientMap.get(paramJson.get("clientId").getAsInt()));
  271. entity.set("hikvision_id", paramJson.get("personId").getAsString());
  272. entity.set("send_data", paramJson.toString());
  273. insertList.add(entity);
  274. }else if("0x00052102".equals(resultJson.get("code").getAsString()) && "PersonId Already In Db".equals(resultJson.get("msg").getAsString())){
  275. Entity entity = Entity.create(tableName);
  276. entity.set("table_name", tableName);
  277. entity.set("create_date", sdf.format(new Date()));
  278. entity.set("source_id", paramJson.get("personId").getAsString());
  279. entity.set("hikvision_id", paramJson.get("personId").getAsString());
  280. entity.set("send_data", paramJson.toString());
  281. insertList.add(entity);
  282. }
  283. }
  284. tableName = "hikvision_data";
  285. if(!insertList.isEmpty()){
  286. SqlRunnerAdapter.db().dynamicInsertBatch(tableName, insertList);
  287. }
  288. apiPath = "/api/resource/v1/person/single/update";
  289. for (JsonObject paramJson : updateList) {
  290. Thread.sleep(500);
  291. String result = apiUtil.doPost(apiPath, paramJson.toString(), null, header);
  292. JsonElement parse = jsonParser.parse(result);
  293. JsonObject resultJson = parse.getAsJsonObject();
  294. if("0".equals(resultJson.get("code").getAsString()) && "success".equals(resultJson.get("msg").getAsString())){
  295. Entity entity = Entity.create(tableName);
  296. entity.set("send_data", paramJson.toString());
  297. Entity where = Entity.create(tableName);
  298. where.set("source_id", paramJson.get("personId").getAsString());
  299. SqlRunnerAdapter.db().dynamicUpdate(tableName, entity, where);
  300. }
  301. }
  302. //插入记录表
  303. return idMap;
  304. }
  305. @Test
  306. void deleteClass() {
  307. String apiPath = "/api/resource/v1/org/batch/delete";
  308. JsonObject jsonObject = new JsonObject();
  309. ApiUtil apiUtil = new ApiUtil();
  310. JsonArray dataArray = new JsonArray();
  311. dataArray.add("1828723343029219330-2024-job_class-FB3001");
  312. dataArray.add("1828723343029219330-2024-job_class-FB3002");
  313. JsonObject paramJson = new JsonObject();
  314. paramJson.add("indexCodes", dataArray);
  315. Map<String, String> headerMap = new HashMap<>();
  316. headerMap.put("tagId", "deleteOrg");
  317. String result = apiUtil.doPost(apiPath, paramJson.toString(), null, headerMap);
  318. System.out.println(result);
  319. }
  320. /**
  321. * 查询资源信息
  322. * @param db
  323. * @throws SQLException
  324. */
  325. void selectResource(Db db) throws SQLException {
  326. String apiPath = "/api/irds/v2/resource/resourcesByParams";
  327. JsonObject jsonObject = new JsonObject();
  328. jsonObject.addProperty("pageNo", 1);
  329. jsonObject.addProperty("pageSize", 500);
  330. jsonObject.addProperty("resourceType", "door");
  331. ApiUtil apiUtil = new ApiUtil();
  332. String result = apiUtil.doPost(apiPath, jsonObject.toString(), null, null);
  333. System.out.println("------------------------------------------------------------------------");
  334. System.out.println(result);
  335. }
  336. @Test
  337. void selecAllPersonById() throws SQLException {
  338. String apiPath = "/api/resource/v2/person/advance/personList";
  339. String sql = "SELECT t1.id, t1.name,t1.user_name,t2.dept_id FROM xjr_user t1\n" +
  340. "INNER JOIN xjr_user_dept_relation t2 ON t1.id = t2.user_id\n" +
  341. " WHERE t1.id IN (\n" +
  342. "SELECT source_id FROM hikvision_data WHERE hikvision_id LIKE '%-%' AND table_name = 'base_teacher'\n" +
  343. ")";
  344. List<Map<String, Object>> maps = SqlRunnerAdapter.db().selectList(sql);
  345. ApiUtil apiUtil = new ApiUtil();
  346. Map<String, String> userInfoMap = new HashMap<>();
  347. JsonParser parser = new JsonParser();
  348. for (Map<String, Object> map : maps) {
  349. String name = map.get("name").toString();
  350. JsonObject jsonObject = new JsonObject();
  351. jsonObject.addProperty("pageNo", 1);
  352. jsonObject.addProperty("pageSize", 500);
  353. jsonObject.addProperty("personName", name);
  354. jsonObject.addProperty("orgIndexCodes", map.get("dept_id").toString());
  355. String result = apiUtil.doPost(apiPath, jsonObject.toString(), null, null);
  356. JsonObject asJsonObject = parser.parse(result).getAsJsonObject();
  357. if("0".equals(asJsonObject.get("code").getAsString())){
  358. JsonArray jsonArray = asJsonObject.get("data").getAsJsonObject().get("list").getAsJsonArray();
  359. if(jsonArray.size() == 0){
  360. continue;
  361. }
  362. JsonObject object = jsonArray.get(0).getAsJsonObject();
  363. if(name.equals(object.get("personName").getAsString()) && map.get("user_name").toString().equals(object.get("jobNo").getAsString())){
  364. userInfoMap.put(map.get("id").toString(), object.get("personId").getAsString());
  365. }
  366. }
  367. }
  368. System.out.println(userInfoMap);
  369. }
  370. void selecAllPerson(Db db) throws SQLException {
  371. String apiPath = "/api/resource/v2/person/personList";
  372. JsonObject jsonObject = new JsonObject();
  373. jsonObject.addProperty("pageNo", 1);
  374. jsonObject.addProperty("pageSize", 500);
  375. ApiUtil apiUtil = new ApiUtil();
  376. String result = apiUtil.doPost(apiPath, jsonObject.toString(), null, null);
  377. System.out.println(result);
  378. String sql = "SELECT t1.* FROM xjr_user t1" +
  379. " INNER JOIN base_teacher t2 ON t1.id = t2.user_id" +
  380. " WHERE t1.delete_mark = 0 AND t2.delete_mark = 0";
  381. List<XjrUser> dataList = db.query(sql, XjrUser.class);
  382. Map<String, Long> phoneMap = new HashMap<>();
  383. for (XjrUser xjrUser : dataList) {
  384. phoneMap.put(xjrUser.getMobile(), xjrUser.getId());
  385. }
  386. JsonParser jsonParser = new JsonParser();
  387. JsonObject resultJson = jsonParser.parse(result).getAsJsonObject();
  388. JsonArray array = resultJson.get("data").getAsJsonObject().get("list").getAsJsonArray();
  389. Map<Long, String> idMap = new HashMap<>();
  390. for (JsonElement jsonElement : array) {
  391. JsonObject element = jsonElement.getAsJsonObject();
  392. if(!element.has("phoneNo") || element.get("phoneNo").isJsonNull()){
  393. System.out.println(element.toString());
  394. continue;
  395. }
  396. String phoneNo = element.get("phoneNo").getAsString();
  397. idMap.put(phoneMap.get(phoneNo), element.get("personId").getAsString());
  398. }
  399. insertRecord(db, "base_teacher", idMap);
  400. }
  401. void selectOrg(Db db, String tableName) throws SQLException {
  402. String sql = "SELECT * FROM " + tableName + " WHERE delete_mark = 0";
  403. List<Department> dataList = db.query(sql, Department.class);
  404. String apiPath = "/api/resource/v1/org/orgList";
  405. JsonObject jsonObject = new JsonObject();
  406. jsonObject.addProperty("pageNo", 1);
  407. jsonObject.addProperty("pageSize", 500);
  408. ApiUtil apiUtil = new ApiUtil();
  409. String result = apiUtil.doPost(apiPath, jsonObject.toString(), null);
  410. System.out.println("-----------------------------------------------------");
  411. System.out.println(result);
  412. }
  413. @Test
  414. void deleteOrg(Db db, String tableName) throws SQLException {
  415. String sql = "SELECT * FROM " + tableName + " WHERE delete_mark = 0 ORDER BY LENGTH(hierarchy)";
  416. List<Department> dataList = db.query(sql, Department.class);
  417. String apiPath = "/api/resource/v1/org/batch/delete";
  418. Map<Long, String> idMap = new HashMap<>();
  419. JsonParser jsonParser = new JsonParser();
  420. ApiUtil apiUtil = new ApiUtil();
  421. JsonArray dataArray = new JsonArray();
  422. for (Department department : dataList) {
  423. dataArray.add(department.getId().toString());
  424. }
  425. JsonObject paramJson = new JsonObject();
  426. paramJson.add("indexCodes", dataArray);
  427. Map<String, String> headerMap = new HashMap<>();
  428. headerMap.put("tagId", "deleteOrg");
  429. String result = apiUtil.doPost(apiPath, paramJson.toString(), null, headerMap);
  430. System.out.println(result);
  431. }
  432. public Map<String, String> insertTeacher(String tableName, Map<String, String> tableData, Map<String, String> deptMap){
  433. 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" +
  434. " INNER JOIN " + tableName + " t2 ON t1.id = t2.user_id" +
  435. " INNER JOIN xjr_user_dept_relation t3 ON t1.id = t3.user_id" +
  436. " INNER JOIN xjr_department t4 ON t3.dept_id = t4.id " +
  437. " WHERE t1.delete_mark = 0 AND t2.delete_mark = 0 AND t4.delete_mark = 0" +
  438. " AND t1.name = '徐海洋'";
  439. List<Map<String, Object>> list = SqlRunnerAdapter.db().selectList(sql, String[].class);
  440. JsonArray paramArray = new JsonArray();
  441. Map<Integer, String> clientMap = new HashMap<>();
  442. for(int i = 0; i < list.size(); i ++){
  443. Map<String, Object> objectMap = list.get(i);
  444. if(tableData != null && tableData.containsKey(objectMap.get("id").toString())){
  445. //continue;
  446. }
  447. JsonObject paramJson = new JsonObject();
  448. paramJson.addProperty("clientId", i);
  449. paramJson.addProperty("personId", objectMap.get("id").toString() + i);
  450. paramJson.addProperty("personName", objectMap.get("name").toString());
  451. paramJson.addProperty("orgIndexCode", objectMap.get("dept_id").toString());
  452. paramJson.addProperty("phoneNo", objectMap.get("mobile")==null?"":objectMap.get("mobile").toString());
  453. paramJson.addProperty("jobNo", objectMap.get("user_name").toString());
  454. paramJson.addProperty("birthday", objectMap.get("birth_date")==null?"":objectMap.get("birth_date").toString());
  455. paramJson.addProperty("personType", 1);
  456. int gender = 1;
  457. switch (objectMap.get("gender").toString()) {
  458. case "SB10001":
  459. gender = 1;
  460. break;
  461. case "SB10002":
  462. gender = 2;
  463. break;
  464. default:
  465. gender = 0;
  466. break;
  467. }
  468. paramJson.addProperty("gender", gender);
  469. paramArray.add(paramJson);
  470. clientMap.put(i, objectMap.get("id").toString());
  471. }
  472. String apiPath = "/api/resource/v1/person/batch/add";
  473. Map<String, String> idMap = new HashMap<>();
  474. JsonParser jsonParser = new JsonParser();
  475. ApiUtil apiUtil = new ApiUtil();
  476. Map<String, String> header = new HashMap<>();
  477. header.put("tagId", "insert_teacher");
  478. String result = apiUtil.doPost(apiPath, paramArray.toString(), null, header);
  479. JsonElement parse = jsonParser.parse(result);
  480. JsonObject resultJson = parse.getAsJsonObject();
  481. if("0".equals(resultJson.get("code").getAsString()) && "success".equals(resultJson.get("msg").getAsString())){
  482. JsonArray array = resultJson.get("data").getAsJsonObject().get("successes").getAsJsonArray();
  483. for (JsonElement jsonElement : array) {
  484. JsonObject jsonObject = jsonElement.getAsJsonObject();
  485. idMap.put(clientMap.get(jsonObject.get("clientId").getAsInt()), jsonObject.get("personId").getAsString());
  486. }
  487. }else{
  488. }
  489. //插入记录表
  490. insertRecord(tableName, idMap);
  491. return idMap;
  492. }
  493. void insertRecord(String tableName, Map<?, String> idsMap){
  494. if (idsMap.isEmpty()) {
  495. return;
  496. }
  497. for (var entry : idsMap.entrySet()) {
  498. var sourceId = entry.getKey();
  499. String sql = "INSERT INTO hikvision_data(create_date, table_name, source_id, hikvision_id) " +
  500. "VALUES(now(), '" + tableName + "', '" + sourceId + "', '" + entry.getValue() + "')";
  501. SqlRunnerAdapter.db().insert(sql);
  502. }
  503. }
  504. @Test
  505. void deletePerson() throws SQLException {
  506. String sql = "SELECT * FROM xjr_user WHERE id IN (\n" +
  507. "SELECT source_id FROM hikvision_data WHERE table_name IN( 'base_teacher','base_student')\n" +
  508. "AND hikvision_id LIKE '%-%'\n" +
  509. "AND hikvision_id NOT IN('14954805813957-14','14954812448325-19','14954812339909-17','14954812392774-18')\n" +
  510. ") AND id NOT IN ('14954807016389', '14954806910277')\n";
  511. List<Map<String, Object>> maps = SqlRunnerAdapter.db().selectList(sql);
  512. String apiPath = "/api/resource/v1/person/batch/delete";
  513. JsonObject paramJson = new JsonObject();
  514. JsonArray personIds = new JsonArray();
  515. for (Map<String, Object> map : maps) {
  516. personIds.add(map.get("id").toString());
  517. }
  518. paramJson.add("personIds", personIds);
  519. ApiUtil apiUtil = new ApiUtil();
  520. Map<String, String> header = new HashMap<>();
  521. header.put("tagId", "delete_teacher");
  522. String result = apiUtil.doPost(apiPath, paramJson.toString(), null, header);
  523. System.out.println(result);
  524. }
  525. /**
  526. * 同步海康学生信息到hikvision_data表中
  527. */
  528. @Test
  529. void syncStudentId(){
  530. String sql = "SELECT t1.* FROM hikvision_data t1" +
  531. " INNER JOIN base_class t2 ON t1.source_id LIKE CONCAT(t2.id, '%')" +
  532. " WHERE t1.table_name = 'base_class'" +
  533. " AND t2.grade_id = 1868532405673439232";
  534. List<Map<String, Object>> list = SqlRunnerAdapter.db().selectList(sql);
  535. List<String> orgIndexCodes = new ArrayList<>();
  536. for (int i = 0; i < list.size(); i ++){
  537. Map<String, Object> dataMap = list.get(i);
  538. orgIndexCodes.add(dataMap.get("hikvision_id").toString());
  539. }
  540. JsonParser jsonParser = new JsonParser();
  541. ApiUtil apiUtil = new ApiUtil();
  542. String apiPath = "/api/resource/v2/person/advance/personList";
  543. Set<String> personIds = new HashSet<>();
  544. for (String orgIndexCode : orgIndexCodes) {
  545. JsonObject jsonObject = new JsonObject();
  546. jsonObject.addProperty("pageNo", 1);
  547. jsonObject.addProperty("pageSize", 500);
  548. jsonObject.addProperty("orgIndexCodes", orgIndexCode);
  549. String result = apiUtil.doPost(apiPath, jsonObject.toString(), null, null);
  550. JsonObject resultJson = jsonParser.parse(result).getAsJsonObject();
  551. JsonArray array = resultJson.get("data").getAsJsonObject().get("list").getAsJsonArray();
  552. for (JsonElement jsonElement : array) {
  553. JsonObject element = jsonElement.getAsJsonObject();
  554. String personId = element.get("personId").getAsString();
  555. personIds.add(personId);
  556. }
  557. }
  558. for (String personId : personIds) {
  559. String insertsql = "INSERT INTO hikvision_data(create_date,TABLE_NAME,source_id,hikvision_id) " +
  560. " SELECT NOW(),'base_student','" + personId + "','" + personId + "'" +
  561. " where not EXISTS (" +
  562. " SELECT 1 FROM hikvision_data " +
  563. " WHERE source_id = '" + personId + "'" +
  564. " );";
  565. SqlRunnerAdapter.db().insert(insertsql);
  566. }
  567. }
  568. }