|
@@ -6,12 +6,14 @@ import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
|
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
|
import com.fdkankan.manage_jp.common.PageInfo;
|
|
import com.fdkankan.manage_jp.common.PageInfo;
|
|
import com.fdkankan.manage_jp.common.ResultCode;
|
|
import com.fdkankan.manage_jp.common.ResultCode;
|
|
|
|
+import com.fdkankan.manage_jp.entity.Company;
|
|
import com.fdkankan.manage_jp.entity.ScenePlus;
|
|
import com.fdkankan.manage_jp.entity.ScenePlus;
|
|
import com.fdkankan.manage_jp.entity.ScenePro;
|
|
import com.fdkankan.manage_jp.entity.ScenePro;
|
|
import com.fdkankan.manage_jp.entity.SceneProEdit;
|
|
import com.fdkankan.manage_jp.entity.SceneProEdit;
|
|
import com.fdkankan.manage_jp.exception.BusinessException;
|
|
import com.fdkankan.manage_jp.exception.BusinessException;
|
|
import com.fdkankan.manage_jp.httpClient.service.LaserService;
|
|
import com.fdkankan.manage_jp.httpClient.service.LaserService;
|
|
import com.fdkankan.manage_jp.mapper.ISceneProMapper;
|
|
import com.fdkankan.manage_jp.mapper.ISceneProMapper;
|
|
|
|
+import com.fdkankan.manage_jp.service.ICompanyService;
|
|
import com.fdkankan.manage_jp.service.IScenePlusService;
|
|
import com.fdkankan.manage_jp.service.IScenePlusService;
|
|
import com.fdkankan.manage_jp.service.ISceneProEditService;
|
|
import com.fdkankan.manage_jp.service.ISceneProEditService;
|
|
import com.fdkankan.manage_jp.service.ISceneProService;
|
|
import com.fdkankan.manage_jp.service.ISceneProService;
|
|
@@ -43,6 +45,8 @@ public class SceneProServiceImpl extends ServiceImpl<ISceneProMapper, ScenePro>
|
|
LaserService laserService;
|
|
LaserService laserService;
|
|
@Autowired
|
|
@Autowired
|
|
ISceneProEditService sceneProEditService;
|
|
ISceneProEditService sceneProEditService;
|
|
|
|
+ @Autowired
|
|
|
|
+ ICompanyService companyService;
|
|
|
|
|
|
|
|
|
|
@Override
|
|
@Override
|
|
@@ -77,6 +81,13 @@ public class SceneProServiceImpl extends ServiceImpl<ISceneProMapper, ScenePro>
|
|
return laserService.pageList(param);
|
|
return laserService.pageList(param);
|
|
}
|
|
}
|
|
Page<SceneVo> page = this.getBaseMapper().pageList(new Page<>(param.getPage(),param.getLimit()),param);
|
|
Page<SceneVo> page = this.getBaseMapper().pageList(new Page<>(param.getPage(),param.getLimit()),param);
|
|
|
|
+ for (SceneVo record : page.getRecords()) {
|
|
|
|
+ String userName = record.getUserName();
|
|
|
|
+ Company company = companyService.getByUserName(userName);
|
|
|
|
+ if(company != null){
|
|
|
|
+ record.setCompanyName(company.getCompanyName());
|
|
|
|
+ }
|
|
|
|
+ }
|
|
return PageInfo.PageInfo(page);
|
|
return PageInfo.PageInfo(page);
|
|
}
|
|
}
|
|
|
|
|