meidu2021/daju/target/classes/cn/com/lzt/warehouse/controller/WarehouseController.java

package cn.com.lzt.warehouse.controller;

import cn.com.lzt.tools.SaltUtil;
import cn.com.lzt.warehouse.entity.WarehouseEntity;
import cn.com.lzt.warehouse.service.WarehouseServiceI;

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

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.RequestMapping;
import org.springframework.web.bind.annotation.ResponseBody;
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.MyClassLoader;
import org.jeecgframework.core.util.StringUtil;
import org.jeecgframework.tag.core.easyui.TagUtil;
import org.jeecgframework.tag.vo.datatable.SortDirection;
import org.jeecgframework.web.system.pojo.base.TSBaseUser;
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.bind.annotation.RequestMethod;
import org.springframework.web.multipart.MultipartFile;
import org.springframework.web.multipart.MultipartHttpServletRequest;

import java.util.Map;

import org.jeecgframework.core.util.ExceptionUtil;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.ResponseStatus;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpStatus;
import org.jeecgframework.core.beanvalidator.BeanValidators;

import java.util.Set;
import javax.validation.ConstraintViolation;
import javax.validation.Validator;
import java.net.URI;

import org.springframework.http.MediaType;
import org.springframework.web.util.UriComponentsBuilder;

/**
 * @author onlineGenerator
 * @version V1.0
 * @Title: Controller
 * @Description: 仓库管理表
 * @date 2017-06-06 17:11:47
 */
@Controller
@RequestMapping("/warehouseController")
public class WarehouseController extends BaseController {
    /**
     * Logger for this class
     */
    private static final Logger logger = Logger.getLogger(WarehouseController.class);

    @Autowired
    private WarehouseServiceI warehouseService;
    @Autowired
    private SystemService systemService;
    @Autowired
    private Validator validator;


    /**
     * 仓库管理表列表 页面跳转
     *
     * @return
     */
    @RequestMapping(params = "list")
    public ModelAndView list(HttpServletRequest request) {
        return new ModelAndView("cn/com/lzt/warehouse/warehouseList");
    }

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

    @RequestMapping(params = "datagrid")
    public void datagrid(WarehouseEntity warehouse, HttpServletRequest request, HttpServletResponse response, DataGrid dataGrid) {
        CriteriaQuery cq = new CriteriaQuery(WarehouseEntity.class, dataGrid);
        if (!StringUtil.isEmpty(warehouse.getWarehouseName())) {
            warehouse.setWarehouseName("*" + warehouse.getWarehouseName() + "*");
        }
        //查询条件组装器
        org.jeecgframework.core.extend.hqlsearch.HqlGenerateUtil.installHql(cq, warehouse, request.getParameterMap());
        try {
            //自定义追加查询条件
            String[] unitstate = new String[]{Globals.Enable_Forbidden, Globals.Enable_Normal};
            cq.in("status", unitstate);
            cq.eq("deleteFlag", Globals.Delete_Normal.toString());
            cq.addOrder("createDate", SortDirection.desc);
        } catch (Exception e) {
            throw new BusinessException(e.getMessage());
        }
        cq.add();
        this.warehouseService.getDataGridReturn(cq, true);
        Collections.sort(dataGrid.getResults());
        TagUtil.datagrid(response, dataGrid);
    }

    /**
     * 删除仓库管理表
     *
     * @return
     */
    @RequestMapping(params = "doDel")
    @ResponseBody
    public AjaxJson doDel(WarehouseEntity warehouse, HttpServletRequest request) {
        String message = null;
        AjaxJson j = new AjaxJson();
        warehouse = systemService.getEntity(WarehouseEntity.class, warehouse.getId());
        message = "仓库管理表删除成功";
        try {
            message = "仓库：" + warehouse.getWarehouseName() + "删除成功";
            warehouse.setDeleteFlag(Globals.Delete_Forbidden.toString());
            warehouseService.updateEntitie(warehouse);
            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 = "lock")
    @ResponseBody
    public AjaxJson lock(WarehouseEntity warehouse, HttpServletRequest request) {
        String message = null;
        AjaxJson j = new AjaxJson();
        message = "停用仓库成功";
        String lockValue = request.getParameter("lockvalue");
        warehouse = warehouseService.get(WarehouseEntity.class, warehouse.getId());
        try {
            warehouse.setStatus(lockValue);
            warehouseService.updateEntitie(warehouse);
            systemService.addLog(message, Globals.Log_Type_UPDATE, Globals.Log_Leavel_INFO);
        } catch (Exception e) {
            e.printStackTrace();
            message = "停用仓库失败";
        }
        j.setMsg(message);
        return j;
    }

    /**
     * 自定义按钮-[启用仓库]业务
     *
     * @return
     */
    @RequestMapping(params = "unLock")
    @ResponseBody
    public AjaxJson unLock(WarehouseEntity warehouse, HttpServletRequest request) {
        String message = null;
        AjaxJson j = new AjaxJson();
        message = "启用仓库成功";
        String lockValue = request.getParameter("lockvalue");
        warehouse = warehouseService.get(WarehouseEntity.class, warehouse.getId());
        try {
            warehouse.setStatus(lockValue);
            warehouseService.updateEntitie(warehouse);
            systemService.addLog(message, Globals.Log_Type_UPDATE, Globals.Log_Leavel_INFO);
        } catch (Exception e) {
            e.printStackTrace();
            message = "启用仓库失败";
        }
        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(",")) {
                WarehouseEntity warehouse = systemService.getEntity(WarehouseEntity.class,
                        id
                );
                warehouseService.delete(warehouse);
                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(WarehouseEntity warehouse, HttpServletRequest request) {
        String message = null;
        AjaxJson j = new AjaxJson();
        message = "仓库档案添加成功";
        String warehouseCode = SaltUtil.randomSalt(10);
        try {
            String projectid = warehouse.getProjectId();
            if (projectid.indexOf(",") > 0) {
                projectid = projectid.replace(",", "");
                warehouse.setProjectId(projectid);
            }
            warehouse.setStatus(Globals.Enable_Normal);// 启用
            warehouse.setDeleteFlag(Globals.Delete_Normal.toString());// 正常
            warehouse.setWarehouseCode(warehouseCode);
            warehouseService.save(warehouse);
            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;
    }

    /**
     * 更新仓库管理表
     *
     * @param ids
     * @return
     */
    @RequestMapping(params = "doUpdate")
    @ResponseBody
    public AjaxJson doUpdate(WarehouseEntity warehouse, HttpServletRequest request) {
        String message = null;
        AjaxJson j = new AjaxJson();
        message = "仓库管理表更新成功";
        WarehouseEntity t = warehouseService.get(WarehouseEntity.class, warehouse.getId());
        try {
            String projectid = warehouse.getProjectId();
            if (projectid.indexOf(",") > 0) {
                projectid = projectid.replace(",", "");
                warehouse.setProjectId(projectid);
            }
            MyBeanUtils.copyBeanNotNull2Bean(warehouse, t);
            warehouseService.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 = "doBtnenable")
    @ResponseBody
    public AjaxJson doBtnenable(WarehouseEntity warehouse, HttpServletRequest request) {
        String message = null;
        AjaxJson j = new AjaxJson();
        message = "停用仓库成功";
        WarehouseEntity t = warehouseService.get(WarehouseEntity.class, warehouse.getId());
        try {
            warehouseService.doBtnenableBus(t);
            systemService.addLog(message, Globals.Log_Type_UPDATE, Globals.Log_Leavel_INFO);
        } catch (Exception e) {
            e.printStackTrace();
            message = "停用仓库失败";
        }
        j.setMsg(message);
        return j;
    }

    /**
     * 仓库管理表新增页面跳转
     *
     * @return
     */
    @RequestMapping(params = "goAdd")
    public ModelAndView goAdd(WarehouseEntity warehouse, HttpServletRequest req) {
        if (StringUtil.isNotEmpty(warehouse.getId())) {
            warehouse = warehouseService.getEntity(WarehouseEntity.class, warehouse.getId());
            req.setAttribute("warehousePage", warehouse);
        }
        return new ModelAndView("cn/com/lzt/warehouse/warehouse-add");
    }

    /**
     * 仓库管理表编辑页面跳转
     *
     * @return
     */
    @RequestMapping(params = "goUpdate")
    public ModelAndView goUpdate(WarehouseEntity warehouse, HttpServletRequest req) {
        if (StringUtil.isNotEmpty(warehouse.getId())) {
            warehouse = warehouseService.getEntity(WarehouseEntity.class, warehouse.getId());
            setReferName(req, warehouse);
            String isView = req.getParameter("viewFlag");
            req.setAttribute("viewFlag", isView);
            req.setAttribute("entity", warehouse);
        }
        return new ModelAndView("cn/com/lzt/warehouse/warehouse-update");
    }

    // 获取参照名称
    public void setReferName(HttpServletRequest req, WarehouseEntity warehouse) {
        TSBaseUser user = systemService.findUniqueByProperty(TSBaseUser.class, "id", warehouse.getResponsiblePerson());
        // 负责人
        String handlerPersonName = "";
        if (user != null) {
            handlerPersonName = user.getRealName();
            req.setAttribute("realName", handlerPersonName);
        }
    }

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

    /**
     * 导出excel
     *
     * @param request
     * @param response
     */
    @SuppressWarnings({"rawtypes", "unchecked"})
    @RequestMapping(params = "exportXls")
    public String exportXls(WarehouseEntity warehouse, HttpServletRequest request, HttpServletResponse response
            , DataGrid dataGrid, ModelMap modelMap) {
        CriteriaQuery cq = new CriteriaQuery(WarehouseEntity.class, dataGrid);
        //自定义追加查询条件
        String[] unitstate = new String[]{Globals.Enable_Forbidden, Globals.Enable_Normal};
        cq.in("status", unitstate);
        cq.eq("deleteFlag", Globals.Delete_Normal.toString());
        org.jeecgframework.core.extend.hqlsearch.HqlGenerateUtil.installHql(cq, warehouse, request.getParameterMap());
        List<WarehouseEntity> warehouses = this.warehouseService.getListByCriteriaQuery(cq, false);
        Class cls = MyClassLoader.getClassByScn("cn.com.lzt.warehouse.entity.WarehouseEntity");
        List<WarehouseEntity> result = new ArrayList<WarehouseEntity>();
        try {
            result = systemService.dealResultShowText4Entities(warehouses, cls);
        } catch (Exception e) {
            e.printStackTrace();
        }
        modelMap.put(NormalExcelConstants.FILE_NAME, "仓库管理表");
        modelMap.put(NormalExcelConstants.CLASS, WarehouseEntity.class);
        modelMap.put(NormalExcelConstants.PARAMS, new ExportParams("仓库管理表列表", "导出人:" + ResourceUtil.getSessionUserName().getRealName(),
                "导出信息"));
        modelMap.put(NormalExcelConstants.DATA_LIST, result);
        return NormalExcelConstants.JEECG_EXCEL_VIEW;
    }

    /**
     * 导出excel 使模板
     *
     * @param request
     * @param response
     */
    @RequestMapping(params = "exportXlsByT")
    public String exportXlsByT(WarehouseEntity warehouse, HttpServletRequest request, HttpServletResponse response
            , DataGrid dataGrid, ModelMap modelMap) {
        modelMap.put(NormalExcelConstants.FILE_NAME, "仓库管理表");
        modelMap.put(NormalExcelConstants.CLASS, WarehouseEntity.class);
        modelMap.put(NormalExcelConstants.PARAMS, new ExportParams("仓库管理表列表", "导出人:" + ResourceUtil.getSessionUserName().getRealName(),
                "导出信息"));
        modelMap.put(NormalExcelConstants.DATA_LIST, new ArrayList());
        return NormalExcelConstants.JEECG_EXCEL_VIEW;
    }

    @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<WarehouseEntity> listWarehouseEntitys = ExcelImportUtil.importExcel(file.getInputStream(), WarehouseEntity.class, params);
                for (WarehouseEntity warehouse : listWarehouseEntitys) {
                    warehouseService.save(warehouse);
                }
                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<WarehouseEntity> list() {
        List<WarehouseEntity> listWarehouses = warehouseService.getList(WarehouseEntity.class);
        return listWarehouses;
    }

    @RequestMapping(value = "/{id}", method = RequestMethod.GET)
    @ResponseBody
    public ResponseEntity<?> get(@PathVariable("id") String id) {
        WarehouseEntity task = warehouseService.get(WarehouseEntity.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 WarehouseEntity warehouse, UriComponentsBuilder uriBuilder) {
        //调用JSR303 Bean Validator进行校验，如果出错返回含400错误码及json格式的错误信息.
        Set<ConstraintViolation<WarehouseEntity>> failures = validator.validate(warehouse);
        if (!failures.isEmpty()) {
            return new ResponseEntity(BeanValidators.extractPropertyAndMessage(failures), HttpStatus.BAD_REQUEST);
        }

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

        //保存
        try {
            warehouseService.saveOrUpdate(warehouse);
        } 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) {
        warehouseService.deleteEntityById(WarehouseEntity.class, id);
    }
}