meidu2021/daju/target/classes/cn/com/lzt/budget/dimfunction/controller/BudgetDimFunctionController.java

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

import cn.com.lzt.budget.data.dto.BudgetRedisCellData;
import cn.com.lzt.budget.data.util.BudgetConstant;
import cn.com.lzt.budget.dimfunction.entity.BudgetDimFunctionDetailEntity;
import cn.com.lzt.budget.dimfunction.entity.BudgetDimFunctionEntity;
import cn.com.lzt.budget.dimfunction.service.BudgetDimFunctionServiceI;

import java.util.*;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import cn.com.lzt.budget.inst.entity.BudgetInstEntity;
import cn.com.lzt.budget.tempsheet.entity.BudgetTempSheetEntity;
import org.apache.commons.lang.xwork.StringUtils;
import org.apache.log4j.Logger;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.ui.ModelMap;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.servlet.ModelAndView;

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.StringUtil;
import org.jeecgframework.tag.core.easyui.TagUtil;
import org.jeecgframework.web.system.service.SystemService;
import org.jeecgframework.core.util.MyBeanUtils;

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.core.util.ResourceUtil;
import java.io.IOException;

import org.springframework.web.multipart.MultipartFile;
import org.springframework.web.multipart.MultipartHttpServletRequest;

import org.jeecgframework.core.util.ExceptionUtil;

import org.springframework.http.ResponseEntity;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpStatus;
import org.jeecgframework.core.beanvalidator.BeanValidators;

import javax.validation.ConstraintViolation;
import javax.validation.Validator;
import java.net.URI;
import org.springframework.http.MediaType;
import org.springframework.web.util.UriComponentsBuilder;

/**   
 * @Title: Controller  
 * @Description: 维度公式
 * @author onlineGenerator
 * @date 2020-08-08 20:20:45
 * @version V1.0   
 *
 */
@Controller
@RequestMapping("/budgetDimFunctionController")
public class BudgetDimFunctionController extends BaseController {
	/**
	 * Logger for this class
	 */
	private static final Logger logger = Logger.getLogger(BudgetDimFunctionController.class);

	@Autowired
	private BudgetDimFunctionServiceI budgetDimFunctionService;
	@Autowired
	private SystemService systemService;
	@Autowired
	private Validator validator;


	@RequestMapping(params = "goDimFunction")
	public ModelAndView goDimFunction(BudgetDimFunctionEntity functionEntity, HttpServletRequest req) {
		String id = functionEntity.getId();
		if(StringUtils.isNotBlank(id)){
			functionEntity = systemService.getEntity(BudgetDimFunctionEntity.class,id );
		}else {
			functionEntity = new BudgetDimFunctionEntity();
			String measureId = req.getParameter("measureId");
            functionEntity.setMeasureId(measureId);
			systemService.save(functionEntity);
			id=functionEntity.getId();
		}
		req.setAttribute("entity",functionEntity);
		String type = req.getParameter("type");
		if(StringUtils.equals(type,"cost")) {
			String hql = "from BudgetDimFunctionDetailEntity where fkId=? ";
			List<BudgetDimFunctionDetailEntity> detailEntityList =systemService.findHql(hql,id);
			BudgetDimFunctionDetailEntity detailEntity = null;
			if(!detailEntityList.isEmpty()){
				detailEntity=detailEntityList.get(0);
			}
			if(detailEntity==null){
				detailEntity = new BudgetDimFunctionDetailEntity();
				detailEntity.setFkId(id);
				systemService.save(detailEntity);
			}
			req.setAttribute("detailEntity", detailEntity);
			return new ModelAndView("cn/com/lzt/budget/dimfunction/dimFunction-costbreakdown");
		}else {
			return new ModelAndView("cn/com/lzt/budget/dimfunction/dimFunction");
		}
	}


	/**
	 * 维度公式列表 页面跳转
	 * 
	 * @return
	 */
	@RequestMapping(params = "list")
	public ModelAndView list(HttpServletRequest request) {
		return new ModelAndView("cn/com/lzt/budget/dimfunction/budgetDimFunctionList");
	}

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

	@RequestMapping(params = "datagrid")
	public void datagrid(BudgetDimFunctionEntity budgetDemoFunction, HttpServletRequest request, HttpServletResponse response, DataGrid dataGrid) {
		CriteriaQuery cq = new CriteriaQuery(BudgetDimFunctionEntity.class, dataGrid);
		//查询条件组装器
		org.jeecgframework.core.extend.hqlsearch.HqlGenerateUtil.installHql(cq, budgetDemoFunction, request.getParameterMap());
		try{
		//自定义追加查询条件
		}catch (Exception e) {
			throw new BusinessException(e.getMessage());
		}
		cq.add();
		this.budgetDimFunctionService.getDataGridReturn(cq, true);
		TagUtil.datagrid(response, dataGrid);
	}
	
	/**
	 * 删除维度公式
	 * 
	 * @return
	 */
	@RequestMapping(params = "doDel")
	@ResponseBody
	public AjaxJson doDel(BudgetDimFunctionEntity budgetDemoFunction, HttpServletRequest request) {
		String message = null;
		AjaxJson j = new AjaxJson();
		budgetDemoFunction = systemService.getEntity(BudgetDimFunctionEntity.class, budgetDemoFunction.getId());
		message = "维度公式删除成功";
		try{
			budgetDimFunctionService.delete(budgetDemoFunction);
			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(",")){
				BudgetDimFunctionEntity budgetDemoFunction = systemService.getEntity(BudgetDimFunctionEntity.class,
				id
				);
				budgetDimFunctionService.delete(budgetDemoFunction);
				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(BudgetDimFunctionEntity budgetDemoFunction, HttpServletRequest request) {
		String message = null;
		AjaxJson j = new AjaxJson();
		message = "维度公式添加成功";
		try{
			budgetDimFunctionService.save(budgetDemoFunction);
			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(BudgetDimFunctionEntity budgetDemoFunction, HttpServletRequest request) {
		String message = null;
		AjaxJson j = new AjaxJson();
		message = "维度公式更新成功";
		BudgetDimFunctionEntity t = budgetDimFunctionService.get(BudgetDimFunctionEntity.class, budgetDemoFunction.getId());
		try {
			MyBeanUtils.copyBeanNotNull2Bean(budgetDemoFunction, t);
			budgetDimFunctionService.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(BudgetDimFunctionEntity budgetDemoFunction, HttpServletRequest req) {
		if (StringUtil.isNotEmpty(budgetDemoFunction.getId())) {
			budgetDemoFunction = budgetDimFunctionService.getEntity(BudgetDimFunctionEntity.class, budgetDemoFunction.getId());
			req.setAttribute("budgetDemoFunctionPage", budgetDemoFunction);
		}
		return new ModelAndView("cn/com/lzt/budget/dimfunction/budgetDimFunction-add");
	}
	/**
	 * 维度公式编辑页面跳转
	 * 
	 * @return
	 */
	@RequestMapping(params = "goUpdate")
	public ModelAndView goUpdate(BudgetDimFunctionEntity budgetDemoFunction, HttpServletRequest req) {
		if (StringUtil.isNotEmpty(budgetDemoFunction.getId())) {
			budgetDemoFunction = budgetDimFunctionService.getEntity(BudgetDimFunctionEntity.class, budgetDemoFunction.getId());
			req.setAttribute("budgetDemoFunctionPage", budgetDemoFunction);
		}
		return new ModelAndView("cn/com/lzt/budget/dimfunction/budgetDimFunction-update");
	}
	
	/**
	 * 导入功能跳转
	 * 
	 * @return
	 */
	@RequestMapping(params = "upload")
	public ModelAndView upload(HttpServletRequest req) {
		req.setAttribute("controller_name","budgetDimFunctionController");
		return new ModelAndView("common/upload/pub_excel_upload");
	}
	
	/**
	 * 导出excel
	 * 
	 * @param request
	 * @param response
	 */
	@RequestMapping(params = "exportXls")
	public String exportXls(BudgetDimFunctionEntity budgetDemoFunction, HttpServletRequest request, HttpServletResponse response
			, DataGrid dataGrid, ModelMap modelMap) {
		CriteriaQuery cq = new CriteriaQuery(BudgetDimFunctionEntity.class, dataGrid);
		org.jeecgframework.core.extend.hqlsearch.HqlGenerateUtil.installHql(cq, budgetDemoFunction, request.getParameterMap());
		List<BudgetDimFunctionEntity> budgetDemoFunctions = this.budgetDimFunctionService.getListByCriteriaQuery(cq,false);
		modelMap.put(NormalExcelConstants.FILE_NAME,"维度公式");
		modelMap.put(NormalExcelConstants.CLASS,BudgetDimFunctionEntity.class);
		modelMap.put(NormalExcelConstants.PARAMS,new ExportParams("维度公式列表", "导出人:"+ResourceUtil.getSessionUser().getRealName(),
			"导出信息"));
		modelMap.put(NormalExcelConstants.DATA_LIST,budgetDemoFunctions);
		return NormalExcelConstants.JEECG_EXCEL_VIEW;
	}
	/**
	 * 导出excel 使模板
	 * 
	 * @param request
	 * @param response
	 */
	@RequestMapping(params = "exportXlsByT")
	public String exportXlsByT(BudgetDimFunctionEntity budgetDemoFunction, HttpServletRequest request, HttpServletResponse response
			, DataGrid dataGrid, ModelMap modelMap) {
    	modelMap.put(NormalExcelConstants.FILE_NAME,"维度公式");
    	modelMap.put(NormalExcelConstants.CLASS,BudgetDimFunctionEntity.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<BudgetDimFunctionEntity> listBudgetDemoFunctionEntitys = ExcelImportUtil.importExcel(file.getInputStream(),BudgetDimFunctionEntity.class,params);
				for (BudgetDimFunctionEntity budgetDemoFunction : listBudgetDemoFunctionEntitys) {
					budgetDimFunctionService.save(budgetDemoFunction);
				}
				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<BudgetDimFunctionEntity> list() {
		List<BudgetDimFunctionEntity> listBudgetDemoFunctions=budgetDimFunctionService.getList(BudgetDimFunctionEntity.class);
		return listBudgetDemoFunctions;
	}
	
	@RequestMapping(value = "/{id}", method = RequestMethod.GET)
	@ResponseBody
	public ResponseEntity<?> get(@PathVariable("id") String id) {
		BudgetDimFunctionEntity task = budgetDimFunctionService.get(BudgetDimFunctionEntity.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 BudgetDimFunctionEntity budgetDemoFunction, UriComponentsBuilder uriBuilder) {
		//调用JSR303 Bean Validator进行校验，如果出错返回含400错误码及json格式的错误信息.
		Set<ConstraintViolation<BudgetDimFunctionEntity>> failures = validator.validate(budgetDemoFunction);
		if (!failures.isEmpty()) {
			return new ResponseEntity(BeanValidators.extractPropertyAndMessage(failures), HttpStatus.BAD_REQUEST);
		}

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

		//保存
		try{
			budgetDimFunctionService.saveOrUpdate(budgetDemoFunction);
		} 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) {
		budgetDimFunctionService.deleteEntityById(BudgetDimFunctionEntity.class, id);
	}

	@RequestMapping(params = "copy")
	@ResponseBody
	public AjaxJson copy(@RequestParam("functionId") String functionId, HttpServletRequest request) {
		String message = null;
		AjaxJson j = new AjaxJson();
		String newFunctionId = budgetDimFunctionService.copy(functionId);
		j.setObj(newFunctionId);
		j.setMsg(message);
		return j;
	}
	@RequestMapping(params = "effect")
	@ResponseBody
	public AjaxJson effect(HttpServletRequest request) {
		String message = null;
		AjaxJson j = new AjaxJson();
		String data = request.getParameter("data");
		String functionFrom = request.getParameter("functionFrom");
		List<String> keyList = new ArrayList<>();
		String id = null;
		String[] dataArr = data.split(",");
		if(StringUtils.equals(functionFrom,"tempsheet")){
			String instId = request.getParameter("instId");
			BudgetInstEntity entity = systemService.getEntity(BudgetInstEntity.class,instId);
			String tableName = entity.getInstMeasureTableName();
			String sql = "select id from `%s` where row=? and col=? and expect=? ";
			sql = String.format(sql,tableName);
			for(String pos :dataArr){
				String[] posArr = pos.split("#");
				Integer row = Integer.valueOf(posArr[0]);
				Integer col = Integer.valueOf(posArr[1]);
				List<String> idList = systemService.findListbySql(sql,row,col,BudgetConstant.EXPECT_YES);
				if(!idList.isEmpty()){
					id = idList.get(0);
					keyList.add(id);
				}
			}
		}else {
			String sql = "select id from t_b_budget_cost_breakdown_measure where cost_breakdown_id =? and row=? and col=? and expect=? ";
			String costEntityId = request.getParameter("costEntityId");
			for(String pos :dataArr) {
				String[] posArr = pos.split("#");
				Integer row = Integer.valueOf(posArr[0]);
				Integer col = Integer.valueOf(posArr[1]);
				List<String> idList = systemService.findListbySql(sql, costEntityId,row, col, BudgetConstant.EXPECT_YES);
				if (!idList.isEmpty()) {
					id = idList.get(0);
					keyList.add(id);
				}
			}
		}
		if(!keyList.isEmpty()){
			Set<BudgetRedisCellData> set = budgetDimFunctionService.effect(keyList);
			j.setObj(set);
		}
//		j.setObj(newFunctionId);
		j.setMsg(message);
		return j;
	}


}