股東表插入、查詢

ace-modules/ace-admin/src/main/java/com/github/wxiaoqi/security/admin/biz/AppShareholderDetailBiz.java

@@ -9,11 +9,9 @@ import com.github.wxiaoqi.security.admin.entity.AppUserPosition;
 import com.github.wxiaoqi.security.admin.mapper.AppShareholderDetailMapper;
 import com.github.wxiaoqi.security.admin.vo.AppShareholderDetailVo;
 import com.github.wxiaoqi.security.common.biz.BaseBiz;
-import com.github.wxiaoqi.security.common.exception.BaseException;
 import com.github.wxiaoqi.security.common.msg.ObjectRestResponse;
 import com.github.wxiaoqi.security.common.util.process.ResultCode;
 import com.github.wxiaoqi.security.common.vo.PageDataVO;
-import com.xxfc.platform.im.utils.StringUtil;
 import com.xxfc.platform.vehicle.common.RestResponse;
 import com.xxfc.platform.vehicle.entity.BranchCompany;
 import com.xxfc.platform.vehicle.feign.VehicleFeign;
@@ -22,17 +20,14 @@ import org.apache.commons.collections.CollectionUtils;
 import org.apache.commons.lang3.StringUtils;
 import org.springframework.beans.BeanUtils;
 import org.springframework.beans.factory.annotation.Autowired;
-import org.springframework.data.mongodb.core.mapreduce.GroupBy;
 import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Propagation;
 import org.springframework.transaction.annotation.Transactional;
-import tk.mybatis.mapper.entity.Example;
 
 import java.text.ParseException;
 import java.text.SimpleDateFormat;
 import java.util.*;
-import java.util.concurrent.CountDownLatch;
 import java.util.stream.Collectors;
 
 
@@ -143,13 +138,8 @@ public class AppShareholderDetailBiz extends BaseBiz<AppShareholderDetailMapper,
         if (Objects.nonNull(appShareholderDetailFindDTO.getPositionId())) {
             positionId = appShareholderDetailFindDTO.getPositionId();
         }
-        Integer finalPositionId = positionId;
-        Integer finalCompanyID = companyID;
-        String finalPhone = phone;
-        String finalName = name;
-        PageDataVO<AppShareholderDetail> pageDataVO = PageDataVO.pageInfo(appShareholderDetailFindDTO.getPage(), appShareholderDetailFindDTO.getLimit(), () -> {
-            mapper.findPage(finalName, finalPhone, finalPositionId, finalCompanyID);
-        });
+
+        PageDataVO<AppShareholderDetail> pageDataVO = findPage(appShareholderDetailFindDTO, positionId, companyID, phone, name);
         List<AppShareholderDetail> data = pageDataVO.getData();
         if (CollectionUtils.isEmpty(data)) {
             dataVO.setData(Collections.EMPTY_LIST);
@@ -179,7 +169,7 @@ public class AppShareholderDetailBiz extends BaseBiz<AppShareholderDetailMapper,
                 }
             }
             if (flag) {
-                AppShareholderDetailMerge.add(appShareholderDetailVoOld);//给整合后集合添加子元素
+                appShareholderDetailMerge.add(appShareholderDetailVoOld);//给整合后集合添加子元素
             }
         }*/
         dataVO.setPageSize(pageDataVO.getPageSize());
@@ -190,13 +180,20 @@ public class AppShareholderDetailBiz extends BaseBiz<AppShareholderDetailMapper,
         return dataVO;
     }
 
+    private PageDataVO<AppShareholderDetail> findPage(AppShareholderDetailFindDTO appShareholderDetailFindDTO, Integer finalPositionId, Integer finalCompanyID, String finalPhone, String finalName) {
+        return PageDataVO.pageInfo(appShareholderDetailFindDTO.getPage(), appShareholderDetailFindDTO.getLimit(), () -> {
+            mapper.findPage(finalName, finalPhone, finalPositionId, finalCompanyID);
+        });
+    }
+
     /**
      * 批量导入
      */
     @Transactional(rollbackFor = Exception.class, propagation = Propagation.REQUIRED)
     public int importShareholder(List<String[]> shareholderData) {
         List<AppShareholderDetail> appShareholderDetail = new ArrayList<>();
-        wrapperData(shareholderData);
+        int size = wrapperData(shareholderData);
+
    /*int threadNums = shareholderData.size() / BORDER_NUM == 0 ? 1 : shareholderData.size() / BORDER_NUM;
         CountDownLatch latch = new CountDownLatch(threadNums);
         for (int i = 0; i < threadNums; i++) {
@@ -213,14 +210,15 @@ public class AppShareholderDetailBiz extends BaseBiz<AppShareholderDetailMapper,
                 throw new BaseException("导入数据失败");
             }
         }*/
-        return appShareholderDetail.size();
+        return size;
     }
 
     /**
      * @param subResultDate
      */
-    private void wrapperData(List<String[]> subResultDate) {
+    private int wrapperData(List<String[]> subResultDate) {
         int size = subResultDate.get(0).length;
+        int total=0;
         subResultDate.remove(0);
         for (String[] data : subResultDate) {
             data = Arrays.copyOf(data, size);
@@ -231,6 +229,7 @@ public class AppShareholderDetailBiz extends BaseBiz<AppShareholderDetailMapper,
             SimpleDateFormat formatter = new SimpleDateFormat("yyyy-MM-dd");
             if (StringUtils.isNotBlank(name) && StringUtils.isNotBlank(phone) && StringUtils.isNotBlank(companyName)) {
                 try {
+                    total++;
                     Date relTime = StringUtils.isNoneBlank(time) ? formatter.parse(time) : null;
                     BranchCompany branchCompany = vehicleFeign.companyId(companyName);
                     if (branchCompany != null) {
@@ -271,5 +270,6 @@ public class AppShareholderDetailBiz extends BaseBiz<AppShareholderDetailMapper,
                 }
             }
         }
+        return total;
     }
 }