feat:排班任务的新增及修改参数内容修改

This commit is contained in:
qiaoqinzheng 2023-05-24 09:08:10 +08:00
parent aaf022b438
commit 7b1c9d1784
3 changed files with 68 additions and 29 deletions

View File

@ -7,8 +7,8 @@ import java.util.List;
public interface ISysTaskStationService extends IService<SysTaskStation> { public interface ISysTaskStationService extends IService<SysTaskStation> {
void create(String taskId, List<SysTaskStation> taskStations); void create(String taskId, String ipAddr, List<SysTaskStation> taskStations);
void updateStation(String taskId, List<SysTaskStation> stationList); void updateStation(String taskId, String ipAddr, List<SysTaskStation> stationList);
} }

View File

@ -175,16 +175,28 @@ public class SysTaskServiceImpl extends ServiceImpl<SysTaskMapper, SysTask> impl
@Transactional @Transactional
public Result create(List<SysTask> sysTasks){ public Result create(List<SysTask> sysTasks){
Result result = new Result(); Result result = new Result();
//获取request
HttpServletRequest request = SpringContextUtils.getHttpServletRequest();
//获取当前操作人用户名
String username = JwtUtil.getUserNameByToken(request);
//设置IP地址
String ipAddr = IpUtils.getIpAddr(request);
//如果新增排班任务不为空 //如果新增排班任务不为空
if (CollectionUtils.isNotEmpty(sysTasks)){ if (CollectionUtils.isNotEmpty(sysTasks)){
//遍历排班任务进行存储 //遍历排班任务进行存储
for (SysTask sysTask:sysTasks) { for (SysTask sysTask:sysTasks) {
//操作人
sysTask.setCreateBy(username);
//操作人的ip
sysTask.setOperateIp(ipAddr);
//id //id
sysTask.setId(String.valueOf(IdWorker.getId())); sysTask.setId(String.valueOf(IdWorker.getId()));
//创建时间
sysTask.setCreateTime(new Date());
this.baseMapper.insert(sysTask); this.baseMapper.insert(sysTask);
//如果排班任务中台站信息不为空 //如果排班任务中台站信息不为空
if (CollectionUtils.isNotEmpty(sysTask.getStationList())){ if (CollectionUtils.isNotEmpty(sysTask.getStationList())){
sysTaskStationService.create(sysTask.getId(), sysTask.getStationList()); sysTaskStationService.create(sysTask.getId(), ipAddr, sysTask.getStationList());
} }
} }
} }
@ -197,6 +209,12 @@ public class SysTaskServiceImpl extends ServiceImpl<SysTaskMapper, SysTask> impl
@Transactional @Transactional
public Result update(List<SysTask> sysTasks) { public Result update(List<SysTask> sysTasks) {
Result result = new Result(); Result result = new Result();
//获取request
HttpServletRequest request = SpringContextUtils.getHttpServletRequest();
//获取当前操作人用户名
String username = JwtUtil.getUserNameByToken(request);
//设置IP地址
String ipAddr = IpUtils.getIpAddr(request);
//如果新增排班任务不为空 //如果新增排班任务不为空
if (CollectionUtils.isNotEmpty(sysTasks)){ if (CollectionUtils.isNotEmpty(sysTasks)){
//获取第一个排班任务的信息 //获取第一个排班任务的信息
@ -204,12 +222,18 @@ public class SysTaskServiceImpl extends ServiceImpl<SysTaskMapper, SysTask> impl
this.deleteByDate(task.getSchedulingDate()); this.deleteByDate(task.getSchedulingDate());
//遍历排班任务进行存储 //遍历排班任务进行存储
for (SysTask sysTask:sysTasks) { for (SysTask sysTask:sysTasks) {
//操作人
sysTask.setCreateBy(username);
//操作人的ip
sysTask.setOperateIp(ipAddr);
//id //id
sysTask.setId(String.valueOf(IdWorker.getId())); sysTask.setId(String.valueOf(IdWorker.getId()));
//创建时间
sysTask.setCreateTime(new Date());
this.baseMapper.insert(sysTask); this.baseMapper.insert(sysTask);
//如果排班任务中台站信息不为空 //如果排班任务中台站信息不为空
if (CollectionUtils.isNotEmpty(sysTask.getStationList())){ if (CollectionUtils.isNotEmpty(sysTask.getStationList())){
sysTaskStationService.create(sysTask.getId(), sysTask.getStationList()); sysTaskStationService.create(sysTask.getId(), ipAddr, sysTask.getStationList());
} }
} }
} }
@ -349,12 +373,15 @@ public class SysTaskServiceImpl extends ServiceImpl<SysTaskMapper, SysTask> impl
} }
@Override @Override
@Transactional
public ImportViewVo importExcel(List<SysTaskExportVo> dataList, int headRow) { public ImportViewVo importExcel(List<SysTaskExportVo> dataList, int headRow) {
ImportViewVo importViewVo = new ImportViewVo(); ImportViewVo importViewVo = new ImportViewVo();
//获取用户信息 //获取用户信息
List<SysUser> userList = sysUserMapper.selectList(new LambdaQueryWrapper<>()); List<SysUser> userList = sysUserMapper.selectList(new LambdaQueryWrapper<>());
//获取台站信息 //获取台站信息
List<GardsStations> stationsList = gardsStationsService.getGardsStations(); List<GardsStations> stationsList = gardsStationsService.getGardsStations();
//获取排班任务信息
List<SysTask> sysTaskList = this.baseMapper.selectList(new LambdaQueryWrapper<SysTask>());
//存储排班任务信息 //存储排班任务信息
List<SysTask> taskList = new ArrayList<>(); List<SysTask> taskList = new ArrayList<>();
//导入失败的结果集 //导入失败的结果集
@ -467,10 +494,7 @@ public class SysTaskServiceImpl extends ServiceImpl<SysTaskMapper, SysTask> impl
} }
//判断当前数据库中的排班任务信息是否为空 //判断当前数据库中的排班任务信息是否为空
if (CollectionUtils.isNotEmpty(taskList)){ if (CollectionUtils.isNotEmpty(taskList)){
//遍历可以新增的数据信息 this.addOrUpdate(sysTaskList,taskList);
for (SysTask sysTask:taskList) {
this.addOrUpdate(sysTask);
}
} }
} }
} catch (ParseException e) { } catch (ParseException e) {
@ -482,25 +506,37 @@ public class SysTaskServiceImpl extends ServiceImpl<SysTaskMapper, SysTask> impl
return importViewVo; return importViewVo;
} }
private void addOrUpdate(SysTask sysTask){ private void addOrUpdate(List<SysTask> sysTaskList,List<SysTask> taskList){
//根据用户名称及排班日期查询对应的数据 //获取request
LambdaQueryWrapper<SysTask> queryWrapper = new LambdaQueryWrapper<>(); HttpServletRequest request = SpringContextUtils.getHttpServletRequest();
queryWrapper.eq(SysTask::getUserId, sysTask.getUserId()); //获取当前操作人用户名
queryWrapper.eq(SysTask::getSchedulingDate, sysTask.getSchedulingDate()); String username = JwtUtil.getUserNameByToken(request);
SysTask task = this.baseMapper.selectOne(queryWrapper); //设置IP地址
//如果查询到的内容不为空则进行修改 String ipAddr = IpUtils.getIpAddr(request);
if (Objects.nonNull(task)){ for (SysTask sysTask:taskList) {
List<SysTaskStation> stationList = sysTask.getStationList(); List<SysTask> tasks = sysTaskList.stream()
if (CollectionUtils.isNotEmpty(stationList)){ .filter(item -> item.getUserId().equals(sysTask.getUserId()) && item.getSchedulingDate().equals(sysTask.getSchedulingDate())).collect(Collectors.toList());
sysTaskStationService.updateStation(task.getId(), stationList); //如果查询到的内容不为空则进行修改
} if (CollectionUtils.isNotEmpty(tasks)){
}else {//如果查询的内容为空则进行新增 SysTask task = tasks.get(0);
//id List<SysTaskStation> stationList = sysTask.getStationList();
sysTask.setId(String.valueOf(IdWorker.getId())); if (CollectionUtils.isNotEmpty(stationList)){
this.baseMapper.insert(sysTask); sysTaskStationService.updateStation(task.getId(), ipAddr, stationList);
//如果排班任务中台站信息不为空 }
if (CollectionUtils.isNotEmpty(sysTask.getStationList())){ }else {//如果查询的内容为空则进行新增
sysTaskStationService.create(sysTask.getId(), sysTask.getStationList()); //操作人
sysTask.setCreateBy(username);
//操作人的ip
sysTask.setOperateIp(ipAddr);
//id
sysTask.setId(String.valueOf(IdWorker.getId()));
//创建时间
sysTask.setCreateTime(new Date());
this.baseMapper.insert(sysTask);
//如果排班任务中台站信息不为空
if (CollectionUtils.isNotEmpty(sysTask.getStationList())){
sysTaskStationService.create(sysTask.getId(), ipAddr, sysTask.getStationList());
}
} }
} }

View File

@ -18,11 +18,12 @@ public class SysTaskStationServiceImpl extends ServiceImpl<SysTaskStationMapper,
@Override @Override
@Transactional @Transactional
public void create(String taskId, List<SysTaskStation> taskStations) { public void create(String taskId, String ipAddr, List<SysTaskStation> taskStations) {
for (SysTaskStation sysTaskStation:taskStations) { for (SysTaskStation sysTaskStation:taskStations) {
if (StringUtils.isBlank(sysTaskStation.getStationId())){ if (StringUtils.isBlank(sysTaskStation.getStationId())){
throw new RuntimeException("台站信息不能为空"); throw new RuntimeException("台站信息不能为空");
} }
sysTaskStation.setOperateIp(ipAddr);
sysTaskStation.setId(String.valueOf(IdWorker.getId())); sysTaskStation.setId(String.valueOf(IdWorker.getId()));
sysTaskStation.setTaskId(taskId); sysTaskStation.setTaskId(taskId);
this.baseMapper.insert(sysTaskStation); this.baseMapper.insert(sysTaskStation);
@ -30,7 +31,8 @@ public class SysTaskStationServiceImpl extends ServiceImpl<SysTaskStationMapper,
} }
@Override @Override
public void updateStation(String taskId, List<SysTaskStation> stationList) { @Transactional
public void updateStation(String taskId, String ipAddr, List<SysTaskStation> stationList) {
LambdaQueryWrapper<SysTaskStation> queryWrapper = new LambdaQueryWrapper<>(); LambdaQueryWrapper<SysTaskStation> queryWrapper = new LambdaQueryWrapper<>();
queryWrapper.eq(SysTaskStation::getTaskId, taskId); queryWrapper.eq(SysTaskStation::getTaskId, taskId);
this.baseMapper.delete(queryWrapper); this.baseMapper.delete(queryWrapper);
@ -38,6 +40,7 @@ public class SysTaskStationServiceImpl extends ServiceImpl<SysTaskStationMapper,
if (StringUtils.isBlank(sysTaskStation.getStationId())){ if (StringUtils.isBlank(sysTaskStation.getStationId())){
throw new RuntimeException("台站信息不能为空"); throw new RuntimeException("台站信息不能为空");
} }
sysTaskStation.setOperateIp(ipAddr);
sysTaskStation.setId(String.valueOf(IdWorker.getId())); sysTaskStation.setId(String.valueOf(IdWorker.getId()));
sysTaskStation.setTaskId(taskId); sysTaskStation.setTaskId(taskId);
sysTaskStation.setCreateTime(new Date()); sysTaskStation.setCreateTime(new Date());