1.ftp

tangzy

2021-08-22 fd7b9aabefdcfa4e2f8504c77d5a4e6f4f955585

 src/main/java/org/springblade/modules/dispatcher/controller/DispatcherController.java

@@ -29,10 +29,15 @@
import org.springblade.core.tool.utils.Func;
import org.springblade.modules.dispatcher.entity.Dispatcher;
import org.springblade.modules.dispatcher.service.IDispatcherService;
import org.springblade.modules.dispatcher.service.IDispatcherUnitService;
import org.springblade.modules.dispatcher.vo.DispatcherVO;
import org.springblade.modules.system.entity.User;
import org.springblade.modules.system.service.IUserService;
import org.springframework.web.bind.annotation.*;

import javax.validation.Valid;
import java.util.Arrays;
import java.util.List;

/**
 *  控制器
@@ -47,6 +52,10 @@
public class DispatcherController extends BladeController {

   private final IDispatcherService dispatcherService;

   private final IDispatcherUnitService dispatcherUnitService;

   private final IUserService userService;

   /**
    * 详情
@@ -88,6 +97,12 @@
   @ApiOperationSupport(order = 4)
   @ApiOperation(value = "新增", notes = "传入dispatcher")
   public R save(@Valid @RequestBody Dispatcher dispatcher) {
      if (null != dispatcher.getUserIds() && dispatcher.getUserIds() != "") {
         //计算派遣人数
         dispatcher.setNum(Arrays.asList(dispatcher.getUserIds().split(",")).size());
         //查询派遣单位的信息,设置行政区
         dispatcher.setJurisdiction(dispatcherUnitService.getById(dispatcher.getDispatcherUnitId()).getJurisdiction().toString());
      }
      return R.status(dispatcherService.save(dispatcher));
   }

@@ -108,6 +123,38 @@
   @ApiOperationSupport(order = 6)
   @ApiOperation(value = "新增或修改", notes = "传入dispatcher")
   public R submit(@Valid @RequestBody Dispatcher dispatcher) {
      if (dispatcher.getId()==null){
         if (null!=dispatcher.getUserIds() && dispatcher.getUserIds()!=""){
            List<String> list = Arrays.asList(dispatcher.getUserIds().split(","));
            //计算派遣人数
            dispatcher.setNum(list.size());
            //修改保安人员的派遣状态
            list.forEach(id -> {
               User user = new User();
               user.setId(Long.parseLong(id));
               user.setDispatch("0");
               userService.updateById(user);
            });
            //查询派遣单位的信息,设置行政区
            dispatcher.setJurisdiction(dispatcherUnitService.getById(dispatcher.getDispatcherUnitId()).getJurisdiction().toString());
         }
      }
      else {
         if (null != dispatcher.getUserIds() && dispatcher.getUserIds() != "") {
            List<String> list = Arrays.asList(dispatcher.getUserIds().split(","));
            //计算派遣人数
            dispatcher.setNum(list.size());
            //修改保安人员的派遣状态
            list.forEach(id -> {
               User user = new User();
               user.setId(Long.parseLong(id));
               user.setDispatch("0");
               userService.updateById(user);
            });
            //查询派遣单位的信息,设置行政区
            dispatcher.setJurisdiction(dispatcherUnitService.getById(dispatcher.getDispatcherUnitId()).getJurisdiction().toString());
         }
      }
      return R.status(dispatcherService.saveOrUpdate(dispatcher));
   }