package com.xt.dsp.controller; import java.util.ArrayList; import java.util.Date; import java.util.List; import org.h2.util.StringUtils; import org.springframework.beans.BeanUtils; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Controller; import org.springframework.ui.Model; import org.springframework.web.bind.annotation.RequestMapping; import org.springframework.web.bind.annotation.ResponseBody; import com.xt.dsp.common.TreeNode; import com.xt.dsp.common.util.CommonUtil; import com.xt.dsp.model.CodeBean; import com.xt.dsp.service.CodeService; import com.xt.dsp.vo.CodeVo; import com.yuanxd.tools.pagehelper.PageHelper; import com.yuanxd.tools.pagehelper.PageInfo; @Controller @RequestMapping("code") public class CodeCtl { @Autowired private CodeService codeService; /** * 主画面 * * @return */ @RequestMapping("main") public String main() { return "sys/code/code"; } /** * 初始化页面加载数据 */ @RequestMapping("initCode") @ResponseBody public PageInfo initCode(Model model, CodeVo vo) { // 初始化参数 if (vo.getPage() < 1) { vo.setPage(1); } if (vo.getRows() < 1) { vo.setRows(1); } String pCode = vo.getpCode(); if (StringUtils.isNullOrEmpty(vo.getpCode())) { pCode = "00"; } PageHelper.startPage(vo.getPage(), vo.getRows()); List list = codeService.selectByPCode(pCode); PageInfo pageResult = new PageInfo<>(list); return pageResult; } /** * 主画面 * * @return */ @RequestMapping("initTree") @ResponseBody public List initTree() { List beanList = codeService.selectAll(); List treeList = new ArrayList(); TreeNode node = null; for(CodeBean bean : beanList) { node = new TreeNode(); node.setId(bean.getCode()); node.settId(bean.getCode()); node.setName(bean.getName()); node.setpId(bean.getpCode()); node.addAttribute("id", bean.getId()); node.addAttribute("code", bean.getCode()); if (StringUtils.isNullOrEmpty(bean.getpCode())) { node.setOpen("true"); node.setIconSkin("home"); } else { node.setOpen("false"); node.setIconSkin("child"); } treeList.add(node); } return treeList; } /** * 保存实体 * * @param vo * @return */ @RequestMapping("save") @ResponseBody public CodeBean saveCode(Model model, CodeVo vo) { CodeBean saveBean = new CodeBean(); // 新增 if (StringUtils.isNullOrEmpty(vo.getId())) { BeanUtils.copyProperties(vo, saveBean); saveBean.setId(CommonUtil.getUUID()); saveBean.setXh(Long.parseLong(vo.getXh())); saveBean.setUpdateTime(new Date()); codeService.insert(saveBean); } else { // 修改 saveBean = codeService.selectByPrimaryKey(vo.getId()); saveBean.setName(vo.getName()); saveBean.setCode(vo.getCode()); saveBean.setMapName(vo.getMapName()); saveBean.setMapCode(vo.getMapCode()); saveBean.setValid(vo.getValid()); saveBean.setXh(Long.parseLong(vo.getXh())); saveBean.setRemark(vo.getRemark()); saveBean.setUpdateTime(new Date()); codeService.updateByPrimaryKey(saveBean); } return saveBean; } /** * 初始化页面加载数据 */ @RequestMapping("initEditCode") @ResponseBody public CodeBean initEditCode(Model model, String id) { CodeBean bean = codeService.selectByPrimaryKey(id); return bean; } /** * 删除数据 */ @RequestMapping("delCode") @ResponseBody public int delCode(Model model, String ids) { int cnt = 0; if (!StringUtils.isNullOrEmpty(ids)) { String[] idArr = ids.split(","); for(String id : idArr) { if(!StringUtils.isNullOrEmpty(id)) { cnt += codeService.deleteByPrimaryKey(id); } } } return cnt; } /** * 获取任务下拉数据 */ @RequestMapping("getCodeForDdl") @ResponseBody public List getCodeForDdl(Model model) { List list = codeService.selectAll(); return list; } }