添加用户陪诊员假数据

pz-system/src/main/java/com/pz/merchant/service/impl/EmployeesServiceImpl.java

@@ -510,15 +510,16 @@ public class EmployeesServiceImpl implements IEmployeesService {
 
     @Override
     public DataViewVo dataEmployees(DataViewBo dataViewBo) {
-        Company company = companyMapper.selectOne(new LambdaQueryWrapper<Company>().eq(Company::getUid, dataViewBo.getUserId()));
-        if(null == company){
-            DataViewVo dataViewVo = new DataViewVo();
-            dataViewVo.setDataSum(0);
-            return dataViewVo;
-        }
-        FalseData falseData = falseDataMapper.selectOne(new LambdaQueryWrapper<FalseData>().eq(FalseData::getCompanyId,company.getId()));
+
         DataViewVo dataViewVo = new DataViewVo();
         if(null != dataViewBo.getUserId()){
+            Company company = companyMapper.selectOne(new LambdaQueryWrapper<Company>().eq(Company::getUid, dataViewBo.getUserId()));
+            if(null == company){
+                dataViewVo.setDataSum(0);
+                return dataViewVo;
+            }
+            FalseData falseData = falseDataMapper.selectOne(new LambdaQueryWrapper<FalseData>().eq(FalseData::getCompanyId,company.getId()));
+
             if(dataViewBo.getCondition() == 2){
                 dataViewVo.setDataSum(falseData.getCompanySeven().intValue());
             }else if(dataViewBo.getCondition() == 3){
@@ -529,14 +530,15 @@ public class EmployeesServiceImpl implements IEmployeesService {
                 dataViewVo.setDataSum(falseData.getCompanySum().intValue());
             }
         }else {
+            List<FalseData> falseData1 = falseDataMapper.selectList();
             if(dataViewBo.getCondition() == 2){
-                dataViewVo.setDataSum(falseData.getCompanySeven().intValue());
+                dataViewVo.setDataSum(falseData1.get(0).getCompanySeven().intValue());
             }else if(dataViewBo.getCondition() == 3){
-                dataViewVo.setDataSum(falseData.getCompanyThirty().intValue());
+                dataViewVo.setDataSum(falseData1.get(0).getCompanyThirty().intValue());
             }else if(dataViewBo.getCondition() == 1){
-                dataViewVo.setDataSum(falseData.getCompanyToday().intValue());
+                dataViewVo.setDataSum(falseData1.get(0).getCompanyToday().intValue());
             }else {
-                dataViewVo.setDataSum(falseData.getCompanySum().intValue());
+                dataViewVo.setDataSum(falseData1.get(0).getCompanySum().intValue());
             }
         }