From ccebbfd35eca5cd24eb3e32f7c1158e0596c471c Mon Sep 17 00:00:00 2001
From: tangzy <tangzy123456>
Date: Tue, 27 Jul 2021 15:34:53 +0800
Subject: [PATCH] 1.统计接口调整

---
 src/main/java/org/springblade/modules/dispatcher/controller/DispatcherController.java |   11 +++++++++++
 1 files changed, 11 insertions(+), 0 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 cfa1422..6c16bd1 100644
--- a/src/main/java/org/springblade/modules/dispatcher/controller/DispatcherController.java
+++ b/src/main/java/org/springblade/modules/dispatcher/controller/DispatcherController.java
@@ -29,6 +29,7 @@
 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.springframework.web.bind.annotation.*;
 
@@ -48,6 +49,8 @@
 public class DispatcherController extends BladeController {
 
 	private final IDispatcherService dispatcherService;
+
+	private final IDispatcherUnitService dispatcherUnitService;
 
 	/**
 	 * 详情
@@ -89,6 +92,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));
 	}
 
@@ -112,6 +121,8 @@
 		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.saveOrUpdate(dispatcher));
 	}

--
Gitblit v1.9.3