content: package com.ruoyi.web.controller.demo.controller;
  
  import java.util.ArrayList;
  import java.util.LinkedHashMap;
  import java.util.List;
  import java.util.Map;
  import org.springframework.stereotype.Controller;
  import org.springframework.ui.ModelMap;
  import org.springframework.web.bind.annotation.GetMapping;
  import org.springframework.web.bind.annotation.PathVariable;
  import org.springframework.web.bind.annotation.PostMapping;
  import org.springframework.web.bind.annotation.RequestMapping;
  import org.springframework.web.bind.annotation.ResponseBody;
  import org.springframework.web.multipart.MultipartFile;
  import com.ruoyi.common.core.controller.BaseController;
  import com.ruoyi.common.core.domain.AjaxResult;
  import com.ruoyi.common.core.page.PageDomain;
  import com.ruoyi.common.core.page.TableDataInfo;
  import com.ruoyi.common.core.page.TableSupport;
  import com.ruoyi.common.core.text.Convert;
  import com.ruoyi.common.exception.BusinessException;
  import com.ruoyi.common.utils.StringUtils;
  import com.ruoyi.common.utils.poi.ExcelUtil;
  import com.ruoyi.web.controller.demo.domain.UserOperateModel;
  
  /**
   * 操作控制
   * 
   * @author ruoyi
   */
  @Controller
  @RequestMapping("/demo/operate")
  public class DemoOperateController extends BaseController
  {
      private String prefix = "demo/operate";
  
      private final static Map<Integer, UserOperateModel> users = new LinkedHashMap<Integer, UserOperateModel>();
      {
          users.put(1, new UserOperateModel(1, "1000001", "测试1", "0", "15888888888", "ry@qq.com", 150.0, "0"));
          users.put(2, new UserOperateModel(2, "1000002", "测试2", "1", "15666666666", "ry@qq.com", 180.0, "1"));
          users.put(3, new UserOperateModel(3, "1000003", "测试3", "0", "15666666666", "ry@qq.com", 110.0, "1"));
          users.put(4, new UserOperateModel(4, "1000004", "测试4", "1", "15666666666", "ry@qq.com", 220.0, "1"));
          users.put(5, new UserOperateModel(5, "1000005", "测试5", "0", "15666666666", "ry@qq.com", 140.0, "1"));
          users.put(6, new UserOperateModel(6, "1000006", "测试6", "1", "15666666666", "ry@qq.com", 330.0, "1"));
          users.put(7, new UserOperateModel(7, "1000007", "测试7", "0", "15666666666", "ry@qq.com", 160.0, "1"));
          users.put(8, new UserOperateModel(8, "1000008", "测试8", "1", "15666666666", "ry@qq.com", 170.0, "1"));
          users.put(9, new UserOperateModel(9, "1000009", "测试9", "0", "15666666666", "ry@qq.com", 180.0, "1"));
          users.put(10, new UserOperateModel(10, "1000010", "测试10", "0", "15666666666", "ry@qq.com", 210.0, "1"));
          users.put(11, new UserOperateModel(11, "1000011", "测试11", "1", "15666666666", "ry@qq.com", 110.0, "1"));
          users.put(12, new UserOperateModel(12, "1000012", "测试12", "0", "15666666666", "ry@qq.com", 120.0, "1"));
          users.put(13, new UserOperateModel(13, "1000013", "测试13", "1", "15666666666", "ry@qq.com", 380.0, "1"));
          users.put(14, new UserOperateModel(14, "1000014", "测试14", "0", "15666666666", "ry@qq.com", 280.0, "1"));
          users.put(15, new UserOperateModel(15, "1000015", "测试15", "0", "15666666666", "ry@qq.com", 570.0, "1"));
          users.put(16, new UserOperateModel(16, "1000016", "测试16", "1", "15666666666", "ry@qq.com", 260.0, "1"));
          users.put(17, new UserOperateModel(17, "1000017", "测试17", "1", "15666666666", "ry@qq.com", 210.0, "1"));
          users.put(18, new UserOperateModel(18, "1000018", "测试18", "1", "15666666666", "ry@qq.com", 340.0, "1"));
          users.put(19, new UserOperateModel(19, "1000019", "测试19", "1", "15666666666", "ry@qq.com", 160.0, "1"));
          users.put(20, new UserOperateModel(20, "1000020", "测试20", "1", "15666666666", "ry@qq.com", 220.0, "1"));
          users.put(21, new UserOperateModel(21, "1000021", "测试21", "1", "15666666666", "ry@qq.com", 120.0, "1"));
          users.put(22, new UserOperateModel(22, "1000022", "测试22", "1", "15666666666", "ry@qq.com", 130.0, "1"));
          users.put(23, new UserOperateModel(23, "1000023", "测试23", "1", "15666666666", "ry@qq.com", 490.0, "1"));
          users.put(24, new UserOperateModel(24, "1000024", "测试24", "1", "15666666666", "ry@qq.com", 570.0, "1"));
          users.put(25, new UserOperateModel(25, "1000025", "测试25", "1", "15666666666", "ry@qq.com", 250.0, "1"));
          users.put(26, new UserOperateModel(26, "1000026", "测试26", "1", "15666666666", "ry@qq.com", 250.0, "1"));
      }
  
      /**
       * 表格
       */
      @GetMapping("/table")
      public String table()
      {
          return prefix + "/table";
      }
  
      /**
       * 其他
       */
      @GetMapping("/other")
      public String other()
      {
          return prefix + "/other";
      }
  
      /**
       * 查询数据
       */
      @PostMapping("/list")
      @ResponseBody
      public TableDataInfo list(UserOperateModel userModel)
      {
          TableDataInfo rspData = new TableDataInfo();
          List<UserOperateModel> userList = new ArrayList<UserOperateModel>(users.values());
          // 查询条件过滤
          if (StringUtils.isNotEmpty(userModel.getSearchValue()))
          {
              userList.clear();
              for (Map.Entry<Integer, UserOperateModel> entry : users.entrySet())
              {
                  if (entry.getValue().getUserName().equals(userModel.getSearchValue()))
                  {
                      userList.add(entry.getValue());
                  }
              }
          }
          else if (StringUtils.isNotEmpty(userModel.getUserName()))
          {
              userList.clear();
              for (Map.Entry<Integer, UserOperateModel> entry : users.entrySet())
              {
                  if (entry.getValue().getUserName().equals(userModel.getUserName()))
                  {
                      userList.add(entry.getValue());
                  }
              }
          }
          PageDomain pageDomain = TableSupport.buildPageRequest();
          if (null == pageDomain.getPageNum() || null == pageDomain.getPageSize())
          {
              rspData.setRows(userList);
              rspData.setTotal(userList.size());
              return rspData;
          }
          Integer pageNum = (pageDomain.getPageNum() - 1) * 10;
          Integer pageSize = pageDomain.getPageNum() * 10;
          if (pageSize > userList.size())
          {
              pageSize = userList.size();
          }
          rspData.setRows(userList.subList(pageNum, pageSize));
          rspData.setTotal(userList.size());
          return rspData;
      }
  
      /**
       * 新增用户
       */
      @GetMapping("/add")
      public String add(ModelMap mmap)
      {
          return prefix + "/add";
      }
  
      /**
       * 新增保存用户
       */
      @PostMapping("/add")
      @ResponseBody
      public AjaxResult addSave(UserOperateModel user)
      {
          Integer userId = users.size() + 1;
          user.setUserId(userId);
          return AjaxResult.success(users.put(userId, user));
      }
  
      /**
       * 修改用户
       */
      @GetMapping("/edit/{userId}")
      public String edit(@PathVariable("userId") Integer userId, ModelMap mmap)
      {
          mmap.put("user", users.get(userId));
          return prefix + "/edit";
      }
  
      /**
       * 修改保存用户
       */
      @PostMapping("/edit")
      @ResponseBody
      public AjaxResult editSave(UserOperateModel user)
      {
          return AjaxResult.success(users.put(user.getUserId(), user));
      }
  
      /**
       * 导出
       */
      @PostMapping("/export")
      @ResponseBody
      public AjaxResult export(UserOperateModel user)
      {
          List<UserOperateModel> list = new ArrayList<UserOperateModel>(users.values());
          ExcelUtil<UserOperateModel> util = new ExcelUtil<UserOperateModel>(UserOperateModel.class);
          return util.exportExcel(list, "用户数据");
      }
  
      /**
       * 下载模板
       */
      @GetMapping("/importTemplate")
      @ResponseBody
      public AjaxResult importTemplate()
      {
          ExcelUtil<UserOperateModel> util = new ExcelUtil<UserOperateModel>(UserOperateModel.class);
          return util.importTemplateExcel("用户数据");
      }
  
      /**
       * 导入数据
       */
      @PostMapping("/importData")
      @ResponseBody
      public AjaxResult importData(MultipartFile file, boolean updateSupport) throws Exception
      {
          ExcelUtil<UserOperateModel> util = new ExcelUtil<UserOperateModel>(UserOperateModel.class);
          List<UserOperateModel> userList = util.importExcel(file.getInputStream());
          String message = importUser(userList, updateSupport);
          return AjaxResult.success(message);
      }
  
      /**
       * 删除用户
       */
      @PostMapping("/remove")
      @ResponseBody
      public AjaxResult remove(String ids)
      {
          Integer[] userIds = Convert.toIntArray(ids);
          for (Integer userId : userIds)
          {
              users.remove(userId);
          }
          return AjaxResult.success();
      }
  
      /**
       * 查看详细
       */
      @GetMapping("/detail/{userId}")
      public String detail(@PathVariable("userId") Integer userId, ModelMap mmap)
      {
          mmap.put("user", users.get(userId));
          return prefix + "/detail";
      }
  
      @PostMapping("/clean")
      @ResponseBody
      public AjaxResult clean()
      {
          users.clear();
          return success();
      }
  
      /**
       * 导入用户数据
       * 
       * @param userList 用户数据列表
       * @param isUpdateSupport 是否更新支持，如果已存在，则进行更新数据
       * @return 结果
       */
      public String importUser(List<UserOperateModel> userList, Boolean isUpdateSupport)
      {
          if (StringUtils.isNull(userList) || userList.size() == 0)
          {
              throw new BusinessException("导入用户数据不能为空！");
          }
          int successNum = 0;
          int failureNum = 0;
          StringBuilder successMsg = new StringBuilder();
          StringBuilder failureMsg = new StringBuilder();
          for (UserOperateModel user : userList)
          {
              try
              {
                  // 验证是否存在这个用户
                  boolean userFlag = false;
                  for (Map.Entry<Integer, UserOperateModel> entry : users.entrySet())
                  {
                      if (entry.getValue().getUserName().equals(user.getUserName()))
                      {
                          userFlag = true;
                          break;
                      }
                  }
                  if (!userFlag)
                  {
                      Integer userId = users.size() + 1;
                      user.setUserId(userId);
                      users.put(userId, user);
                      successNum++;
                      successMsg.append("<br/>" + successNum + "、用户 " + user.getUserName() + " 导入成功");
                  }
                  else if (isUpdateSupport)
                  {
                      users.put(user.getUserId(), user);
                      successNum++;
                      successMsg.append("<br/>" + successNum + "、用户 " + user.getUserName() + " 更新成功");
                  }
                  else
                  {
                      failureNum++;
                      failureMsg.append("<br/>" + failureNum + "、用户 " + user.getUserName() + " 已存在");
                  }
              }
              catch (Exception e)
              {
                  failureNum++;
                  String msg = "<br/>" + failureNum + "、账号 " + user.getUserName() + " 导入失败：";
                  failureMsg.append(msg + e.getMessage());
              }
          }
          if (failureNum > 0)
          {
              failureMsg.insert(0, "很抱歉，导入失败！共 " + failureNum + " 条数据格式不正确，错误如下：");
              throw new BusinessException(failureMsg.toString());
          }
          else
          {
              successMsg.insert(0, "恭喜您，数据已全部导入成功！共 " + successNum + " 条，数据如下：");
          }
          return successMsg.toString();
      }
  }
  