meidu2021/daju/target/classes/cn/com/lzt/budget/measure/controller/BudgetMeasureController.java

package cn.com.lzt.budget.measure.controller;

import cn.com.lzt.budget.measure.entity.BudgetMeasureEntity;
import cn.com.lzt.budget.measure.service.BudgetMeasureServiceI;
import cn.com.lzt.budget.tempsheet.entity.BudgetTempSheetEntity;
import org.apache.commons.lang.StringUtils;
import org.apache.log4j.Logger;
import org.hibernate.criterion.Disjunction;
import org.hibernate.criterion.Restrictions;
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.*;
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.tag.vo.datatable.SortDirection;
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.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.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 2020-06-28 16:31:21
 * @version V1.0   
 *
 */
@Controller
@RequestMapping("/budgetMeasureController")
public class BudgetMeasureController extends BaseController {
	/**
	 * Logger for this class
	 */
	private static final Logger logger = Logger.getLogger(BudgetMeasureController.class);

	@Autowired
	private BudgetMeasureServiceI budgetMeasureService;
	@Autowired
	private SystemService systemService;
	@Autowired
	private Validator validator;
	


	/**
	 * 预算指标列表 页面跳转
	 * 
	 * @return
	 */
	@RequestMapping(params = "list")
	public ModelAndView list(HttpServletRequest request) {
		return new ModelAndView("cn/com/lzt/budget/measure/budgetMeasureList");
	}

	/**
	 * easyui AJAX请求数据
	 * 
	 * @param request
	 * @param response
	 * @param dataGrid
	 */

	@RequestMapping(params = "datagrid")
	public void datagrid(BudgetMeasureEntity budgetGoals, HttpServletRequest request, HttpServletResponse response, DataGrid dataGrid) {
		CriteriaQuery cq = new CriteriaQuery(BudgetMeasureEntity.class, dataGrid);
		if(StringUtil.isEmpty(budgetGoals.getId())){
			cq.isNull("parentId");
		}else{
			cq.eq("parentId", budgetGoals.getId());
			budgetGoals.setId(null);
		}
		//查询条件组装器
		org.jeecgframework.core.extend.hqlsearch.HqlGenerateUtil.installHql(cq, budgetGoals, request.getParameterMap());
		try{
		//自定义追加查询条件
		}catch (Exception e) {
			throw new BusinessException(e.getMessage());
		}
		cq.add();
		this.budgetMeasureService.getDataGridReturn(cq, false);
		TagUtil.treegrid(response, dataGrid);
	}
	
	/**
	 * 删除预算指标
	 * 
	 * @return
	 */
	@RequestMapping(params = "doDel")
	@ResponseBody
	public AjaxJson doDel(BudgetMeasureEntity budgetMeasure, HttpServletRequest request) {
		String message = null;
		AjaxJson j = new AjaxJson();
		budgetMeasure = systemService.getEntity(BudgetMeasureEntity.class, budgetMeasure.getId());
		message = "预算指标删除成功";
		try{
			budgetMeasureService.delete(budgetMeasure);
			systemService.addLog(message, Globals.Log_Type_DEL, Globals.Log_Leavel_INFO);
		}catch (IllegalStateException e){
			message =e.getMessage();
		}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(",")){
				BudgetMeasureEntity budgetGoals = systemService.getEntity(BudgetMeasureEntity.class,
				id
				);
				budgetMeasureService.delete(budgetGoals);
				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 = "doAdd")
	@ResponseBody
	public AjaxJson doAdd(BudgetMeasureEntity budgetGoals, HttpServletRequest request) {
		if(StringUtils.isNotBlank(budgetGoals.getId())){
			return  doUpdate(budgetGoals,request);
		}
		String message = null;
		AjaxJson j = new AjaxJson();
		message = "预算指标添加成功";
		try{
			String parentId = budgetGoals.getParentId();
			if(StringUtil.isEmpty(parentId)){
				budgetGoals.setParentId(null);
			}
			budgetMeasureService.save(budgetGoals);
			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;
	}
	
	/**
	 * 更新预算指标
	 * 
	 * @return
	 */
	@RequestMapping(params = "doUpdate")
	@ResponseBody
	public AjaxJson doUpdate(BudgetMeasureEntity budgetGoals, HttpServletRequest request) {
		String message = null;
		AjaxJson j = new AjaxJson();
		message = "预算指标更新成功";
		BudgetMeasureEntity t = budgetMeasureService.get(BudgetMeasureEntity.class, budgetGoals.getId());
		try {
			MyBeanUtils.copyBeanNotNull2Bean(budgetGoals, t);
			if(StringUtil.isEmpty(t.getParentId())){
				t.setParentId(null);
			}
			if(budgetGoals.getThousandth()==null) t.setThousandth(0);
			budgetMeasureService.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;
	}
	

	/**
	 * 预算指标新增页面跳转
	 * 
	 * @return
	 */
	@RequestMapping(params = "goAdd")
	public ModelAndView goAdd(BudgetMeasureEntity budgetGoals, HttpServletRequest req) {
		return new ModelAndView("cn/com/lzt/budget/measure/budgetMeasure-add");
	}
	/**
	 * 预算指标编辑页面跳转
	 * 
	 * @return
	 */
	@RequestMapping(params = "goUpdate")
	public ModelAndView goUpdate(BudgetMeasureEntity budgetGoals, HttpServletRequest req) {
		if (StringUtil.isNotEmpty(budgetGoals.getId())) {
			budgetGoals = budgetMeasureService.getEntity(BudgetMeasureEntity.class, budgetGoals.getId());
			req.setAttribute("entity", budgetGoals);
		}
		return new ModelAndView("cn/com/lzt/budget/measure/budgetMeasure-add");
	}
	
	/**
	 * 导入功能跳转
	 * 
	 * @return
	 */
	@RequestMapping(params = "upload")
	public ModelAndView upload(HttpServletRequest req) {
		req.setAttribute("controller_name","budgetMeasureController");
		return new ModelAndView("common/upload/pub_excel_upload");
	}
	
	/**
	 * 导出excel
	 * 
	 * @param request
	 * @param response
	 */
	@RequestMapping(params = "exportXls")
	public String exportXls(BudgetMeasureEntity budgetGoals, HttpServletRequest request, HttpServletResponse response
			, DataGrid dataGrid, ModelMap modelMap) {
		CriteriaQuery cq = new CriteriaQuery(BudgetMeasureEntity.class, dataGrid);
		org.jeecgframework.core.extend.hqlsearch.HqlGenerateUtil.installHql(cq, budgetGoals, request.getParameterMap());
		List<BudgetMeasureEntity> budgetGoalss = this.budgetMeasureService.getListByCriteriaQuery(cq,false);
		modelMap.put(NormalExcelConstants.FILE_NAME,"预算指标");
		modelMap.put(NormalExcelConstants.CLASS,BudgetMeasureEntity.class);
		modelMap.put(NormalExcelConstants.PARAMS,new ExportParams("预算指标列表", "导出人:"+ResourceUtil.getSessionUser().getRealName(),
			"导出信息"));
		modelMap.put(NormalExcelConstants.DATA_LIST,budgetGoalss);
		return NormalExcelConstants.JEECG_EXCEL_VIEW;
	}
	/**
	 * 导出excel 使模板
	 * 
	 * @param request
	 * @param response
	 */
	@RequestMapping(params = "exportXlsByT")
	public String exportXlsByT(BudgetMeasureEntity budgetGoals, HttpServletRequest request, HttpServletResponse response
			, DataGrid dataGrid, ModelMap modelMap) {
    	modelMap.put(NormalExcelConstants.FILE_NAME,"预算指标");
    	modelMap.put(NormalExcelConstants.CLASS,BudgetMeasureEntity.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<BudgetMeasureEntity> listBudgetMeasureEntities = ExcelImportUtil.importExcel(file.getInputStream(),BudgetMeasureEntity.class,params);
				for (BudgetMeasureEntity budgetGoals : listBudgetMeasureEntities) {
					budgetMeasureService.save(budgetGoals);
				}
				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<BudgetMeasureEntity> list() {
		List<BudgetMeasureEntity> listBudgetGoalss=budgetMeasureService.getList(BudgetMeasureEntity.class);
		return listBudgetGoalss;
	}
	
	@RequestMapping(value = "/{id}", method = RequestMethod.GET)
	@ResponseBody
	public ResponseEntity<?> get(@PathVariable("id") String id) {
		BudgetMeasureEntity task = budgetMeasureService.get(BudgetMeasureEntity.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 BudgetMeasureEntity budgetGoals, UriComponentsBuilder uriBuilder) {
		//调用JSR303 Bean Validator进行校验，如果出错返回含400错误码及json格式的错误信息.
		Set<ConstraintViolation<BudgetMeasureEntity>> failures = validator.validate(budgetGoals);
		if (!failures.isEmpty()) {
			return new ResponseEntity(BeanValidators.extractPropertyAndMessage(failures), HttpStatus.BAD_REQUEST);
		}

		//保存
		try{
			budgetMeasureService.save(budgetGoals);
		} catch (Exception e) {
			e.printStackTrace();
			return new ResponseEntity(HttpStatus.NO_CONTENT);
		}
		//按照Restful风格约定，创建指向新任务的url, 也可以直接返回id或对象.
		String id = budgetGoals.getId();
		URI uri = uriBuilder.path("/rest/budgetMeasureController/" + 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 BudgetMeasureEntity budgetGoals) {
		//调用JSR303 Bean Validator进行校验，如果出错返回含400错误码及json格式的错误信息.
		Set<ConstraintViolation<BudgetMeasureEntity>> failures = validator.validate(budgetGoals);
		if (!failures.isEmpty()) {
			return new ResponseEntity(BeanValidators.extractPropertyAndMessage(failures), HttpStatus.BAD_REQUEST);
		}

		//保存
		try{
			budgetMeasureService.saveOrUpdate(budgetGoals);
		} 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) {
		budgetMeasureService.deleteEntityById(BudgetMeasureEntity.class, id);
	}

	@RequestMapping(params = "select")
	public ModelAndView select(HttpServletRequest request) {
		return new ModelAndView("cn/com/lzt/budget/measure/budgetMeasure-select");
	}

	/**
	 * easyui AJAX请求数据
	 *
	 * @param request
	 * @param response
	 * @param dataGrid
	 */

	@RequestMapping(params = "selectData")
	public void selectData(BudgetMeasureEntity budgetGoals, HttpServletRequest request, HttpServletResponse response, DataGrid dataGrid) {
		CriteriaQuery cq = new CriteriaQuery(BudgetMeasureEntity.class, dataGrid);
		if(StringUtil.isEmpty(budgetGoals.getId())){
			cq.isNull("parentId");
		}else{
			cq.eq("parentId", budgetGoals.getId());
			budgetGoals.setId(null);
		}
		//查询条件组装器
		org.jeecgframework.core.extend.hqlsearch.HqlGenerateUtil.installHql(cq, budgetGoals, request.getParameterMap());
		try{
			//自定义追加查询条件
		}catch (Exception e) {
			throw new BusinessException(e.getMessage());
		}
		cq.add();
		this.budgetMeasureService.getDataGridReturn(cq, false);
		TagUtil.treegrid(response, dataGrid);
	}

	@RequestMapping(params = "selectByTempSheet")
	public ModelAndView selectByTempSheet(HttpServletRequest request) {
		return new ModelAndView("cn/com/lzt/budget/measure/budgetMeasure-selectByTempSheet");
	}

	/**
	 * easyui AJAX请求数据
	 *
	 * @param request
	 * @param response
	 * @param dataGrid
	 */

	@RequestMapping(params = "selectByTempSheetDatagrid")
	public void selectByTempSheetDatagrid(BudgetMeasureEntity budgetMeasureEntity, HttpServletRequest request, HttpServletResponse response, DataGrid dataGrid) {
		CriteriaQuery cq = new CriteriaQuery(BudgetMeasureEntity.class, dataGrid);
		//查询条件组装器
//		org.jeecgframework.core.extend.hqlsearch.HqlGenerateUtil.installHql(cq, budgetMeasureEntity, request.getParameterMap());
		try{
			String type = request.getParameter("type");
			if(StringUtils.equals(type,"cost")){
				String costBreakdownId = request.getParameter("costBreakdownId");
				String sql = "select  distinct measure_id from t_b_budget_cost_breakdown_measure   " +
						" where cost_breakdown_id=? ";
				List<String> idList = systemService.findListbySql(sql, costBreakdownId);
				if (idList.isEmpty()) {
					cq.eq("id", "-1");
				} else {
					cq.in("id", idList.toArray());
				}
			}else {
				String tempSheetId = request.getParameter("tempSheetId");
				BudgetTempSheetEntity tempSheetEntity = systemService.getEntity(BudgetTempSheetEntity.class, tempSheetId);
				String sql = "select distinct measure_id from %s where tempsheet_id='%s'";
				sql = String.format(sql, tempSheetEntity.getTempsheetMeasureTable(), tempSheetId);
				List<String> idList = systemService.findListbySql(sql);
				if (idList.isEmpty()) {
					cq.eq("id", "-1");
				} else {
					cq.in("id", idList.toArray());
				}
			}
		}catch (Exception e) {
			throw new BusinessException(e.getMessage());
		}
		cq.add();
		this.systemService.getDataGridReturn(cq, false);
		List<BudgetMeasureEntity> list = dataGrid.getResults();
		for(BudgetMeasureEntity measureEntity:list){
			String parentId = measureEntity.getParentId();
			if(StringUtils.isNotBlank(parentId)){
				BudgetMeasureEntity parent= systemService.get(BudgetMeasureEntity.class,parentId);
				String showName = measureEntity.getName();
				if(parent!=null){
					showName=parent.getName()+"-"+showName;
				}
				measureEntity.setShowName(showName);
			}
		}
		TagUtil.treegrid(response, dataGrid);
	}

	@RequestMapping(params = "getCode")
	@ResponseBody
	public AjaxJson getCode( HttpServletRequest request) {
		AjaxJson j = new AjaxJson();
		String parentId = request.getParameter("parentId");
		String code = budgetMeasureService.getCode(parentId);
		j.setObj(code);
		return j;
	}

	@RequestMapping(params = "checkCode")
	@ResponseBody
	public AjaxJson checkCode( HttpServletRequest request) {
		String customCode = request.getParameter("customCode");
		String id = request.getParameter("id");
		if(id==null) id="";
		String sql = "select count(*) c from t_b_budget_measure where custom_code=? and id<>? ";
		Map<String,Object>  map =systemService.findOneForJdbc(sql,customCode,id);
		AjaxJson j = new AjaxJson();
		Integer count = oConvertUtils.getInt(map.get("c"),0);
		if(count>0) {
			j.setSuccess(false);
			j.setMsg("编码已存在");
		}
		return j;
	}

	/**
	 * 财务补录选择
	 * @param request
	 * @return
	 */
	@RequestMapping(params = "select4financeWriteback")
	public ModelAndView select4financeWriteback(HttpServletRequest request) {
		return new ModelAndView("cn/com/lzt/budget/measure/budgetMeasure-select4financeWriteback");
	}

	/**
	 * easyui AJAX请求数据
	 *
	 * @param request
	 * @param response
	 * @param dataGrid
	 */

	@RequestMapping(params = "select4financeWritebackDatagrid")
	public void select4financeWritebackDatagrid(BudgetMeasureEntity budgetMeasureEntity, HttpServletRequest request, HttpServletResponse response, DataGrid dataGrid) {
		CriteriaQuery cq = new CriteriaQuery(BudgetMeasureEntity.class, dataGrid);
		//查询条件组装器
		if(StringUtils.isNotBlank(budgetMeasureEntity.getName())){
			cq.like("name","%"+budgetMeasureEntity.getName()+"%" );
			budgetMeasureEntity.setName(null);
		}
		org.jeecgframework.core.extend.hqlsearch.HqlGenerateUtil.installHql(cq, budgetMeasureEntity, request.getParameterMap());

		try{
			Disjunction dis = Restrictions.disjunction();
			dis.add(Restrictions.like("code", "A01A01___" ));
			dis.add(Restrictions.eq("code", "A01A01A05A01" ));
			dis.add(Restrictions.eq("code", "A01A01A05A02" ));
			dis.add(Restrictions.eq("code", "A01A01A05A03" ));

			dis.add(Restrictions.eq("code", "A01A02A01" ));
			dis.add(Restrictions.eq("code", "A01A02A02" ));

			dis.add(Restrictions.like("code", "A01A03___" ));
			dis.add(Restrictions.like("code", "A01A04___" ));
			dis.add(Restrictions.like("code", "A01A05_%" ));
			dis.add(Restrictions.like("code", "A01A06___" ));
			dis.add(Restrictions.eq("code", "A01A07" ));
			dis.add(Restrictions.like("code", "A01A08___" ));
			dis.add(Restrictions.like("code", "A01A09___" ));
			dis.add(Restrictions.like("code", "A01A10___" ));
			dis.add(Restrictions.like("code", "A01A11___" ));
			dis.add(Restrictions.like("code", "A01A13___" ));
			dis.add(Restrictions.like("code", "A01A14___" ));
			dis.add(Restrictions.eq("code", "A01A15" ));
			dis.add(Restrictions.like("code", "A01A16___" ));
			dis.add(Restrictions.eq("code", "A01A17" ));
			dis.add(Restrictions.like("code", "A01A18___" ));
			cq.add(dis);
			cq.notEq("code","A01A01A05" );

			//自定义追加查询条件
		}catch (Exception e) {
			throw new BusinessException(e.getMessage());
		}
		cq.add();
		this.budgetMeasureService.getDataGridReturn(cq, false);
		TagUtil.treegrid(response, dataGrid);
	}

}