package com.dingzhuo.energy.basic.data.workforce.controller;
|
|
import com.dingzhuo.energy.basic.data.workforce.domain.Rostering;
|
import com.dingzhuo.energy.basic.data.workforce.domain.RosteringCopy;
|
import com.dingzhuo.energy.basic.data.workforce.domain.RosteringScheme;
|
import com.dingzhuo.energy.basic.data.workforce.service.IRosteringService;
|
import com.dingzhuo.energy.common.utils.DateUtils;
|
import com.dingzhuo.energy.common.utils.SecurityUtils;
|
import com.dingzhuo.energy.common.utils.poi.ExcelUtil;
|
import com.dingzhuo.energy.framework.aspectj.lang.annotation.Log;
|
import com.dingzhuo.energy.framework.aspectj.lang.enums.BusinessType;
|
import com.dingzhuo.energy.framework.web.controller.BaseController;
|
import com.dingzhuo.energy.framework.web.domain.AjaxResult;
|
import com.dingzhuo.energy.framework.web.page.TableDataInfo;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.security.access.prepost.PreAuthorize;
|
import org.springframework.web.bind.annotation.*;
|
|
import java.text.ParseException;
|
import java.text.SimpleDateFormat;
|
import java.util.*;
|
|
/**
|
* 排班表查询Controller
|
*
|
* @author liuli
|
* @date 2020-05-13
|
*/
|
@RestController
|
@RequestMapping("/workforce/rosteringManagement")
|
public class RosteringController extends BaseController
|
{
|
@Autowired
|
private IRosteringService rosteringService;
|
/**
|
*查询轮值方案名称集合
|
*/
|
@PreAuthorize("@ss.hasPermi('workforce:rosteringManagement:schemeNamelist')")
|
@GetMapping("/schemeNamelist")
|
public TableDataInfo schemeNamelist(Rostering rostering) {
|
startPage();
|
List<RosteringScheme> list = rosteringService.selectSchemeNameList();
|
return getDataTable(list);
|
}
|
|
/**
|
* 查询排班表查询列表
|
*/
|
@PreAuthorize("@ss.hasPermi('workforce:rosteringManagement:list')")
|
@GetMapping("/list")
|
public TableDataInfo list(Rostering rostering)
|
{
|
startPage();
|
List<Rostering> list = rosteringService.selectRosteringList(rostering);
|
return getDataTable(list);
|
}
|
|
/**
|
* 导出排班表查询列表
|
*/
|
@PreAuthorize("@ss.hasPermi('workforce:rosteringManagement:export')")
|
@Log(title = "排班表查询", businessType = BusinessType.EXPORT)
|
@GetMapping("/export")
|
public AjaxResult export(Rostering rostering)
|
{
|
List<Rostering> list = rosteringService.selectRosteringList(rostering);
|
ExcelUtil<Rostering> util = new ExcelUtil<Rostering>(Rostering.class);
|
return util.exportExcel(list, "rosteringManagement");
|
}
|
|
/**
|
* 获取排班表查询详细信息
|
*/
|
@PreAuthorize("@ss.hasPermi('workforce:rosteringManagement:query')")
|
@GetMapping(value = "/{id}")
|
public AjaxResult getInfo(@PathVariable("id") String id)
|
{
|
return AjaxResult.success(rosteringService.selectRosteringById(id));
|
}
|
|
/**
|
* 新增排班表查询（生成）
|
*/
|
@PreAuthorize("@ss.hasPermi('workforce:rosteringManagement:add')")
|
@PostMapping
|
public AjaxResult add(@RequestBody Rostering rostering) throws ParseException {
|
//根据轮值方案id查询班次id、值次id等list
|
List<RosteringCopy> list=rosteringService.selectList(rostering);
|
//设置转换的日期格式
|
SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd");
|
//建一个新的list存放insert的list
|
List<Rostering> dataItems = new ArrayList<>();
|
//定义一个count来完成code（编码）的自增
|
int count=1;
|
//定义从前台获取的开始时间 startdate为开始时间
|
String startdate=sdf.format(rostering.getBeginTimes());
|
/* Long timenum =(rostering.getEndTimes().getTime()-rostering.getBeginTimes().getTime())/(60*60*24*1000);//时间差*/
|
for(RosteringCopy b:list){
|
Date loopstarttime=sdf.parse(startdate);
|
long long3 =loopstarttime.getTime();
|
long long4= rostering.getEndTimes().getTime();
|
if(long3 > long4){//循环后的时间>从页面获取的结束时间则停止循环
|
break;
|
}else {
|
startdate=sdf.format(loopstarttime.getTime());
|
for (RosteringCopy a:list) {
|
Rostering rosterings =new Rostering();
|
rosterings.setId(UUID.randomUUID().toString());
|
String code=String.valueOf(count);
|
rosterings.setCode(code);
|
rosterings.setName("排班"+code);
|
rosterings.setSchemeId(a.getSchemeId());
|
rosterings.setShiftId(a.getShiftId());
|
rosterings.setDutyId(a.getDutyId());
|
rosterings.setDepartMemberId(rostering.getDepartMemberId());
|
rosterings.setCreateBy(SecurityUtils.getUsername());
|
rosterings.setCreateTime(DateUtils.getNowDate());
|
SimpleDateFormat df = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
|
String bengintime=startdate +" "+a.getStartTime();//bengintime为拼接完的开始时间
|
Date looptime=sdf.parse(startdate);//循环后的开始时间
|
Date endtime =rostering.getEndTimes();//从前台获取的结束时间
|
long long1 =looptime.getTime();
|
long long2= endtime.getTime();
|
if(long1 > long2){//若当前时间>从前台获取的结束时间则结束循环
|
break;
|
}else {
|
if(a.getIsCrossDay().equals("Y")) {//是否跨天
|
Date starttime = sdf.parse(startdate);
|
Calendar calendar = Calendar.getInstance();
|
calendar.setTime(starttime);
|
calendar.add(Calendar.DAY_OF_MONTH, 1);
|
Date looptimes=sdf.parse(startdate);//循环后的开始时间
|
Date endtimes =rostering.getEndTimes();//从前台获取的结束时间
|
long long5 =looptime.getTime();
|
long long6= endtime.getTime();
|
Date beginTimes = df.parse(bengintime);
|
String enddate=startdate+" "+a.getEnddTime();
|
Date endTimes = df.parse(enddate);
|
rosterings.setBeginTimes(beginTimes);
|
rosterings.setEndTimes(endTimes);
|
startdate=sdf.format(calendar.getTime());//给开始时间赋值跨天后的时间
|
}else{
|
Date beginTimes = df.parse(bengintime);
|
String enddate=startdate+" "+a.getEnddTime();
|
Date endTimes = df.parse(enddate);
|
rosterings.setBeginTimes(beginTimes);
|
rosterings.setEndTimes(endTimes);
|
}
|
}
|
dataItems.add(rosterings);
|
count ++;
|
}
|
}
|
}
|
|
this.rosteringService.saveRostering(dataItems);
|
return AjaxResult.success("保存成功！");
|
}
|
|
/**
|
* 修改排班表查询
|
*/
|
@PreAuthorize("@ss.hasPermi('workforce:rosteringManagement:edit')")
|
@Log(title = "排班表查询", businessType = BusinessType.UPDATE)
|
@PutMapping
|
public AjaxResult edit(@RequestBody Rostering rostering)
|
{
|
rostering.setUpdateBy(SecurityUtils.getUsername());
|
rostering.setUpdateTime(DateUtils.getNowDate());
|
return toAjax(rosteringService.updateRostering(rostering));
|
}
|
|
/**
|
* 删除排班表查询
|
*/
|
@PreAuthorize("@ss.hasPermi('workforce:rosteringManagement:remove')")
|
@Log(title = "排班表查询", businessType = BusinessType.DELETE)
|
@DeleteMapping("/{ids}")
|
public AjaxResult remove(@PathVariable String[] ids)
|
{
|
return toAjax(rosteringService.deleteRosteringByIds(ids));
|
}
|
public static <E> List<List<E>> splitList(List<E> targetList, Integer splitSize) {
|
if (targetList == null) {
|
return new ArrayList<>();
|
}
|
|
int size = targetList.size();
|
List<List<E>> resultList = new ArrayList<>();
|
if (size <= splitSize) {
|
resultList.add(targetList);
|
} else {
|
for (int i = 0; i < size; i += splitSize) {
|
//用于限制最后一部分size小于splitSize的list
|
int limit = i + splitSize;
|
if (limit > size) {
|
limit = size;
|
}
|
resultList.add(targetList.subList(i, limit));
|
}
|
}
|
return resultList;
|
}
|
}
|
