批量获取数据

server/src/main/java/com/pica/cloud/account/account/server/service/impl/PatHealthPackServiceIml.java

@@ -382,44 +382,56 @@ public class PatHealthPackServiceIml implements PatHealthPackService {
         AcctPatIdResp resp = new AcctPatIdResp();
         Map<Integer, Integer> patFamId = new HashMap<>();
 
-        if (acctPatIdReq != null && acctPatIdReq.getPatIds().size() > 1000) {
-            throw new AccountException(AccountExceptionEnum.PAT_QUERY_MAX);
+        if (acctPatIdReq == null) {
+            throw new AccountException(AccountExceptionEnum.PICA_PARAMS_ERROR);
         }
 
-        /**先获取是主账号本人的居民id*/
-        List<AcctPatFamily> acctMembers = patFamilyMapper.getListByPatIdSelfs(acctPatIdReq.getPatIds());
-        for (AcctPatFamily family : acctMembers) {
-            patFamId.put(family.getPatientId(), family.getPatientId());
-        }
-
-        /** 本人为监护人的patIds*/
-        List<Integer> acctMemberPatIds = acctMembers.stream().map(o -> o.getPatientId()).collect(toList());
-
-        /**作为成员的记录*/
-        List<Integer> memberIds = acctPatIdReq.getPatIds().stream().filter(reqPatId -> acctMemberPatIds.stream().noneMatch(patId -> reqPatId.equals(patId))).collect(toList());
+        /***批量处理*/
+        List<Integer> allPatIds = acctPatIdReq.getPatIds();
+        int once = 500;
+        for (int i = 0; i < allPatIds.size(); i += once) {
+            List<Integer> patIds = null;
+            if (allPatIds.size() < i + once) {
+                patIds = allPatIds.subList(i, allPatIds.size());
+            } else {
+                patIds = allPatIds.subList(i, i + once);
+            }
 
-        if (CollectionUtils.isNotEmpty(memberIds)) {
+            /**先获取是主账号本人的居民id*/
+            List<AcctPatFamily> acctMembers = patFamilyMapper.getListByPatIdSelfs(patIds);
+            for (AcctPatFamily family : acctMembers) {
+                patFamId.put(family.getPatientId(), family.getPatientId());
+            }
 
-            /**获取不是本人的主账号的本人居民id*/
-            List<AcctPatFamily> members = patFamilyMapper.getListByPatIdNotSelfs(memberIds);
-            if (CollectionUtils.isNotEmpty(members)) {
-                List<Integer> memberAcctIds = members.stream().map(o -> o.getAcctId()).collect(toList());
-                if (CollectionUtils.isNotEmpty(memberAcctIds)) {
-                    List<AcctPatFamily> acctPats = patFamilyMapper.getSelfListByAcctIds(memberAcctIds);
-                    for (AcctPatFamily family : acctPats) {
-                        for (AcctPatFamily member : members) {
-                            if (StringUtils.equals(String.valueOf(family.getAcctId()), String.valueOf(member.getAcctId()))) {
-                                patFamId.put(member.getPatientId(), family.getPatientId());
+            /** 本人为监护人的patIds*/
+            List<Integer> acctMemberPatIds = acctMembers.stream().map(o -> o.getPatientId()).collect(toList());
+
+            /**作为成员的记录*/
+            List<Integer> memberIds = patIds.stream().filter(reqPatId -> acctMemberPatIds.stream().noneMatch(patId -> reqPatId.equals(patId))).collect(toList());
+
+            if (CollectionUtils.isNotEmpty(memberIds)) {
+
+                /**获取不是本人的主账号的本人居民id*/
+                List<AcctPatFamily> members = patFamilyMapper.getListByPatIdNotSelfs(memberIds);
+                if (CollectionUtils.isNotEmpty(members)) {
+                    List<Integer> memberAcctIds = members.stream().map(o -> o.getAcctId()).collect(toList());
+                    if (CollectionUtils.isNotEmpty(memberAcctIds)) {
+                        List<AcctPatFamily> acctPats = patFamilyMapper.getSelfListByAcctIds(memberAcctIds);
+                        for (AcctPatFamily family : acctPats) {
+                            for (AcctPatFamily member : members) {
+                                if (StringUtils.equals(String.valueOf(family.getAcctId()), String.valueOf(member.getAcctId()))) {
+                                    patFamId.put(member.getPatientId(), family.getPatientId());
+                                }
                             }
                         }
                     }
-                }
 
-                /**没有主账号的patIds*/
-                memberIds.removeAll(members.stream().map(o -> o.getPatientId()).collect(toList()));
-                if (CollectionUtils.isNotEmpty(memberIds)) {
-                    for(Integer member : memberIds) {
-                        patFamId.put(member, member);
+                    /**没有主账号的patIds*/
+                    memberIds.removeAll(members.stream().map(o -> o.getPatientId()).collect(toList()));
+                    if (CollectionUtils.isNotEmpty(memberIds)) {
+                        for (Integer member : memberIds) {
+                            patFamId.put(member, member);
+                        }
                     }
                 }
             }