From bdbd092755deb8792f091c9a9bb7b9b698472390 Mon Sep 17 00:00:00 2001
From: Administrator <admin>
Date: Thu, 29 Jul 2021 20:12:28 +0800
Subject: [PATCH] 派遣记录接口修改,人员被派遣后,修改人员派遣状态为已派遣

---
 src/main/java/org/springblade/modules/dispatcher/controller/DispatcherController.java |   41 ++++++++++++++++++++++++++++++++++++-----
 1 files changed, 36 insertions(+), 5 deletions(-)

diff --git a/src/main/java/org/springblade/modules/dispatcher/controller/DispatcherController.java b/src/main/java/org/springblade/modules/dispatcher/controller/DispatcherController.java
index 6c16bd1..b994696 100644
--- a/src/main/java/org/springblade/modules/dispatcher/controller/DispatcherController.java
+++ b/src/main/java/org/springblade/modules/dispatcher/controller/DispatcherController.java
@@ -31,10 +31,13 @@
 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;
 
 /**
  *  控制器
@@ -51,6 +54,8 @@
 	private final IDispatcherService dispatcherService;
 
 	private final IDispatcherUnitService dispatcherUnitService;
+
+	private final IUserService userService;
 
 	/**
 	 * 详情
@@ -118,11 +123,37 @@
 	@ApiOperationSupport(order = 6)
 	@ApiOperation(value = "新增或修改", notes = "传入dispatcher")
 	public R submit(@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());
+		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));
 	}

--
Gitblit v1.9.3