1

src/main/java/com/fdkankan/ucenter/service/impl/SceneCooperationCountServiceImpl.java

@@ -50,10 +50,11 @@ public class SceneCooperationCountServiceImpl extends ServiceImpl<ISceneCooperat
         for (String num : addNumList) {
             if(!dbNumList.contains(num)){
                 List<User> users = userMap.get(num);
+                Integer totalSize = users == null ?size :users.size() + size;
                 SceneCooperationCount count = new SceneCooperationCount();
                 count.setSceneType(sceneType);
                 count.setNum(num);
-                count.setCount(users.size() + size);
+                count.setCount(totalSize );
                 saveList.add(count);
             }
         }
@@ -63,10 +64,11 @@ public class SceneCooperationCountServiceImpl extends ServiceImpl<ISceneCooperat
 
         for (SceneCooperationCount sceneCooperationCount : sceneCooperationCounts) {
             List<User> users = userMap.get(sceneCooperationCount.getNum());
-            if(sceneCooperationCount.getCount() < users.size() + size){
+            Integer totalSize = users == null ?size :users.size() + size;
+            if(sceneCooperationCount.getCount() < totalSize){
                 LambdaUpdateWrapper<SceneCooperationCount> wrapper = new LambdaUpdateWrapper<>();
                 wrapper.eq(SceneCooperationCount::getId,sceneCooperationCount.getId());
-                wrapper.set(SceneCooperationCount::getCount,users.size() + size);
+                wrapper.set(SceneCooperationCount::getCount,totalSize);
                 this.update(wrapper);
             }
         }