meidu2021/daju/target/classes/cn/com/lzt/car/schedule/controller/ScheduleArrangeController.java

package cn.com.lzt.car.schedule.controller;

import cn.com.lzt.car.cardoc.entity.CarEntity;
import cn.com.lzt.car.schedule.entity.CarScheduleEntity;
import cn.com.lzt.car.schedule.service.CarScheduleServiceI;
import cn.com.lzt.car.scheduledetail.entity.CarScheduleRouteEntity;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.daju.mix.dao.entity.TBCarSchedule;
import com.daju.mix.dao.entity.TBCarScheduleArrange;
import com.daju.mix.dao.entity.TBCarScheduleArrangeSection;
import com.daju.mix.dao.service.impl.TBCarScheduleArrangeSectionServiceImpl;
import com.daju.mix.dao.service.impl.TBCarScheduleArrangeServiceImpl;
import com.daju.mix.dao.service.impl.TBCarScheduleSectionServiceImpl;
import com.daju.mix.dao.service.impl.TBCarScheduleServiceImpl;
import com.daju.mix.service.ScheduleService;
import org.apache.commons.lang3.StringUtils;
import org.apache.log4j.Logger;
import org.jeecgframework.core.beanvalidator.BeanValidators;
import org.jeecgframework.core.common.controller.BaseController;
import org.jeecgframework.core.common.exception.BusinessException;
import org.jeecgframework.core.common.hibernate.qbc.CriteriaQuery;
import org.jeecgframework.core.common.model.json.AjaxJson;
import org.jeecgframework.core.common.model.json.DataGrid;
import org.jeecgframework.core.constant.Globals;
import org.jeecgframework.core.util.ExceptionUtil;
import org.jeecgframework.core.util.MyBeanUtils;
import org.jeecgframework.core.util.ResourceUtil;
import org.jeecgframework.core.util.StringUtil;
import org.jeecgframework.poi.excel.ExcelImportUtil;
import org.jeecgframework.poi.excel.entity.ExportParams;
import org.jeecgframework.poi.excel.entity.ImportParams;
import org.jeecgframework.poi.excel.entity.vo.NormalExcelConstants;
import org.jeecgframework.tag.core.easyui.TagUtil;
import org.jeecgframework.web.system.pojo.base.TSBaseUser;
import org.jeecgframework.web.system.service.SystemService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpStatus;
import org.springframework.http.MediaType;
import org.springframework.http.ResponseEntity;
import org.springframework.stereotype.Controller;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.ui.ModelMap;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;
import org.springframework.web.multipart.MultipartHttpServletRequest;
import org.springframework.web.servlet.ModelAndView;
import org.springframework.web.util.UriComponentsBuilder;

import javax.annotation.Resource;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.validation.ConstraintViolation;
import javax.validation.Validator;
import java.io.IOException;
import java.net.URI;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.Set;

/**
 * @Title: Controller
 * @Description: 车辆作业计划
 * @author onlineGenerator
 * @date 2019-10-15 09:47:08
 * @version V1.0
 *
 */
@Controller
@RequestMapping("/scheduleArrangeController")
public class ScheduleArrangeController extends BaseController {
	/**
	 * Logger for this class
	 */
	private static final Logger logger = Logger.getLogger(ScheduleArrangeController.class);

	@Autowired
	private CarScheduleServiceI carScheduleService;
	@Resource
	private TBCarScheduleServiceImpl tbCarScheduleService;
	@Resource
	private TBCarScheduleArrangeServiceImpl arrangeService;
	@Resource
	private TBCarScheduleArrangeSectionServiceImpl arrangeSectionService;
	@Resource
	private TBCarScheduleSectionServiceImpl scheduleSectionService;
	@Resource
	private ScheduleService scheduleService;
	@Autowired
	private SystemService systemService;
	@Autowired
	private Validator validator;

	/**
	 * easyui AJAX请求数据
	 *
	 * @param request
	 * @param response
	 * @param dataGrid
	 */
	@RequestMapping(params = "datagrid")
	public void datagrid(TBCarScheduleArrange object, HttpServletRequest request, HttpServletResponse response, DataGrid dataGrid) {
		QueryWrapper<?> queryWrapper = new QueryWrapper<>();
		if (StringUtils.isNotBlank(object.getScheduleId())) {
			queryWrapper.eq("schedule_id", object.getScheduleId());
		}
		queryWrapper.orderByAsc("date");
		String selectSql = "select a.*,b.carid carid,b.carname carname,b.start,b.end,u.realname user_name " +
				"from t_b_car_schedule_arrange a " +
				"left join t_b_car_schedule b on a.schedule_id=b.id " +
				"left join t_s_base_user u on a.user_id=u.id";
//		IPage<Map<String, Object>> mapPage = arrangeService.getBaseMapper().queryMapPageBySql(
//				new Page<>(dataGrid.getPage(), dataGrid.getRows()), queryWrapper, selectSql);
		IPage<TBCarScheduleArrange> dataPage = arrangeService.queryPage(
				new Page<>(dataGrid.getPage(), dataGrid.getRows()), queryWrapper, selectSql);
//		DataPage<TBCarScheduleArrangeView> dataPage = PageUtil.transformMapPage(mapPage, TBCarScheduleArrangeView.class);
		TagUtil.datagrid(response, dataGrid, dataPage);
	}

	@RequestMapping(params = "listDetail")
	public ModelAndView listDetail(HttpServletRequest request) {
		String arrangeId = request.getParameter("arrangeId");
		TBCarScheduleArrange entity = arrangeService.getById(arrangeId);
		TSBaseUser user = systemService.get(TSBaseUser.class, entity.getUserId());
		request.setAttribute("arrange", entity);
		request.setAttribute("user", user);
		request.setAttribute("arrangeId", arrangeId);
//		if(entity.getType().equals("route")) {
//			return new ModelAndView("cn/com/lzt/car/schedule/scheduleArrangeSectionList");
//		} else if(entity.getType().equals("section")) {
//			return new ModelAndView("cn/com/lzt/car/schedule/scheduleArrangeSectionList");
//		} else if(entity.getType().equals("place")) {
//			return new ModelAndView("cn/com/lzt/car/schedule/scheduleArrangeSectionList");
//		}
		return new ModelAndView("cn/com/lzt/car/schedule/scheduleArrangeSectionList");
	}

	@RequestMapping(params = "datagridSectionList")
	public void datagridSectionList(TBCarScheduleArrangeSection object, HttpServletRequest request,
						 HttpServletResponse response, DataGrid dataGrid) {
		QueryWrapper<?> queryWrapper = new QueryWrapper<>();
		if (object.getArrangeId() != null) {
			queryWrapper.eq("a.arrange_id", object.getArrangeId());
		}
		queryWrapper.orderByAsc("a.id");
		String selectSql =
				"select a.*,s.name section_name,b.start,b.end,b.carname,b.description,ss.plan_end_time,u.realname user_name " +
				"from t_b_car_schedule_arrange_section a " +
				"left join t_b_archives_road_section s on a.section_id=s.id " +
				"left join t_b_car_schedule_section ss on a.schedule_id=ss.schedule_id and a.section_id=ss.section_id " +
				"left join t_b_car_schedule b on a.schedule_id=b.id " +
				"left join t_s_base_user u on a.user_id=u.id";
//		IPage<Map<String, Object>> mapPage = arrangeService.getBaseMapper().queryMapPageBySql(
//				new Page<>(dataGrid.getPage(), dataGrid.getRows()), queryWrapper, selectSql);
		IPage<TBCarScheduleArrangeSection> dataPage = arrangeSectionService.queryPage(
				new Page<>(dataGrid.getPage(), dataGrid.getRows()), queryWrapper, selectSql);
//		DataPage<TBCarScheduleArrangeView> dataPage = PageUtil.transformMapPage(mapPage, TBCarScheduleArrangeView.class);
		TagUtil.datagrid(response, dataGrid, dataPage);
	}

	/**
	 * 删除排班和排班明细
	 *
	 * @return
	 */
	@RequestMapping(params = "doDel")
	@ResponseBody
	public AjaxJson doDel(TBCarScheduleArrange object, HttpServletRequest request) {
		String message = null;
		AjaxJson j = new AjaxJson();
		message = "删除成功";
		try{
			TBCarScheduleArrange arrange = arrangeService.getById(object.getId());
			arrangeService.removeById(arrange);
			LambdaQueryWrapper<TBCarScheduleArrangeSection> wrapper = new LambdaQueryWrapper<>();
			wrapper.eq(TBCarScheduleArrangeSection::getArrangeId, object.getId());
			arrangeSectionService.remove(wrapper);
			systemService.addLog(message, Globals.Log_Type_DEL, Globals.Log_Leavel_INFO);
		}catch(Exception e){
			e.printStackTrace();
			message = "删除失败";
			throw new BusinessException(e.getMessage());
		}
		j.setMsg(message);
		return j;
	}

	/**
	 * 批量删除车辆作业计划
	 *
	 * @return
	 */
	@RequestMapping(params = "doBatchDel")
	@ResponseBody
	public AjaxJson doBatchDel(String ids,HttpServletRequest request){
		String message = null;
		AjaxJson j = new AjaxJson();
		message = "车辆作业计划删除成功";
		try{
			for(String id:ids.split(",")){
				CarScheduleEntity carSchedule = systemService.getEntity(CarScheduleEntity.class,
						id
				);
				carScheduleService.delete(carSchedule);
				systemService.addLog(message, Globals.Log_Type_DEL, Globals.Log_Leavel_INFO);
			}
		}catch(Exception e){
			e.printStackTrace();
			message = "车辆作业计划删除失败";
			throw new BusinessException(e.getMessage());
		}
		j.setMsg(message);
		return j;
	}

	/**
	 * 添加车辆作业计划
	 *
	 * @return
	 */
	@RequestMapping(params = "doAddArrange")
	@ResponseBody
	@Transactional
	public AjaxJson doAddArrange(String scheduleId, String start, String end, HttpServletRequest request) {
		String message = null;
		AjaxJson j = new AjaxJson();
		message = "添加成功";
		try{
			logger.debug("doAddArrange scheduleId=" + scheduleId + " start=" + start + " end=" + end);
			scheduleService.generateScheduleArrange(scheduleId, start, end);
			systemService.addLog(message, Globals.Log_Type_INSERT, Globals.Log_Leavel_INFO);
		}catch(Exception e){
			e.printStackTrace();
			message = "添加失败";
			throw new BusinessException(e.getMessage());
		}
		j.setMsg(message);
		return j;
	}

	private boolean hasCarSchedule(CarScheduleEntity carSchedule) {
		boolean hasSchedule = false;
		String sql = "select id From t_b_Car_Schedule sche where carid =? and ( (sche.start > ?  and sche.start < ? ) or ( sche.end > ?  and sche.end < ? ) )";
		List<Map<String, Object>> scheList = systemService.findForJdbc(sql, carSchedule.getCarid(),
				carSchedule.getStart(),carSchedule.getEnd(),
				carSchedule.getStart(),carSchedule.getEnd());
		if(scheList.size() > 0)
			hasSchedule = true;
		return hasSchedule;
	}

	/**
	 * 更新车辆作业计划
	 *
	 * @param ids
	 * @return
	 */
	@RequestMapping(params = "doUpdate")
	@ResponseBody
	public AjaxJson doUpdate(CarScheduleEntity carSchedule, HttpServletRequest request) {
		String message = null;
		AjaxJson j = new AjaxJson();
		message = "车辆作业计划更新成功";
		CarScheduleEntity t = carScheduleService.get(CarScheduleEntity.class, carSchedule.getId());
		try {
			MyBeanUtils.copyBeanNotNull2Bean(carSchedule, t);
			CarEntity carEntity = systemService.get(CarEntity.class, carSchedule.getCarid());
			t.setUserId(carEntity.getDriverid());
			t.setCarname(carEntity.getName());
			carScheduleService.saveOrUpdate(t);
			systemService.addLog(message, Globals.Log_Type_UPDATE, Globals.Log_Leavel_INFO);
		} catch (Exception e) {
			e.printStackTrace();
			message = "车辆作业计划更新失败";
			throw new BusinessException(e.getMessage());
		}
		j.setMsg(message);
		return j;
	}
	/**
	 * 复制车辆作业计划
	 *
	 * @param ids
	 * @return
	 */
	@RequestMapping(params = "doClone")
	@ResponseBody
	public AjaxJson doClone(String scheduleid, HttpServletRequest request) {
		String message = null;
		AjaxJson j = new AjaxJson();
		message = "车辆作业计划复制成功";
		CarScheduleEntity t = carScheduleService.get(CarScheduleEntity.class, scheduleid);
		List<CarScheduleRouteEntity> details = carScheduleService.findByProperty(CarScheduleRouteEntity.class, "scheduleid", scheduleid);
		try {
			CarScheduleEntity carSchedule = new CarScheduleEntity();
			MyBeanUtils.copyBean2Bean(carSchedule, t);
			carSchedule.setId(null);
			carScheduleService.save(carSchedule);
			for(CarScheduleRouteEntity detail : details) {
				CarScheduleRouteEntity newDetail = new CarScheduleRouteEntity();
				MyBeanUtils.copyBean2Bean(newDetail, detail);
				newDetail.setScheduleid(carSchedule.getId());
				newDetail.setId(null);
				systemService.save(newDetail);
			}
			systemService.addLog(message, Globals.Log_Type_UPDATE, Globals.Log_Leavel_INFO);
		} catch (Exception e) {
			e.printStackTrace();
			message = "车辆作业计划复制失败";
			throw new BusinessException(e.getMessage());
		}
		j.setMsg(message);
		return j;
	}

	/**
	 * 车辆排班新增页面跳转
	 *
	 * @return
	 */
	@RequestMapping(params = "goAddArrange")
	public ModelAndView goAddArrange(TBCarScheduleArrange arrange, HttpServletRequest req) {
		if (StringUtil.isNotEmpty(arrange.getScheduleId())) {
			TBCarSchedule schedule = tbCarScheduleService.getById(arrange.getScheduleId());
			req.setAttribute("schedule", schedule);
		}
		return new ModelAndView("cn/com/lzt/car/schedule/scheduleArrange-add");
	}

	/**
	 * 车辆作业计划编辑页面跳转
	 *
	 * @return
	 */
	@RequestMapping(params = "goUpdate")
	public ModelAndView goUpdate(CarScheduleEntity carSchedule, HttpServletRequest req) {
		if (StringUtil.isNotEmpty(carSchedule.getId())) {
			carSchedule = carScheduleService.getEntity(CarScheduleEntity.class, carSchedule.getId());
			req.setAttribute("carSchedulePage", carSchedule);
		}
		return new ModelAndView("cn/com/lzt/car/schedule/carSchedule-update");
	}

	/**
	 * 导入功能跳转
	 *
	 * @return
	 */
	@RequestMapping(params = "upload")
	public ModelAndView upload(HttpServletRequest req) {
		req.setAttribute("controller_name","carScheduleController");
		return new ModelAndView("common/upload/pub_excel_upload");
	}

	/**
	 * 导出excel
	 *
	 * @param request
	 * @param response
	 */
	@RequestMapping(params = "exportXls")
	public String exportXls(CarScheduleEntity carSchedule,HttpServletRequest request,HttpServletResponse response
			, DataGrid dataGrid,ModelMap modelMap) {
		CriteriaQuery cq = new CriteriaQuery(CarScheduleEntity.class, dataGrid);
		org.jeecgframework.core.extend.hqlsearch.HqlGenerateUtil.installHql(cq, carSchedule, request.getParameterMap());
		List<CarScheduleEntity> carSchedules = this.carScheduleService.getListByCriteriaQuery(cq,false);
		modelMap.put(NormalExcelConstants.FILE_NAME,"车辆作业计划");
		modelMap.put(NormalExcelConstants.CLASS,CarScheduleEntity.class);
		modelMap.put(NormalExcelConstants.PARAMS,new ExportParams("车辆作业计划列表", "导出人:"+ResourceUtil.getSessionUser().getRealName(),
				"导出信息"));
		modelMap.put(NormalExcelConstants.DATA_LIST,carSchedules);
		return NormalExcelConstants.JEECG_EXCEL_VIEW;
	}
	/**
	 * 导出excel 使模板
	 *
	 * @param request
	 * @param response
	 */
	@RequestMapping(params = "exportXlsByT")
	public String exportXlsByT(CarScheduleEntity carSchedule,HttpServletRequest request,HttpServletResponse response
			, DataGrid dataGrid,ModelMap modelMap) {
		modelMap.put(NormalExcelConstants.FILE_NAME,"车辆作业计划");
		modelMap.put(NormalExcelConstants.CLASS,CarScheduleEntity.class);
		modelMap.put(NormalExcelConstants.PARAMS,new ExportParams("车辆作业计划列表", "导出人:"+ResourceUtil.getSessionUser().getRealName(),
				"导出信息"));
		modelMap.put(NormalExcelConstants.DATA_LIST,new ArrayList());
		return NormalExcelConstants.JEECG_EXCEL_VIEW;
	}

	@SuppressWarnings("unchecked")
	@RequestMapping(params = "importExcel", method = RequestMethod.POST)
	@ResponseBody
	public AjaxJson importExcel(HttpServletRequest request, HttpServletResponse response) {
		AjaxJson j = new AjaxJson();

		MultipartHttpServletRequest multipartRequest = (MultipartHttpServletRequest) request;
		Map<String, MultipartFile> fileMap = multipartRequest.getFileMap();
		for (Map.Entry<String, MultipartFile> entity : fileMap.entrySet()) {
			MultipartFile file = entity.getValue();// 获取上传文件对象
			ImportParams params = new ImportParams();
			params.setTitleRows(2);
			params.setHeadRows(1);
			params.setNeedSave(true);
			try {
				List<CarScheduleEntity> listCarScheduleEntitys = ExcelImportUtil.importExcel(file.getInputStream(),CarScheduleEntity.class,params);
				for (CarScheduleEntity carSchedule : listCarScheduleEntitys) {
					carScheduleService.save(carSchedule);
				}
				j.setMsg("文件导入成功！");
			} catch (Exception e) {
				j.setMsg("文件导入失败！");
				logger.error(ExceptionUtil.getExceptionMessage(e));
			}finally{
				try {
					file.getInputStream().close();
				} catch (IOException e) {
					e.printStackTrace();
				}
			}
		}
		return j;
	}

	@RequestMapping(method = RequestMethod.GET)
	@ResponseBody
	public List<CarScheduleEntity> list() {
		List<CarScheduleEntity> listCarSchedules=carScheduleService.getList(CarScheduleEntity.class);
		return listCarSchedules;
	}

	@RequestMapping(value = "/{id}", method = RequestMethod.GET)
	@ResponseBody
	public ResponseEntity<?> get(@PathVariable("id") String id) {
		CarScheduleEntity task = carScheduleService.get(CarScheduleEntity.class, id);
		if (task == null) {
			return new ResponseEntity(HttpStatus.NOT_FOUND);
		}
		return new ResponseEntity(task, HttpStatus.OK);
	}

	@RequestMapping(method = RequestMethod.POST, consumes = MediaType.APPLICATION_JSON_VALUE)
	@ResponseBody
	public ResponseEntity<?> create(@RequestBody CarScheduleEntity carSchedule, UriComponentsBuilder uriBuilder) {
		//调用JSR303 Bean Validator进行校验，如果出错返回含400错误码及json格式的错误信息.
		Set<ConstraintViolation<CarScheduleEntity>> failures = validator.validate(carSchedule);
		if (!failures.isEmpty()) {
			return new ResponseEntity(BeanValidators.extractPropertyAndMessage(failures), HttpStatus.BAD_REQUEST);
		}

		//保存
		try{
			carScheduleService.save(carSchedule);
		} catch (Exception e) {
			e.printStackTrace();
			return new ResponseEntity(HttpStatus.NO_CONTENT);
		}
		//按照Restful风格约定，创建指向新任务的url, 也可以直接返回id或对象.
		String id = carSchedule.getId();
		URI uri = uriBuilder.path("/rest/carScheduleController/" + id).build().toUri();
		HttpHeaders headers = new HttpHeaders();
		headers.setLocation(uri);

		return new ResponseEntity(headers, HttpStatus.CREATED);
	}

	@RequestMapping(value = "/{id}", method = RequestMethod.PUT, consumes = MediaType.APPLICATION_JSON_VALUE)
	public ResponseEntity<?> update(@RequestBody CarScheduleEntity carSchedule) {
		//调用JSR303 Bean Validator进行校验，如果出错返回含400错误码及json格式的错误信息.
		Set<ConstraintViolation<CarScheduleEntity>> failures = validator.validate(carSchedule);
		if (!failures.isEmpty()) {
			return new ResponseEntity(BeanValidators.extractPropertyAndMessage(failures), HttpStatus.BAD_REQUEST);
		}

		//保存
		try{
			carScheduleService.saveOrUpdate(carSchedule);
		} catch (Exception e) {
			e.printStackTrace();
			return new ResponseEntity(HttpStatus.NO_CONTENT);
		}

		//按Restful约定，返回204状态码, 无内容. 也可以返回200状态码.
		return new ResponseEntity(HttpStatus.NO_CONTENT);
	}

	@RequestMapping(value = "/{id}", method = RequestMethod.DELETE)
	@ResponseStatus(HttpStatus.NO_CONTENT)
	public void delete(@PathVariable("id") String id) {
		carScheduleService.deleteEntityById(CarScheduleEntity.class, id);
	}
}