|
@@ -1,6 +1,7 @@
|
|
|
package cn.iselab.mooctest.site.service.impl;
|
|
|
|
|
|
import cn.iselab.mooctest.site.configure.MongoDBConfiguration;
|
|
|
+import cn.iselab.mooctest.site.data.PageData;
|
|
|
import cn.iselab.mooctest.site.service.CaughtNodeService;
|
|
|
import cn.iselab.mooctest.site.web.data.forMongo.NodeCatch.CaughtNodeDTO;
|
|
|
import cn.iselab.mooctest.site.web.util.mongodb.MongoAPIUtils;
|
|
@@ -85,6 +86,28 @@ public class CaughtNodeServiceImpl implements CaughtNodeService {
|
|
|
return oldcnDTO;
|
|
|
}
|
|
|
|
|
|
+ @Override
|
|
|
+ public Long getAllCaughtNodeNumbers() {
|
|
|
+ HttpHeaders headers = MongoAPIUtils.createAuthHeaderForMongo();
|
|
|
+ HttpEntity<String> entity = new HttpEntity<>(headers);
|
|
|
+
|
|
|
+ RestTemplate rt = new RestTemplate();
|
|
|
+
|
|
|
+ String url = MongoAPIUtils.generatePagingUrl(mongoDBConfig.getDb(),
|
|
|
+ mongoDBConfig.getCaughtNodeCollection());
|
|
|
+
|
|
|
+ String sortBy = "DESC";
|
|
|
+ ResponseEntity<JSONObject> dto = rt.exchange(url, HttpMethod.GET, entity, JSONObject.class,
|
|
|
+ activePage, pageSize, sortBy);
|
|
|
+
|
|
|
+ if (dto.getBody().getJSONObject("_embedded").isNullObject() || dto.getBody().getJSONObject("_embedded").isEmpty()){
|
|
|
+ return 0L;
|
|
|
+ }
|
|
|
+ PageData<CaughtNodeDTO> pageData = new PageData<>(dto.getBody(), activePage);
|
|
|
+ long totalElements = pageData.getTotalElements();
|
|
|
+ return totalElements;
|
|
|
+ }
|
|
|
+
|
|
|
private CaughtNodeDTO extractCaughtNodeData(ResponseEntity<JSONObject> dto) {
|
|
|
JSONArray resultArray = dto.getBody().getJSONArray("_embedded");
|
|
|
if (resultArray.size() == 0) {
|