Administrator
2021-08-23 98b253d649a8c41c43ed0317923ed50392bbd0a5
src/main/java/org/springblade/modules/dispatcher/controller/DispatcherUnitController.java
@@ -22,21 +22,31 @@
import io.swagger.annotations.ApiOperation;
import io.swagger.annotations.ApiParam;
import lombok.AllArgsConstructor;
import org.springblade.common.utils.arg;
import org.springblade.core.boot.ctrl.BladeController;
import org.springblade.core.excel.util.ExcelUtil;
import org.springblade.core.mp.support.Condition;
import org.springblade.core.mp.support.Query;
import org.springblade.core.tool.api.R;
import org.springblade.core.tool.utils.Func;
import org.springblade.modules.dispatcher.entity.Dispatcher;
import org.springblade.modules.dispatcher.entity.DispatcherUnit;
import org.springblade.modules.dispatcher.excel.DispatcherUnitExcel;
import org.springblade.modules.dispatcher.excel.DispatcherUnitImporter;
import org.springblade.modules.dispatcher.service.IDispatcherService;
import org.springblade.modules.dispatcher.service.IDispatcherUnitService;
import org.springblade.modules.dispatcher.vo.DispatcherUnitVO;
import org.springblade.modules.dispatcher.vo.DispatcherVO;
import org.springblade.modules.exam.excel.ExamScoreExcel;
import org.springblade.modules.exam.excel.ExamScoreImporter;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;
import javax.servlet.http.HttpServletResponse;
import javax.validation.Valid;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
/**
 *  控制器
@@ -80,6 +90,18 @@
      return R.data(pages);
   }
   /**
    * 自定义树
    * @param dispatcher
    * @return
    */
   @GetMapping("/page-tree")
   public R pageTree(DispatcherUnitVO dispatcher) {
      List<DispatcherUnitVO> pages = dispatcherUnitService.selectDispatcherUnitPageTree(dispatcher);
      return R.data(pages);
   }
   /**
    * 新增
    */
@@ -100,8 +122,8 @@
    * 新增或修改
    */
   @PostMapping("/submit")
   public R submit(@Valid @RequestBody DispatcherUnit dispatcherUnit) {
      if (null==dispatcherUnit.getId()){
   public R submit(@Valid @RequestBody DispatcherUnit dispatcherUnit) throws Exception {
      if (null==dispatcherUnit.getId()) {
         dispatcherUnit.setCreateTime(new Date());
      }
      return R.status(dispatcherUnitService.saveOrUpdate(dispatcherUnit));
@@ -113,9 +135,31 @@
    */
   @PostMapping("/remove")
   @ApiOperation(value = "删除", notes = "传入ids")
   public R remove(@ApiParam(value = "主键集合", required = true) @RequestParam String ids) {
   public R remove(@ApiParam(value = "主键集合", required = true) @RequestParam String ids) throws Exception {
      return R.status(dispatcherUnitService.removeByIds(Func.toLongList(ids)));
   }
   /**
    * 导入派遣单位数据
    * @param isCovered 1 覆盖  0不覆盖
    * @return
    */
   @PostMapping("import-dispatcherUnit")
   public R importDispatcherUnit(MultipartFile file, Integer isCovered) {
      DispatcherUnitImporter dispatcherImporter = new DispatcherUnitImporter(dispatcherUnitService, false);
      ExcelUtil.save(file, dispatcherImporter, DispatcherUnitExcel.class);
      return R.success("操作成功");
   }
   /**
    * 导出模板
    */
   @GetMapping("export-template")
   @ApiOperation(value = "导出模板")
   public void exportUser(HttpServletResponse response) {
      List<DispatcherUnitExcel> list = new ArrayList<>();
      ExcelUtil.export(response, "派遣单位数据导入模板", "派遣单位数据表", list, DispatcherUnitExcel.class);
   }
}